Geografic-Information-System/gui/wxpython/iscatt/toolbars.py

"""
@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 <stepan.turek seznam.cz> (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


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._data:
                i_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._data:
            i_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._data:
                i_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._data:
            i_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(),
                ),
            )
        )