gdialogs.py 50 KB

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