Numerical modelling of friction stir welding: a grain size evolution model