def switch_on_off(self, obj, trait_name, old, new):
if not self.signal._plot.is_active(): return
if new is True and old is False:
self._line = DraggableVerticalLine(self.axes_manager)
self._line.add_axes(self.signal._plot.signal_plot.ax)
self._line.patch.set_linewidth(2)
self._color_changed("black", "black")
# There is not need to call draw because setting the
# color calls it.
elif new is False and old is True:
self._line.close()
self._line = None
self.draw()
def switch_on_off(self, obj, trait_name, old, new):
if not self.signal._plot.is_active(): return
if new is True and old is False:
self._line = DraggableVerticalLine(self.axes_manager)
self._line.add_axes(self.signal._plot.signal_plot.ax)
self._line.patch.set_linewidth(2)
self._color_changed("black","black")
# There is not need to call draw because setting the
# color calls it.
elif new is False and old is True:
self._line.close()
self._line = None
self.draw()
class LineInSpectrum(t.HasTraits):
"""Adds a vertical draggable line to a spectrum that reports its
position to the position attribute of the class.
Attributes:
-----------
position : float
The position of the vertical line in the spectrum. Moving the
line changes the position but the reverse is not true.
on : bool
Turns on and off the line
color : wx.Colour
The color of the line. It automatically redraws the line.
"""
position = t.Float()
is_ok = t.Bool(False)
on = t.Bool(False)
color = t.Color("black")
def __init__(self, signal):
if signal.axes_manager.signal_dimension != 1:
raise SignalDimensionError(
signal.axes_manager.signal_dimension, 1)
self.signal = signal
self.signal.plot()
axis_dict = signal.axes_manager.signal_axes[0].get_axis_dictionary()
am = AxesManager([axis_dict, ])
am._axes[0].navigate = True
# Set the position of the line in the middle of the spectral
# range by default
am._axes[0].index = int(round(am._axes[0].size / 2))
self.axes_manager = am
self.axes_manager.connect(self.update_position)
self.on_trait_change(self.switch_on_off, 'on')
def draw(self):
self.signal._plot.signal_plot.figure.canvas.draw()
def switch_on_off(self, obj, trait_name, old, new):
if not self.signal._plot.is_active():
return
if new is True and old is False:
self._line = DraggableVerticalLine(self.axes_manager)
self._line.add_axes(self.signal._plot.signal_plot.ax)
self._line.patch.set_linewidth(2)
self._color_changed("black", "black")
# There is not need to call draw because setting the
# color calls it.
elif new is False and old is True:
self._line.close()
self._line = None
self.draw()
def update_position(self, *args, **kwargs):
if not self.signal._plot.is_active():
return
self.position = self.axes_manager.coordinates[0]
def _color_changed(self, old, new):
if self.on is False:
return
self._line.patch.set_color((self.color.Red() / 255.,
self.color.Green() / 255.,
self.color.Blue() / 255.,))
self.draw()
class LineInSpectrum(t.HasTraits):
"""Adds a vertical draggable line to a spectrum that reports its
position to the position attribute of the class.
Attributes:
-----------
position : float
The position of the vertical line in the spectrum. Moving the
line changes the position but the reverse is not true.
on : bool
Turns on and off the line
color : wx.Colour
The color of the line. It automatically redraws the line.
"""
position = t.Float()
is_ok = t.Bool(False)
on = t.Bool(False)
color = t.Color("black")
def __init__(self, signal):
if signal.axes_manager.signal_dimension != 1:
raise SignalDimensionError(
signal.axes_manager.signal_dimension, 1)
self.signal = signal
self.signal.plot()
axis_dict = signal.axes_manager.signal_axes[0].get_axis_dictionary()
am = AxesManager([axis_dict, ])
am._axes[0].navigate = True
# Set the position of the line in the middle of the spectral
# range by default
am._axes[0].index = int(round(am._axes[0].size / 2))
self.axes_manager = am
self.axes_manager.connect(self.update_position)
self.on_trait_change(self.switch_on_off, 'on')
def draw(self):
self.signal._plot.signal_plot.figure.canvas.draw()
def switch_on_off(self, obj, trait_name, old, new):
if not self.signal._plot.is_active():
return
if new is True and old is False:
self._line = DraggableVerticalLine(self.axes_manager)
self._line.set_mpl_ax(self.signal._plot.signal_plot.ax)
self._line.patch.set_linewidth(2)
self._color_changed("black", "black")
# There is not need to call draw because setting the
# color calls it.
elif new is False and old is True:
self._line.close()
self._line = None
self.draw()
def update_position(self, *args, **kwargs):
if not self.signal._plot.is_active():
return
self.position = self.axes_manager.coordinates[0]
def _color_changed(self, old, new):
if self.on is False:
return
self._line.patch.set_color((self.color.Red() / 255.,
self.color.Green() / 255.,
self.color.Blue() / 255.,))
self.draw()
