Bedload transport for heterogeneous sediments