override the predefined colors about d3plus-viz HOT 2 OPEN

let me know if this example helps: http://d3plus.org/examples/d3plus-hierarchy/custom-color/

from d3plus-viz.

Hello,
I am trying to override your predefined colors in a pie/treemap. I need to use my own set of colors. My set of Colors are NOT RELATED TO THE DATA. I just need that the Treemap/Pie take my colors instead of your default colors.
I tried to create a JS in the shapeConfig. You find My example here : https://jsfiddle.net/ubxLemfy/
It works but the legend donesn't correspond to the colors in the Treeamp/Pie, in my example, sweden is red in the pie while it is purple in the legend.
Am I doing somethig wrong? is there another way to do it?
In version 1 I just needed to add
.color({"scale": [ Array of Colors]})

from d3plus-viz.