|
12 | 12 | matplotlib.use('Agg') |
13 | 13 |
|
14 | 14 | ### Plotting Parameters ### |
15 | | -LINEWIDTH = 1.0 |
| 15 | +LINEWIDTH = 1.5 |
16 | 16 | LINE_COLOR = 'darkgrey' |
17 | 17 | ALPHA = 0.4 |
18 | 18 | CIRCLE = 4.0 |
@@ -123,21 +123,24 @@ def draw(self, path, color=True, plotIpe=False, saveFile=True): |
123 | 123 | points = [(self.G.nodes[source]['X'], self.G.nodes[source]['Y'])] + data['Spline'].points + [(self.G.nodes[target]['X'], self.G.nodes[target]['Y'])] |
124 | 124 |
|
125 | 125 | X, Y = self.approxBezier(points, 50) |
| 126 | + data['X'] = X |
| 127 | + data['Y'] = Y |
| 128 | + ax.plot(X, Y, color=cmap(data['Angle']), alpha=ALPHA, lw = LINEWIDTH) |
126 | 129 | else: |
127 | 130 | X = [self.G.nodes[source]['X'], self.G.nodes[target]['X']] |
128 | 131 | Y = [self.G.nodes[source]['Y'], self.G.nodes[target]['Y']] |
129 | 132 |
|
130 | 133 | data['X'] = X |
131 | 134 | data['Y'] = Y |
132 | | - ax.plot(X, Y, color=cmap(data['Angle']), alpha=ALPHA, lw = LINEWIDTH) |
| 135 | + # ax.plot(X, Y, color=cmap(data['Angle']), alpha=ALPHA, lw = LINEWIDTH) |
133 | 136 |
|
134 | 137 | X = [] |
135 | 138 | Y = [] |
136 | 139 | for id, data in self.G.nodes().data(): |
137 | 140 | X.append(data['X']) |
138 | 141 | Y.append(data['Y']) |
139 | 142 |
|
140 | | - ax.plot(X, Y, linestyle="none", color=CIRCLE_COLOR_LIGHT, marker='.', markersize = CIRCLE) |
| 143 | + # ax.plot(X, Y, linestyle="none", color=CIRCLE_COLOR_LIGHT, marker='.', markersize = CIRCLE) |
141 | 144 |
|
142 | 145 | if saveFile: |
143 | 146 | plt.savefig(f'{path}{self.name}.png') |
|
0 commit comments