gdialogs.py 51 KB

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