This is another improved version without using the library:
It’s not ideal since it’s using a slow if-then statement in the shader, and I guess could be even smoother.
I would still love to know how to solve the issue with the library, as it would be more flexible and potentially faster. Any help is very much appreciated! 