123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392 |
- """!
- @package gdialogs.py
- @brief Various dialogs used in wxGUI.
- List of classes:
- - ElementDialog
- - LocationDialog
- - MapsetDialog
- - NewVectorDialog
- - SavedRegion
- - DecorationDialog
- - TextLayerDialog
- - LoadMapLayersDialog
- - MultiImportDialog
- - LayersList (used by MultiImport)
- - SetOpacityDialog
- (C) 2008-2009 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 Martin Landa <landa.martin gmail.com>
- """
- import os
- import sys
- import re
- import glob
- import wx
- import wx.lib.filebrowsebutton as filebrowse
- import wx.lib.mixins.listctrl as listmix
- from grass.script import core as grass
- import gcmd
- import globalvar
- import gselect
- import menuform
- import utils
- from preferences import globalSettings as UserSettings
- class ElementDialog(wx.Dialog):
- """!General dialog to choose given element (location, mapset, vector map, etc.)"""
- def __init__(self, parent, title, label, id = wx.ID_ANY,
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
- **kwargs):
-
- wx.Dialog.__init__(self, parent, id, title, style = style, **kwargs)
-
- self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
-
- self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
- self.btnOK = wx.Button(parent = self.panel, id = wx.ID_OK)
- self.btnOK.SetDefault()
- self.btnOK.Enable(False)
-
- self.label = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = label)
-
- self.element = None # must be defined
-
- self.__Layout()
-
- def PostInit(self):
- self.element.SetFocus()
- self.element.Bind(wx.EVT_TEXT, self.OnElement)
-
- def OnElement(self, event):
- """!Name for vector map layer given"""
- if len(event.GetString()) > 0:
- self.btnOK.Enable(True)
- else:
- self.btnOK.Enable(False)
-
- def __Layout(self):
- """!Do layout"""
- self.sizer = wx.BoxSizer(wx.VERTICAL)
-
- self.dataSizer = wx.BoxSizer(wx.VERTICAL)
- self.dataSizer.Add(self.label, proportion=0,
- flag=wx.ALL, border=1)
-
- # buttons
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(self.btnCancel)
- btnSizer.AddButton(self.btnOK)
- btnSizer.Realize()
-
- self.sizer.Add(item=self.dataSizer, proportion=1,
- flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
-
- self.sizer.Add(item=btnSizer, proportion=0,
- flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
-
- def GetElement(self):
- """!Return (mapName, overwrite)"""
- return self.element.GetValue()
-
- class LocationDialog(ElementDialog):
- """!Dialog used to select location"""
- def __init__(self, parent, title = _("Select GRASS location and mapset"), id = wx.ID_ANY):
- ElementDialog.__init__(self, parent, title, label = _("Name of GRASS location:"))
- self.element = gselect.LocationSelect(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_GSELECT_SIZE)
- self.element1 = gselect.MapsetSelect(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_GSELECT_SIZE,
- setItems = False)
-
- self.PostInit()
-
- self.__Layout()
- self.SetMinSize(self.GetSize())
- def __Layout(self):
- """!Do layout"""
- self.dataSizer.Add(self.element, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.dataSizer.Add(wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = _("Name of mapset:")), proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
- self.dataSizer.Add(self.element1, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.panel.SetSizer(self.sizer)
- self.sizer.Fit(self)
- def OnElement(self, event):
- """!Select mapset given location name"""
- location = event.GetString()
-
- if location:
- dbase = grass.gisenv()['GISDBASE']
- self.element1.SetItems(utils.GetListOfMapsets(dbase, location, selectable = True))
- self.element1.SetSelection(0)
- mapset = self.element1.GetStringSelection()
-
- if location and mapset:
- self.btnOK.Enable(True)
- else:
- self.btnOK.Enable(False)
- def GetValues(self):
- """!Get location, mapset"""
- return (self.GetElement(), self.element1.GetStringSelection())
-
- class MapsetDialog(ElementDialog):
- """!Dialog used to select mapset"""
- def __init__(self, parent, title = _("Select mapset in GRASS location"),
- location = None, id = wx.ID_ANY):
- ElementDialog.__init__(self, parent, title, label = _("Name of mapset:"))
- if location:
- self.SetTitle(self.GetTitle() + '<%s>' % location)
- else:
- self.SetTitle(self.GetTitle() + '<%s>' % grass.gisenv()['LOCATION_NAME'])
-
- self.element = gselect.MapsetSelect(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_GSELECT_SIZE)
-
- self.PostInit()
-
- self.__Layout()
- self.SetMinSize(self.GetSize())
- def __Layout(self):
- """!Do layout"""
- self.dataSizer.Add(self.element, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.panel.SetSizer(self.sizer)
- self.sizer.Fit(self)
- def GetMapset(self):
- return self.GetElement()
-
- class NewVectorDialog(ElementDialog):
- """!Dialog for creating new vector map"""
- def __init__(self, parent, id, title, disableAdd=False, disableTable=False,
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
-
- ElementDialog.__init__(self, parent, title, label = _("Name for new vector map:"))
-
- self.element = gselect.Select(parent=self.panel, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
- type='vector', mapsets=[grass.gisenv()['MAPSET'],])
-
- self.table = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
- label = _("Create attribute table"))
- self.table.SetValue(True)
- if disableTable:
- self.table.Enable(False)
-
- self.addbox = wx.CheckBox(parent = self.panel,
- label = _('Add created map into layer tree'), style = wx.NO_BORDER)
- if disableAdd:
- self.addbox.SetValue(True)
- self.addbox.Enable(False)
- else:
- self.addbox.SetValue(UserSettings.Get(group = 'cmd', key = 'addNewLayer', subkey = 'enabled'))
-
- self.PostInit()
-
- self.__Layout()
- self.SetMinSize(self.GetSize())
-
- def OnMapName(self, event):
- """!Name for vector map layer given"""
- self.OnElement(event)
-
- def __Layout(self):
- """!Do layout"""
- self.dataSizer.Add(self.element, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.dataSizer.Add(self.table, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.dataSizer.AddSpacer(5)
-
- self.dataSizer.Add(item=self.addbox, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=1)
-
- self.panel.SetSizer(self.sizer)
- self.sizer.Fit(self)
- def GetName(self):
- """!Return (mapName, overwrite)"""
- return self.GetElement().split('@', 1)[0]
-
- def CreateNewVector(parent, cmd, title=_('Create new vector map'),
- exceptMap=None, log=None, disableAdd=False, disableTable=False):
- """!Create new vector map layer
- @cmd cmd (prog, **kwargs)
-
- @return tuple (name of create vector map, add to layer tree)
- @return None of failure
- """
- dlg = NewVectorDialog(parent, wx.ID_ANY, title,
- disableAdd, disableTable)
- if dlg.ShowModal() == wx.ID_OK:
- outmap = dlg.GetName()
- if outmap == exceptMap:
- wx.MessageBox(parent=parent,
- message=_("Unable to create vector map <%s>.") % outmap,
- caption=_("Error"),
- style=wx.ID_OK | wx.ICON_ERROR | wx.CENTRE)
- return False
-
- if outmap == '': # should not happen
- return False
-
- cmd[1][cmd[2]] = outmap
-
- try:
- listOfVectors = grass.list_grouped('vect')[grass.gisenv()['MAPSET']]
- except KeyError:
- listOfVectors = []
-
- overwrite = False
- if not UserSettings.Get(group='cmd', key='overwrite', subkey='enabled') and \
- outmap in listOfVectors:
- dlgOw = wx.MessageDialog(parent, message=_("Vector map <%s> already exists "
- "in the current mapset. "
- "Do you want to overwrite it?") % outmap,
- caption=_("Overwrite?"),
- style=wx.YES_NO | wx.YES_DEFAULT | wx.ICON_QUESTION)
- if dlgOw.ShowModal() == wx.ID_YES:
- overwrite = True
- else:
- dlgOw.Destroy()
- return False
- if UserSettings.Get(group='cmd', key='overwrite', subkey='enabled') is True:
- overwrite = True
-
- try:
- gcmd.RunCommand(prog = cmd[0],
- overwrite = overwrite,
- **cmd[1])
- except gcmd.CmdError, e:
- e.Show()
- return None
- #
- # create attribute table
- #
- if dlg.table.IsEnabled() and dlg.table.IsChecked():
- key = UserSettings.Get(group='atm', key='keycolumn', subkey='value')
- sql = 'CREATE TABLE %s (%s INTEGER)' % (outmap, key)
-
- gcmd.RunCommand('db.execute',
- quiet = True,
- parent = parent,
- input = '-',
- stdin = sql)
- gcmd.RunCommand('v.db.connect',
- quiet = True,
- parent = parent,
- map = outmap,
- table = outmap,
- key = key,
- layer = '1')
-
- # return fully qualified map name
- if '@' not in outmap:
- outmap += '@' + grass.gisenv()['MAPSET']
- if log:
- log.WriteLog(_("New vector map <%s> created") % outmap)
- return (outmap, dlg.addbox.IsChecked())
-
- return (None, dlg.addbox.IsChecked())
- class SavedRegion(wx.Dialog):
- def __init__(self, parent, id = wx.ID_ANY, title="", loadsave='load',
- **kwargs):
- """!Loading and saving of display extents to saved region file
- @param loadsave load or save region?
- """
- wx.Dialog.__init__(self, parent, id, title, **kwargs)
- self.loadsave = loadsave
- self.wind = ''
-
- sizer = wx.BoxSizer(wx.VERTICAL)
-
- box = wx.BoxSizer(wx.HORIZONTAL)
- label = wx.StaticText(parent=self, id=wx.ID_ANY)
- box.Add(item=label, proportion=0, flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
- if loadsave == 'load':
- label.SetLabel(_("Load region:"))
- selection = gselect.Select(parent=self, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
- type='windows')
- elif loadsave == 'save':
- label.SetLabel(_("Save region:"))
- selection = gselect.Select(parent=self, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
- type='windows', mapsets = [grass.gisenv()['MAPSET']])
-
- box.Add(item=selection, proportion=0, flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
- selection.SetFocus()
- selection.Bind(wx.EVT_TEXT, self.OnRegion)
-
- sizer.Add(item=box, proportion=0, flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL,
- border=5)
-
- line = wx.StaticLine(parent=self, id=wx.ID_ANY, size=(20, -1), style=wx.LI_HORIZONTAL)
- sizer.Add(item=line, proportion=0,
- flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.LEFT|wx.RIGHT, border=5)
-
- btnsizer = wx.StdDialogButtonSizer()
-
- btn = wx.Button(parent = self, id = wx.ID_OK)
- btn.SetDefault()
- btnsizer.AddButton(btn)
-
- btn = wx.Button(parent = self, id = wx.ID_CANCEL)
- btnsizer.AddButton(btn)
- btnsizer.Realize()
-
- sizer.Add(item=btnsizer, proportion=0, flag=wx.ALIGN_RIGHT | wx.ALL, border=5)
-
- self.SetSizer(sizer)
- sizer.Fit(self)
- self.Layout()
-
- def OnRegion(self, event):
- self.wind = event.GetString()
-
- class DecorationDialog(wx.Dialog):
- """
- Controls setting options and displaying/hiding map overlay decorations
- """
- def __init__(self, parent, ovlId, title, cmd, name=None,
- pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE,
- checktxt='', ctrltxt=''):
- wx.Dialog.__init__(self, parent, wx.ID_ANY, title, pos, size, style)
- self.ovlId = ovlId # PseudoDC id
- self.cmd = cmd
- self.name = name # overlay name
- self.parent = parent # MapFrame
- sizer = wx.BoxSizer(wx.VERTICAL)
- box = wx.BoxSizer(wx.HORIZONTAL)
- self.chkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=checktxt)
- if self.parent.Map.GetOverlay(self.ovlId) is None:
- self.chkbox.SetValue(True)
- else:
- self.chkbox.SetValue(self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive())
- box.Add(item=self.chkbox, proportion=0,
- flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
- sizer.Add(item=box, proportion=0,
- flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
- box = wx.BoxSizer(wx.HORIZONTAL)
- optnbtn = wx.Button(parent=self, id=wx.ID_ANY, label=_("Set options"))
- box.Add(item=optnbtn, proportion=0, flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
- sizer.Add(item=box, proportion=0,
- flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
- box = wx.BoxSizer(wx.HORIZONTAL)
- label = wx.StaticText(parent=self, id=wx.ID_ANY,
- label=_("Drag %s with mouse in pointer mode to position.\n"
- "Double-click to change options." % ctrltxt))
- if self.name == 'legend':
- label.SetLabel(label.GetLabel() + _('\nDefine raster map name for legend in '
- 'properties dialog.'))
- box.Add(item=label, proportion=0,
- flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
- sizer.Add(item=box, proportion=0,
- flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
- line = wx.StaticLine(parent=self, id=wx.ID_ANY, size=(20,-1), style=wx.LI_HORIZONTAL)
- sizer.Add(item=line, proportion=0,
- flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
- # buttons
- btnsizer = wx.StdDialogButtonSizer()
- self.btnOK = wx.Button(parent=self, id=wx.ID_OK)
- self.btnOK.SetDefault()
- if self.name == 'legend':
- self.btnOK.Enable(False)
- btnsizer.AddButton(self.btnOK)
- btnCancel = wx.Button(parent=self, id=wx.ID_CANCEL)
- btnsizer.AddButton(btnCancel)
- btnsizer.Realize()
- sizer.Add(item=btnsizer, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
- #
- # bindings
- #
- self.Bind(wx.EVT_BUTTON, self.OnOptions, optnbtn)
- self.Bind(wx.EVT_BUTTON, self.OnCancel, btnCancel)
- self.Bind(wx.EVT_BUTTON, self.OnOK, self.btnOK)
- self.SetSizer(sizer)
- sizer.Fit(self)
- # create overlay if doesn't exist
- self._CreateOverlay()
- if len(self.parent.MapWindow.overlays[self.ovlId]['cmd']) > 1:
- mapName = utils.GetLayerNameFromCmd(self.parent.MapWindow.overlays[self.ovlId]['cmd'])
- if self.parent.MapWindow.overlays[self.ovlId]['propwin'] is None and mapName:
- # build properties dialog
- menuform.GUI().ParseCommand(cmd=self.cmd,
- completed=(self.GetOptData, self.name, ''),
- parentframe=self.parent, show=False)
- if mapName:
- # enable 'OK' button
- self.btnOK.Enable()
- if name == 'legend':
- # set title
- self.SetTitle(_('Legend of raster map <%s>') % \
- mapName)
-
- def _CreateOverlay(self):
- if not self.parent.Map.GetOverlay(self.ovlId):
- overlay = self.parent.Map.AddOverlay(id=self.ovlId, type=self.name,
- command=self.cmd,
- l_active=False, l_render=False, l_hidden=True)
- self.parent.MapWindow.overlays[self.ovlId] = {}
- self.parent.MapWindow.overlays[self.ovlId] = { 'layer' : overlay,
- 'params' : None,
- 'propwin' : None,
- 'cmd' : self.cmd,
- 'coords': (10, 10),
- 'pdcType': 'image' }
- else:
- if self.parent.MapWindow.overlays[self.ovlId]['propwin'] == None:
- return
- self.parent.MapWindow.overlays[self.ovlId]['propwin'].get_dcmd = self.GetOptData
- def OnOptions(self, event):
- """ self.SetSizer(sizer)
- sizer.Fit(self)
- Sets option for decoration map overlays
- """
- if self.parent.MapWindow.overlays[self.ovlId]['propwin'] is None:
- # build properties dialog
- menuform.GUI().ParseCommand(cmd=self.cmd,
- completed=(self.GetOptData, self.name, ''),
- parentframe=self.parent)
- else:
- if self.parent.MapWindow.overlays[self.ovlId]['propwin'].IsShown():
- self.parent.MapWindow.overlays[self.ovlId]['propwin'].SetFocus()
- else:
- self.parent.MapWindow.overlays[self.ovlId]['propwin'].Show()
- def OnCancel(self, event):
- """!Cancel dialog"""
- self.parent.dialogs['barscale'] = None
- self.Destroy()
- def OnOK(self, event):
- """!Button 'OK' pressed"""
- # enable or disable overlay
- self.parent.Map.GetOverlay(self.ovlId).SetActive(self.chkbox.IsChecked())
- # update map
- self.parent.MapWindow.UpdateMap()
- # close dialog
- self.OnCancel(None)
- def GetOptData(self, dcmd, layer, params, propwin):
- """!Process decoration layer data"""
- # update layer data
- if params:
- self.parent.MapWindow.overlays[self.ovlId]['params'] = params
- if dcmd:
- self.parent.MapWindow.overlays[self.ovlId]['cmd'] = dcmd
- self.parent.MapWindow.overlays[self.ovlId]['propwin'] = propwin
- # change parameters for item in layers list in render.Map
- # "Use mouse..." (-m) flag causes GUI freeze and is pointless here, trac #119
-
- try:
- self.parent.MapWindow.overlays[self.ovlId]['cmd'].remove('-m')
- except ValueError:
- pass
-
- self.parent.Map.ChangeOverlay(id=self.ovlId, type=self.name,
- command=self.parent.MapWindow.overlays[self.ovlId]['cmd'],
- l_active=self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive(),
- l_render=False, l_hidden=True)
- if self.name == 'legend':
- if params and not self.btnOK.IsEnabled():
- self.btnOK.Enable()
-
- class TextLayerDialog(wx.Dialog):
- """
- Controls setting options and displaying/hiding map overlay decorations
- """
- def __init__(self, parent, ovlId, title, name='text',
- pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE):
- wx.Dialog.__init__(self, parent, wx.ID_ANY, title, pos, size, style)
- from wx.lib.expando import ExpandoTextCtrl, EVT_ETC_LAYOUT_NEEDED
- self.ovlId = ovlId
- self.parent = parent
- if self.ovlId in self.parent.MapWindow.textdict.keys():
- self.currText = self.parent.MapWindow.textdict[self.ovlId]['text']
- self.currFont = self.parent.MapWindow.textdict[self.ovlId]['font']
- self.currClr = self.parent.MapWindow.textdict[self.ovlId]['color']
- self.currRot = self.parent.MapWindow.textdict[self.ovlId]['rotation']
- self.currCoords = self.parent.MapWindow.textdict[self.ovlId]['coords']
- else:
- self.currClr = wx.BLACK
- self.currText = ''
- self.currFont = self.GetFont()
- self.currRot = 0.0
- self.currCoords = [10, 10, 10, 10]
- self.sizer = wx.BoxSizer(wx.VERTICAL)
- box = wx.GridBagSizer(vgap=5, hgap=5)
- # show/hide
- self.chkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, \
- label='Show text object')
- if self.parent.Map.GetOverlay(self.ovlId) is None:
- self.chkbox.SetValue(True)
- else:
- self.chkbox.SetValue(self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive())
- box.Add(item=self.chkbox, span=(1,2),
- flag=wx.ALIGN_LEFT|wx.ALL, border=5,
- pos=(0, 0))
- # text entry
- label = wx.StaticText(parent=self, id=wx.ID_ANY, label=_("Enter text:"))
- box.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(1, 0))
- self.textentry = ExpandoTextCtrl(parent=self, id=wx.ID_ANY, value="", size=(300,-1))
- self.textentry.SetFont(self.currFont)
- self.textentry.SetForegroundColour(self.currClr)
- self.textentry.SetValue(self.currText)
- # get rid of unneeded scrollbar when text box first opened
- self.textentry.SetClientSize((300,-1))
-
- box.Add(item=self.textentry,
- pos=(1, 1))
- # rotation
- label = wx.StaticText(parent=self, id=wx.ID_ANY, label=_("Rotation:"))
- box.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(2, 0))
- self.rotation = wx.SpinCtrl(parent=self, id=wx.ID_ANY, value="", pos=(30, 50),
- size=(75,-1), style=wx.SP_ARROW_KEYS)
- self.rotation.SetRange(-360, 360)
- self.rotation.SetValue(int(self.currRot))
- box.Add(item=self.rotation,
- flag=wx.ALIGN_RIGHT,
- pos=(2, 1))
- # font
- fontbtn = wx.Button(parent=self, id=wx.ID_ANY, label=_("Set font"))
- box.Add(item=fontbtn,
- flag=wx.ALIGN_RIGHT,
- pos=(3, 1))
- self.sizer.Add(item=box, proportion=1,
- flag=wx.ALL, border=10)
- # note
- box = wx.BoxSizer(wx.HORIZONTAL)
- label = wx.StaticText(parent=self, id=wx.ID_ANY,
- label=_("Drag text with mouse in pointer mode "
- "to position.\nDouble-click to change options"))
- box.Add(item=label, proportion=0,
- flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
- self.sizer.Add(item=box, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER | wx.ALL, border=5)
- line = wx.StaticLine(parent=self, id=wx.ID_ANY,
- size=(20,-1), style=wx.LI_HORIZONTAL)
- self.sizer.Add(item=line, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTRE | wx.ALL, border=5)
- btnsizer = wx.StdDialogButtonSizer()
- btn = wx.Button(parent=self, id=wx.ID_OK)
- btn.SetDefault()
- btnsizer.AddButton(btn)
- btn = wx.Button(parent=self, id=wx.ID_CANCEL)
- btnsizer.AddButton(btn)
- btnsizer.Realize()
- self.sizer.Add(item=btnsizer, proportion=0,
- flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
- self.SetSizer(self.sizer)
- self.sizer.Fit(self)
- # bindings
- self.Bind(EVT_ETC_LAYOUT_NEEDED, self.OnRefit, self.textentry)
- self.Bind(wx.EVT_BUTTON, self.OnSelectFont, fontbtn)
- self.Bind(wx.EVT_TEXT, self.OnText, self.textentry)
- self.Bind(wx.EVT_SPINCTRL, self.OnRotation, self.rotation)
- def OnRefit(self, event):
- """!Resize text entry to match text"""
- self.sizer.Fit(self)
- def OnText(self, event):
- """!Change text string"""
- self.currText = event.GetString()
- def OnRotation(self, event):
- """!Change rotation"""
- self.currRot = event.GetInt()
- event.Skip()
- def OnSelectFont(self, event):
- """!Change font"""
- data = wx.FontData()
- data.EnableEffects(True)
- data.SetColour(self.currClr) # set colour
- data.SetInitialFont(self.currFont)
- dlg = wx.FontDialog(self, data)
- if dlg.ShowModal() == wx.ID_OK:
- data = dlg.GetFontData()
- self.currFont = data.GetChosenFont()
- self.currClr = data.GetColour()
- self.textentry.SetFont(self.currFont)
- self.textentry.SetForegroundColour(self.currClr)
- self.Layout()
- dlg.Destroy()
- def GetValues(self):
- """!Get text properties"""
- return { 'text' : self.currText,
- 'font' : self.currFont,
- 'color' : self.currClr,
- 'rotation' : self.currRot,
- 'coords' : self.currCoords,
- 'active' : self.chkbox.IsChecked() }
- class LoadMapLayersDialog(wx.Dialog):
- """!Load selected map layers (raster, vector) into layer tree"""
- def __init__(self, parent, title, style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
- wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
- self.parent = parent # GMFrame
- #
- # dialog body
- #
- self.bodySizer = self.__createDialogBody()
- # update list of layer to be loaded
- self.map_layers = [] # list of map layers (full list type/mapset)
- self.LoadMapLayers(self.layerType.GetStringSelection()[:4],
- self.mapset.GetStringSelection())
- #
- # buttons
- #
- btnCancel = wx.Button(self, wx.ID_CANCEL)
- btnOk = wx.Button(self, wx.ID_OK, _("&Load") )
- btnOk.SetDefault()
-
- #
- # sizers & do layout
- #
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnCancel)
- btnSizer.AddButton(btnOk)
- btnSizer.Realize()
-
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- mainSizer.Add(item=self.bodySizer, proportion=1,
- flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(item=btnSizer, proportion=0,
- flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
- # set dialog min size
- self.SetMinSize(self.GetSize())
-
- def __createDialogBody(self):
- bodySizer = wx.GridBagSizer(vgap=3, hgap=3)
- bodySizer.AddGrowableCol(1)
- bodySizer.AddGrowableRow(3)
-
- # layer type
- bodySizer.Add(item=wx.StaticText(parent=self, label=_("Map layer type:")),
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(0,0))
- self.layerType = wx.Choice(parent=self, id=wx.ID_ANY,
- choices=['raster', 'vector'], size=(100,-1))
- self.layerType.SetSelection(0)
- bodySizer.Add(item=self.layerType,
- pos=(0,1))
- # select toggle
- self.toggle = wx.CheckBox(parent=self, id=wx.ID_ANY,
- label=_("Select toggle"))
- self.toggle.SetValue(True)
- bodySizer.Add(item=self.toggle,
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(0,2))
-
- # mapset filter
- bodySizer.Add(item=wx.StaticText(parent=self, label=_("Mapset:")),
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(1,0))
- self.mapset = gselect.MapsetSelect(parent = self)
- self.mapset.SetStringSelection(grass.gisenv()['MAPSET'])
- bodySizer.Add(item=self.mapset,
- pos=(1,1), span=(1, 2))
- # map name filter
- bodySizer.Add(item=wx.StaticText(parent=self, label=_("Filter:")),
- flag=wx.ALIGN_CENTER_VERTICAL,
- pos=(2,0))
- self.filter = wx.TextCtrl(parent=self, id=wx.ID_ANY,
- value="",
- size=(250,-1))
- bodySizer.Add(item=self.filter,
- flag=wx.EXPAND,
- pos=(2,1), span=(1, 2))
- # layer list
- bodySizer.Add(item=wx.StaticText(parent=self, label=_("List of maps:")),
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_TOP,
- pos=(3,0))
- self.layers = wx.CheckListBox(parent=self, id=wx.ID_ANY,
- size=(250, 100),
- choices=[])
- bodySizer.Add(item=self.layers,
- flag=wx.EXPAND,
- pos=(3,1), span=(1, 2))
- # bindings
- self.layerType.Bind(wx.EVT_CHOICE, self.OnChangeParams)
- self.mapset.Bind(wx.EVT_COMBOBOX, self.OnChangeParams)
- self.layers.Bind(wx.EVT_RIGHT_DOWN, self.OnMenu)
- self.filter.Bind(wx.EVT_TEXT, self.OnFilter)
- self.toggle.Bind(wx.EVT_CHECKBOX, self.OnToggle)
- return bodySizer
- def LoadMapLayers(self, type, mapset):
- """!Load list of map layers
- @param type layer type ('raster' or 'vector')
- @param mapset mapset name
- """
- try:
- self.map_layers = grass.mlist_grouped(type=type, mapset=mapset)[mapset]
- except KeyError:
- self.map_layers = []
-
- self.layers.Set(self.map_layers)
-
- # check all items by default
- for item in range(self.layers.GetCount()):
- self.layers.Check(item)
- def OnChangeParams(self, event):
- """!Filter parameters changed by user"""
- # update list of layer to be loaded
- self.LoadMapLayers(self.layerType.GetStringSelection()[:4],
- self.mapset.GetStringSelection())
-
- event.Skip()
- def OnMenu(self, event):
- """!Table description area, context menu"""
- if not hasattr(self, "popupID1"):
- self.popupDataID1 = wx.NewId()
- self.popupDataID2 = wx.NewId()
- self.popupDataID3 = wx.NewId()
- self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
- self.Bind(wx.EVT_MENU, self.OnSelectInvert, id=self.popupDataID2)
- self.Bind(wx.EVT_MENU, self.OnDeselectAll, id=self.popupDataID3)
- # generate popup-menu
- menu = wx.Menu()
- menu.Append(self.popupDataID1, _("Select all"))
- menu.Append(self.popupDataID2, _("Invert selection"))
- menu.Append(self.popupDataID3, _("Deselect all"))
- self.PopupMenu(menu)
- menu.Destroy()
- def OnSelectAll(self, event):
- """!Select all map layer from list"""
- for item in range(self.layers.GetCount()):
- self.layers.Check(item, True)
- def OnSelectInvert(self, event):
- """!Invert current selection"""
- for item in range(self.layers.GetCount()):
- if self.layers.IsChecked(item):
- self.layers.Check(item, False)
- else:
- self.layers.Check(item, True)
-
- def OnDeselectAll(self, event):
- """!Select all map layer from list"""
- for item in range(self.layers.GetCount()):
- self.layers.Check(item, False)
- def OnFilter(self, event):
- """!Apply filter for map names"""
- if len(event.GetString()) == 0:
- self.layers.Set(self.map_layers)
- return
- list = []
- for layer in self.map_layers:
- try:
- if re.compile('^' + event.GetString()).search(layer):
- list.append(layer)
- except:
- pass
- self.layers.Set(list)
- self.OnSelectAll(None)
-
- event.Skip()
- def OnToggle(self, event):
- """!Select toggle (check or uncheck all layers)"""
- check = event.Checked()
- for item in range(self.layers.GetCount()):
- self.layers.Check(item, check)
- event.Skip()
-
- def GetMapLayers(self):
- """!Return list of checked map layers"""
- layerNames = []
- for indx in self.layers.GetSelections():
- # layers.append(self.layers.GetStringSelec(indx))
- pass
- # return fully qualified map names
- mapset = self.mapset.GetStringSelection()
- for item in range(self.layers.GetCount()):
- if not self.layers.IsChecked(item):
- continue
- layerNames.append(self.layers.GetString(item) + '@' + mapset)
- return layerNames
-
- def GetLayerType(self):
- """!Get selected layer type"""
- return self.layerType.GetStringSelection()
-
- class MultiImportDialog(wx.Dialog):
- """!Import dxf layers"""
- def __init__(self, parent, type,
- id=wx.ID_ANY, title=_("Multiple import"),
- link = False,
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
- self.parent = parent # GMFrame
- self.inputType = type
- self.link = link # Link or import data (only for GDAL/OGR)
-
- wx.Dialog.__init__(self, parent, id, title, style=style)
- self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
- if self.inputType == 'dxf':
- self.inputTitle = _("Input DXF file")
- else:
- self.inputTitle = _("Input directory")
-
- self.inputBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
- label=" %s " % self.inputTitle)
- self.layerBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
- label=_(" List of %s layers ") % self.inputType.upper())
- #
- # input
- #
- if self.inputType == 'dxf':
- self.inputText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Choose DXF file:"))
- self.input = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
- size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
- dialogTitle=_('Choose DXF file to import'),
- buttonText=_('Browse'),
- startDirectory=os.getcwd(), fileMode=0,
- changeCallback=self.OnSetInput,
- fileMask="*.dxf")
- else:
- self.inputText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Choose directory:"))
- self.input = filebrowse.DirBrowseButton(parent=self.panel, id=wx.ID_ANY,
- size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
- dialogTitle=_('Choose input directory'),
- buttonText=_('Browse'),
- startDirectory=os.getcwd(),
- changeCallback=self.OnSetInput)
- self.formatText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Select file extension:"))
- self.format = wx.TextCtrl(parent=self.panel, id=wx.ID_ANY, size=(100, -1),
- value="")
- if self.inputType == 'gdal':
- self.format.SetValue('tif')
- else: # ogr
- self.format.SetValue('shp')
- self.format.Bind(wx.EVT_TEXT, self.OnSetInput)
-
- #
- # list of layers
- #
- self.list = LayersList(self.panel)
- self.list.LoadData()
- self.add = wx.CheckBox(parent=self.panel, id=wx.ID_ANY)
- if link:
- self.add.SetLabel(_("Add linked layers into layer tree"))
- else:
- self.add.SetLabel(_("Add imported layers into layer tree"))
- if not link and self.inputType in ('gdal', 'ogr'):
- self.overrideCheck = wx.CheckBox(parent=self.panel, id=wx.ID_ANY,
- label=_("Override projection (use location's projection)"))
- self.overrideCheck.SetValue(True)
-
- self.add.SetValue(UserSettings.Get(group='cmd', key='addNewLayer', subkey='enabled'))
- self.overwrite = wx.CheckBox(parent=self.panel, id=wx.ID_ANY,
- label=_("Allow output files to overwrite existing files"))
- self.overwrite.SetValue(UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'))
-
- #
- # buttons
- #
- # cancel
- self.btn_cancel = wx.Button(parent=self.panel, id=wx.ID_CANCEL)
- self.btn_cancel.SetToolTipString(_("Close dialog"))
- self.btn_cancel.Bind(wx.EVT_BUTTON, self.OnCancel)
- # run
- if link:
- self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Link"))
- self.btn_run.SetToolTipString(_("Link selected layers"))
- else:
- self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Import"))
- self.btn_run.SetToolTipString(_("Import selected layers"))
- self.btn_run.SetDefault()
- self.btn_run.Enable(False)
- self.btn_run.Bind(wx.EVT_BUTTON, self.OnRun)
-
- self.__doLayout()
- self.Layout()
- def __doLayout(self):
- dialogSizer = wx.BoxSizer(wx.VERTICAL)
- #
- # input
- #
- inputSizer = wx.StaticBoxSizer(self.inputBox, wx.HORIZONTAL)
-
- gridSizer = wx.FlexGridSizer(cols=2, vgap=5, hgap=5)
-
- gridSizer.Add(item=self.inputText,
- flag=wx.ALIGN_CENTER_VERTICAL)
- gridSizer.AddGrowableCol(1)
- gridSizer.Add(item=self.input,
- flag=wx.EXPAND | wx.ALL)
-
- if self.inputType != 'dxf':
- gridSizer.Add(item=self.formatText,
- flag=wx.ALIGN_CENTER_VERTICAL)
- gridSizer.Add(item=self.format)
-
- inputSizer.Add(item=gridSizer, proportion=1,
- flag=wx.EXPAND | wx.ALL)
-
- dialogSizer.Add(item=inputSizer, proportion=0,
- flag=wx.ALL | wx.EXPAND, border=5)
- #
- # list of DXF layers
- #
- layerSizer = wx.StaticBoxSizer(self.layerBox, wx.HORIZONTAL)
- layerSizer.Add(item=self.list, proportion=1,
- flag=wx.ALL | wx.EXPAND, border=5)
-
- dialogSizer.Add(item=layerSizer, proportion=1,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
- if hasattr(self, "overrideCheck"):
- dialogSizer.Add(item=self.overrideCheck, proportion=0,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
- dialogSizer.Add(item=self.overwrite, proportion=0,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
-
- dialogSizer.Add(item=self.add, proportion=0,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
- #
- # buttons
- #
- btnsizer = wx.BoxSizer(orient=wx.HORIZONTAL)
-
- btnsizer.Add(item=self.btn_cancel, proportion=0,
- flag=wx.ALL | wx.ALIGN_CENTER,
- border=10)
-
- btnsizer.Add(item=self.btn_run, proportion=0,
- flag=wx.ALL | wx.ALIGN_CENTER,
- border=10)
-
- dialogSizer.Add(item=btnsizer, proportion=0,
- flag=wx.ALIGN_CENTER)
-
- # dialogSizer.SetSizeHints(self.panel)
- self.panel.SetAutoLayout(True)
- self.panel.SetSizer(dialogSizer)
- dialogSizer.Fit(self.panel)
-
- self.Layout()
- # auto-layout seems not work here - FIXME
- self.SetMinSize((globalvar.DIALOG_GSELECT_SIZE[0] + 175, 400))
- width = self.GetSize()[0]
- self.list.SetColumnWidth(col=1, width=width/2 - 50)
-
- def OnCancel(self, event=None):
- """!Close dialog"""
- self.Close()
- def OnRun(self, event):
- """!Import/Link data (each layes as separate vector map)"""
- data = self.list.GetLayers()
-
- # hide dialog
- self.Hide()
-
- for layer, output in data:
- if self.inputType == 'dxf':
- cmd = ['v.in.dxf',
- 'input=%s' % self.input.GetValue(),
- 'layers=%s' % layer,
- 'output=%s' % output]
- elif self.inputType == 'ogr':
- if self.link:
- cmd = ['v.external',
- 'dsn=%s' % os.path.join(self.input.GetValue()),
- 'output=%s' % output,
- 'layer=%s' % layer.rstrip('.' + self.format.GetValue())
- ]
- else:
- cmd = ['v.in.ogr',
- 'dsn=%s' % (os.path.join(self.input.GetValue(), layer)),
- 'output=%s' % output]
- else: # gdal
- if self.link:
- cmd = ['r.external',
- 'input=%s' % (os.path.join(self.input.GetValue(), layer)),
- 'output=%s' % output]
- else:
- cmd = ['r.in.gdal',
- 'input=%s' % (os.path.join(self.input.GetValue(), layer)),
- 'output=%s' % output]
- if self.overwrite.IsChecked():
- cmd.append('--overwrite')
-
- if hasattr(self, "overrideCheck") and self.overrideCheck.IsChecked():
- cmd.append('-o')
-
- if UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'):
- cmd.append('--overwrite')
-
- # run in Layer Manager
- self.parent.goutput.RunCmd(cmd, switchPage=True)
- if self.add.IsChecked():
- maptree = self.parent.curr_page.maptree
- for layer, output in data:
- if '@' not in output:
- name = output + '@' + grass.gisenv()['MAPSET']
- else:
- name = output
- # add imported layers into layer tree
- if self.inputType == 'gdal':
- cmd = ['d.rast',
- 'map=%s' % name]
- if UserSettings.Get(group='cmd', key='rasterOverlay', subkey='enabled'):
- cmd.append('-o')
- maptree.AddLayer(ltype='raster',
- lname=name,
- lcmd=cmd)
- else:
- maptree.AddLayer(ltype='vector',
- lname=name,
- lcmd=['d.vect',
- 'map=%s' % name])
-
- ### wx.CallAfter(self.parent.notebook.SetSelection, 0)
-
- self.OnCancel()
-
- def OnAbort(self, event):
- """!Abort running import
- @todo not yet implemented
- """
- pass
-
- def OnSetInput(self, event):
- """!Input DXF file/OGR dsn defined, update list of layer widget"""
- path = event.GetString()
- if self.inputType == 'dxf':
- ret = gcmd.RunCommand('v.in.dxf',
- quiet = True,
- parent = self,
- read = True,
- flags = 'l',
- input = path)
- if not ret:
- self.list.LoadData()
- self.btn_run.Enable(False)
- return
- data = []
- if self.inputType == 'dxf':
- for line in ret.splitlines():
- layerId = line.split(':')[0].split(' ')[1]
- layerName = line.split(':')[1].strip()
- grassName = utils.GetValidLayerName(layerName)
- data.append((layerId, layerName.strip(), grassName.strip()))
-
- else: # gdal/ogr (for ogr maybe to use v.in.ogr -l)
- layerId = 1
- for file in glob.glob(os.path.join(self.input.GetValue(), "*.%s") % self.format.GetValue()):
- baseName = os.path.basename(file)
- grassName = utils.GetValidLayerName(baseName.split('.', -1)[0])
- data.append((layerId, baseName, grassName))
- layerId += 1
-
- self.list.LoadData(data)
- if len(data) > 0:
- self.btn_run.Enable(True)
- else:
- self.btn_run.Enable(False)
-
- class LayersList(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin,
- listmix.CheckListCtrlMixin):
- # listmix.CheckListCtrlMixin, listmix.TextEditMixin):
- """!List of layers to be imported (dxf, shp...)"""
- def __init__(self, parent, pos=wx.DefaultPosition,
- log=None):
- self.parent = parent
-
- wx.ListCtrl.__init__(self, parent, wx.ID_ANY,
- style=wx.LC_REPORT)
- listmix.CheckListCtrlMixin.__init__(self)
- self.log = log
- # setup mixins
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- # listmix.TextEditMixin.__init__(self)
-
- self.InsertColumn(0, _('Layer'))
- self.InsertColumn(1, _('Layer name'))
- self.InsertColumn(2, _('Name for GRASS map'))
-
- self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) #wxMSW
- self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) #wxGTK
- def LoadData(self, data=None):
- """!Load data into list"""
- if data is None:
- return
- self.DeleteAllItems()
-
- for id, name, grassName in data:
- index = self.InsertStringItem(sys.maxint, str(id))
- self.SetStringItem(index, 1, "%s" % str(name))
- self.SetStringItem(index, 2, "%s" % str(grassName))
- # check by default
- self.CheckItem(index, True)
-
- self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
- def OnPopupMenu(self, event):
- """!Show popup menu"""
- if self.GetItemCount() < 1:
- return
-
- if not hasattr(self, "popupDataID1"):
- self.popupDataID1 = wx.NewId()
- self.popupDataID2 = wx.NewId()
-
- self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
- self.Bind(wx.EVT_MENU, self.OnSelectNone, id=self.popupDataID2)
-
- # generate popup-menu
- menu = wx.Menu()
- menu.Append(self.popupDataID1, _("Select all"))
- menu.Append(self.popupDataID2, _("Deselect all"))
- self.PopupMenu(menu)
- menu.Destroy()
- def OnSelectAll(self, event):
- """!Select all items"""
- item = -1
-
- while True:
- item = self.GetNextItem(item)
- if item == -1:
- break
- self.CheckItem(item, True)
- event.Skip()
-
- def OnSelectNone(self, event):
- """!Deselect items"""
- item = -1
-
- while True:
- item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
- if item == -1:
- break
- self.CheckItem(item, False)
- event.Skip()
-
- def GetLayers(self):
- """!Get list of layers (layer name, output name)"""
- data = []
- item = -1
- while True:
- item = self.GetNextItem(item)
- if item == -1:
- break
- if self.IsChecked(item):
- # layer / output name
- data.append((self.GetItem(item, 1).GetText(),
- self.GetItem(item, 2).GetText()))
- return data
- class SetOpacityDialog(wx.Dialog):
- """!Set opacity of map layers"""
- def __init__(self, parent, id=wx.ID_ANY, title=_("Set Map Layer Opacity"),
- size=wx.DefaultSize, pos=wx.DefaultPosition,
- style=wx.DEFAULT_DIALOG_STYLE, opacity=100):
- self.parent = parent # GMFrame
- self.opacity = opacity # current opacity
- super(SetOpacityDialog, self).__init__(parent, id=id, pos=pos,
- size=size, style=style, title=title)
- panel = wx.Panel(parent=self, id=wx.ID_ANY)
-
- sizer = wx.BoxSizer(wx.VERTICAL)
- box = wx.GridBagSizer(vgap=5, hgap=5)
- self.value = wx.Slider(panel, id=wx.ID_ANY, value=self.opacity,
- style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS | \
- wx.SL_TOP | wx.SL_LABELS,
- minValue=0, maxValue=100,
- size=(350, -1))
- box.Add(item=self.value,
- flag=wx.ALIGN_CENTRE, pos=(0, 0), span=(1, 2))
- box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
- label=_("transparent")),
- pos=(1, 0))
- box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
- label=_("opaque")),
- flag=wx.ALIGN_RIGHT,
- pos=(1, 1))
- sizer.Add(item=box, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
- line = wx.StaticLine(parent=panel, id=wx.ID_ANY,
- style=wx.LI_HORIZONTAL)
- sizer.Add(item=line, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
- # buttons
- btnsizer = wx.StdDialogButtonSizer()
- btnOK = wx.Button(parent=panel, id=wx.ID_OK)
- btnOK.SetDefault()
- btnsizer.AddButton(btnOK)
- btnCancel = wx.Button(parent=panel, id=wx.ID_CANCEL)
- btnsizer.AddButton(btnCancel)
- btnsizer.Realize()
- sizer.Add(item=btnsizer, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
- panel.SetSizer(sizer)
- sizer.Fit(panel)
- self.SetSize(self.GetBestSize())
- self.Layout()
- def GetOpacity(self):
- """!Button 'OK' pressed"""
- # return opacity value
- opacity = float(self.value.GetValue()) / 100
- return opacity
|