""" @package iscatt.toolbars @brief Scatter plot - toolbars Classes: - toolbars::MainToolbar (C) 2013 by the GRASS Development Team This program is free software under the GNU General Public License (>=v2). Read the file COPYING that comes with GRASS for details. @author Stepan Turek (mentor: Martin Landa) """ import wx from icons.icon import MetaIcon from gui_core.toolbars import BaseToolbar, BaseIcons from core.gcmd import RunCommand from iscatt.dialogs import SettingsDialog def get_tool_name(tool_name, tool_name_type=tuple): """Get tool name :param str|tuple tool_name: tool name :param type tool_name_type: tool name type with default tuple type :return str: tool name """ if isinstance(tool_name, tool_name_type): return tool_name[0] return tool_name class MainToolbar(BaseToolbar): """Main toolbar""" def __init__(self, parent, scatt_mgr, opt_tools=None): BaseToolbar.__init__(self, parent) self.scatt_mgr = scatt_mgr self.opt_tools = opt_tools self.InitToolbar(self._toolbarData()) # realize the toolbar self.Realize() self.scatt_mgr.modeSet.connect(self.ModeSet) def _toolbarData(self): icons = { "selectGroup": MetaIcon( img="layer-group-add", label=_("Select imagery group") ), "settings": BaseIcons["settings"], "help": BaseIcons["help"], "add_scatt_pl": MetaIcon( img="layer-raster-analyze", label=_("Add scatter plot") ), "selCatPol": MetaIcon(img="polygon", label=_("Select area with polygon")), "pan": MetaIcon(img="pan", label=_("Pan mode for scatter plots")), "zoomIn": MetaIcon( img="zoom-in", label=_("Zoom mode for scatter plots (left mouse button, wheel)"), ), "zoomExtent": MetaIcon( img="zoom-extent", label=_( "Zoom to scatter plot data extend mode (click on scatter plot for zooming to extend)" ), ), "cats_mgr": MetaIcon( img="table-manager", label=_("Show/hide class manager") ), } tools = [ ( ("add_scatt", icons["add_scatt_pl"].label), icons["add_scatt_pl"], lambda event: self.scatt_mgr.AddScattPlot(), ), (None,), ( ("cats_mgr", icons["cats_mgr"].label), icons["cats_mgr"], lambda event: self.parent.ShowCategoryPanel(event.IsChecked()), wx.ITEM_CHECK, ), (None,), ( ("pan", icons["pan"].label), icons["pan"], lambda event: self.SetPloltsMode(event, "pan"), wx.ITEM_CHECK, ), ( ("zoom", icons["zoomIn"].label), icons["zoomIn"], lambda event: self.SetPloltsMode(event, "zoom"), wx.ITEM_CHECK, ), ( ("zoom_extend", icons["zoomExtent"].label), icons["zoomExtent"], lambda event: self.SetPloltsMode(event, "zoom_extend"), wx.ITEM_CHECK, ), (None,), ( ("sel_pol_mode", icons["selCatPol"].label), icons["selCatPol"], self.ActivateSelectionPolygonMode, wx.ITEM_CHECK, ), (None,), ( ("settings", icons["settings"].label), icons["settings"], self.OnSettings, ), ( ("help", icons["help"].label), icons["help"], self.OnHelp, ), ] if self.opt_tools and "add_group" in self.opt_tools: tools.insert( 0, ( ("selectGroup", icons["selectGroup"].label), icons["selectGroup"], lambda event: self.scatt_mgr.SetData(), ), ) return self._getToolbarData(tools) def GetToolId(self, toolName): # TODO can be useful in base return vars(self)[toolName] def SetPloltsMode(self, event, tool_name): self.scatt_mgr.modeSet.disconnect(self.ModeSet) if event.IsChecked(): for i_tool_data in self.controller.data: i_tool_name = get_tool_name(i_tool_data[0]) if not i_tool_name or i_tool_name in ["cats_mgr", "sel_pol_mode"]: continue if i_tool_name == tool_name: continue i_tool_id = vars(self)[i_tool_name] self.ToggleTool(i_tool_id, False) self.scatt_mgr.SetPlotsMode(mode=tool_name) else: self.scatt_mgr.SetPlotsMode(mode=None) self.scatt_mgr.modeSet.connect(self.ModeSet) def ActivateSelectionPolygonMode(self, event): activated = self.scatt_mgr.ActivateSelectionPolygonMode(event.IsChecked()) self.parent.ShowPlotEditingToolbar(activated) i_tool_id = vars(self)["sel_pol_mode"] self.ToggleTool(i_tool_id, activated) def ModeSet(self, mode): self.UnsetMode() def UnsetMode(self): for i_tool_data in self.controller.data: i_tool_name = get_tool_name(i_tool_data[0]) if not i_tool_name or i_tool_name in ["cats_mgr", "sel_pol_mode"]: continue i_tool_id = vars(self)[i_tool_name] self.ToggleTool(i_tool_id, False) def OnSettings(self, event): dlg = SettingsDialog( parent=self, id=wx.ID_ANY, title=_("Settings"), scatt_mgr=self.scatt_mgr ) dlg.ShowModal() dlg.Destroy() def OnHelp(self, event): RunCommand("g.manual", entry="wxGUI.iscatt") class EditingToolbar(BaseToolbar): """Main toolbar""" def __init__(self, parent, scatt_mgr): BaseToolbar.__init__(self, parent) self.scatt_mgr = scatt_mgr self.InitToolbar(self._toolbarData()) # realize the toolbar self.Realize() self.scatt_mgr.modeSet.connect(self.ModeSet) def _toolbarData(self): """Toolbar data""" self.icons = { "sel_add": MetaIcon( img="layer-add", label=_("Include selected area to class"), desc=_("Include selected area to class"), ), "sel_remove": MetaIcon( img="layer-remove", label=_("Exclude selected area from class"), desc=_("Exclude selected area from class"), ), "addVertex": MetaIcon( img="vertex-create", label=_("Add new vertex"), desc=_("Add new vertex to polygon boundary scatter plot"), ), "editLine": MetaIcon( img="polygon-create", label=_("Create selection polygon"), desc=_("Add new vertex between last and first points of the boundary"), ), "moveVertex": MetaIcon( img="vertex-move", label=_("Move vertex"), desc=_("Move boundary vertex"), ), "removeVertex": MetaIcon( img="vertex-delete", label=_("Remove vertex"), desc=_("Remove boundary vertex"), ), "delete": MetaIcon( img="polygon-delete", label=_( "Remove polygon (click on scatter plot for removing it's polygon)" ), ), } return self._getToolbarData( ( ( ("sel_add", self.icons["sel_add"].label), self.icons["sel_add"], lambda event: self.scatt_mgr.ProcessSelectionPolygons("add"), ), ( ("sel_remove", self.icons["sel_remove"].label), self.icons["sel_remove"], lambda event: self.scatt_mgr.ProcessSelectionPolygons("remove"), ), (None,), ( ("add_vertex", self.icons["editLine"].label), self.icons["editLine"], lambda event: self.SetMode(event, "add_vertex"), wx.ITEM_CHECK, ), ( ("add_boundary_vertex", self.icons["addVertex"].label), self.icons["addVertex"], lambda event: self.SetMode(event, "add_boundary_vertex"), wx.ITEM_CHECK, ), ( ("move_vertex", self.icons["moveVertex"].label), self.icons["moveVertex"], lambda event: self.SetMode(event, "move_vertex"), wx.ITEM_CHECK, ), ( ("delete_vertex", self.icons["removeVertex"].label), self.icons["removeVertex"], lambda event: self.SetMode(event, "delete_vertex"), wx.ITEM_CHECK, ), ( ("remove_polygon", self.icons["delete"].label), self.icons["delete"], lambda event: self.SetMode(event, "remove_polygon"), wx.ITEM_CHECK, ), ) ) def SetMode(self, event, tool_name): self.scatt_mgr.modeSet.disconnect(self.ModeSet) if event.IsChecked(): for i_tool_data in self.controller.data: i_tool_name = get_tool_name(i_tool_data[0]) if not i_tool_name: continue if i_tool_name == tool_name: continue i_tool_id = vars(self)[i_tool_name] self.ToggleTool(i_tool_id, False) self.scatt_mgr.SetPlotsMode(tool_name) else: self.scatt_mgr.SetPlotsMode(None) self.scatt_mgr.modeSet.connect(self.ModeSet) def ModeSet(self, mode): if mode in ["zoom", "pan", "zoom_extend", None]: self.UnsetMode() def UnsetMode(self): for i_tool_data in self.controller.data: i_tool_name = get_tool_name(i_tool_data[0]) if not i_tool_name: continue i_tool_id = vars(self)[i_tool_name] self.ToggleTool(i_tool_id, False) def GetToolId(self, toolName): return vars(self)[toolName] class CategoryToolbar(BaseToolbar): """Main toolbar""" def __init__(self, parent, scatt_mgr, cats_list): BaseToolbar.__init__(self, parent) self.scatt_mgr = scatt_mgr self.cats_mgr = self.scatt_mgr.GetCategoriesManager() self.cats_list = cats_list self.InitToolbar(self._toolbarData()) # realize the toolbar self.Realize() def _toolbarData(self): """Toolbar data""" self.icons = { "add_class": MetaIcon(img="layer-add", label=_("Add class")), "remove_class": MetaIcon( img="layer-remove", label=_("Remove selected class") ), } return self._getToolbarData( ( ( ("add_class", self.icons["add_class"].label), self.icons["add_class"], lambda event: self.cats_mgr.AddCategory(), ), ( ("remove_class", self.icons["remove_class"].label), self.icons["remove_class"], lambda event: self.cats_list.DeleteCategory(), ), ) )