I think the color should be base on tracks rather than notes

[simonmysun]

RT
Using same color may seems ugly but will be more conductive to visualization.

[simonmysun]

Maybe a radio box can meet everyone's need.