Edge color encoding

ecomfe · Aug 4, 2016 · 4dcbfa3 · 4dcbfa3
1 parent 788c51f
commit 4dcbfa3
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 3 deletions.
diff --git a/dist/echarts-graph-modularity.js b/dist/echarts-graph-modularity.js
@@ -91,6 +91,25 @@ return /******/ (function(modules) { // webpackBootstrap
 	                var comm = result[id];
 	                graph.data.setItemVisual(idIndexMap[id], 'color', seriesModel.getColorFromPalette(comm, paletteScope));
 	            }
+
+	            graph.edgeData.each(function (idx) {
+	                var itemModel = graph.edgeData.getItemModel(idx);
+	                var edge = graph.getEdgeByIndex(idx);
+	                var color = itemModel.get('lineStyle.normal.color');
+
+	                switch (color) {
+	                    case 'source':
+	                        color = edge.node1.getVisual('color');
+	                        break;
+	                    case 'target':
+	                        color = edge.node2.getVisual('color');
+	                        break;
+	                }
+
+	                if (color != null) {
+	                    edge.setVisual('color', color);
+	                }
+	            });
 	        }
 	    });
 	});

diff --git a/dist/echarts-graph-modularity.min.js b/dist/echarts-graph-modularity.min.js
diff --git a/example/example.html b/example/example.html
@@ -65,7 +65,7 @@
                     width: '50%',
                     bottom: 0
                 }, {
-                    name: 'Without Modularity',
+                    name: 'With Modularity',
                     type: 'graph',
                     layout: 'force',
                     modularity: true,
@@ -75,7 +75,12 @@
                     left: '50%',
                     top: 0,
                     width: '50%',
-                    bottom: 0
+                    bottom: 0,
+                    lineStyle: {
+                        normal: {
+                            color: 'source'
+                        }
+                    }
                 }]
             };
 

diff --git a/src/main.js b/src/main.js
@@ -29,6 +29,25 @@ echarts.registerVisual(echarts.PRIORITY.VISUAL.CHART + 1, function (ecModel, api
                 var comm = result[id];
                 graph.data.setItemVisual(idIndexMap[id], 'color', seriesModel.getColorFromPalette(comm, paletteScope));
             }
+
+            graph.edgeData.each(function (idx) {
+                var itemModel = graph.edgeData.getItemModel(idx);
+                var edge = graph.getEdgeByIndex(idx);
+                var color = itemModel.get('lineStyle.normal.color');
+
+                switch (color) {
+                    case 'source':
+                        color = edge.node1.getVisual('color');
+                        break;
+                    case 'target':
+                        color = edge.node2.getVisual('color');
+                        break;
+                }
+
+                if (color != null) {
+                    edge.setVisual('color', color);
+                }
+            });
         }
     });
 });