"""! @package mapdisp.toolbars @brief Map display frame - toolbars Classes: - toolbars::MapToolbar (C) 2007-2011 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 Michael Barton @author Jachym Cepicky @author Martin Landa """ import wx from gui_core.toolbars import BaseToolbar, BaseIcons from nviz.main import haveNviz from vdigit.main import haveVDigit from icons.icon import MetaIcon from core.utils import _ MapIcons = { 'query' : MetaIcon(img = 'info', label = _('Query raster/vector map(s)'), desc = _('Query selected raster/vector map(s)')), 'addBarscale': MetaIcon(img = 'scalebar-add', label = _('Show/hide scale bar')), 'addLegend' : MetaIcon(img = 'legend-add', label = _('Show/hide legend')), 'addNorthArrow': MetaIcon(img = 'north-arrow-add', label = _('Show/hide north arrow')), 'analyze' : MetaIcon(img = 'layer-raster-analyze', label = _('Analyze map'), desc = _('Measuring, profiling, histogramming, ...')), 'measure' : MetaIcon(img = 'measure-length', label = _('Measure distance')), 'profile' : MetaIcon(img = 'layer-raster-profile', label = _('Profile surface map')), 'scatter' : MetaIcon(img = 'layer-raster-profile', label = _("Create bivariate scatterplot of raster maps")), 'addText' : MetaIcon(img = 'text-add', label = _('Add text layer')), 'histogram' : MetaIcon(img = 'layer-raster-histogram', label = _('Create histogram of raster map')), 'vnet' : MetaIcon(img = 'vector-tools', label = _('Vector network analysis tool')), 'interact_scatter' : MetaIcon(img = 'layer-raster-profile', label = _("Interactive scatter plot tool")), } NvizIcons = { 'rotate' : MetaIcon(img = '3d-rotate', label = _('Rotate 3D scene'), desc = _('Drag with mouse to rotate 3D scene')), 'flyThrough': MetaIcon(img = 'flythrough', label = _('Fly-through mode'), desc = _('Drag with mouse, hold Ctrl down for different mode' ' or Shift to accelerate')), 'zoomIn' : BaseIcons['zoomIn'].SetLabel(desc = _('Click mouse to zoom')), 'zoomOut' : BaseIcons['zoomOut'].SetLabel(desc = _('Click mouse to unzoom')) } class MapToolbar(BaseToolbar): """!Map Display toolbar """ def __init__(self, parent, toolSwitcher): """!Map Display constructor @param parent reference to MapFrame """ BaseToolbar.__init__(self, parent=parent, toolSwitcher=toolSwitcher) # MapFrame self.InitToolbar(self._toolbarData()) self._default = self.pointer # optional tools choices = [ _('2D view'), ] self.toolId = { '2d' : 0 } if self.parent.GetLayerManager(): log = self.parent.GetLayerManager().GetLogWindow() if haveNviz: choices.append(_('3D view')) self.toolId['3d'] = 1 else: from nviz.main import errorMsg if self.parent.GetLayerManager(): log.WriteCmdLog(_('3D view mode not available')) log.WriteWarning(_('Reason: %s') % str(errorMsg)) self.toolId['3d'] = -1 if haveVDigit: choices.append(_('Digitize')) if self.toolId['3d'] > -1: self.toolId['vdigit'] = 2 else: self.toolId['vdigit'] = 1 else: from vdigit.main import errorMsg if self.parent.GetLayerManager(): log.WriteCmdLog(_('Vector digitizer not available')) log.WriteWarning(_('Reason: %s') % errorMsg) log.WriteLog(_('Note that the wxGUI\'s vector digitizer is currently disabled ' '(hopefully this will be fixed soon). ' 'Please keep an eye out for updated versions of GRASS. ' 'In the meantime you can use "v.digit" from the Develop Vector menu.'), wrap = 60) self.toolId['vdigit'] = -1 self.combo = wx.ComboBox(parent = self, id = wx.ID_ANY, choices = choices, style = wx.CB_READONLY, size = (110, -1)) self.combo.SetSelection(0) self.comboid = self.AddControl(self.combo) self.parent.Bind(wx.EVT_COMBOBOX, self.OnSelectTool, self.comboid) # realize the toolbar self.Realize() # workaround for Mac bug. May be fixed by 2.8.8, but not before then. self.combo.Hide() self.combo.Show() for tool in (self.pointer, self.query, self.pan, self.zoomIn, self.zoomOut): self.toolSwitcher.AddToolToGroup(group='mouseUse', toolbar=self, tool=tool) self.EnableTool(self.zoomBack, False) self.FixSize(width = 90) def _toolbarData(self): """!Toolbar data""" return self._getToolbarData((('displayMap', BaseIcons['display'], self.parent.OnDraw), ('renderMap', BaseIcons['render'], self.parent.OnRender), ('erase', BaseIcons['erase'], self.parent.OnErase), (None, ), ('pointer', BaseIcons['pointer'], self.parent.OnPointer, wx.ITEM_CHECK), ('query', MapIcons['query'], self.parent.OnQuery, wx.ITEM_CHECK), ('pan', BaseIcons['pan'], self.parent.OnPan, wx.ITEM_CHECK), ('zoomIn', BaseIcons['zoomIn'], self.parent.OnZoomIn, wx.ITEM_CHECK), ('zoomOut', BaseIcons['zoomOut'], self.parent.OnZoomOut, wx.ITEM_CHECK), ('zoomExtent', BaseIcons['zoomExtent'], self.parent.OnZoomToMap), ('zoomBack', BaseIcons['zoomBack'], self.parent.OnZoomBack), ('zoomMenu', BaseIcons['zoomMenu'], self.parent.OnZoomMenu), (None, ), ('analyze', MapIcons['analyze'], self.OnAnalyze), (None, ), ('overlay', BaseIcons['overlay'], self.OnDecoration), (None, ), ('saveFile', BaseIcons['saveFile'], self.parent.SaveToFile), ('printMap', BaseIcons['print'], self.parent.PrintMenu), (None, )) ) def InsertTool(self, data): """!Insert tool to toolbar @param data toolbar data""" data = self._getToolbarData(data) for tool in data: self.CreateTool(*tool) self.Realize() self.parent._mgr.GetPane('mapToolbar').BestSize(self.GetBestSize()) self.parent._mgr.Update() def RemoveTool(self, tool): """!Remove tool from toolbar @param tool tool id""" self.DeleteTool(tool) self.parent._mgr.GetPane('mapToolbar').BestSize(self.GetBestSize()) self.parent._mgr.Update() def ChangeToolsDesc(self, mode2d): """!Change description of zoom tools for 2D/3D view""" if mode2d: icons = BaseIcons else: icons = NvizIcons for i, data in enumerate(self._data): for tool in (('zoomIn', 'zoomOut')): if data[0] == tool: tmp = list(data) tmp[4] = icons[tool].GetDesc() self._data[i] = tuple(tmp) def OnSelectTool(self, event): """!Select / enable tool available in tools list """ tool = event.GetSelection() if tool == self.toolId['2d']: self.ExitToolbars() self.Enable2D(True) self.ChangeToolsDesc(mode2d = True) elif tool == self.toolId['3d'] and \ not (self.parent.MapWindow3D and self.parent.IsPaneShown('3d')): self.ExitToolbars() self.parent.AddNviz() elif tool == self.toolId['vdigit'] and \ not self.parent.GetToolbar('vdigit'): self.ExitToolbars() self.parent.AddToolbar("vdigit") self.parent.MapWindow.SetFocus() def OnAnalyze(self, event): """!Analysis tools menu """ self._onMenu(((MapIcons["measure"], self.parent.OnMeasure), (MapIcons["profile"], self.parent.OnProfile), (MapIcons["scatter"], self.parent.OnScatterplot), (MapIcons["histogram"], self.parent.OnHistogramPyPlot), (BaseIcons["histogramD"], self.parent.OnHistogram), (MapIcons["vnet"], self.parent.OnVNet), (MapIcons["interact_scatter"], self.parent.OnIScatt))) def OnDecoration(self, event): """!Decorations overlay menu """ self._onMenu(((MapIcons["addLegend"], lambda evt: self.parent.AddLegend()), (MapIcons["addBarscale"], lambda evt: self.parent.AddBarscale()), (MapIcons["addNorthArrow"], lambda evt: self.parent.AddArrow()), (MapIcons["addText"], self.parent.OnAddText))) def ExitToolbars(self): if self.parent.GetToolbar('vdigit'): self.parent.toolbars['vdigit'].OnExit() if self.parent.GetLayerManager() and \ self.parent.GetLayerManager().IsPaneShown('toolbarNviz'): self.parent.RemoveNviz() def Enable2D(self, enabled): """!Enable/Disable 2D display mode specific tools""" for tool in (self.zoomMenu, self.analyze, self.printMap): self.EnableTool(tool, enabled)