Hello!
Another solution would be to use a PRIVATE VARIABLE defined on the mesh to identify the different zones. This is what is done for defining the CN coefficient for the rainfall-runoff model implemented in runoff_scs_cn.f for example. Have a look at my paper from the TUC 2016 for more details.
If you only apply runoff and don't need a rainfall-runoff model you can actually do this directly in prosou.f, the rainfall depth applied at each node is RAIN_MPS.
Good luck!
Best regards
PL