Unity WaterPro Shader Follow Mesh Direction?

Is it possible to have the WaterPro shader’s waves follow the direction of a mesh? So the waves would flow down a curved river instead of a straight river. I’ve tried a curved mesh but it seems like the Wave speed values on the shader make the waves move in a world/camera/shader space ignoring the underlying mesh.

What you’re looking for is a “flow map”.

http://teckartist.com/?page_id=107

http://scrawkblog.com/2013/12/12/tiled-directional-flow-in-unity/

Thanks. I have the Unity shader looking the way I want and was hoping to tweak that, but Flow looks worth investigating. Are there any big issues with Flow and Unity 5.0