Modeling Segregation in Granular Flows.