class CrossSectionWidget(PanelWidget):
PANEL_ID = None
PANEL_TITLE = _("Cross section tool")
PANEL_ICON = "csection.png"
CrossSectionPlotKlass = None
__implements__ = (IPanel,)
def __init__(self, parent=None):
super(CrossSectionWidget, self).__init__(parent)
self.export_ac = None
self.autoscale_ac = None
self.refresh_ac = None
self.autorefresh_ac = None
self.manager = None # manager for the associated image plot
self.local_manager = PlotManager(self)
self.cs_plot = self.CrossSectionPlotKlass(parent)
self.cs_plot.SIG_CS_CURVE_CHANGED.connect(self.cs_curve_has_changed)
self.export_tool = None
self.setup_plot()
self.toolbar = QToolBar(self)
self.toolbar.setOrientation(Qt.Vertical)
self.setup_widget()
def set_options(self, autoscale=None, autorefresh=None):
assert self.manager is not None, "Panel '%s' must be registered to plot manager before changing options" % self.PANEL_ID
if autoscale is not None:
self.autoscale_ac.setChecked(autoscale)
if autorefresh is not None:
self.autorefresh_ac.setChecked(autorefresh)
def setup_plot(self):
# Configure the local manager
lman = self.local_manager
lman.add_plot(self.cs_plot)
lman.register_all_curve_tools()
self.export_tool = lman.get_tool(ExportItemDataTool)
def setup_widget(self):
layout = QHBoxLayout()
layout.addWidget(self.cs_plot)
layout.addWidget(self.toolbar)
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
def cs_curve_has_changed(self, curve):
"""Cross section curve has just changed"""
# Do something with curve's data for example
pass
def register_panel(self, manager):
"""Register panel to plot manager"""
self.manager = manager
for plot in manager.get_plots():
self.cs_plot.connect_plot(plot)
self.setup_actions()
self.add_actions_to_toolbar()
def configure_panel(self):
"""Configure panel"""
pass
def get_plot(self):
return self.manager.get_active_plot()
def setup_actions(self):
self.export_ac = self.export_tool.action
self.autoscale_ac = create_action(self, _("Auto-scale"),
icon=get_icon('csautoscale.png'),
toggled=self.cs_plot.toggle_autoscale)
self.autoscale_ac.setChecked(self.cs_plot.autoscale_mode)
self.refresh_ac = create_action(self, _("Refresh"),
icon=get_icon('refresh.png'),
triggered=lambda: self.cs_plot.update_plot())
self.autorefresh_ac = create_action(self, _("Auto-refresh"),
icon=get_icon('autorefresh.png'),
toggled=self.cs_plot.toggle_autorefresh)
self.autorefresh_ac.setChecked(self.cs_plot.autorefresh_mode)
def add_actions_to_toolbar(self):
add_actions(self.toolbar, (self.export_ac, self.autoscale_ac,
None, self.refresh_ac, self.autorefresh_ac))
def register_shape(self, shape, final, refresh=True):
plot = self.get_plot()
self.cs_plot.register_shape(plot, shape, final, refresh)
def unregister_shape(self, shape):
self.cs_plot.unregister_shape(shape)
def update_plot(self, obj=None):
"""
Update cross section curve(s) associated to object *obj*
*obj* may be a marker or a rectangular shape
(see :py:class:`guiqwt.tools.CrossSectionTool`
and :py:class:`guiqwt.tools.AverageCrossSectionTool`)
If obj is None, update the cross sections of the last active object
"""
self.cs_plot.update_plot(obj)
class CrossSectionWidget(PanelWidget):
PANEL_ID = None
PANEL_TITLE = _("Cross section tool")
PANEL_ICON = "csection.png"
CrossSectionPlotKlass = None
__implements__ = (IPanel, )
def __init__(self, parent=None):
super(CrossSectionWidget, self).__init__(parent)
self.export_ac = None
self.autoscale_ac = None
self.refresh_ac = None
self.autorefresh_ac = None
self.manager = None # manager for the associated image plot
self.local_manager = PlotManager(self)
self.cs_plot = self.CrossSectionPlotKlass(parent)
self.cs_plot.SIG_CS_CURVE_CHANGED.connect(self.cs_curve_has_changed)
self.export_tool = None
self.setup_plot()
self.toolbar = QToolBar(self)
self.toolbar.setOrientation(Qt.Vertical)
self.setup_widget()
def set_options(self, autoscale=None, autorefresh=None):
assert self.manager is not None, "Panel '%s' must be registered to plot manager before changing options" % self.PANEL_ID
if autoscale is not None:
self.autoscale_ac.setChecked(autoscale)
if autorefresh is not None:
self.autorefresh_ac.setChecked(autorefresh)
def setup_plot(self):
# Configure the local manager
lman = self.local_manager
lman.add_plot(self.cs_plot)
lman.register_all_curve_tools()
self.export_tool = lman.get_tool(ExportItemDataTool)
def setup_widget(self):
layout = QHBoxLayout()
layout.addWidget(self.cs_plot)
layout.addWidget(self.toolbar)
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
def cs_curve_has_changed(self, curve):
"""Cross section curve has just changed"""
# Do something with curve's data for example
pass
def register_panel(self, manager):
"""Register panel to plot manager"""
self.manager = manager
for plot in manager.get_plots():
self.cs_plot.connect_plot(plot)
self.setup_actions()
self.add_actions_to_toolbar()
def configure_panel(self):
"""Configure panel"""
pass
def get_plot(self):
return self.manager.get_active_plot()
def setup_actions(self):
self.export_ac = self.export_tool.action
self.autoscale_ac = create_action(
self,
_("Auto-scale"),
icon=get_icon('csautoscale.png'),
toggled=self.cs_plot.toggle_autoscale)
self.autoscale_ac.setChecked(self.cs_plot.autoscale_mode)
self.refresh_ac = create_action(
self,
_("Refresh"),
icon=get_icon('refresh.png'),
triggered=lambda: self.cs_plot.update_plot())
self.autorefresh_ac = create_action(
self,
_("Auto-refresh"),
icon=get_icon('autorefresh.png'),
toggled=self.cs_plot.toggle_autorefresh)
self.autorefresh_ac.setChecked(self.cs_plot.autorefresh_mode)
def add_actions_to_toolbar(self):
add_actions(self.toolbar, (self.export_ac, self.autoscale_ac, None,
self.refresh_ac, self.autorefresh_ac))
def register_shape(self, shape, final, refresh=True):
plot = self.get_plot()
self.cs_plot.register_shape(plot, shape, final, refresh)
def unregister_shape(self, shape):
self.cs_plot.unregister_shape(shape)
def update_plot(self, obj=None):
"""
Update cross section curve(s) associated to object *obj*
*obj* may be a marker or a rectangular shape
(see :py:class:`guiqwt.tools.CrossSectionTool`
and :py:class:`guiqwt.tools.AverageCrossSectionTool`)
If obj is None, update the cross sections of the last active object
"""
self.cs_plot.update_plot(obj)
