fix(sankey): use node color for link color (#1615)

plouc · Jun 22, 2021 · 1002edb · 1002edb
1 parent 232b613
commit 1002edb
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/packages/sankey/src/hooks.js b/packages/sankey/src/hooks.js
@@ -19,7 +19,6 @@ export const computeNodeAndLinks = ({
     width,
     height,
     getColor,
-    getLinkColor,
     getLabel,
 }) => {
     const sankey = d3Sankey()
@@ -61,7 +60,7 @@ export const computeNodeAndLinks = ({
     })
 
     data.links.forEach(link => {
-        link.color = getLinkColor(link)
+        link.color = link.source.color
         link.pos0 = link.y0
         link.pos1 = link.y1
         link.thickness = link.width
@@ -111,7 +110,6 @@ export const useSankey = ({
 
     const getColor = useOrdinalColorScale(colors, 'id')
     const getNodeBorderColor = useInheritedColor(nodeBorderColor, theme)
-    const getLinkColor = useOrdinalColorScale(colors, 'source.id')
 
     const getLabel = useMemo(() => getLabelGenerator(label, labelFormat), [label, labelFormat])
     const getLabelTextColor = useInheritedColor(labelTextColor, theme)
@@ -130,7 +128,6 @@ export const useSankey = ({
                 width,
                 height,
                 getColor,
-                getLinkColor,
                 getLabel,
             }),
         [
@@ -145,7 +142,6 @@ export const useSankey = ({
             width,
             height,
             getColor,
-            getLinkColor,
             getLabel,
         ]
     )