Optimising Water Purification: A 3D Modelling Approach