This is absolutely stunning. Wonderful some function of the state of a point can give it colour.

Author here, there is a setting to pick colour mode. I implemented it after similar suggestion by someone on twitter. Give it a try.

It already supports colour!