I wrote a blog on creating a stylized waterfall inspired by Simon Trümpler's work for RiME which got very positive responses. You can read it on my blog, or read it on 80 Level! For this project I used Unity and the Amplify Shader Editor plugin, but this would work in UE4 as well.
Link to the first part on 80.lvl
I hope it might teach you something!