Paint different paint for different elements

I want to paint different paint for different elements (column, structure beam, wall, version)

Just like revit this function

What kind of node should I use?

Please advise, thank you