Modelling the anisotropy of turbulence with the SWASH model: Heterogeneous roughness conditions in open channel flows