gdialogs.py 50 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370
  1. """!
  2. @package gdialogs.py
  3. @brief Various dialogs used in wxGUI.
  4. List of classes:
  5. - ElementDialog
  6. - LocationDialog
  7. - MapsetDialog
  8. - NewVectorDialog
  9. - SavedRegion
  10. - DecorationDialog
  11. - TextLayerDialog
  12. - LoadMapLayersDialog
  13. - MultiImportDialog
  14. - LayersList (used by MultiImport)
  15. - SetOpacityDialog
  16. (C) 2008-2009 by the GRASS Development Team
  17. This program is free software under the GNU General Public
  18. License (>=v2). Read the file COPYING that comes with GRASS
  19. for details.
  20. @author Martin Landa <landa.martin gmail.com>
  21. """
  22. import os
  23. import sys
  24. import re
  25. import glob
  26. import wx
  27. import wx.lib.filebrowsebutton as filebrowse
  28. import wx.lib.mixins.listctrl as listmix
  29. from grass.script import core as grass
  30. import gcmd
  31. import globalvar
  32. import gselect
  33. import menuform
  34. import utils
  35. from preferences import globalSettings as UserSettings
  36. class ElementDialog(wx.Dialog):
  37. """!General dialog to choose given element (location, mapset, vector map, etc.)"""
  38. def __init__(self, parent, title, label, id = wx.ID_ANY,
  39. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  40. **kwargs):
  41. wx.Dialog.__init__(self, parent, id, title, style = style, **kwargs)
  42. self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
  43. self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
  44. self.btnOK = wx.Button(parent = self.panel, id = wx.ID_OK)
  45. self.btnOK.SetDefault()
  46. self.btnOK.Enable(False)
  47. self.label = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  48. label = label)
  49. self.element = None # must be defined
  50. self.__Layout()
  51. def PostInit(self):
  52. self.element.SetFocus()
  53. self.element.Bind(wx.EVT_TEXT, self.OnElement)
  54. def OnElement(self, event):
  55. """!Name for vector map layer given"""
  56. if len(event.GetString()) > 0:
  57. self.btnOK.Enable(True)
  58. else:
  59. self.btnOK.Enable(False)
  60. def __Layout(self):
  61. """!Do layout"""
  62. self.sizer = wx.BoxSizer(wx.VERTICAL)
  63. self.dataSizer = wx.BoxSizer(wx.VERTICAL)
  64. self.dataSizer.Add(self.label, proportion=0,
  65. flag=wx.ALL, border=1)
  66. # buttons
  67. btnSizer = wx.StdDialogButtonSizer()
  68. btnSizer.AddButton(self.btnCancel)
  69. btnSizer.AddButton(self.btnOK)
  70. btnSizer.Realize()
  71. self.sizer.Add(item=self.dataSizer, proportion=1,
  72. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  73. self.sizer.Add(item=btnSizer, proportion=0,
  74. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  75. def GetElement(self):
  76. """!Return (mapName, overwrite)"""
  77. return self.element.GetValue()
  78. class LocationDialog(ElementDialog):
  79. """!Dialog used to select location"""
  80. def __init__(self, parent, title = _("Select GRASS location and mapset"), id = wx.ID_ANY):
  81. ElementDialog.__init__(self, parent, title, label = _("Name of GRASS location:"))
  82. self.element = gselect.LocationSelect(parent = self.panel, id = wx.ID_ANY,
  83. size = globalvar.DIALOG_GSELECT_SIZE)
  84. self.element1 = gselect.MapsetSelect(parent = self.panel, id = wx.ID_ANY,
  85. size = globalvar.DIALOG_GSELECT_SIZE,
  86. setItems = False)
  87. self.PostInit()
  88. self.__Layout()
  89. self.SetMinSize(self.GetSize())
  90. def __Layout(self):
  91. """!Do layout"""
  92. self.dataSizer.Add(self.element, proportion=0,
  93. flag=wx.EXPAND | wx.ALL, border=1)
  94. self.dataSizer.Add(wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  95. label = _("Name of mapset:")), proportion=0,
  96. flag=wx.EXPAND | wx.ALL, border=1)
  97. self.dataSizer.Add(self.element1, proportion=0,
  98. flag=wx.EXPAND | wx.ALL, border=1)
  99. self.panel.SetSizer(self.sizer)
  100. self.sizer.Fit(self)
  101. def OnElement(self, event):
  102. """!Name for vector map layer given"""
  103. location = event.GetString()
  104. mapset = self.element1.GetValue()
  105. if location:
  106. dbase = grass.gisenv()['GISDBASE']
  107. self.element1.SetItems(utils.GetListOfMapsets(dbase, location, selectable = True))
  108. self.element1.SetSelection(0)
  109. mapset = self.element1.GetStringSelection()
  110. if location and mapset:
  111. self.btnOK.Enable(True)
  112. else:
  113. self.btnOK.Enable(False)
  114. def GetValues(self):
  115. """!Get location, mapset"""
  116. return (self.GetElement(), self.element1.GetValue())
  117. class MapsetDialog(ElementDialog):
  118. """!Dialog used to select mapset"""
  119. def __init__(self, parent, title = _("Select mapset in GRASS location"),
  120. location = None, id = wx.ID_ANY):
  121. ElementDialog.__init__(self, parent, title, label = _("Name of mapset:"))
  122. if location:
  123. self.SetTitle(self.GetTitle() + '<%s>' % location)
  124. else:
  125. self.SetTitle(self.GetTitle() + '<%s>' % grass.gisenv()['LOCATION_NAME'])
  126. self.element = gselect.MapsetSelect(parent = self.panel, id = wx.ID_ANY,
  127. size = globalvar.DIALOG_GSELECT_SIZE)
  128. self.PostInit()
  129. self.__Layout()
  130. self.SetMinSize(self.GetSize())
  131. def __Layout(self):
  132. """!Do layout"""
  133. self.dataSizer.Add(self.element, proportion=0,
  134. flag=wx.EXPAND | wx.ALL, border=1)
  135. self.panel.SetSizer(self.sizer)
  136. self.sizer.Fit(self)
  137. def GetMapset(self):
  138. return self.GetElement()
  139. class NewVectorDialog(ElementDialog):
  140. """!Dialog for creating new vector map"""
  141. def __init__(self, parent, id, title, disableAdd=False, disableTable=False,
  142. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  143. ElementDialog.__init__(self, parent, title, label = _("Name for new vector map:"))
  144. self.element = gselect.Select(parent=self.panel, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
  145. type='vector', mapsets=[grass.gisenv()['MAPSET'],])
  146. self.table = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
  147. label = _("Create attribute table"))
  148. self.table.SetValue(True)
  149. if disableTable:
  150. self.table.Enable(False)
  151. self.addbox = wx.CheckBox(parent = self.panel,
  152. label = _('Add created map into layer tree'), style = wx.NO_BORDER)
  153. if disableAdd:
  154. self.addbox.SetValue(True)
  155. self.addbox.Enable(False)
  156. else:
  157. self.addbox.SetValue(UserSettings.Get(group = 'cmd', key = 'addNewLayer', subkey = 'enabled'))
  158. self.PostInit()
  159. self.__Layout()
  160. self.SetMinSize(self.GetSize())
  161. def OnMapName(self, event):
  162. """!Name for vector map layer given"""
  163. self.OnElement(event)
  164. def __Layout(self):
  165. """!Do layout"""
  166. self.dataSizer.Add(self.element, proportion=0,
  167. flag=wx.EXPAND | wx.ALL, border=1)
  168. self.dataSizer.Add(self.table, proportion=0,
  169. flag=wx.EXPAND | wx.ALL, border=1)
  170. self.dataSizer.AddSpacer(5)
  171. self.dataSizer.Add(item=self.addbox, proportion=0,
  172. flag=wx.EXPAND | wx.ALL, border=1)
  173. self.panel.SetSizer(self.sizer)
  174. self.sizer.Fit(self)
  175. def GetName(self):
  176. """!Return (mapName, overwrite)"""
  177. return self.GetElement().split('@', 1)[0]
  178. def CreateNewVector(parent, cmd, title=_('Create new vector map'),
  179. exceptMap=None, log=None, disableAdd=False, disableTable=False):
  180. """!Create new vector map layer
  181. @cmd cmd (prog, **kwargs)
  182. @return tuple (name of create vector map, add to layer tree)
  183. @return None of failure
  184. """
  185. dlg = NewVectorDialog(parent, wx.ID_ANY, title,
  186. disableAdd, disableTable)
  187. if dlg.ShowModal() == wx.ID_OK:
  188. outmap = dlg.GetName()
  189. if outmap == exceptMap:
  190. wx.MessageBox(parent=parent,
  191. message=_("Unable to create vector map <%s>.") % outmap,
  192. caption=_("Error"),
  193. style=wx.ID_OK | wx.ICON_ERROR | wx.CENTRE)
  194. return False
  195. if outmap == '': # should not happen
  196. return False
  197. cmd[1][cmd[2]] = outmap
  198. try:
  199. listOfVectors = grass.list_grouped('vect')[grass.gisenv()['MAPSET']]
  200. except KeyError:
  201. listOfVectors = []
  202. overwrite = False
  203. if not UserSettings.Get(group='cmd', key='overwrite', subkey='enabled') and \
  204. outmap in listOfVectors:
  205. dlgOw = wx.MessageDialog(parent, message=_("Vector map <%s> already exists "
  206. "in the current mapset. "
  207. "Do you want to overwrite it?") % outmap,
  208. caption=_("Overwrite?"),
  209. style=wx.YES_NO | wx.YES_DEFAULT | wx.ICON_QUESTION)
  210. if dlgOw.ShowModal() == wx.ID_YES:
  211. overwrite = True
  212. else:
  213. dlgOw.Destroy()
  214. return False
  215. if UserSettings.Get(group='cmd', key='overwrite', subkey='enabled') is True:
  216. overwrite = True
  217. try:
  218. gcmd.RunCommand(prog = cmd[0],
  219. overwrite = overwrite,
  220. **cmd[1])
  221. except gcmd.CmdError, e:
  222. e.Show()
  223. return None
  224. #
  225. # create attribute table
  226. #
  227. if dlg.table.IsEnabled() and dlg.table.IsChecked():
  228. key = UserSettings.Get(group='atm', key='keycolumn', subkey='value')
  229. sql = 'CREATE TABLE %s (%s INTEGER)' % (outmap, key)
  230. gcmd.RunCommand('db.execute',
  231. quiet = True,
  232. parent = parent,
  233. input = '-',
  234. stdin = sql)
  235. gcmd.RunCommand('v.db.connect',
  236. quiet = True,
  237. parent = parent,
  238. map = outmap,
  239. table = outmap,
  240. key = key,
  241. layer = '1')
  242. # return fully qualified map name
  243. if '@' not in outmap:
  244. outmap += '@' + grass.gisenv()['MAPSET']
  245. if log:
  246. log.WriteLog(_("New vector map <%s> created") % outmap)
  247. return (outmap, dlg.addbox.IsChecked())
  248. return (None, dlg.addbox.IsChecked())
  249. class SavedRegion(wx.Dialog):
  250. def __init__(self, parent, id = wx.ID_ANY, title="", loadsave='load',
  251. **kwargs):
  252. """!Loading and saving of display extents to saved region file
  253. @param loadsave load or save region?
  254. """
  255. wx.Dialog.__init__(self, parent, id, title, **kwargs)
  256. self.loadsave = loadsave
  257. self.wind = ''
  258. sizer = wx.BoxSizer(wx.VERTICAL)
  259. box = wx.BoxSizer(wx.HORIZONTAL)
  260. label = wx.StaticText(parent=self, id=wx.ID_ANY)
  261. box.Add(item=label, proportion=0, flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
  262. if loadsave == 'load':
  263. label.SetLabel(_("Load region:"))
  264. selection = gselect.Select(parent=self, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
  265. type='windows')
  266. elif loadsave == 'save':
  267. label.SetLabel(_("Save region:"))
  268. selection = gselect.Select(parent=self, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
  269. type='windows', mapsets = [grass.gisenv()['MAPSET']])
  270. box.Add(item=selection, proportion=0, flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
  271. selection.SetFocus()
  272. selection.Bind(wx.EVT_TEXT, self.OnRegion)
  273. sizer.Add(item=box, proportion=0, flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL,
  274. border=5)
  275. line = wx.StaticLine(parent=self, id=wx.ID_ANY, size=(20, -1), style=wx.LI_HORIZONTAL)
  276. sizer.Add(item=line, proportion=0,
  277. flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.LEFT|wx.RIGHT, border=5)
  278. btnsizer = wx.StdDialogButtonSizer()
  279. btn = wx.Button(parent = self, id = wx.ID_OK)
  280. btn.SetDefault()
  281. btnsizer.AddButton(btn)
  282. btn = wx.Button(parent = self, id = wx.ID_CANCEL)
  283. btnsizer.AddButton(btn)
  284. btnsizer.Realize()
  285. sizer.Add(item=btnsizer, proportion=0, flag=wx.ALIGN_RIGHT | wx.ALL, border=5)
  286. self.SetSizer(sizer)
  287. sizer.Fit(self)
  288. self.Layout()
  289. def OnRegion(self, event):
  290. self.wind = event.GetString()
  291. class DecorationDialog(wx.Dialog):
  292. """
  293. Controls setting options and displaying/hiding map overlay decorations
  294. """
  295. def __init__(self, parent, ovlId, title, cmd, name=None,
  296. pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE,
  297. checktxt='', ctrltxt=''):
  298. wx.Dialog.__init__(self, parent, wx.ID_ANY, title, pos, size, style)
  299. self.ovlId = ovlId # PseudoDC id
  300. self.cmd = cmd
  301. self.name = name # overlay name
  302. self.parent = parent # MapFrame
  303. sizer = wx.BoxSizer(wx.VERTICAL)
  304. box = wx.BoxSizer(wx.HORIZONTAL)
  305. self.chkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=checktxt)
  306. if self.parent.Map.GetOverlay(self.ovlId) is None:
  307. self.chkbox.SetValue(True)
  308. else:
  309. self.chkbox.SetValue(self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive())
  310. box.Add(item=self.chkbox, proportion=0,
  311. flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
  312. sizer.Add(item=box, proportion=0,
  313. flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
  314. box = wx.BoxSizer(wx.HORIZONTAL)
  315. optnbtn = wx.Button(parent=self, id=wx.ID_ANY, label=_("Set options"))
  316. box.Add(item=optnbtn, proportion=0, flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
  317. sizer.Add(item=box, proportion=0,
  318. flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
  319. box = wx.BoxSizer(wx.HORIZONTAL)
  320. label = wx.StaticText(parent=self, id=wx.ID_ANY,
  321. label=_("Drag %s with mouse in pointer mode to position.\n"
  322. "Double-click to change options." % ctrltxt))
  323. if self.name == 'legend':
  324. label.SetLabel(label.GetLabel() + _('\nDefine raster map name for legend in '
  325. 'properties dialog.'))
  326. box.Add(item=label, proportion=0,
  327. flag=wx.ALIGN_CENTRE|wx.ALL, border=5)
  328. sizer.Add(item=box, proportion=0,
  329. flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
  330. line = wx.StaticLine(parent=self, id=wx.ID_ANY, size=(20,-1), style=wx.LI_HORIZONTAL)
  331. sizer.Add(item=line, proportion=0,
  332. flag=wx.GROW|wx.ALIGN_CENTER_VERTICAL|wx.ALL, border=5)
  333. # buttons
  334. btnsizer = wx.StdDialogButtonSizer()
  335. self.btnOK = wx.Button(parent=self, id=wx.ID_OK)
  336. self.btnOK.SetDefault()
  337. if self.name == 'legend':
  338. self.btnOK.Enable(False)
  339. btnsizer.AddButton(self.btnOK)
  340. btnCancel = wx.Button(parent=self, id=wx.ID_CANCEL)
  341. btnsizer.AddButton(btnCancel)
  342. btnsizer.Realize()
  343. sizer.Add(item=btnsizer, proportion=0,
  344. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  345. #
  346. # bindings
  347. #
  348. self.Bind(wx.EVT_BUTTON, self.OnOptions, optnbtn)
  349. self.Bind(wx.EVT_BUTTON, self.OnCancel, btnCancel)
  350. self.Bind(wx.EVT_BUTTON, self.OnOK, self.btnOK)
  351. self.SetSizer(sizer)
  352. sizer.Fit(self)
  353. # create overlay if doesn't exist
  354. self._CreateOverlay()
  355. if len(self.parent.MapWindow.overlays[self.ovlId]['cmd']) > 1:
  356. mapName = utils.GetLayerNameFromCmd(self.parent.MapWindow.overlays[self.ovlId]['cmd'])
  357. if self.parent.MapWindow.overlays[self.ovlId]['propwin'] is None and mapName:
  358. # build properties dialog
  359. menuform.GUI().ParseCommand(cmd=self.cmd,
  360. completed=(self.GetOptData, self.name, ''),
  361. parentframe=self.parent, show=False)
  362. if mapName:
  363. # enable 'OK' button
  364. self.btnOK.Enable()
  365. if name == 'legend':
  366. # set title
  367. self.SetTitle(_('Legend of raster map <%s>') % \
  368. mapName)
  369. def _CreateOverlay(self):
  370. if not self.parent.Map.GetOverlay(self.ovlId):
  371. overlay = self.parent.Map.AddOverlay(id=self.ovlId, type=self.name,
  372. command=self.cmd,
  373. l_active=False, l_render=False, l_hidden=True)
  374. self.parent.MapWindow.overlays[self.ovlId] = {}
  375. self.parent.MapWindow.overlays[self.ovlId] = { 'layer' : overlay,
  376. 'params' : None,
  377. 'propwin' : None,
  378. 'cmd' : self.cmd,
  379. 'coords': (10, 10),
  380. 'pdcType': 'image' }
  381. else:
  382. if self.parent.MapWindow.overlays[self.ovlId]['propwin'] == None:
  383. return
  384. self.parent.MapWindow.overlays[self.ovlId]['propwin'].get_dcmd = self.GetOptData
  385. def OnOptions(self, event):
  386. """ self.SetSizer(sizer)
  387. sizer.Fit(self)
  388. Sets option for decoration map overlays
  389. """
  390. if self.parent.MapWindow.overlays[self.ovlId]['propwin'] is None:
  391. # build properties dialog
  392. menuform.GUI().ParseCommand(cmd=self.cmd,
  393. completed=(self.GetOptData, self.name, ''),
  394. parentframe=self.parent)
  395. else:
  396. if self.parent.MapWindow.overlays[self.ovlId]['propwin'].IsShown():
  397. self.parent.MapWindow.overlays[self.ovlId]['propwin'].SetFocus()
  398. else:
  399. self.parent.MapWindow.overlays[self.ovlId]['propwin'].Show()
  400. def OnCancel(self, event):
  401. """!Cancel dialog"""
  402. self.parent.dialogs['barscale'] = None
  403. self.Destroy()
  404. def OnOK(self, event):
  405. """!Button 'OK' pressed"""
  406. # enable or disable overlay
  407. self.parent.Map.GetOverlay(self.ovlId).SetActive(self.chkbox.IsChecked())
  408. # update map
  409. self.parent.MapWindow.UpdateMap()
  410. # close dialog
  411. self.OnCancel(None)
  412. def GetOptData(self, dcmd, layer, params, propwin):
  413. """!Process decoration layer data"""
  414. # update layer data
  415. if params:
  416. self.parent.MapWindow.overlays[self.ovlId]['params'] = params
  417. if dcmd:
  418. self.parent.MapWindow.overlays[self.ovlId]['cmd'] = dcmd
  419. self.parent.MapWindow.overlays[self.ovlId]['propwin'] = propwin
  420. # change parameters for item in layers list in render.Map
  421. # "Use mouse..." (-m) flag causes GUI freeze and is pointless here, trac #119
  422. try:
  423. self.parent.MapWindow.overlays[self.ovlId]['cmd'].remove('-m')
  424. except ValueError:
  425. pass
  426. self.parent.Map.ChangeOverlay(id=self.ovlId, type=self.name,
  427. command=self.parent.MapWindow.overlays[self.ovlId]['cmd'],
  428. l_active=self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive(),
  429. l_render=False, l_hidden=True)
  430. if self.name == 'legend':
  431. if params and not self.btnOK.IsEnabled():
  432. self.btnOK.Enable()
  433. class TextLayerDialog(wx.Dialog):
  434. """
  435. Controls setting options and displaying/hiding map overlay decorations
  436. """
  437. def __init__(self, parent, ovlId, title, name='text',
  438. pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.DEFAULT_DIALOG_STYLE):
  439. wx.Dialog.__init__(self, parent, wx.ID_ANY, title, pos, size, style)
  440. from wx.lib.expando import ExpandoTextCtrl, EVT_ETC_LAYOUT_NEEDED
  441. self.ovlId = ovlId
  442. self.parent = parent
  443. if self.ovlId in self.parent.MapWindow.textdict.keys():
  444. self.currText = self.parent.MapWindow.textdict[self.ovlId]['text']
  445. self.currFont = self.parent.MapWindow.textdict[self.ovlId]['font']
  446. self.currClr = self.parent.MapWindow.textdict[self.ovlId]['color']
  447. self.currRot = self.parent.MapWindow.textdict[self.ovlId]['rotation']
  448. self.currCoords = self.parent.MapWindow.textdict[self.ovlId]['coords']
  449. else:
  450. self.currClr = wx.BLACK
  451. self.currText = ''
  452. self.currFont = self.GetFont()
  453. self.currRot = 0.0
  454. self.currCoords = [10, 10, 10, 10]
  455. self.sizer = wx.BoxSizer(wx.VERTICAL)
  456. box = wx.GridBagSizer(vgap=5, hgap=5)
  457. # show/hide
  458. self.chkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, \
  459. label='Show text object')
  460. if self.parent.Map.GetOverlay(self.ovlId) is None:
  461. self.chkbox.SetValue(True)
  462. else:
  463. self.chkbox.SetValue(self.parent.MapWindow.overlays[self.ovlId]['layer'].IsActive())
  464. box.Add(item=self.chkbox, span=(1,2),
  465. flag=wx.ALIGN_LEFT|wx.ALL, border=5,
  466. pos=(0, 0))
  467. # text entry
  468. label = wx.StaticText(parent=self, id=wx.ID_ANY, label=_("Enter text:"))
  469. box.Add(item=label,
  470. flag=wx.ALIGN_CENTER_VERTICAL,
  471. pos=(1, 0))
  472. self.textentry = ExpandoTextCtrl(parent=self, id=wx.ID_ANY, value="", size=(300,-1))
  473. self.textentry.SetFont(self.currFont)
  474. self.textentry.SetForegroundColour(self.currClr)
  475. self.textentry.SetValue(self.currText)
  476. # get rid of unneeded scrollbar when text box first opened
  477. self.textentry.SetClientSize((300,-1))
  478. box.Add(item=self.textentry,
  479. pos=(1, 1))
  480. # rotation
  481. label = wx.StaticText(parent=self, id=wx.ID_ANY, label=_("Rotation:"))
  482. box.Add(item=label,
  483. flag=wx.ALIGN_CENTER_VERTICAL,
  484. pos=(2, 0))
  485. self.rotation = wx.SpinCtrl(parent=self, id=wx.ID_ANY, value="", pos=(30, 50),
  486. size=(75,-1), style=wx.SP_ARROW_KEYS)
  487. self.rotation.SetRange(-360, 360)
  488. self.rotation.SetValue(int(self.currRot))
  489. box.Add(item=self.rotation,
  490. flag=wx.ALIGN_RIGHT,
  491. pos=(2, 1))
  492. # font
  493. fontbtn = wx.Button(parent=self, id=wx.ID_ANY, label=_("Set font"))
  494. box.Add(item=fontbtn,
  495. flag=wx.ALIGN_RIGHT,
  496. pos=(3, 1))
  497. self.sizer.Add(item=box, proportion=1,
  498. flag=wx.ALL, border=10)
  499. # note
  500. box = wx.BoxSizer(wx.HORIZONTAL)
  501. label = wx.StaticText(parent=self, id=wx.ID_ANY,
  502. label=_("Drag text with mouse in pointer mode "
  503. "to position.\nDouble-click to change options"))
  504. box.Add(item=label, proportion=0,
  505. flag=wx.ALIGN_CENTRE | wx.ALL, border=5)
  506. self.sizer.Add(item=box, proportion=0,
  507. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER | wx.ALL, border=5)
  508. line = wx.StaticLine(parent=self, id=wx.ID_ANY,
  509. size=(20,-1), style=wx.LI_HORIZONTAL)
  510. self.sizer.Add(item=line, proportion=0,
  511. flag=wx.EXPAND | wx.ALIGN_CENTRE | wx.ALL, border=5)
  512. btnsizer = wx.StdDialogButtonSizer()
  513. btn = wx.Button(parent=self, id=wx.ID_OK)
  514. btn.SetDefault()
  515. btnsizer.AddButton(btn)
  516. btn = wx.Button(parent=self, id=wx.ID_CANCEL)
  517. btnsizer.AddButton(btn)
  518. btnsizer.Realize()
  519. self.sizer.Add(item=btnsizer, proportion=0,
  520. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  521. self.SetSizer(self.sizer)
  522. self.sizer.Fit(self)
  523. # bindings
  524. self.Bind(EVT_ETC_LAYOUT_NEEDED, self.OnRefit, self.textentry)
  525. self.Bind(wx.EVT_BUTTON, self.OnSelectFont, fontbtn)
  526. self.Bind(wx.EVT_TEXT, self.OnText, self.textentry)
  527. self.Bind(wx.EVT_SPINCTRL, self.OnRotation, self.rotation)
  528. def OnRefit(self, event):
  529. """!Resize text entry to match text"""
  530. self.sizer.Fit(self)
  531. def OnText(self, event):
  532. """!Change text string"""
  533. self.currText = event.GetString()
  534. def OnRotation(self, event):
  535. """!Change rotation"""
  536. self.currRot = event.GetInt()
  537. event.Skip()
  538. def OnSelectFont(self, event):
  539. """!Change font"""
  540. data = wx.FontData()
  541. data.EnableEffects(True)
  542. data.SetColour(self.currClr) # set colour
  543. data.SetInitialFont(self.currFont)
  544. dlg = wx.FontDialog(self, data)
  545. if dlg.ShowModal() == wx.ID_OK:
  546. data = dlg.GetFontData()
  547. self.currFont = data.GetChosenFont()
  548. self.currClr = data.GetColour()
  549. self.textentry.SetFont(self.currFont)
  550. self.textentry.SetForegroundColour(self.currClr)
  551. self.Layout()
  552. dlg.Destroy()
  553. def GetValues(self):
  554. """!Get text properties"""
  555. return { 'text' : self.currText,
  556. 'font' : self.currFont,
  557. 'color' : self.currClr,
  558. 'rotation' : self.currRot,
  559. 'coords' : self.currCoords,
  560. 'active' : self.chkbox.IsChecked() }
  561. class LoadMapLayersDialog(wx.Dialog):
  562. """!Load selected map layers (raster, vector) into layer tree"""
  563. def __init__(self, parent, title, style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  564. wx.Dialog.__init__(self, parent=parent, id=wx.ID_ANY, title=title, style=style)
  565. self.parent = parent # GMFrame
  566. #
  567. # dialog body
  568. #
  569. self.bodySizer = self.__createDialogBody()
  570. # update list of layer to be loaded
  571. self.map_layers = [] # list of map layers (full list type/mapset)
  572. self.LoadMapLayers(self.layerType.GetStringSelection()[:4],
  573. self.mapset.GetStringSelection())
  574. #
  575. # buttons
  576. #
  577. btnCancel = wx.Button(self, wx.ID_CANCEL)
  578. btnOk = wx.Button(self, wx.ID_OK, _("Load") )
  579. btnOk.SetDefault()
  580. #
  581. # sizers & do layout
  582. #
  583. btnSizer = wx.StdDialogButtonSizer()
  584. btnSizer.AddButton(btnCancel)
  585. btnSizer.AddButton(btnOk)
  586. btnSizer.Realize()
  587. mainSizer = wx.BoxSizer(wx.VERTICAL)
  588. mainSizer.Add(item=self.bodySizer, proportion=1,
  589. flag=wx.EXPAND | wx.ALL, border=5)
  590. mainSizer.Add(item=btnSizer, proportion=0,
  591. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  592. self.SetSizer(mainSizer)
  593. mainSizer.Fit(self)
  594. # set dialog min size
  595. self.SetMinSize(self.GetSize())
  596. def __createDialogBody(self):
  597. bodySizer = wx.GridBagSizer(vgap=3, hgap=3)
  598. bodySizer.AddGrowableCol(1)
  599. bodySizer.AddGrowableRow(3)
  600. # layer type
  601. bodySizer.Add(item=wx.StaticText(parent=self, label=_("Map layer type:")),
  602. flag=wx.ALIGN_CENTER_VERTICAL,
  603. pos=(0,0))
  604. self.layerType = wx.Choice(parent=self, id=wx.ID_ANY,
  605. choices=['raster', 'vector'], size=(100,-1))
  606. self.layerType.SetSelection(0)
  607. bodySizer.Add(item=self.layerType,
  608. pos=(0,1))
  609. # select toggle
  610. self.toggle = wx.CheckBox(parent=self, id=wx.ID_ANY,
  611. label=_("Select toggle"))
  612. self.toggle.SetValue(True)
  613. bodySizer.Add(item=self.toggle,
  614. flag=wx.ALIGN_CENTER_VERTICAL,
  615. pos=(0,2))
  616. # mapset filter
  617. bodySizer.Add(item=wx.StaticText(parent=self, label=_("Mapset:")),
  618. flag=wx.ALIGN_CENTER_VERTICAL,
  619. pos=(1,0))
  620. self.mapset = gselect.MapsetSelect(parent = self)
  621. self.mapset.SetStringSelection(grass.gisenv()['MAPSET'])
  622. bodySizer.Add(item=self.mapset,
  623. pos=(1,1), span=(1, 2))
  624. # map name filter
  625. bodySizer.Add(item=wx.StaticText(parent=self, label=_("Filter:")),
  626. flag=wx.ALIGN_CENTER_VERTICAL,
  627. pos=(2,0))
  628. self.filter = wx.TextCtrl(parent=self, id=wx.ID_ANY,
  629. value="",
  630. size=(250,-1))
  631. bodySizer.Add(item=self.filter,
  632. flag=wx.EXPAND,
  633. pos=(2,1), span=(1, 2))
  634. # layer list
  635. bodySizer.Add(item=wx.StaticText(parent=self, label=_("List of maps:")),
  636. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_TOP,
  637. pos=(3,0))
  638. self.layers = wx.CheckListBox(parent=self, id=wx.ID_ANY,
  639. size=(250, 100),
  640. choices=[])
  641. bodySizer.Add(item=self.layers,
  642. flag=wx.EXPAND,
  643. pos=(3,1), span=(1, 2))
  644. # bindings
  645. self.layerType.Bind(wx.EVT_CHOICE, self.OnChangeParams)
  646. self.mapset.Bind(wx.EVT_COMBOBOX, self.OnChangeParams)
  647. self.layers.Bind(wx.EVT_RIGHT_DOWN, self.OnMenu)
  648. self.filter.Bind(wx.EVT_TEXT, self.OnFilter)
  649. self.toggle.Bind(wx.EVT_CHECKBOX, self.OnToggle)
  650. return bodySizer
  651. def LoadMapLayers(self, type, mapset):
  652. """!Load list of map layers
  653. @param type layer type ('raster' or 'vector')
  654. @param mapset mapset name
  655. """
  656. try:
  657. self.map_layers = grass.mlist_grouped(type=type, mapset=mapset)[mapset]
  658. except KeyError:
  659. self.map_layers = []
  660. self.layers.Set(self.map_layers)
  661. # check all items by default
  662. for item in range(self.layers.GetCount()):
  663. self.layers.Check(item)
  664. def OnChangeParams(self, event):
  665. """!Filter parameters changed by user"""
  666. # update list of layer to be loaded
  667. self.LoadMapLayers(self.layerType.GetStringSelection()[:4],
  668. self.mapset.GetStringSelection())
  669. event.Skip()
  670. def OnMenu(self, event):
  671. """!Table description area, context menu"""
  672. if not hasattr(self, "popupID1"):
  673. self.popupDataID1 = wx.NewId()
  674. self.popupDataID2 = wx.NewId()
  675. self.popupDataID3 = wx.NewId()
  676. self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
  677. self.Bind(wx.EVT_MENU, self.OnSelectInvert, id=self.popupDataID2)
  678. self.Bind(wx.EVT_MENU, self.OnDeselectAll, id=self.popupDataID3)
  679. # generate popup-menu
  680. menu = wx.Menu()
  681. menu.Append(self.popupDataID1, _("Select all"))
  682. menu.Append(self.popupDataID2, _("Invert selection"))
  683. menu.Append(self.popupDataID3, _("Deselect all"))
  684. self.PopupMenu(menu)
  685. menu.Destroy()
  686. def OnSelectAll(self, event):
  687. """!Select all map layer from list"""
  688. for item in range(self.layers.GetCount()):
  689. self.layers.Check(item, True)
  690. def OnSelectInvert(self, event):
  691. """!Invert current selection"""
  692. for item in range(self.layers.GetCount()):
  693. if self.layers.IsChecked(item):
  694. self.layers.Check(item, False)
  695. else:
  696. self.layers.Check(item, True)
  697. def OnDeselectAll(self, event):
  698. """!Select all map layer from list"""
  699. for item in range(self.layers.GetCount()):
  700. self.layers.Check(item, False)
  701. def OnFilter(self, event):
  702. """!Apply filter for map names"""
  703. if len(event.GetString()) == 0:
  704. self.layers.Set(self.map_layers)
  705. return
  706. list = []
  707. for layer in self.map_layers:
  708. try:
  709. if re.compile('^' + event.GetString()).search(layer):
  710. list.append(layer)
  711. except:
  712. pass
  713. self.layers.Set(list)
  714. self.OnSelectAll(None)
  715. event.Skip()
  716. def OnToggle(self, event):
  717. """!Select toggle (check or uncheck all layers)"""
  718. check = event.Checked()
  719. for item in range(self.layers.GetCount()):
  720. self.layers.Check(item, check)
  721. event.Skip()
  722. def GetMapLayers(self):
  723. """!Return list of checked map layers"""
  724. layerNames = []
  725. for indx in self.layers.GetSelections():
  726. # layers.append(self.layers.GetStringSelec(indx))
  727. pass
  728. # return fully qualified map names
  729. mapset = self.mapset.GetStringSelection()
  730. for item in range(self.layers.GetCount()):
  731. if not self.layers.IsChecked(item):
  732. continue
  733. layerNames.append(self.layers.GetString(item) + '@' + mapset)
  734. return layerNames
  735. def GetLayerType(self):
  736. """!Get selected layer type"""
  737. return self.layerType.GetStringSelection()
  738. class MultiImportDialog(wx.Dialog):
  739. """!Import dxf layers"""
  740. def __init__(self, parent, type,
  741. id=wx.ID_ANY, title=_("Multiple import"),
  742. link = False,
  743. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  744. self.parent = parent # GMFrame
  745. self.inputType = type
  746. self.link = link # Link or import data (only for GDAL/OGR)
  747. wx.Dialog.__init__(self, parent, id, title, style=style)
  748. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  749. if self.inputType == 'dxf':
  750. self.inputTitle = _("Input DXF file")
  751. else:
  752. self.inputTitle = _("Input directory")
  753. self.inputBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
  754. label=" %s " % self.inputTitle)
  755. self.layerBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
  756. label=_(" List of %s layers ") % self.inputType.upper())
  757. #
  758. # input
  759. #
  760. if self.inputType == 'dxf':
  761. self.inputText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Choose DXF file:"))
  762. self.input = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
  763. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  764. dialogTitle=_('Choose DXF file to import'),
  765. buttonText=_('Browse'),
  766. startDirectory=os.getcwd(), fileMode=0,
  767. changeCallback=self.OnSetInput,
  768. fileMask="*.dxf")
  769. else:
  770. self.inputText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Choose directory:"))
  771. self.input = filebrowse.DirBrowseButton(parent=self.panel, id=wx.ID_ANY,
  772. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  773. dialogTitle=_('Choose input directory'),
  774. buttonText=_('Browse'),
  775. startDirectory=os.getcwd(),
  776. changeCallback=self.OnSetInput)
  777. self.formatText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Select file extension:"))
  778. self.format = wx.TextCtrl(parent=self.panel, id=wx.ID_ANY, size=(100, -1),
  779. value="")
  780. if self.inputType == 'gdal':
  781. self.format.SetValue('tif')
  782. else: # ogr
  783. self.format.SetValue('shp')
  784. self.format.Bind(wx.EVT_TEXT, self.OnSetInput)
  785. #
  786. # list of layers
  787. #
  788. self.list = LayersList(self.panel)
  789. self.list.LoadData()
  790. self.add = wx.CheckBox(parent=self.panel, id=wx.ID_ANY)
  791. if link:
  792. self.add.SetLabel(_("Add linked layers into layer tree"))
  793. else:
  794. self.add.SetLabel(_("Add imported layers into layer tree"))
  795. self.add.SetValue(UserSettings.Get(group='cmd', key='addNewLayer', subkey='enabled'))
  796. #
  797. # buttons
  798. #
  799. # cancel
  800. self.btn_cancel = wx.Button(parent=self.panel, id=wx.ID_CANCEL)
  801. self.btn_cancel.SetToolTipString(_("Close dialog"))
  802. self.btn_cancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  803. # run
  804. if link:
  805. self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Link"))
  806. self.btn_run.SetToolTipString(_("Link selected layers"))
  807. else:
  808. self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Import"))
  809. self.btn_run.SetToolTipString(_("Import selected layers"))
  810. self.btn_run.SetDefault()
  811. self.btn_run.Enable(False)
  812. self.btn_run.Bind(wx.EVT_BUTTON, self.OnRun)
  813. self.__doLayout()
  814. self.Layout()
  815. def __doLayout(self):
  816. dialogSizer = wx.BoxSizer(wx.VERTICAL)
  817. #
  818. # input
  819. #
  820. inputSizer = wx.StaticBoxSizer(self.inputBox, wx.HORIZONTAL)
  821. gridSizer = wx.FlexGridSizer(cols=2, vgap=5, hgap=5)
  822. gridSizer.Add(item=self.inputText,
  823. flag=wx.ALIGN_CENTER_VERTICAL)
  824. gridSizer.AddGrowableCol(1)
  825. gridSizer.Add(item=self.input,
  826. flag=wx.EXPAND | wx.ALL)
  827. if self.inputType != 'dxf':
  828. gridSizer.Add(item=self.formatText,
  829. flag=wx.ALIGN_CENTER_VERTICAL)
  830. gridSizer.Add(item=self.format)
  831. inputSizer.Add(item=gridSizer, proportion=1,
  832. flag=wx.EXPAND | wx.ALL)
  833. dialogSizer.Add(item=inputSizer, proportion=0,
  834. flag=wx.ALL | wx.EXPAND, border=5)
  835. #
  836. # list of DXF layers
  837. #
  838. layerSizer = wx.StaticBoxSizer(self.layerBox, wx.HORIZONTAL)
  839. layerSizer.Add(item=self.list, proportion=1,
  840. flag=wx.ALL | wx.EXPAND, border=5)
  841. dialogSizer.Add(item=layerSizer, proportion=1,
  842. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  843. dialogSizer.Add(item=self.add, proportion=0,
  844. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  845. #
  846. # buttons
  847. #
  848. btnsizer = wx.BoxSizer(orient=wx.HORIZONTAL)
  849. btnsizer.Add(item=self.btn_cancel, proportion=0,
  850. flag=wx.ALL | wx.ALIGN_CENTER,
  851. border=10)
  852. btnsizer.Add(item=self.btn_run, proportion=0,
  853. flag=wx.ALL | wx.ALIGN_CENTER,
  854. border=10)
  855. dialogSizer.Add(item=btnsizer, proportion=0,
  856. flag=wx.ALIGN_CENTER)
  857. # dialogSizer.SetSizeHints(self.panel)
  858. self.panel.SetAutoLayout(True)
  859. self.panel.SetSizer(dialogSizer)
  860. dialogSizer.Fit(self.panel)
  861. self.Layout()
  862. # auto-layout seems not work here - FIXME
  863. self.SetMinSize((globalvar.DIALOG_GSELECT_SIZE[0] + 175, 300))
  864. width = self.GetSize()[0]
  865. self.list.SetColumnWidth(col=1, width=width/2 - 50)
  866. def OnCancel(self, event=None):
  867. """!Close dialog"""
  868. self.Close()
  869. def OnRun(self, event):
  870. """!Import/Link data (each layes as separate vector map)"""
  871. data = self.list.GetLayers()
  872. # hide dialog
  873. self.Hide()
  874. for layer, output in data:
  875. if self.inputType == 'dxf':
  876. cmd = ['v.in.dxf',
  877. 'input=%s' % self.input.GetValue(),
  878. 'layers=%s' % layer,
  879. 'output=%s' % output]
  880. elif self.inputType == 'ogr':
  881. if self.link:
  882. cmd = ['v.external',
  883. 'dsn=%s' % os.path.join(self.input.GetValue()),
  884. 'output=%s' % output,
  885. 'layer=%s' % layer.rstrip('.' + self.format.GetValue())
  886. ]
  887. else:
  888. cmd = ['v.in.ogr',
  889. 'dsn=%s' % (os.path.join(self.input.GetValue(), layer)),
  890. 'output=%s' % output]
  891. else: # gdal
  892. if self.link:
  893. cmd = ['r.external', '-o', # override projection by default
  894. 'input=%s' % (os.path.join(self.input.GetValue(), layer)),
  895. 'output=%s' % output]
  896. else:
  897. cmd = ['r.in.gdal', '-o', # override projection by default
  898. 'input=%s' % (os.path.join(self.input.GetValue(), layer)),
  899. 'output=%s' % output]
  900. if UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'):
  901. cmd.append('--overwrite')
  902. # run in Layer Manager
  903. self.parent.goutput.RunCmd(cmd, switchPage=True)
  904. if self.add.IsChecked():
  905. maptree = self.parent.curr_page.maptree
  906. for layer, output in data:
  907. if '@' not in output:
  908. name = output + '@' + grass.gisenv()['MAPSET']
  909. else:
  910. name = output
  911. # add imported layers into layer tree
  912. if self.inputType == 'gdal':
  913. cmd = ['d.rast',
  914. 'map=%s' % name]
  915. if UserSettings.Get(group='cmd', key='rasterOverlay', subkey='enabled'):
  916. cmd.append('-o')
  917. maptree.AddLayer(ltype='raster',
  918. lname=name,
  919. lcmd=cmd)
  920. else:
  921. maptree.AddLayer(ltype='vector',
  922. lname=name,
  923. lcmd=['d.vect',
  924. 'map=%s' % name])
  925. ### wx.CallAfter(self.parent.notebook.SetSelection, 0)
  926. self.OnCancel()
  927. def OnAbort(self, event):
  928. """!Abort running import
  929. @todo not yet implemented
  930. """
  931. pass
  932. def OnSetInput(self, event):
  933. """!Input DXF file/OGR dsn defined, update list of layer widget"""
  934. path = event.GetString()
  935. if self.inputType == 'dxf':
  936. ret = gcmd.RunCommand('v.in.dxf',
  937. quiet = True,
  938. parent = self,
  939. read = True,
  940. flags = 'l',
  941. input = path)
  942. if not ret:
  943. self.list.LoadData()
  944. self.btn_run.Enable(False)
  945. return
  946. data = []
  947. if self.inputType == 'dxf':
  948. for line in ret.splitlines():
  949. layerId = line.split(':')[0].split(' ')[1]
  950. layerName = line.split(':')[1].strip()
  951. grassName = utils.GetValidLayerName(layerName)
  952. data.append((layerId, layerName.strip(), grassName.strip()))
  953. else: # gdal/ogr (for ogr maybe to use v.in.ogr -l)
  954. layerId = 1
  955. for file in glob.glob(os.path.join(self.input.GetValue(), "*.%s") % self.format.GetValue()):
  956. baseName = os.path.basename(file)
  957. grassName = utils.GetValidLayerName(baseName.split('.', -1)[0])
  958. data.append((layerId, baseName, grassName))
  959. layerId += 1
  960. self.list.LoadData(data)
  961. if len(data) > 0:
  962. self.btn_run.Enable(True)
  963. else:
  964. self.btn_run.Enable(False)
  965. class LayersList(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin,
  966. listmix.CheckListCtrlMixin):
  967. # listmix.CheckListCtrlMixin, listmix.TextEditMixin):
  968. """!List of layers to be imported (dxf, shp...)"""
  969. def __init__(self, parent, pos=wx.DefaultPosition,
  970. log=None):
  971. self.parent = parent
  972. wx.ListCtrl.__init__(self, parent, wx.ID_ANY,
  973. style=wx.LC_REPORT)
  974. listmix.CheckListCtrlMixin.__init__(self)
  975. self.log = log
  976. # setup mixins
  977. listmix.ListCtrlAutoWidthMixin.__init__(self)
  978. # listmix.TextEditMixin.__init__(self)
  979. self.InsertColumn(0, _('Layer'))
  980. self.InsertColumn(1, _('Layer name'))
  981. self.InsertColumn(2, _('Name for GRASS map'))
  982. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) #wxMSW
  983. self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) #wxGTK
  984. def LoadData(self, data=None):
  985. """!Load data into list"""
  986. if data is None:
  987. return
  988. self.DeleteAllItems()
  989. for id, name, grassName in data:
  990. index = self.InsertStringItem(sys.maxint, str(id))
  991. self.SetStringItem(index, 1, "%s" % str(name))
  992. self.SetStringItem(index, 2, "%s" % str(grassName))
  993. # check by default
  994. self.CheckItem(index, True)
  995. self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
  996. def OnPopupMenu(self, event):
  997. """!Show popup menu"""
  998. if self.GetItemCount() < 1:
  999. return
  1000. if not hasattr(self, "popupDataID1"):
  1001. self.popupDataID1 = wx.NewId()
  1002. self.popupDataID2 = wx.NewId()
  1003. self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
  1004. self.Bind(wx.EVT_MENU, self.OnSelectNone, id=self.popupDataID2)
  1005. # generate popup-menu
  1006. menu = wx.Menu()
  1007. menu.Append(self.popupDataID1, _("Select all"))
  1008. menu.Append(self.popupDataID2, _("Deselect all"))
  1009. self.PopupMenu(menu)
  1010. menu.Destroy()
  1011. def OnSelectAll(self, event):
  1012. """!Select all items"""
  1013. item = -1
  1014. while True:
  1015. item = self.GetNextItem(item)
  1016. if item == -1:
  1017. break
  1018. self.CheckItem(item, True)
  1019. event.Skip()
  1020. def OnSelectNone(self, event):
  1021. """!Deselect items"""
  1022. item = -1
  1023. while True:
  1024. item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
  1025. if item == -1:
  1026. break
  1027. self.CheckItem(item, False)
  1028. event.Skip()
  1029. def GetLayers(self):
  1030. """!Get list of layers (layer name, output name)"""
  1031. data = []
  1032. item = -1
  1033. while True:
  1034. item = self.GetNextItem(item)
  1035. if item == -1:
  1036. break
  1037. if self.IsChecked(item):
  1038. # layer / output name
  1039. data.append((self.GetItem(item, 1).GetText(),
  1040. self.GetItem(item, 2).GetText()))
  1041. return data
  1042. class SetOpacityDialog(wx.Dialog):
  1043. """!Set opacity of map layers"""
  1044. def __init__(self, parent, id=wx.ID_ANY, title=_("Set Map Layer Opacity"),
  1045. size=wx.DefaultSize, pos=wx.DefaultPosition,
  1046. style=wx.DEFAULT_DIALOG_STYLE, opacity=100):
  1047. self.parent = parent # GMFrame
  1048. self.opacity = opacity # current opacity
  1049. super(SetOpacityDialog, self).__init__(parent, id=id, pos=pos,
  1050. size=size, style=style, title=title)
  1051. panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1052. sizer = wx.BoxSizer(wx.VERTICAL)
  1053. box = wx.GridBagSizer(vgap=5, hgap=5)
  1054. self.value = wx.Slider(panel, id=wx.ID_ANY, value=self.opacity,
  1055. style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS | \
  1056. wx.SL_TOP | wx.SL_LABELS,
  1057. minValue=0, maxValue=100,
  1058. size=(350, -1))
  1059. box.Add(item=self.value,
  1060. flag=wx.ALIGN_CENTRE, pos=(0, 0), span=(1, 2))
  1061. box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1062. label=_("transparent")),
  1063. pos=(1, 0))
  1064. box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1065. label=_("opaque")),
  1066. flag=wx.ALIGN_RIGHT,
  1067. pos=(1, 1))
  1068. sizer.Add(item=box, proportion=0,
  1069. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1070. line = wx.StaticLine(parent=panel, id=wx.ID_ANY,
  1071. style=wx.LI_HORIZONTAL)
  1072. sizer.Add(item=line, proportion=0,
  1073. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1074. # buttons
  1075. btnsizer = wx.StdDialogButtonSizer()
  1076. btnOK = wx.Button(parent=panel, id=wx.ID_OK)
  1077. btnOK.SetDefault()
  1078. btnsizer.AddButton(btnOK)
  1079. btnCancel = wx.Button(parent=panel, id=wx.ID_CANCEL)
  1080. btnsizer.AddButton(btnCancel)
  1081. btnsizer.Realize()
  1082. sizer.Add(item=btnsizer, proportion=0,
  1083. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1084. panel.SetSizer(sizer)
  1085. sizer.Fit(panel)
  1086. self.SetSize(self.GetBestSize())
  1087. self.Layout()
  1088. def GetOpacity(self):
  1089. """!Button 'OK' pressed"""
  1090. # return opacity value
  1091. opacity = float(self.value.GetValue()) / 100
  1092. return opacity