You can actually choose what colours get reflected from each material you use outside of the function editor, in the advanced material editor. 

I am sure it can be done in the function editor as well, I just don't know how.