Updated to handle new numpy/scipy.

This commit is contained in:
Einar Ryeng 2007-07-24 12:19:13 +00:00
parent 05274b4f0b
commit 524c90fcf7

View File

@ -13,6 +13,7 @@ from matplotlib.mlab import prctile
from matplotlib.colors import ColorConverter
import networkx
import scipy
from numpy import matlib
import fluents
import logger
@ -692,7 +693,7 @@ class NetworkPlot(Plot):
ax=self.axes,
edge_color=edge_color)
# edge color rgba-arrays
self._edge_color_rgba = scipy.repmat(ColorConverter().to_rgba(edge_color),
self._edge_color_rgba = matlib.repmat(ColorConverter().to_rgba(edge_color),
self.graph.number_of_edges(),1)
self._edge_color_selected = ColorConverter().to_rgba(self.default_props['edge_color_selected'])
if self._with_labels:
@ -736,7 +737,7 @@ class NetworkPlot(Plot):
edge_color_rgba = self._edge_color_rgba.copy()
index = self.get_index_from_selection(self.dataset, selection)
if len(index) > 0:
linewidth.put(2, index)
linewidth[index] = 2
idents = selection[self.current_dim]
edge_index = [i for i,edge in enumerate(self.graph.edges()) if (edge[0] in idents and edge[1] in idents)]
if len(edge_index)>0: