1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632 |
- """!
- @package gui_core.dialogs
- @brief Various dialogs used in wxGUI.
- List of classes:
- - dialogs::ElementDialog
- - dialogs::LocationDialog
- - dialogs::MapsetDialog
- - dialogs::NewVectorDialog
- - dialogs::SavedRegion
- - dialogs::DecorationDialog
- - dialogs::TextLayerDialog
- - dialogs::GroupDialog
- - dialogs::MapLayersDialog
- - dialogs::ImportDialog
- - dialogs::GdalImportDialog
- - dialogs::GdalOutputDialog
- - dialogs::DxfImportDialog
- - dialogs::LayersList (used by MultiImport)
- - dialogs::SetOpacityDialog
- - dialogs::ImageSizeDialog
- - dialogs::SqlQueryFrame
- - dialogs::SymbolDialog
- (C) 2008-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 Martin Landa <landa.martin gmail.com>
- @author Anna Kratochvilova <kratochanna gmail.com> (GroupDialog, SymbolDialog)
- """
- import os
- import sys
- import re
- from bisect import bisect
- import wx
- import wx.lib.filebrowsebutton as filebrowse
- import wx.lib.mixins.listctrl as listmix
- from wx.lib.newevent import NewEvent
- from grass.script import core as grass
- from grass.script import task as gtask
- from core import globalvar
- from core.gcmd import GError, RunCommand, GMessage
- from gui_core.gselect import ElementSelect, LocationSelect, MapsetSelect, Select, OgrTypeSelect, GdalSelect, MapsetSelect
- from gui_core.forms import GUI
- from gui_core.widgets import SingleSymbolPanel, EVT_SYMBOL_SELECTION_CHANGED, GListCtrl
- from core.utils import GetLayerNameFromCmd, GetValidLayerName
- from core.settings import UserSettings, GetDisplayVectSettings
- from core.debug import Debug
- wxApplyMapLayers, EVT_APPLY_MAP_LAYERS = NewEvent()
- wxApplyOpacity, EVT_APPLY_OPACITY = NewEvent()
- class ElementDialog(wx.Dialog):
- def __init__(self, parent, title, label, id = wx.ID_ANY,
- etype = False, style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
- **kwargs):
- """!General dialog to choose given element (location, mapset, vector map, etc.)
-
- @param parent window
- @param title window title
- @param label element label
- @param etype show also ElementSelect
- """
- wx.Dialog.__init__(self, parent, id, title, style = style, **kwargs)
-
- self.etype = etype
- self.label = label
-
- 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()
-
- if self.etype:
- self.typeSelect = ElementSelect(parent = self.panel,
- size = globalvar.DIALOG_GSELECT_SIZE)
- self.typeSelect.Bind(wx.EVT_CHOICE, self.OnType)
-
- self.element = None # must be defined
-
- self.__layout()
-
- def PostInit(self):
- self.element.SetFocus()
- self.element.Bind(wx.EVT_TEXT, self.OnElement)
- if not self.element.GetValue():
- self.btnOK.Disable()
-
- def OnType(self, event):
- """!Select element type"""
- if not self.etype:
- return
- evalue = self.typeSelect.GetValue(event.GetString())
- self.element.SetType(evalue)
-
- 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)
-
- if self.etype:
- self.dataSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = _("Type of element:")),
- proportion = 0, flag = wx.ALL, border = 1)
- self.dataSizer.Add(item = self.typeSelect,
- proportion = 0, flag = wx.ALL, border = 1)
-
- self.dataSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = 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()
-
- def GetType(self):
- """!Get element type"""
- return self.element.tcp.GetType()
-
- 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 = LocationSelect(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_GSELECT_SIZE)
-
- self.element1 = MapsetSelect(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_GSELECT_SIZE,
- setItems = False, skipCurrent = True)
-
- 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.UpdateItems(dbase = dbase, location = location)
- 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 = MapsetSelect(parent = self.panel, id = wx.ID_ANY, skipCurrent = True,
- 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):
- def __init__(self, parent, id = wx.ID_ANY, title = _('Create new vector map'),
- disableAdd = False, disableTable = False, showType = False,
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, *kwargs):
- """!Dialog for creating new vector map
- @param parent parent window
- @param id window id
- @param title window title
- @param disableAdd disable 'add layer' checkbox
- @param disableTable disable 'create table' checkbox
- @param showType True to show feature type selector (used for creating new empty OGR layers)
- @param style window style
- @param kwargs other argumentes for ElementDialog
-
- @return dialog instance
- """
- ElementDialog.__init__(self, parent, title, label = _("Name for new vector map:"))
-
- self.element = Select(parent = self.panel, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
- type = 'vector', mapsets = [grass.gisenv()['MAPSET'],])
-
- # determine output format
- if showType:
- self.ftype = OgrTypeSelect(parent = self, panel = self.panel)
- else:
- self.ftype = None
-
- 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)
-
- if showType:
- self.keycol = None
- else:
- self.keycol = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY,
- size = globalvar.DIALOG_SPIN_SIZE)
- self.keycol.SetValue(UserSettings.Get(group = 'atm', key = 'keycolumn', subkey = 'value'))
- if disableTable:
- self.keycol.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.table.Bind(wx.EVT_CHECKBOX, self.OnTable)
-
- self.PostInit()
-
- self._layout()
- self.SetMinSize(self.GetSize())
-
- def OnMapName(self, event):
- """!Name for vector map layer given"""
- self.OnElement(event)
-
- def OnTable(self, event):
- if self.keycol:
- self.keycol.Enable(event.IsChecked())
-
- def _layout(self):
- """!Do layout"""
- self.dataSizer.Add(item = self.element, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 1)
- if self.ftype:
- self.dataSizer.AddSpacer(1)
- self.dataSizer.Add(item = self.ftype, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 1)
-
- self.dataSizer.Add(item = self.table, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 1)
-
- if self.keycol:
- keySizer = wx.BoxSizer(wx.HORIZONTAL)
- keySizer.Add(item = wx.StaticText(parent = self.panel, label = _("Key column:")),
- proportion = 0,
- flag = wx.ALIGN_CENTER_VERTICAL)
- keySizer.AddSpacer(10)
- keySizer.Add(item = self.keycol, proportion = 0,
- flag = wx.ALIGN_RIGHT)
- self.dataSizer.Add(item = keySizer, proportion = 1,
- 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, full = False):
- """!Get name of vector map to be created
- @param full True to get fully qualified name
- """
- name = self.GetElement()
- if full:
- if '@' in name:
- return name
- else:
- return name + '@' + grass.gisenv()['MAPSET']
-
- return name.split('@', 1)[0]
- def GetKey(self):
- """!Get key column name"""
- if self.keycol:
- return self.keycol.GetValue()
- return UserSettings.Get(group = 'atm', key = 'keycolumn', subkey = 'value')
-
- def IsChecked(self, key):
- """!Get dialog properties
- @param key window key ('add', 'table')
- @return True/False
- @return None on error
- """
- if key == 'add':
- return self.addbox.IsChecked()
- elif key == 'table':
- return self.table.IsChecked()
-
- return None
-
- def GetFeatureType(self):
- """!Get feature type for OGR
- @return feature type as string
- @return None for native format
- """
- if self.ftype:
- return self.ftype.GetType()
-
- return None
- def CreateNewVector(parent, cmd, title = _('Create new vector map'),
- exceptMap = None, log = None,
- disableAdd = False, disableTable = False):
- """!Create new vector map layer
-
- @param cmd (prog, **kwargs)
- @param title window title
- @param exceptMap list of maps to be excepted
- @param log
- @param disableAdd disable 'add layer' checkbox
- @param disableTable disable 'create table' checkbox
-
- @return dialog instance
- @return None on error
- """
- vExternalOut = grass.parse_command('v.external.out', flags = 'g')
- isNative = vExternalOut['format'] == 'native'
- if cmd[0] == 'v.edit' and not isNative:
- showType = True
- else:
- showType = False
- dlg = NewVectorDialog(parent, title = title,
- disableAdd = disableAdd, disableTable = disableTable,
- showType = showType)
-
- if dlg.ShowModal() != wx.ID_OK:
- dlg.Destroy()
- return None
- outmap = dlg.GetName()
- key = dlg.GetKey()
- if outmap == exceptMap:
- GError(parent = parent,
- message = _("Unable to create vector map <%s>.") % outmap)
- dlg.Destroy()
- return None
- if dlg.table.IsEnabled() and not key:
- GError(parent = parent,
- message = _("Invalid or empty key column.\n"
- "Unable to create vector map <%s>.") % outmap)
- dlg.Destroy()
- return
-
- if outmap == '': # should not happen
- dlg.Destroy()
- return None
- # update cmd -> output name defined
- cmd[1][cmd[2]] = outmap
- if showType:
- cmd[1]['type'] = dlg.GetFeatureType()
-
- if isNative:
- listOfVectors = grass.list_grouped('vect')[grass.gisenv()['MAPSET']]
- else:
- listOfVectors = RunCommand('v.external',
- quiet = True,
- parent = parent,
- read = True,
- flags = 'l',
- dsn = vExternalOut['dsn']).splitlines()
-
- 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()
- dlg.Destroy()
- return None
-
- if UserSettings.Get(group = 'cmd', key = 'overwrite', subkey = 'enabled'):
- overwrite = True
-
- ret = RunCommand(prog = cmd[0],
- parent = parent,
- overwrite = overwrite,
- **cmd[1])
- if ret != 0:
- dlg.Destroy()
- return None
-
- if not isNative:
- # create link for OGR layers
- RunCommand('v.external',
- overwrite = overwrite,
- parent = parent,
- dsn = vExternalOut['dsn'],
- layer = outmap)
-
- # create attribute table
- if dlg.table.IsEnabled() and dlg.table.IsChecked():
- if isNative:
- sql = 'CREATE TABLE %s (%s INTEGER)' % (outmap, key)
-
- RunCommand('db.connect',
- flags = 'c')
-
- Debug.msg(1, "SQL: %s" % sql)
- RunCommand('db.execute',
- quiet = True,
- parent = parent,
- input = '-',
- stdin = sql)
-
- RunCommand('v.db.connect',
- quiet = True,
- parent = parent,
- map = outmap,
- table = outmap,
- key = key,
- layer = '1')
- # TODO: how to deal with attribute tables for OGR layers?
-
- # return fully qualified map name
- if '@' not in outmap:
- outmap += '@' + grass.gisenv()['MAPSET']
-
- if log:
- log.WriteLog(_("New vector map <%s> created") % outmap)
-
- return dlg
- class SavedRegion(wx.Dialog):
- def __init__(self, parent, title, id = wx.ID_ANY, loadsave = 'load',
- **kwargs):
- """!Loading or 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 = Select(parent = self, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
- type = 'windows')
- elif loadsave == 'save':
- label.SetLabel(_("Save region:"))
- selection = Select(parent = self, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
- type = 'windows', mapsets = [grass.gisenv()['MAPSET']], fullyQualified = False)
-
- 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()
- def GetName(self):
- """!Return region name"""
- return self.wind
-
- 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)
- if self.name == 'legend':
- box = wx.BoxSizer(wx.HORIZONTAL)
- resize = wx.ToggleButton(parent = self, id = wx.ID_ANY, label = _("Set size and position"))
- resize.SetToolTipString(_("Click and drag on the map display to set legend"
- " size and position and then press OK"))
- resize.SetName('resize')
- resize.Disable()
- box.Add(item = resize, 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)
- if self.name == 'legend':
- self.Bind(wx.EVT_TOGGLEBUTTON, self.OnResize, resize)
- 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:
- if name == 'legend':
- mapName, found = GetLayerNameFromCmd(self.parent.MapWindow.overlays[self.ovlId]['cmd'])
- if found:
- # enable 'OK' and 'Resize' button
- self.btnOK.Enable()
- if not self.parent.IsPaneShown('3d'):
- self.FindWindowByName('resize').Enable()
-
- # set title
- self.SetTitle(_('Legend of raster map <%s>') % \
- mapName)
-
-
- def _createOverlay(self):
- """!Creates overlay"""
- if not self.parent.GetMap().GetOverlay(self.ovlId):
- self.newOverlay = self.parent.Map.AddOverlay(id = self.ovlId, type = self.name,
- command = self.cmd,
- l_active = False, l_render = False, l_hidden = True)
- prop = { 'layer' : self.newOverlay,
- 'params' : None,
- 'propwin' : None,
- 'cmd' : self.cmd,
- 'coords': (0, 0),
- 'pdcType': 'image' }
- self.parent.MapWindow2D.overlays[self.ovlId] = prop
- if self.parent.MapWindow3D:
- self.parent.MapWindow3D.overlays[self.ovlId] = prop
-
- 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):
- """!Sets option for decoration map overlays
- """
- if self.parent.MapWindow.overlays[self.ovlId]['propwin'] is None:
- # build properties dialog
- GUI(parent = self.parent).ParseCommand(cmd = self.cmd,
- completed = (self.GetOptData, self.name, ''))
-
- 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 OnResize(self, event):
- if self.FindWindowByName('resize').GetValue():
- self.parent.SwitchTool(self.parent.toolbars['map'], event)
- self.parent.MapWindow.SetCursor(self.parent.cursors["cross"])
- self.parent.MapWindow.mouse['use'] = 'legend'
- self.parent.MapWindow.mouse['box'] = 'box'
- self.parent.MapWindow.pen = wx.Pen(colour = 'Black', width = 2, style = wx.SHORT_DASH)
- else:
- self.parent.MapWindow.SetCursor(self.parent.cursors["default"])
- self.parent.MapWindow.mouse['use'] = 'pointer'
-
- def OnCancel(self, event):
- """!Cancel dialog"""
- if self.name == 'legend' and self.FindWindowByName('resize').GetValue():
- self.FindWindowByName('resize').SetValue(False)
- self.OnResize(None)
-
- self.parent.dialogs['barscale'] = None
- if event and hasattr(self, 'newOverlay'):
- self.parent.Map.DeleteOverlay(self.newOverlay)
- 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
- if self.parent.IsPaneShown('3d'):
- self.parent.MapWindow.UpdateOverlays()
-
- 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()
- if not self.parent.IsPaneShown('3d'):
- self.FindWindowByName('resize').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']
- self.currBB = self.parent.MapWindow.textdict[self.ovlId]['bbox']
- else:
- self.currClr = wx.BLACK
- self.currText = ''
- self.currFont = self.GetFont()
- self.currRot = 0.0
- self.currCoords = [10, 10]
- self.currBB = wx.Rect()
- 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 GroupDialog(wx.Dialog):
- """!Dialog for creating/editing groups"""
- def __init__(self, parent = None, defaultGroup = None,
- title = _("Create or edit imagery groups"),
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
-
- wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title,
- style = style, **kwargs)
-
- self.parent = parent
- self.defaultGroup = defaultGroup
- self.currentGroup = self.defaultGroup
- self.groupChanged = False
-
- self.bodySizer = self._createDialogBody()
-
- # buttons
- btnOk = wx.Button(parent = self, id = wx.ID_OK)
- btnApply = wx.Button(parent = self, id = wx.ID_APPLY)
- btnClose = wx.Button(parent = self, id = wx.ID_CANCEL)
-
- btnOk.SetToolTipString(_("Apply changes to selected group and close dialog"))
- btnApply.SetToolTipString(_("Apply changes to selected group"))
- btnClose.SetToolTipString(_("Close dialog, changes are not applied"))
- btnOk.SetDefault()
-
- # sizers & do layout
- # btnSizer = wx.BoxSizer(wx.HORIZONTAL)
- # btnSizer.Add(item = btnClose, proportion = 0,
- # flag = wx.RIGHT | wx.ALIGN_RIGHT | wx.EXPAND, border = 5)
- # btnSizer.Add(item = btnApply, proportion = 0,
- # flag = wx.LEFT, border = 5)
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnOk)
- btnSizer.AddButton(btnApply)
- btnSizer.AddButton(btnClose)
- btnSizer.Realize()
-
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- mainSizer.Add(item = self.bodySizer, proportion = 1,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 10)
- mainSizer.Add(item = wx.StaticLine(parent = self, id = wx.ID_ANY,
- style = wx.LI_HORIZONTAL), proportion = 0,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 10)
-
- mainSizer.Add(item = btnSizer, proportion = 0,
- flag = wx.ALL | wx.ALIGN_RIGHT, border = 10)
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
-
- btnOk.Bind(wx.EVT_BUTTON, self.OnOk)
- btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
- btnClose.Bind(wx.EVT_BUTTON, self.OnClose)
- # set dialog min size
- self.SetMinSize(self.GetSize())
-
- def _createDialogBody(self):
- bodySizer = wx.BoxSizer(wx.VERTICAL)
-
- # group selection
- bodySizer.Add(item = wx.StaticText(parent = self, id = wx.ID_ANY,
- label = _("Select the group you want to edit or "
- "enter name of new group:")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.TOP, border = 10)
- self.groupSelect = Select(parent = self, type = 'group',
- mapsets = [grass.gisenv()['MAPSET']],
- size = globalvar.DIALOG_GSELECT_SIZE) # searchpath?
-
- bodySizer.Add(item = self.groupSelect, flag = wx.TOP | wx.EXPAND, border = 5)
-
- bodySizer.AddSpacer(10)
- # layers in group
- bodySizer.Add(item = wx.StaticText(parent = self, label = _("Layers in selected group:")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.BOTTOM, border = 5)
-
- gridSizer = wx.GridBagSizer(vgap = 5, hgap = 5)
- gridSizer.AddGrowableCol(0)
-
- self.layerBox = wx.ListBox(parent = self, id = wx.ID_ANY, size = (-1, 150),
- style = wx.LB_MULTIPLE | wx.LB_NEEDED_SB)
-
- gridSizer.Add(item = self.layerBox, pos = (0, 0), span = (2, 1), flag = wx.EXPAND)
-
- self.addLayer = wx.Button(self, id = wx.ID_ADD)
- self.addLayer.SetToolTipString(_("Select map layers and add them to the list."))
- gridSizer.Add(item = self.addLayer, pos = (0, 1), flag = wx.EXPAND)
-
- self.removeLayer = wx.Button(self, id = wx.ID_REMOVE)
- self.removeLayer.SetToolTipString(_("Remove selected layer(s) from list."))
- gridSizer.Add(item = self.removeLayer, pos = (1, 1))
-
- bodySizer.Add(item = gridSizer, proportion = 1, flag = wx.EXPAND)
-
- self.infoLabel = wx.StaticText(parent = self, id = wx.ID_ANY)
- bodySizer.Add(item = self.infoLabel,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.TOP | wx.BOTTOM, border = 5)
-
- self.subGroup = wx.CheckBox(parent = self, id = wx.ID_ANY,
- label = _("Define also sub-group (same name as group)"))
- self.subGroup.SetValue(True) # most of imagery modules requires also subgroup
- bodySizer.Add(item = self.subGroup, flag = wx.BOTTOM | wx.EXPAND, border = 5)
- # bindings
- self.groupSelect.GetTextCtrl().Bind(wx.EVT_TEXT, self.OnGroupSelected)
- self.addLayer.Bind(wx.EVT_BUTTON, self.OnAddLayer)
- self.removeLayer.Bind(wx.EVT_BUTTON, self.OnRemoveLayer)
-
- if self.defaultGroup:
- self.groupSelect.SetValue(self.defaultGroup)
-
- return bodySizer
-
- def OnAddLayer(self, event):
- """!Add new layer to listbox"""
- dlg = MapLayersDialogForGroups(parent = self, title = _("Add selected map layers into group"))
-
- if dlg.ShowModal() != wx.ID_OK:
- dlg.Destroy()
- return
-
- layers = dlg.GetMapLayers()
- for layer in layers:
- if layer not in self.GetLayers():
- self.layerBox.Append(layer)
- self.groupChanged = True
-
-
- def OnRemoveLayer(self, event):
- """!Remove layer from listbox"""
- while self.layerBox.GetSelections():
- sel = self.layerBox.GetSelections()[0]
- self.layerBox.Delete(sel)
- self.groupChanged = True
-
- def GetLayers(self):
- """!Get layers"""
- return self.layerBox.GetItems()
-
- def OnGroupSelected(self, event):
- """!Text changed in group selector"""
- # callAfter must be called to close popup before other actions
- wx.CallAfter(self.GroupSelected)
-
- def GroupSelected(self):
- """!Group was selected, check if changes were apllied"""
- group = self.GetSelectedGroup()
- if self.groupChanged:
- dlg = wx.MessageDialog(self, message = _("Group <%s> was changed, "
- "do you want to apply changes?") % self.currentGroup,
- caption = _("Unapplied changes"),
- style = wx.YES_NO | wx.ICON_QUESTION | wx.YES_DEFAULT)
- if dlg.ShowModal() == wx.ID_YES:
- self.ApplyChanges()
-
- dlg.Destroy()
-
-
-
- groups = self.GetExistGroups()
- if group in groups:
- self.ShowGroupLayers(self.GetGroupLayers(group))
-
- self.currentGroup = group
- self.groupChanged = False
-
- self.ClearNotification()
-
- def ShowGroupLayers(self, mapList):
- """!Show map layers in currently selected group"""
- self.layerBox.Set(mapList)
-
-
- def EditGroup(self, group):
- """!Edit selected group"""
- layersNew = self.GetLayers()
- layersOld = self.GetGroupLayers(group)
-
- add = []
- remove = []
- for layerNew in layersNew:
- if layerNew not in layersOld:
- add.append(layerNew)
-
- for layerOld in layersOld:
- if layerOld not in layersNew:
- remove.append(layerOld)
-
- kwargs = {}
- if self.subGroup.IsChecked():
- kwargs['subgroup'] = group
-
- ret = None
- if remove:
- ret = RunCommand('i.group',
- parent = self,
- group = group,
- flags = 'r',
- input = ','.join(remove),
- **kwargs)
-
- if add:
- ret = RunCommand('i.group',
- parent = self,
- group = group,
- input = ','.join(add),
- **kwargs)
-
- return ret
-
- def CreateNewGroup(self, group):
- """!Create new group"""
- layers = self.GetLayers()
-
- kwargs = {}
- if self.subGroup.IsChecked():
- kwargs['subgroup'] = group
-
- return RunCommand('i.group',
- parent = self,
- group = group,
- input = layers,
- **kwargs)
-
- def GetExistGroups(self):
- """!Returns existing groups in current mapset"""
- return grass.list_grouped('group')[grass.gisenv()['MAPSET']]
-
- def ShowResult(self, group, returnCode, create):
- """!Show if operation was successfull."""
- group += '@' + grass.gisenv()['MAPSET']
- if returnCode is None:
- label = _("No changes to apply in group <%s>.") % group
- elif returnCode == 0:
- if create:
- label = _("Group <%s> was successfully created.") % group
- else:
- label = _("Group <%s> was successfully changed.") % group
- else:
- if create:
- label = _("Creating of new group <%s> failed.") % group
- else:
- label = _("Changing of group <%s> failed.") % group
-
- self.infoLabel.SetLabel(label)
- wx.FutureCall(4000, self.ClearNotification)
-
- def GetSelectedGroup(self):
- """!Return currently selected group (without mapset)"""
- return self.groupSelect.GetValue().split('@')[0]
-
- def GetGroupLayers(self, group):
- """!Get layers in group"""
- res = RunCommand('i.group',
- parent = self,
- flags = 'g',
- group = group,
- read = True).strip()
- if res.split('\n')[0]:
- return res.split('\n')
- return []
-
- def ClearNotification(self):
- """!Clear notification string"""
- self.infoLabel.SetLabel("")
-
- def ApplyChanges(self):
- """!Create or edit group"""
- group = self.currentGroup
- if not group:
- GMessage(parent = self,
- message = _("No group selected."))
- return False
-
- groups = self.GetExistGroups()
- if group in groups:
- ret = self.EditGroup(group)
- self.ShowResult(group = group, returnCode = ret, create = False)
-
- else:
- ret = self.CreateNewGroup(group)
- self.ShowResult(group = group, returnCode = ret, create = True)
-
- self.groupChanged = False
-
- return True
-
- def OnApply(self, event):
- """!Apply changes"""
- self.ApplyChanges()
-
- def OnOk(self, event):
- """!Apply changes and close dialog"""
- if self.ApplyChanges():
- self.OnClose(event)
-
- def OnClose(self, event):
- """!Close dialog"""
- if not self.IsModal():
- self.Destroy()
- event.Skip()
-
- class MapLayersDialogBase(wx.Dialog):
- """!Base dialog for selecting map layers (raster, vector).
- There are 3 subclasses: MapLayersDialogForGroups, MapLayersDialogForModeler,
- MapLayersDialog. Base class contains core functionality.
- """
- def __init__(self, parent, title,
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
- wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title,
- style = style, **kwargs)
-
- self.parent = parent # GMFrame or ?
- self.mainSizer = wx.BoxSizer(wx.VERTICAL)
-
- # dialog body
- self.bodySizer = self._createDialogBody()
- self.mainSizer.Add(item = self.bodySizer, proportion = 1,
- flag = wx.EXPAND | wx.ALL, border = 5)
-
- # update list of layer to be loaded
- self.map_layers = [] # list of map layers (full list type/mapset)
- self.LoadMapLayers(self.GetLayerType(cmd = True),
- self.mapset.GetStringSelection())
- self._fullyQualifiedNames()
- self._modelerDSeries()
- # buttons
- btnCancel = wx.Button(parent = self, id = wx.ID_CANCEL)
- btnOk = wx.Button(parent = self, id = wx.ID_OK)
- btnOk.SetDefault()
-
- # sizers & do layout
- self.btnSizer = wx.StdDialogButtonSizer()
- self.btnSizer.AddButton(btnCancel)
- self.btnSizer.AddButton(btnOk)
- self._addApplyButton()
- self.btnSizer.Realize()
-
- self.mainSizer.Add(item = self.btnSizer, proportion = 0,
- flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
- self.SetSizer(self.mainSizer)
- self.mainSizer.Fit(self)
- # set dialog min size
- self.SetMinSize(self.GetSize())
- def _modelerDSeries(self):
- """!Method used only by MapLayersDialogForModeler,
- for other subclasses does nothing.
- """
- pass
- def _addApplyButton(self):
- """!Method used only by MapLayersDialog,
- for other subclasses does nothing.
- """
- pass
- def _fullyQualifiedNames(self):
- """!Adds CheckBox which determines is fully qualified names are retuned.
- """
- self.fullyQualified = wx.CheckBox(parent = self, id = wx.ID_ANY,
- label = _("Use fully-qualified map names"))
- self.fullyQualified.SetValue(True)
- self.mainSizer.Add(item = self.fullyQualified, proportion = 0,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 5)
- def _useFullyQualifiedNames(self):
- return self.fullyQualified.IsChecked()
- def _layerTypes(self):
- """!Determines which layer types can be chosen.
- Valid values:
- - raster
- - raster3d
- - vector
- """
- return [_('raster'), _('3D raster'), _('vector')]
- def _selectAll(self):
- """!Check all layers by default"""
- return True
- 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 type:")),
- flag = wx.ALIGN_CENTER_VERTICAL,
- pos = (0,0))
-
- self.layerType = wx.Choice(parent = self, id = wx.ID_ANY,
- choices = self._layerTypes(), size = (100,-1))
- self.layerType.SetSelection(0)
-
- bodySizer.Add(item = self.layerType,
- pos = (0,1))
- self.layerType.Bind(wx.EVT_CHOICE, self.OnChangeParams)
- # select toggle
- self.toggle = wx.CheckBox(parent = self, id = wx.ID_ANY,
- label = _("Select toggle"))
- self.toggle.SetValue(self._selectAll())
- 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 = MapsetSelect(parent = self, searchPath = True)
- 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 = _("Pattern:")),
- 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))
- self.filter.SetFocus()
- self.filter.SetToolTipString(_("Put here a regular expression."
- " Characters '.*' stand for anything,"
- " character '^' stands for the beginning"
- " and '$' for the end."))
- # 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.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
- """
- self.map_layers = grass.mlist_grouped(type = type)[mapset]
- self.layers.Set(self.map_layers)
-
- # check all items by default
- for item in range(self.layers.GetCount()):
-
- self.layers.Check(item, check = self._selectAll())
-
- def OnChangeParams(self, event):
- """!Filter parameters changed by user"""
- # update list of layer to be loaded
- self.LoadMapLayers(self.GetLayerType(cmd = True),
- 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
- mapset = self.mapset.GetStringSelection()
- for item in range(self.layers.GetCount()):
- if not self.layers.IsChecked(item):
- continue
- if self._useFullyQualifiedNames():
- layerNames.append(self.layers.GetString(item) + '@' + mapset)
- else:
- layerNames.append(self.layers.GetString(item))
-
- return layerNames
-
- def GetLayerType(self, cmd = False):
- """!Get selected layer type
- @param cmd True for g.mlist
- """
- if not cmd:
- return self.layerType.GetStringSelection()
-
- sel = self.layerType.GetSelection()
- if sel == 0:
- ltype = 'rast'
- elif sel == 1:
- ltype = 'rast3d'
- else:
- ltype = 'vect'
-
- return ltype
- class MapLayersDialog(MapLayersDialogBase):
- """!Subclass of MapLayersDialogBase used in Layer Manager.
- Contains apply button, which sends wxApplyMapLayers event.
- """
- def __init__(self, parent, title, **kwargs):
- MapLayersDialogBase.__init__(self, parent = parent, title = title, **kwargs)
- def _addApplyButton(self):
- btnApply = wx.Button(parent = self, id = wx.ID_APPLY)
- self.btnSizer.AddButton(btnApply)
- btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
- def OnApply(self, event):
- event = wxApplyMapLayers(mapLayers = self.GetMapLayers(), ltype = self.GetLayerType(cmd = True))
- wx.PostEvent(self, event)
- class MapLayersDialogForGroups(MapLayersDialogBase):
- """!Subclass of MapLayersDialogBase used for specyfying maps in an imagery group.
- Shows only raster maps.
- """
- def __init__(self, parent, title, **kwargs):
- MapLayersDialogBase.__init__(self, parent = parent, title = title, **kwargs)
- def _layerTypes(self):
- return [_('raster'),]
- def _selectAll(self):
- """!Could be overriden"""
- return False
- def _fullyQualifiedNames(self):
- pass
- def _useFullyQualifiedNames(self):
- return True
- class MapLayersDialogForModeler(MapLayersDialogBase):
- """!Subclass of MapLayersDialogBase used in Modeler.
- """
- def __init__(self, parent, title, **kwargs):
- MapLayersDialogBase.__init__(self, parent = parent, title = title, **kwargs)
- def _modelerDSeries(self):
- self.dseries = wx.CheckBox(parent = self, id = wx.ID_ANY,
- label = _("Dynamic series (%s)") % 'g.mlist')
- self.dseries.SetValue(False)
- self.mainSizer.Add(item = self.dseries, proportion = 0,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 5)
- def GetDSeries(self):
- """!Used by modeler only
- @return g.mlist command
- """
- if not self.dseries or not self.dseries.IsChecked():
- return ''
-
- cond = 'map in `g.mlist type=%s ' % self.GetLayerType(cmd = True)
- patt = self.filter.GetValue()
- if patt:
- cond += 'pattern=%s ' % patt
- cond += 'mapset=%s`' % self.mapset.GetStringSelection()
-
- return cond
-
- class ImportDialog(wx.Dialog):
- """!Dialog for bulk import of various data (base class)"""
- def __init__(self, parent, itype,
- id = wx.ID_ANY, title = _("Multiple import"),
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
- self.parent = parent # GMFrame
- self.importType = itype
- self.options = dict() # list of options
-
- self.commandId = -1 # id of running command
-
- wx.Dialog.__init__(self, parent, id, title, style = style,
- name = "MultiImportDialog")
-
- self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
-
- self.layerBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY)
- if self.importType == 'gdal':
- label = _("List of raster layers")
- elif self.importType == 'ogr':
- label = _("List of vector layers")
- else:
- label = _("List of %s layers") % self.importType.upper()
- self.layerBox.SetLabel(" %s - %s " % (label, _("right click to (un)select all")))
-
- # list of layers
- columns = [_('Layer id'),
- _('Layer name'),
- _('Name for output GRASS map (editable)')]
- if itype == 'ogr':
- columns.insert(2, _('Feature type'))
- self.list = LayersList(parent = self.panel, columns = columns)
- self.list.LoadData()
- self.optionBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
- label = "%s" % _("Options"))
-
- cmd = self._getCommand()
- task = gtask.parse_interface(cmd)
- for f in task.get_options()['flags']:
- name = f.get('name', '')
- desc = f.get('label', '')
- if not desc:
- desc = f.get('description', '')
- if not name and not desc:
- continue
- if cmd == 'r.in.gdal' and name not in ('o', 'e', 'l', 'k'):
- continue
- elif cmd == 'r.external' and name not in ('o', 'e', 'r', 'h', 'v'):
- continue
- elif cmd == 'v.in.ogr' and name not in ('c', 'z', 't', 'o', 'r', 'e', 'w'):
- continue
- elif cmd == 'v.external' and name not in ('b'):
- continue
- elif cmd == 'v.in.dxf' and name not in ('e', 't', 'b', 'f', 'i'):
- continue
- self.options[name] = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
- label = desc)
-
-
- 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'))
-
- self.add = wx.CheckBox(parent = self.panel, id = wx.ID_ANY)
- self.closeOnFinish = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
- label = _("Close dialog on finish"))
- self.closeOnFinish.SetValue(UserSettings.Get(group = 'cmd', key = 'closeDlg', subkey = 'enabled'))
-
- #
- # buttons
- #
- # cancel
- self.btn_close = wx.Button(parent = self.panel, id = wx.ID_CLOSE)
- self.btn_close.SetToolTipString(_("Close dialog"))
- self.btn_close.Bind(wx.EVT_BUTTON, self.OnClose)
- # run
- 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)
- # run command dialog
- self.btn_cmd = wx.Button(parent = self.panel, id = wx.ID_ANY,
- label = _("Command dialog"))
- self.btn_cmd.Bind(wx.EVT_BUTTON, self.OnCmdDialog)
-
- def doLayout(self):
- """!Do layout"""
- dialogSizer = wx.BoxSizer(wx.VERTICAL)
-
- # dsn input
- dialogSizer.Add(item = self.dsnInput, proportion = 0,
- flag = wx.EXPAND)
-
- #
- # 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)
- # options
- optionSizer = wx.StaticBoxSizer(self.optionBox, wx.VERTICAL)
- for key in self.options.keys():
- optionSizer.Add(item = self.options[key], proportion = 0)
-
- dialogSizer.Add(item = optionSizer, 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, border = 5)
-
- dialogSizer.Add(item = self.add, proportion = 0,
- flag = wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
-
- dialogSizer.Add(item = self.closeOnFinish, proportion = 0,
- flag = wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
- #
- # buttons
- #
- btnsizer = wx.BoxSizer(orient = wx.HORIZONTAL)
-
- btnsizer.Add(item = self.btn_cmd, proportion = 0,
- flag = wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
-
- btnsizer.Add(item = self.btn_close, proportion = 0,
- flag = wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
-
- btnsizer.Add(item = self.btn_run, proportion = 0,
- flag = wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
-
- dialogSizer.Add(item = btnsizer, proportion = 0,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.BOTTOM | wx.ALIGN_RIGHT,
- border = 10)
-
- # dialogSizer.SetSizeHints(self.panel)
- self.panel.SetAutoLayout(True)
- self.panel.SetSizer(dialogSizer)
- dialogSizer.Fit(self.panel)
-
- # auto-layout seems not work here - FIXME
- size = wx.Size(globalvar.DIALOG_GSELECT_SIZE[0] + 225, 550)
- self.SetMinSize(size)
- self.SetSize((size.width, size.height + 100))
- # width = self.GetSize()[0]
- # self.list.SetColumnWidth(col = 1, width = width / 2 - 50)
- self.Layout()
- def _getCommand(self):
- """!Get command"""
- return ''
-
- def OnClose(self, event = None):
- """!Close dialog"""
- self.Close()
- def OnRun(self, event):
- """!Import/Link data (each layes as separate vector map)"""
- pass
- def OnCmdDialog(self, event):
- """!Show command dialog"""
- pass
-
- def AddLayers(self, returncode, cmd = None):
- """!Add imported/linked layers into layer tree"""
- if not self.add.IsChecked() or returncode != 0:
- return
-
- self.commandId += 1
- maptree = self.parent.GetLayerTree()
-
- layer, output = self.list.GetLayers()[self.commandId]
-
- if '@' not in output:
- name = output + '@' + grass.gisenv()['MAPSET']
- else:
- name = output
-
- # add imported layers into layer tree
- if self.importType == 'gdal':
- cmd = ['d.rast',
- 'map=%s' % name]
- if UserSettings.Get(group = 'rasterLayer', key = 'opaque', subkey = 'enabled'):
- cmd.append('-n')
-
- item = maptree.AddLayer(ltype = 'raster',
- lname = name, lchecked = False,
- lcmd = cmd, multiple = False)
- else:
- item = maptree.AddLayer(ltype = 'vector',
- lname = name, lchecked = False,
- lcmd = ['d.vect',
- 'map=%s' % name] + GetDisplayVectSettings(),
- multiple = False)
-
- maptree.mapdisplay.MapWindow.ZoomToMap()
-
- def OnAbort(self, event):
- """!Abort running import
- @todo not yet implemented
- """
- pass
- class GdalImportDialog(ImportDialog):
- def __init__(self, parent, giface, ogr = False, link = False):
- """!Dialog for bulk import of various raster/vector data
- @todo Split into GdalImportDialog and OgrImportDialog
- @param parent parent window
- @param ogr True for OGR (vector) otherwise GDAL (raster)
- @param link True for linking data otherwise importing data
- """
- self._giface = giface
- self.link = link
- self.ogr = ogr
-
- if ogr:
- ImportDialog.__init__(self, parent, itype = 'ogr')
- if link:
- self.SetTitle(_("Link external vector data"))
- else:
- self.SetTitle(_("Import vector data"))
- else:
- ImportDialog.__init__(self, parent, itype = 'gdal')
- if link:
- self.SetTitle(_("Link external raster data"))
- else:
- self.SetTitle(_("Import raster data"))
-
- self.dsnInput = GdalSelect(parent = self, panel = self.panel,
- ogr = ogr, link = link)
-
- if link:
- self.add.SetLabel(_("Add linked layers into layer tree"))
- else:
- self.add.SetLabel(_("Add imported layers into layer tree"))
-
- self.add.SetValue(UserSettings.Get(group = 'cmd', key = 'addNewLayer', subkey = 'enabled'))
- if link:
- self.btn_run.SetLabel(_("&Link"))
- self.btn_run.SetToolTipString(_("Link selected layers"))
- if ogr:
- self.btn_cmd.SetToolTipString(_('Open %s dialog') % 'v.external')
- else:
- self.btn_cmd.SetToolTipString(_('Open %s dialog') % 'r.external')
- else:
- self.btn_run.SetLabel(_("&Import"))
- self.btn_run.SetToolTipString(_("Import selected layers"))
- if ogr:
- self.btn_cmd.SetToolTipString(_('Open %s dialog') % 'v.in.ogr')
- else:
- self.btn_cmd.SetToolTipString(_('Open %s dialog') % 'r.in.gdal')
-
- self.doLayout()
- def OnRun(self, event):
- """!Import/Link data (each layes as separate vector map)"""
- self.commandId = -1
- data = self.list.GetLayers()
- if not data:
- GMessage(_("No layers selected. Operation canceled."),
- parent = self)
- return
-
- dsn = self.dsnInput.GetDsn()
- ext = self.dsnInput.GetFormatExt()
-
- # determine data driver for PostGIS links
- popOGR = False
- if self.importType == 'ogr' and \
- self.dsnInput.GetType() == 'db' and \
- self.dsnInput.GetFormat() == 'PostgreSQL' and \
- 'GRASS_VECTOR_OGR' not in os.environ:
- popOGR = True
- os.environ['GRASS_VECTOR_OGR'] = '1'
-
- for layer, output in data:
- if self.importType == 'ogr':
- if ext and layer.rfind(ext) > -1:
- layer = layer.replace('.' + ext, '')
- if self.link:
- cmd = ['v.external',
- 'dsn=%s' % dsn,
- 'output=%s' % output,
- 'layer=%s' % layer]
- else:
- cmd = ['v.in.ogr',
- 'dsn=%s' % dsn,
- 'layer=%s' % layer,
- 'output=%s' % output]
- else: # gdal
- if self.dsnInput.GetType() == 'dir':
- idsn = os.path.join(dsn, layer)
- else:
- idsn = dsn
-
- if self.link:
- cmd = ['r.external',
- 'input=%s' % idsn,
- 'output=%s' % output]
- else:
- cmd = ['r.in.gdal',
- 'input=%s' % idsn,
- 'output=%s' % output]
-
- if self.overwrite.IsChecked():
- cmd.append('--overwrite')
-
- for key in self.options.keys():
- if self.options[key].IsChecked():
- cmd.append('-%s' % key)
-
- if UserSettings.Get(group = 'cmd', key = 'overwrite', subkey = 'enabled') and \
- '--overwrite' not in cmd:
- cmd.append('--overwrite')
-
- # run in Layer Manager
- self._giface.RunCmd(cmd, switchPage = True, onDone = self.AddLayers)
-
- if popOGR:
- os.environ.pop('GRASS_VECTOR_OGR')
- if self.closeOnFinish.IsChecked():
- self.Close()
-
- def _getCommand(self):
- """!Get command"""
- if self.link:
- if self.ogr:
- return 'v.external'
- else:
- return 'r.external'
- else:
- if self.ogr:
- return 'v.in.ogr'
- else:
- return 'r.in.gdal'
-
- return ''
-
- def OnCmdDialog(self, event):
- """!Show command dialog"""
- name = self._getCommand()
- GUI(parent = self, modal = False).ParseCommand(cmd = [name])
- class GdalOutputDialog(wx.Dialog):
- def __init__(self, parent, id = wx.ID_ANY, ogr = False,
- style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, *kwargs):
- """!Dialog for setting output format for rasters/vectors
- @todo Split into GdalOutputDialog and OgrOutputDialog
- @param parent parent window
- @param id window id
- @param ogr True for OGR (vector) otherwise GDAL (raster)
- @param style window style
- @param *kwargs other wx.Dialog's arguments
- """
- self.parent = parent # GMFrame
- self.ogr = ogr
- wx.Dialog.__init__(self, parent, id = id, style = style, *kwargs)
- if self.ogr:
- self.SetTitle(_("Define output format for vector data"))
- else:
- self.SetTitle(_("Define output format for raster data"))
-
- self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
- # buttons
- self.btnCmd = wx.Button(parent = self.panel, id = wx.ID_ANY,
- label = _("Command dialog"))
- self.btnCmd.Bind(wx.EVT_BUTTON, self.OnCmdDialog)
- self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
- self.btnCancel.SetToolTipString(_("Close dialog"))
- self.btnOk = wx.Button(parent = self.panel, id = wx.ID_OK)
- self.btnOk.SetToolTipString(_("Set external format and close dialog"))
- self.btnOk.SetDefault()
- self.btnOk.Enable(False)
-
- self.dsnInput = GdalSelect(parent = self, panel = self.panel,
- ogr = ogr,
- exclude = ['file', 'protocol'], dest = True)
-
- self.Bind(wx.EVT_BUTTON, self.OnCancel, self.btnCancel)
- self.Bind(wx.EVT_BUTTON, self.OnOK, self.btnOk)
-
- self._layout()
- def _layout(self):
- dialogSizer = wx.BoxSizer(wx.VERTICAL)
-
- dialogSizer.Add(item = self.dsnInput, proportion = 0,
- flag = wx.EXPAND)
- btnSizer = wx.BoxSizer(orient = wx.HORIZONTAL)
- btnSizer.Add(item = self.btnCmd, proportion = 0,
- flag = wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
- btnSizer.Add(item = self.btnCancel, proportion = 0,
- flag = wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
- btnSizer.Add(item = self.btnOk, proportion = 0,
- flag = wx.RIGHT | wx.ALIGN_CENTER,
- border = 10)
-
- dialogSizer.Add(item = btnSizer, proportion = 0,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.BOTTOM | wx.TOP | wx.ALIGN_RIGHT,
- border = 10)
-
- self.panel.SetAutoLayout(True)
- self.panel.SetSizer(dialogSizer)
- dialogSizer.Fit(self.panel)
- size = wx.Size(globalvar.DIALOG_GSELECT_SIZE[0] + 225, self.GetBestSize()[1])
- self.SetMinSize(size)
- self.SetSize((size.width, size.height))
- self.Layout()
-
- def OnCmdDialog(self, event):
- GUI(parent = self, modal = True).ParseCommand(cmd = ['v.external.out'])
-
- def OnCancel(self, event):
- self.Destroy()
-
- def OnOK(self, event):
- if self.dsnInput.GetType() == 'native':
- RunCommand('v.external.out',
- parent = self,
- flags = 'r')
- else:
- dsn = self.dsnInput.GetDsn()
- frmt = self.dsnInput.GetFormat()
- options = self.dsnInput.GetOptions()
-
- RunCommand('v.external.out',
- parent = self,
- dsn = dsn, format = frmt,
- options = options)
- self.Close()
-
- class DxfImportDialog(ImportDialog):
- """!Dialog for bulk import of DXF layers"""
- def __init__(self, parent, giface):
- ImportDialog.__init__(self, parent, itype = 'dxf',
- title = _("Import DXF layers"))
- self._giface = giface
- self.dsnInput = 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.OnSetDsn,
- fileMask = "DXF File (*.dxf)|*.dxf")
-
- self.add.SetLabel(_("Add imported layers into layer tree"))
-
- self.add.SetValue(UserSettings.Get(group = 'cmd', key = 'addNewLayer', subkey = 'enabled'))
-
- self.doLayout()
- def _getCommand(self):
- """!Get command"""
- return 'v.in.dxf'
-
- def OnRun(self, event):
- """!Import/Link data (each layes as separate vector map)"""
- data = self.list.GetLayers()
-
- # hide dialog
- self.Hide()
-
- inputDxf = self.dsnInput.GetValue()
-
- for layer, output in data:
- cmd = ['v.in.dxf',
- 'input=%s' % inputDxf,
- 'layers=%s' % layer,
- 'output=%s' % output]
- for key in self.options.keys():
- if self.options[key].IsChecked():
- cmd.append('-%s' % key)
-
- if self.overwrite.IsChecked() or \
- UserSettings.Get(group = 'cmd', key = 'overwrite', subkey = 'enabled'):
- cmd.append('--overwrite')
-
- # run in Layer Manager
- self._giface.RunCmd(cmd, switchPage = True, onDone = self.AddLayers)
-
- self.OnCancel()
- def OnSetDsn(self, event):
- """!Input DXF file defined, update list of layer widget"""
- path = event.GetString()
- if not path:
- return
-
- data = list()
- ret = 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
-
- for line in ret.splitlines():
- layerId = line.split(':')[0].split(' ')[1]
- layerName = line.split(':')[1].strip()
- grassName = GetValidLayerName(layerName)
- data.append((layerId, layerName.strip(), grassName.strip()))
-
- self.list.LoadData(data)
- if len(data) > 0:
- self.btn_run.Enable(True)
- else:
- self.btn_run.Enable(False)
- def OnCmdDialog(self, event):
- """!Show command dialog"""
- GUI(parent = self, modal = True).ParseCommand(cmd = ['v.in.dxf'])
-
- class LayersList(GListCtrl, listmix.TextEditMixin):
- """!List of layers to be imported (dxf, shp...)"""
- def __init__(self, parent, columns, log = None):
- GListCtrl.__init__(self, parent)
-
- self.log = log
-
- # setup mixins
- listmix.TextEditMixin.__init__(self)
-
- for i in range(len(columns)):
- self.InsertColumn(i, columns[i])
-
- if len(columns) == 3:
- width = (65, 200)
- else:
- width = (65, 180, 110)
-
- for i in range(len(width)):
- self.SetColumnWidth(col = i, width = width[i])
-
- def LoadData(self, data = None):
- """!Load data into list"""
- self.DeleteAllItems()
- if data is None:
- return
-
- for item in data:
- index = self.InsertStringItem(sys.maxint, str(item[0]))
- for i in range(1, len(item)):
- self.SetStringItem(index, i, "%s" % str(item[i]))
-
- # check by default only on one item
- if len(data) == 1:
- self.CheckItem(index, True)
-
- def OnLeftDown(self, event):
- """!Allow editing only output name
-
- Code taken from TextEditMixin class.
- """
- x, y = event.GetPosition()
-
- colLocs = [0]
- loc = 0
- for n in range(self.GetColumnCount()):
- loc = loc + self.GetColumnWidth(n)
- colLocs.append(loc)
-
- col = bisect(colLocs, x + self.GetScrollPos(wx.HORIZONTAL)) - 1
-
- if col == self.GetColumnCount() - 1:
- listmix.TextEditMixin.OnLeftDown(self, event)
- else:
- 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 not self.IsChecked(item):
- continue
- # layer / output name
- data.append((self.GetItem(item, 1).GetText(),
- self.GetItem(item, self.GetColumnCount() - 1).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)
- btnApply = wx.Button(parent = panel, id = wx.ID_APPLY)
- btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
- btnsizer.AddButton(btnApply)
- 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
- def OnApply(self, event):
- event = wxApplyOpacity(value = self.GetOpacity())
- wx.PostEvent(self, event)
- def GetImageHandlers(image):
- """!Get list of supported image handlers"""
- lext = list()
- ltype = list()
- for h in image.GetHandlers():
- lext.append(h.GetExtension())
-
- filetype = ''
- if 'png' in lext:
- filetype += "PNG file (*.png)|*.png|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_PNG,
- 'ext' : 'png' })
- filetype += "BMP file (*.bmp)|*.bmp|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_BMP,
- 'ext' : 'bmp' })
- if 'gif' in lext:
- filetype += "GIF file (*.gif)|*.gif|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_GIF,
- 'ext' : 'gif' })
-
- if 'jpg' in lext:
- filetype += "JPG file (*.jpg)|*.jpg|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_JPEG,
- 'ext' : 'jpg' })
- if 'pcx' in lext:
- filetype += "PCX file (*.pcx)|*.pcx|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_PCX,
- 'ext' : 'pcx' })
-
- if 'pnm' in lext:
- filetype += "PNM file (*.pnm)|*.pnm|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_PNM,
- 'ext' : 'pnm' })
- if 'tif' in lext:
- filetype += "TIF file (*.tif)|*.tif|"
- ltype.append({ 'type' : wx.BITMAP_TYPE_TIF,
- 'ext' : 'tif' })
- if 'xpm' in lext:
- filetype += "XPM file (*.xpm)|*.xpm"
- ltype.append({ 'type' : wx.BITMAP_TYPE_XPM,
- 'ext' : 'xpm' })
-
- return filetype, ltype
- class ImageSizeDialog(wx.Dialog):
- """!Set size for saved graphic file"""
- def __init__(self, parent, id = wx.ID_ANY, title = _("Set image size"),
- style = wx.DEFAULT_DIALOG_STYLE, **kwargs):
- self.parent = parent
-
- wx.Dialog.__init__(self, parent, id = id, style = style, title = title, **kwargs)
-
- self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
-
- self.box = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
- label = ' % s' % _("Image size"))
-
- size = self.parent.GetWindow().GetClientSize()
- self.width = wx.SpinCtrl(parent = self.panel, id = wx.ID_ANY,
- style = wx.SP_ARROW_KEYS)
- self.width.SetRange(20, 1e6)
- self.width.SetValue(size.width)
- wx.CallAfter(self.width.SetFocus)
- self.height = wx.SpinCtrl(parent = self.panel, id = wx.ID_ANY,
- style = wx.SP_ARROW_KEYS)
- self.height.SetRange(20, 1e6)
- self.height.SetValue(size.height)
- self.template = wx.Choice(parent = self.panel, id = wx.ID_ANY,
- size = (125, -1),
- choices = [ "",
- "640x480",
- "800x600",
- "1024x768",
- "1280x960",
- "1600x1200",
- "1920x1440" ])
-
- self.btnOK = wx.Button(parent = self.panel, id = wx.ID_OK)
- self.btnOK.SetDefault()
- self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
-
- self.template.Bind(wx.EVT_CHOICE, self.OnTemplate)
-
- self._layout()
- self.SetSize(self.GetBestSize())
-
- def _layout(self):
- """!Do layout"""
- sizer = wx.BoxSizer(wx.VERTICAL)
-
- # body
- box = wx.StaticBoxSizer(self.box, wx.HORIZONTAL)
- fbox = wx.FlexGridSizer(cols = 2, vgap = 5, hgap = 5)
- fbox.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = _("Width:")),
- flag = wx.ALIGN_CENTER_VERTICAL)
- fbox.Add(item = self.width)
- fbox.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = _("Height:")),
- flag = wx.ALIGN_CENTER_VERTICAL)
- fbox.Add(item = self.height)
- fbox.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
- label = _("Template:")),
- flag = wx.ALIGN_CENTER_VERTICAL)
- fbox.Add(item = self.template)
-
- box.Add(item = fbox, proportion = 1,
- flag = wx.EXPAND | wx.ALL, border = 5)
- sizer.Add(item = box, proportion = 1,
- flag=wx.EXPAND | wx.ALL, border = 3)
-
- # buttons
- btnsizer = wx.StdDialogButtonSizer()
- btnsizer.AddButton(self.btnOK)
- btnsizer.AddButton(self.btnCancel)
- btnsizer.Realize()
- sizer.Add(item = btnsizer, proportion = 0,
- flag = wx.EXPAND | wx.ALIGN_RIGHT | wx.ALL, border=5)
-
- self.panel.SetSizer(sizer)
- sizer.Fit(self.panel)
- self.Layout()
-
- def GetValues(self):
- """!Get width/height values"""
- return self.width.GetValue(), self.height.GetValue()
-
- def OnTemplate(self, event):
- """!Template selected"""
- sel = event.GetString()
- if not sel:
- width, height = self.parent.GetWindow().GetClientSize()
- else:
- width, height = map(int, sel.split('x'))
- self.width.SetValue(width)
- self.height.SetValue(height)
-
- class SqlQueryFrame(wx.Frame):
- def __init__(self, parent, id = wx.ID_ANY,
- title = _("GRASS GIS SQL Query Utility"),
- *kwargs):
- """!SQL Query Utility window
- """
- self.parent = parent
- wx.Frame.__init__(self, parent = parent, id = id, title = title, *kwargs)
- self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass_sql.ico'), wx.BITMAP_TYPE_ICO))
- self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
-
- self.sqlBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
- label = _(" SQL statement "))
- self.sql = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY,
- style = wx.TE_MULTILINE)
-
- self.btnApply = wx.Button(parent = self.panel, id = wx.ID_APPLY)
- self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
- self.Bind(wx.EVT_BUTTON, self.OnCloseWindow, self.btnCancel)
-
- self._layout()
- self.SetMinSize(wx.Size(300, 150))
- self.SetSize(wx.Size(500, 200))
-
- def _layout(self):
- """!Do layout"""
- sizer = wx.BoxSizer(wx.VERTICAL)
-
- sqlSizer = wx.StaticBoxSizer(self.sqlBox, wx.HORIZONTAL)
- sqlSizer.Add(item = self.sql, proportion = 1,
- flag = wx.EXPAND)
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(self.btnApply)
- btnSizer.AddButton(self.btnCancel)
- btnSizer.Realize()
-
- sizer.Add(item = sqlSizer, proportion = 1,
- flag = wx.EXPAND | wx.ALL, border = 5)
- sizer.Add(item = btnSizer, proportion = 0,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
-
- self.panel.SetSizer(sizer)
-
- self.Layout()
- def OnCloseWindow(self, event):
- """!Close window
- """
- self.Close()
- class SymbolDialog(wx.Dialog):
- """!Dialog for GRASS symbols selection.
-
- Dialog is called in gui_core::forms module.
- """
- def __init__(self, parent, symbolPath, currentSymbol = None, title = _("Symbols")):
- """!Dialog constructor.
-
- It is assumed that symbolPath contains folders with symbols.
-
- @param parent dialog parent
- @param symbolPath absolute path to symbols
- @param currentSymbol currently selected symbol (e.g. 'basic/x')
- @param title dialog title
- """
- wx.Dialog.__init__(self, parent = parent, title = title, id = wx.ID_ANY)
-
- self.symbolPath = symbolPath
- self.currentSymbol = currentSymbol # default basic/x
- self.selected = None
- self.selectedDir = None
-
- self._layout()
-
- def _layout(self):
- mainPanel = wx.Panel(self, id = wx.ID_ANY)
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- vSizer = wx.BoxSizer( wx.VERTICAL)
- fgSizer = wx.FlexGridSizer(rows = 2, vgap = 5, hgap = 5)
- self.folderChoice = wx.Choice(mainPanel, id = wx.ID_ANY, choices = os.listdir(self.symbolPath))
- self.folderChoice.Bind(wx.EVT_CHOICE, self.OnFolderSelect)
-
- fgSizer.Add(item = wx.StaticText(mainPanel, id = wx.ID_ANY, label = _("Symbol directory:")),
- proportion = 0,
- flag = wx.ALIGN_CENTER_VERTICAL)
-
- fgSizer.Add(item = self.folderChoice, proportion = 0,
- flag = wx.ALIGN_CENTER, border = 0)
-
- self.infoLabel = wx.StaticText(mainPanel, id = wx.ID_ANY)
- fgSizer.Add(wx.StaticText(mainPanel, id = wx.ID_ANY, label = _("Symbol name:")),
- flag = wx.ALIGN_CENTRE_VERTICAL)
- fgSizer.Add(self.infoLabel, proportion = 0,
- flag = wx.ALIGN_CENTRE_VERTICAL)
- vSizer.Add(fgSizer, proportion = 0, flag = wx.ALL, border = 5)
-
- self.panels = self._createSymbolPanels(mainPanel)
- for panel in self.panels:
- vSizer.Add(panel, proportion = 0, flag = wx.ALL | wx.EXPAND, border = 5)
- panel.Bind(EVT_SYMBOL_SELECTION_CHANGED, self.SelectionChanged)
-
- mainSizer.Add(vSizer, proportion = 1, flag = wx.ALL| wx.EXPAND, border = 5)
- self.btnCancel = wx.Button(parent = mainPanel, id = wx.ID_CANCEL)
- self.btnOK = wx.Button(parent = mainPanel, id = wx.ID_OK)
- self.btnOK.SetDefault()
- self.btnOK.Enable(False)
-
- # buttons
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(self.btnCancel)
- btnSizer.AddButton(self.btnOK)
- btnSizer.Realize()
- mainSizer.Add(item = btnSizer, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 5)
-
- # show panel with the largest number of images and fit size
- count = []
- for folder in os.listdir(self.symbolPath):
- count.append(len(os.listdir(os.path.join(self.symbolPath, folder))))
-
- index = count.index(max(count))
- self.folderChoice.SetSelection(index)
- self.OnFolderSelect(None)
- self.infoLabel.Show()
-
- mainPanel.SetSizerAndFit(mainSizer)
- self.SetSize(self.GetBestSize())
-
- # show currently selected symbol
- if self.currentSymbol:
- # set directory
- self.selectedDir, self.selected = os.path.split(self.currentSymbol)
- self.folderChoice.SetStringSelection(self.selectedDir)
- # select symbol
- panelIdx = self.folderChoice.GetSelection()
- for panel in self.symbolPanels[panelIdx]:
- if panel.GetName() == self.selected:
- panel.Select()
- else:
- self.folderChoice.SetSelection(0)
-
- self.OnFolderSelect(None)
-
- def _createSymbolPanels(self, parent):
- """!Creates multiple panels with symbols.
-
- Panels are shown/hidden according to selected folder."""
- folders = os.listdir(self.symbolPath)
-
- panels = []
- self.symbolPanels = []
-
- for folder in folders:
- panel = wx.Panel(parent, style = wx.BORDER_RAISED)
- sizer = wx.GridSizer(cols = 6, vgap = 3, hgap = 3)
- images = self._getSymbols(path = os.path.join(self.symbolPath, folder))
-
- symbolPanels = []
- for img in images:
- iP = SingleSymbolPanel(parent = panel, symbolPath = img)
- sizer.Add(item = iP, proportion = 0, flag = wx.ALIGN_CENTER)
- symbolPanels.append(iP)
-
- panel.SetSizerAndFit(sizer)
- panel.Hide()
- panels.append(panel)
- self.symbolPanels.append(symbolPanels)
-
- return panels
-
- def _getSymbols(self, path):
- # we assume that images are in subfolders (1 level only)
- imageList = []
- for image in os.listdir(path):
- imageList.append(os.path.join(path, image))
-
- return sorted(imageList)
-
- def OnFolderSelect(self, event):
- """!Selected folder with symbols changed."""
- idx = self.folderChoice.GetSelection()
- for i in range(len(self.panels)):
- sizer = self.panels[i].GetContainingSizer()
- sizer.Show(self.panels[i], i == idx, recursive = True)
- sizer.Layout()
-
- if self.selectedDir == self.folderChoice.GetStringSelection():
- self.btnOK.Enable()
- self.infoLabel.SetLabel(self.selected)
- else:
- self.btnOK.Disable()
- self.infoLabel.SetLabel('')
-
- def SelectionChanged(self, event):
- """!Selected symbol changed."""
- if event.doubleClick:
- self.EndModal(wx.ID_OK)
- # deselect all
- for i in range(len(self.panels)):
- for panel in self.symbolPanels[i]:
- if panel.GetName() != event.name:
- panel.Deselect()
-
- self.btnOK.Enable()
-
- self.selected = event.name
- self.selectedDir = self.folderChoice.GetStringSelection()
-
- self.infoLabel.SetLabel(event.name)
-
- def GetSelectedSymbolName(self):
- """!Returns currently selected symbol name (e.g. 'basic/x').
- """
- # separator must be '/' and not dependent on OS
- return self.selectedDir + '/' + self.selected
- def GetSelectedSymbolPath(self):
- """!Returns currently selected symbol full path.
- """
- return os.path.join(self.symbolPath, self.selectedDir, self.selected)
|