def __init__(self,
parent=None,
base=None,
data=None,
id=-1,
*args,
**kwds):
kwds["style"] = wx.DEFAULT_DIALOG_STYLE
kwds["size"] = wx.Size(_STATICBOX_WIDTH * 0.8, PANEL_SIZE)
wx.Dialog.__init__(self, parent, id=id, *args, **kwds)
if data != None:
# Font size
kwds = []
self.SetWindowVariant(variant=FONT_VARIANT)
self.SetTitle("Mask Editor for " + data.name)
self.parent = base
self.data = data
self.str = self.data.__str__()
# # mask for 2D
self.mask = data.mask
self.default_mask = copy.deepcopy(data.mask)
# # masked data from GUI
self.slicer_mask = None
self.slicer = None
self.slicer_z = 5
self.data.interactive = True
# # when 2 data have the same id override the 1 st plotted
self.name = self.data.name
# Panel for 2D plot
self.plotpanel = Maskplotpanel(self,
-1,
style=wx.TRANSPARENT_WINDOW)
self.cmap = DEFAULT_CMAP
# # Create Artist and bind it
self.subplot = self.plotpanel.subplot
self.connect = BindArtist(self.subplot.figure)
self._setup_layout()
self.newplot = Data2D(image=self.data.data)
self.newplot.setValues(self.data)
self.plotpanel.add_image(self.newplot)
self._update_mask(self.mask)
self.Centre()
self.Layout()
# bind evt_close to _draw in fitpage
self.Bind(wx.EVT_CLOSE, self.OnClose)
def __init__(self, ax, spline):
self.ax = ax
self.fig = ax.figure
self.spline = spline
# Initialize callbacks
self.callback = BindArtist(self.fig)
self.callback.clearall()
self.ax.cla()
self.callback('dclick',self.ax,self.onAppend)
# Add artists to the canvas
[self.hspline] = self.ax.plot([],[],marker='',
linestyle='-',color='green')
self.hpoints = []
for x,y in self.spline:
self._appendknot(x,y)
# Draw the canvas
self.draw()
class BSplinePlot:
"""Edit a parametric B-Spline profile"""
def __init__(self, ax, spline):
self.ax = ax
self.fig = ax.figure
self.spline = spline
# Initialize callbacks
self.callback = BindArtist(self.fig)
self.callback.clearall()
self.ax.cla()
self.callback('dclick',self.ax,self.onAppend)
# Add artists to the canvas
[self.hspline] = self.ax.plot([],[],marker='',
linestyle='-',color='green')
self.hpoints = []
for x,y in self.spline:
self._appendknot(x,y)
# Draw the canvas
self.draw()
def _appendknot(self,x,y):
"""
Append a knot to the list of knots on the canvas. This had better
maintain order within the spline if callbacks are to work properly.
"""
[h] = self.ax.plot([x],[y],marker='s',markersize=8,
linestyle='',color='yellow',pickradius=5)
self.callback('enter',h,self.onHilite)
self.callback('leave',h,self.onHilite)
self.callback('drag',h,self.onDrag)
self.callback('dclick',h,self.onRemove)
self.hpoints.append(h)
return True
def _removeknot(self, artist):
"""
Remove the knot associated with the artist.
"""
i = self.hpoints.index(artist)
del self.spline[i]
del self.hpoints[i]
artist.remove()
def draw(self):
"""
Recompute the spline curve and show the canvas.
"""
x,y = self.spline.sample()
self.hspline.set_data(x,y)
self.fig.canvas.draw_idle()
def onHilite(self,ev):
"""
Let the user know which point is being edited.
"""
if ev.action == 'enter':
ev.artist.set_color('lightblue')
else:
ev.artist.set_color('yellow')
self.fig.canvas.draw_idle()
return True
def onDrag(self,ev):
"""
Move the selected control point.
"""
if ev.inaxes == self.ax:
i = self.hpoints.index(ev.artist)
self.spline[i] = ev.xdata, ev.ydata
ev.artist.set_data([ev.xdata],[ev.ydata])
self.draw()
return True
def onRemove(self,ev):
"""
Remove the selected point.
"""
# Don't delete the last one
if len(self.spline) > 1:
self._removeknot(ev.artist)
self.draw()
return True
def onAppend(self,ev):
"""
Append a new control point to the end of the spline.
"""
self.spline.append(ev.xdata,ev.ydata)
self._appendknot(ev.xdata,ev.ydata)
self.draw()
return True
