gdialogs.py 64 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684
  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-2010 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.importType = 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. self.inputTitle = _("Source name")
  749. self.inputBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
  750. label=" %s " % self.inputTitle)
  751. self.layerBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
  752. label=_(" List of %s layers ") % self.importType.upper())
  753. #
  754. # input
  755. #
  756. if self.importType == 'dxf':
  757. inputFile = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
  758. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  759. dialogTitle=_('Choose DXF file to import'),
  760. buttonText=_('Browse'),
  761. startDirectory=os.getcwd(), fileMode=0,
  762. changeCallback=self.OnSetInput,
  763. fileMask="DXF File (*.dxf)|*.dxf")
  764. self.input = { 'file' : [_("DXF file:"),
  765. inputFile,
  766. list()] }
  767. self.inputType = 'file'
  768. else:
  769. self.typeRadio = wx.RadioBox(parent = self.panel, id = wx.ID_ANY,
  770. label = _('Source type'),
  771. style = wx.RA_SPECIFY_COLS,
  772. choices = [_("File"),
  773. _("Directory"),
  774. _("Database"),
  775. _("Protocol")])
  776. self.typeRadio.SetSelection(0)
  777. self.Bind(wx.EVT_RADIOBOX, self.OnChangeType)
  778. # input widgets
  779. if self.importType == 'gdal':
  780. filemask = 'GeoTIFF (*.tif)|*.tif'
  781. else:
  782. filemask = 'ESRI Shapefile (*.shp)|*.shp'
  783. inputFile = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
  784. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  785. dialogTitle=_('Choose input file'),
  786. buttonText=_('Browse'),
  787. startDirectory=os.getcwd(),
  788. changeCallback=self.OnSetInput,
  789. fileMask=filemask)
  790. inputDir = filebrowse.DirBrowseButton(parent=self.panel, id=wx.ID_ANY,
  791. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  792. dialogTitle=_('Choose input directory'),
  793. buttonText=_('Browse'),
  794. startDirectory=os.getcwd(),
  795. changeCallback=self.OnSetInput)
  796. inputDir.Hide()
  797. inputDb = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY)
  798. inputDb.Hide()
  799. inputDb.Bind(wx.EVT_TEXT, self.OnSetInput)
  800. inputPro = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY)
  801. inputPro.Hide()
  802. inputPro.Bind(wx.EVT_TEXT, self.OnSetInput)
  803. # format widget
  804. self.formatText = wx.StaticText(self.panel, id=wx.ID_ANY, label=_("Format:"))
  805. self.format = wx.Choice(parent = self.panel, id = wx.ID_ANY, size=(300, -1))
  806. self.format.Bind(wx.EVT_CHOICE, self.OnSetFormat)
  807. if self.importType == 'gdal':
  808. ret = gcmd.RunCommand('r.in.gdal',
  809. quiet = True, read = True,
  810. flags = 'f')
  811. else: # ogr
  812. ret = gcmd.RunCommand('v.in.ogr',
  813. quiet = True, read = True,
  814. flags = 'f')
  815. self.input = { 'file' : [_("File:"),
  816. inputFile,
  817. list()],
  818. 'dir' : [_("Directory:"),
  819. inputDir,
  820. list()],
  821. 'db' : [_("Database:"),
  822. inputDb,
  823. list()],
  824. 'pro' : [_("Protocol:"),
  825. inputPro,
  826. list()] }
  827. self.formatToExt = {
  828. # raster
  829. 'GeoTIFF' : 'tif',
  830. 'Erdas Imagine Images (.img)' : '.img',
  831. 'Ground-based SAR Applications Testbed File Format (.gff)' : '.gff',
  832. 'Arc/Info Binary Grid' : 'adf',
  833. 'Portable Network Graphics' : 'png',
  834. 'JPEG JFIF' : 'jpg',
  835. 'Japanese DEM (.mem)' : 'mem',
  836. 'Graphics Interchange Format (.gif)' : 'gif',
  837. 'X11 PixMap Format' : 'xpm',
  838. 'MS Windows Device Independent Bitmap' : 'bmp',
  839. 'SPOT DIMAP' : '.dim',
  840. 'RadarSat 2 XML Product' : 'xml',
  841. 'EarthWatch .TIL' : '.til',
  842. 'ERMapper .ers Labelled' : '.ers',
  843. 'ERMapper Compressed Wavelets' : 'ecw',
  844. 'GRIdded Binary (.grb)' : 'grb',
  845. 'EUMETSAT Archive native (.nat)' : '.nat',
  846. 'Idrisi Raster A.1' : 'rst',
  847. 'Golden Software ASCII Grid (.grd)' : '.grd',
  848. 'Golden Software Binary Grid (.grd)' : 'grd',
  849. 'Golden Software 7 Binary Grid (.grd)' : 'grd',
  850. 'R Object Data Store' : 'r',
  851. 'USGS DOQ (Old Style)' : 'doq',
  852. 'USGS DOQ (New Style)' : 'doq',
  853. 'ENVI .hdr Labelled' : 'hdr',
  854. 'ESRI .hdr Labelled' : 'hdr',
  855. 'Generic Binary (.hdr Labelled)' : 'hdr',
  856. 'PCI .aux Labelled' : 'aux',
  857. 'EOSAT FAST Format' : 'fst',
  858. 'VTP .bt (Binary Terrain) 1.3 Format' : 'bt',
  859. 'FARSITE v.4 Landscape File (.lcp)' : 'lcp',
  860. 'Swedish Grid RIK (.rik)' : 'rik',
  861. 'USGS Optional ASCII DEM (and CDED)' : '.dem',
  862. 'Northwood Numeric Grid Format .grd/.tab' : '',
  863. 'Northwood Classified Grid Format .grc/.tab' : '',
  864. 'ARC Digitized Raster Graphics' : 'arc',
  865. 'Magellan topo (.blx)' : 'blx',
  866. 'SAGA GIS Binary Grid (.sdat)' : 'sdat',
  867. # vector
  868. 'ESRI Shapefile' : 'shp',
  869. 'UK .NTF' : 'ntf',
  870. 'SDTS' : 'ddf',
  871. 'DGN' : 'dgn',
  872. 'VRT' : 'vrt',
  873. 'REC' : 'rec',
  874. 'BNA' : 'bna',
  875. 'CSV' : 'csv',
  876. 'GML' : 'gml',
  877. 'GPX' : 'gpx',
  878. 'KML' : 'kml',
  879. 'GMT' : 'gmt',
  880. 'PGeo' : 'mdb',
  881. 'XPlane' : 'dat',
  882. 'AVCBin' : 'adf',
  883. 'AVCE00' : 'e00',
  884. 'DXF' : 'dxf',
  885. 'Geoconcept' : 'gxt',
  886. 'GeoRSS' : 'xml',
  887. 'GPSTrackMaker' : 'gtm',
  888. 'VFK' : 'vfk' }
  889. if ret:
  890. for line in ret.splitlines():
  891. format = line.strip().rsplit(':', -1)[1].strip()
  892. if format in ('Memory', 'Virtual Raster', 'In Memory Raster'):
  893. continue
  894. if format in ('PostgreSQL', 'SQLite',
  895. 'ODBC', 'ESRI Personal GeoDatabase',
  896. 'Rasterlite',
  897. 'PostGIS WKT Raster driver'):
  898. self.input['db'][2].append(format)
  899. elif format in ('GeoJSON',
  900. 'OGC Web Coverage Service',
  901. 'OGC Web Map Service',
  902. 'HTTP Fetching Wrapper'):
  903. self.input['pro'][2].append(format)
  904. else:
  905. self.input['file'][2].append(format)
  906. self.input['dir'][2].append(format)
  907. self.inputType = 'file'
  908. self.format.SetItems(self.input[self.inputType][2])
  909. if self.importType == 'gdal':
  910. self.format.SetStringSelection('GeoTIFF')
  911. elif self.importType == 'ogr':
  912. self.format.SetStringSelection('ESRI Shapefile')
  913. self.inputText = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  914. label = self.input[self.inputType][0],
  915. size = (75, -1))
  916. #
  917. # list of layers
  918. #
  919. self.list = LayersList(self.panel)
  920. self.list.LoadData()
  921. self.add = wx.CheckBox(parent=self.panel, id=wx.ID_ANY)
  922. if link:
  923. self.add.SetLabel(_("Add linked layers into layer tree"))
  924. else:
  925. self.add.SetLabel(_("Add imported layers into layer tree"))
  926. if not link and self.importType in ('gdal', 'ogr'):
  927. self.overrideCheck = wx.CheckBox(parent=self.panel, id=wx.ID_ANY,
  928. label=_("Override projection (use location's projection)"))
  929. self.overrideCheck.SetValue(True)
  930. self.add.SetValue(UserSettings.Get(group='cmd', key='addNewLayer', subkey='enabled'))
  931. self.overwrite = wx.CheckBox(parent=self.panel, id=wx.ID_ANY,
  932. label=_("Allow output files to overwrite existing files"))
  933. self.overwrite.SetValue(UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'))
  934. #
  935. # buttons
  936. #
  937. # cancel
  938. self.btn_cancel = wx.Button(parent=self.panel, id=wx.ID_CANCEL)
  939. self.btn_cancel.SetToolTipString(_("Close dialog"))
  940. self.btn_cancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  941. # run
  942. if link:
  943. self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Link"))
  944. self.btn_run.SetToolTipString(_("Link selected layers"))
  945. else:
  946. self.btn_run = wx.Button(parent=self.panel, id=wx.ID_OK, label= _("&Import"))
  947. self.btn_run.SetToolTipString(_("Import selected layers"))
  948. self.btn_run.SetDefault()
  949. self.btn_run.Enable(False)
  950. self.btn_run.Bind(wx.EVT_BUTTON, self.OnRun)
  951. self.__doLayout()
  952. self.Layout()
  953. def __doLayout(self):
  954. dialogSizer = wx.BoxSizer(wx.VERTICAL)
  955. #
  956. # input
  957. #
  958. inputSizer = wx.StaticBoxSizer(self.inputBox, wx.HORIZONTAL)
  959. gridSizer = wx.FlexGridSizer(cols=2, vgap=5, hgap=5)
  960. gridSizer.Add(item=self.inputText,
  961. flag=wx.ALIGN_CENTER_VERTICAL)
  962. gridSizer.AddGrowableCol(1)
  963. self.inputTypeSizer = wx.BoxSizer(wx.HORIZONTAL)
  964. self.inputTypeSizer.Add(item=self.input[self.inputType][1], proportion = 1,
  965. flag = wx.ALIGN_CENTER_VERTICAL)
  966. gridSizer.Add(item=self.inputTypeSizer,
  967. flag=wx.EXPAND | wx.ALL)
  968. if self.importType != 'dxf':
  969. gridSizer.Add(item=self.formatText,
  970. flag=wx.ALIGN_CENTER_VERTICAL)
  971. gridSizer.Add(item=self.format)
  972. inputSizer.Add(item=gridSizer, proportion=1,
  973. flag=wx.EXPAND | wx.ALL)
  974. if self.importType != 'dxf':
  975. dialogSizer.Add(item=self.typeRadio, proportion=0,
  976. flag=wx.ALL | wx.EXPAND, border=5)
  977. dialogSizer.Add(item=inputSizer, proportion=0,
  978. flag=wx.ALL | wx.EXPAND, border=5)
  979. #
  980. # list of DXF layers
  981. #
  982. layerSizer = wx.StaticBoxSizer(self.layerBox, wx.HORIZONTAL)
  983. layerSizer.Add(item=self.list, proportion=1,
  984. flag=wx.ALL | wx.EXPAND, border=5)
  985. dialogSizer.Add(item=layerSizer, proportion=1,
  986. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  987. if hasattr(self, "overrideCheck"):
  988. dialogSizer.Add(item=self.overrideCheck, proportion=0,
  989. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  990. dialogSizer.Add(item=self.overwrite, proportion=0,
  991. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  992. dialogSizer.Add(item=self.add, proportion=0,
  993. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=5)
  994. #
  995. # buttons
  996. #
  997. btnsizer = wx.BoxSizer(orient=wx.HORIZONTAL)
  998. btnsizer.Add(item=self.btn_cancel, proportion=0,
  999. flag=wx.ALL | wx.ALIGN_CENTER,
  1000. border=10)
  1001. btnsizer.Add(item=self.btn_run, proportion=0,
  1002. flag=wx.ALL | wx.ALIGN_CENTER,
  1003. border=10)
  1004. dialogSizer.Add(item=btnsizer, proportion=0,
  1005. flag=wx.ALIGN_CENTER)
  1006. # dialogSizer.SetSizeHints(self.panel)
  1007. self.panel.SetAutoLayout(True)
  1008. self.panel.SetSizer(dialogSizer)
  1009. dialogSizer.Fit(self.panel)
  1010. self.Layout()
  1011. # auto-layout seems not work here - FIXME
  1012. self.SetMinSize((globalvar.DIALOG_GSELECT_SIZE[0] + 175, 400))
  1013. width = self.GetSize()[0]
  1014. self.list.SetColumnWidth(col=1, width=width/2 - 50)
  1015. def OnChangeType(self, event):
  1016. """!Datasource type changed"""
  1017. sel = event.GetSelection()
  1018. win = self.input[self.inputType][1]
  1019. self.inputTypeSizer.Remove(win)
  1020. win.Hide()
  1021. if sel == 0: # file
  1022. self.inputType = 'file'
  1023. format = self.input[self.inputType][2][0]
  1024. try:
  1025. ext = self.formatToExt[format]
  1026. if not ext:
  1027. raise KeyError
  1028. format += ' (*.%s)|*.%s' % (ext, ext)
  1029. except KeyError:
  1030. format += ' (*.*)|*.*'
  1031. win = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
  1032. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1033. dialogTitle=_('Choose input file'),
  1034. buttonText=_('Browse'),
  1035. startDirectory=os.getcwd(),
  1036. changeCallback=self.OnSetInput,
  1037. fileMask = format)
  1038. self.input[self.inputType][1] = win
  1039. elif sel == 1: # directory
  1040. self.inputType = 'dir'
  1041. elif sel == 2: # database
  1042. self.inputType = 'db'
  1043. elif sel == 3: # protocol
  1044. self.inputType = 'pro'
  1045. win = self.input[self.inputType][1]
  1046. self.inputTypeSizer.Add(item = win, proportion = 1,
  1047. flag = wx.ALIGN_CENTER_VERTICAL)
  1048. win.SetValue('')
  1049. self.list.DeleteAllItems()
  1050. win.Show()
  1051. self.inputText.SetLabel(self.input[self.inputType][0])
  1052. self.format.SetItems(self.input[self.inputType][2])
  1053. self.format.SetSelection(0)
  1054. self.inputTypeSizer.Layout()
  1055. def OnCancel(self, event=None):
  1056. """!Close dialog"""
  1057. self.Close()
  1058. def OnRun(self, event):
  1059. """!Import/Link data (each layes as separate vector map)"""
  1060. data = self.list.GetLayers()
  1061. # hide dialog
  1062. self.Hide()
  1063. if self.importType == 'dxf':
  1064. inputDxf = self.input[self.inputType][1].GetValue()
  1065. else:
  1066. if self.inputType == 'file':
  1067. dsn = os.path.dirname(self.input[self.inputType][1].GetValue())
  1068. else:
  1069. dsn = self.input[self.inputType][1].GetValue()
  1070. ext = self.formatToExt[self.format.GetStringSelection()]
  1071. for layer, output in data:
  1072. if self.importType == 'dxf':
  1073. cmd = ['v.in.dxf',
  1074. 'input=%s' % inputDxf,
  1075. 'layers=%s' % layer,
  1076. 'output=%s' % output]
  1077. elif self.importType == 'ogr':
  1078. if self.link:
  1079. cmd = ['v.external',
  1080. 'dsn=%s' % dsn,
  1081. 'output=%s' % output,
  1082. 'layer=%s' % layer.rstrip('.' + ext)]
  1083. else:
  1084. cmd = ['v.in.ogr',
  1085. 'dsn=%s' % (os.path.join(dsn, layer)),
  1086. 'output=%s' % output]
  1087. else: # gdal
  1088. if self.link:
  1089. cmd = ['r.external',
  1090. 'input=%s' % (os.path.join(dsn, layer)),
  1091. 'output=%s' % output]
  1092. else:
  1093. cmd = ['r.in.gdal',
  1094. 'input=%s' % (os.path.join(dsn, layer)),
  1095. 'output=%s' % output]
  1096. if self.overwrite.IsChecked():
  1097. cmd.append('--overwrite')
  1098. if hasattr(self, "overrideCheck") and self.overrideCheck.IsChecked():
  1099. cmd.append('-o')
  1100. if UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'):
  1101. cmd.append('--overwrite')
  1102. # run in Layer Manager
  1103. self.parent.goutput.RunCmd(cmd, switchPage=True)
  1104. if self.add.IsChecked():
  1105. maptree = self.parent.curr_page.maptree
  1106. for layer, output in data:
  1107. if '@' not in output:
  1108. name = output + '@' + grass.gisenv()['MAPSET']
  1109. else:
  1110. name = output
  1111. # add imported layers into layer tree
  1112. if self.importType == 'gdal':
  1113. cmd = ['d.rast',
  1114. 'map=%s' % name]
  1115. if UserSettings.Get(group='cmd', key='rasterOverlay', subkey='enabled'):
  1116. cmd.append('-o')
  1117. maptree.AddLayer(ltype='raster',
  1118. lname=name,
  1119. lcmd=cmd)
  1120. else:
  1121. maptree.AddLayer(ltype='vector',
  1122. lname=name,
  1123. lcmd=['d.vect',
  1124. 'map=%s' % name])
  1125. ### wx.CallAfter(self.parent.notebook.SetSelection, 0)
  1126. self.OnCancel()
  1127. def OnAbort(self, event):
  1128. """!Abort running import
  1129. @todo not yet implemented
  1130. """
  1131. pass
  1132. def OnSetFormat(self, event):
  1133. """!Format changed"""
  1134. if self.inputType != 'file':
  1135. return
  1136. win = self.input[self.inputType][1]
  1137. self.inputTypeSizer.Remove(win)
  1138. win.Destroy()
  1139. format = event.GetString()
  1140. try:
  1141. ext = self.formatToExt[format]
  1142. if not ext:
  1143. raise KeyError
  1144. format += ' (*.%s)|*.%s' % (ext, ext)
  1145. except KeyError:
  1146. format += ' (*.*)|*.*'
  1147. win = filebrowse.FileBrowseButton(parent=self.panel, id=wx.ID_ANY,
  1148. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1149. dialogTitle=_('Choose file'),
  1150. buttonText=_('Browse'),
  1151. startDirectory=os.getcwd(),
  1152. changeCallback=self.OnSetInput,
  1153. fileMask = format)
  1154. self.input[self.inputType][1] = win
  1155. self.inputTypeSizer.Add(item = win, proportion = 1,
  1156. flag = wx.ALIGN_CENTER_VERTICAL)
  1157. self.inputTypeSizer.Layout()
  1158. def OnSetInput(self, event):
  1159. """!Input DXF file/OGR dsn defined, update list of layer widget"""
  1160. path = event.GetString()
  1161. if not path:
  1162. return
  1163. if self.importType == 'dxf':
  1164. ret = gcmd.RunCommand('v.in.dxf',
  1165. quiet = True,
  1166. parent = self,
  1167. read = True,
  1168. flags = 'l',
  1169. input = path)
  1170. if not ret:
  1171. self.list.LoadData()
  1172. self.btn_run.Enable(False)
  1173. return
  1174. data = list()
  1175. if self.importType == 'dxf':
  1176. for line in ret.splitlines():
  1177. layerId = line.split(':')[0].split(' ')[1]
  1178. layerName = line.split(':')[1].strip()
  1179. grassName = utils.GetValidLayerName(layerName)
  1180. data.append((layerId, layerName.strip(), grassName.strip()))
  1181. else: # gdal/ogr (for ogr maybe to use v.in.ogr -l)
  1182. layerId = 1
  1183. dsn = self.input[self.inputType][1].GetValue()
  1184. if self.inputType == 'file':
  1185. baseName = os.path.basename(dsn)
  1186. grassName = utils.GetValidLayerName(baseName.split('.', -1)[0])
  1187. data.append((layerId, baseName, grassName))
  1188. elif self.inputType == 'dir':
  1189. try:
  1190. ext = self.formatToExt[self.format.GetStringSelection()]
  1191. except KeyError:
  1192. ext = ''
  1193. for file in glob.glob(os.path.join(dsn, "*.%s") % ext):
  1194. baseName = os.path.basename(file)
  1195. grassName = utils.GetValidLayerName(baseName.split('.', -1)[0])
  1196. data.append((layerId, baseName, grassName))
  1197. layerId += 1
  1198. self.list.LoadData(data)
  1199. if len(data) > 0:
  1200. self.btn_run.Enable(True)
  1201. else:
  1202. self.btn_run.Enable(False)
  1203. class LayersList(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin,
  1204. listmix.CheckListCtrlMixin):
  1205. # listmix.CheckListCtrlMixin, listmix.TextEditMixin):
  1206. """!List of layers to be imported (dxf, shp...)"""
  1207. def __init__(self, parent, pos=wx.DefaultPosition,
  1208. log=None):
  1209. self.parent = parent
  1210. wx.ListCtrl.__init__(self, parent, wx.ID_ANY,
  1211. style=wx.LC_REPORT)
  1212. listmix.CheckListCtrlMixin.__init__(self)
  1213. self.log = log
  1214. # setup mixins
  1215. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1216. # listmix.TextEditMixin.__init__(self)
  1217. self.InsertColumn(0, _('Layer'))
  1218. self.InsertColumn(1, _('Layer name'))
  1219. self.InsertColumn(2, _('Name for GRASS map'))
  1220. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) #wxMSW
  1221. self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) #wxGTK
  1222. def LoadData(self, data=None):
  1223. """!Load data into list"""
  1224. if data is None:
  1225. return
  1226. self.DeleteAllItems()
  1227. for id, name, grassName in data:
  1228. index = self.InsertStringItem(sys.maxint, str(id))
  1229. self.SetStringItem(index, 1, "%s" % str(name))
  1230. self.SetStringItem(index, 2, "%s" % str(grassName))
  1231. # check by default
  1232. self.CheckItem(index, True)
  1233. self.SetColumnWidth(col=0, width=wx.LIST_AUTOSIZE_USEHEADER)
  1234. def OnPopupMenu(self, event):
  1235. """!Show popup menu"""
  1236. if self.GetItemCount() < 1:
  1237. return
  1238. if not hasattr(self, "popupDataID1"):
  1239. self.popupDataID1 = wx.NewId()
  1240. self.popupDataID2 = wx.NewId()
  1241. self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
  1242. self.Bind(wx.EVT_MENU, self.OnSelectNone, id=self.popupDataID2)
  1243. # generate popup-menu
  1244. menu = wx.Menu()
  1245. menu.Append(self.popupDataID1, _("Select all"))
  1246. menu.Append(self.popupDataID2, _("Deselect all"))
  1247. self.PopupMenu(menu)
  1248. menu.Destroy()
  1249. def OnSelectAll(self, event):
  1250. """!Select all items"""
  1251. item = -1
  1252. while True:
  1253. item = self.GetNextItem(item)
  1254. if item == -1:
  1255. break
  1256. self.CheckItem(item, True)
  1257. event.Skip()
  1258. def OnSelectNone(self, event):
  1259. """!Deselect items"""
  1260. item = -1
  1261. while True:
  1262. item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
  1263. if item == -1:
  1264. break
  1265. self.CheckItem(item, False)
  1266. event.Skip()
  1267. def GetLayers(self):
  1268. """!Get list of layers (layer name, output name)"""
  1269. data = []
  1270. item = -1
  1271. while True:
  1272. item = self.GetNextItem(item)
  1273. if item == -1:
  1274. break
  1275. if self.IsChecked(item):
  1276. # layer / output name
  1277. data.append((self.GetItem(item, 1).GetText(),
  1278. self.GetItem(item, 2).GetText()))
  1279. return data
  1280. class SetOpacityDialog(wx.Dialog):
  1281. """!Set opacity of map layers"""
  1282. def __init__(self, parent, id=wx.ID_ANY, title=_("Set Map Layer Opacity"),
  1283. size=wx.DefaultSize, pos=wx.DefaultPosition,
  1284. style=wx.DEFAULT_DIALOG_STYLE, opacity=100):
  1285. self.parent = parent # GMFrame
  1286. self.opacity = opacity # current opacity
  1287. super(SetOpacityDialog, self).__init__(parent, id=id, pos=pos,
  1288. size=size, style=style, title=title)
  1289. panel = wx.Panel(parent=self, id=wx.ID_ANY)
  1290. sizer = wx.BoxSizer(wx.VERTICAL)
  1291. box = wx.GridBagSizer(vgap=5, hgap=5)
  1292. self.value = wx.Slider(panel, id=wx.ID_ANY, value=self.opacity,
  1293. style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS | \
  1294. wx.SL_TOP | wx.SL_LABELS,
  1295. minValue=0, maxValue=100,
  1296. size=(350, -1))
  1297. box.Add(item=self.value,
  1298. flag=wx.ALIGN_CENTRE, pos=(0, 0), span=(1, 2))
  1299. box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1300. label=_("transparent")),
  1301. pos=(1, 0))
  1302. box.Add(item=wx.StaticText(parent=panel, id=wx.ID_ANY,
  1303. label=_("opaque")),
  1304. flag=wx.ALIGN_RIGHT,
  1305. pos=(1, 1))
  1306. sizer.Add(item=box, proportion=0,
  1307. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1308. line = wx.StaticLine(parent=panel, id=wx.ID_ANY,
  1309. style=wx.LI_HORIZONTAL)
  1310. sizer.Add(item=line, proportion=0,
  1311. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1312. # buttons
  1313. btnsizer = wx.StdDialogButtonSizer()
  1314. btnOK = wx.Button(parent=panel, id=wx.ID_OK)
  1315. btnOK.SetDefault()
  1316. btnsizer.AddButton(btnOK)
  1317. btnCancel = wx.Button(parent=panel, id=wx.ID_CANCEL)
  1318. btnsizer.AddButton(btnCancel)
  1319. btnsizer.Realize()
  1320. sizer.Add(item=btnsizer, proportion=0,
  1321. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, border=5)
  1322. panel.SetSizer(sizer)
  1323. sizer.Fit(panel)
  1324. self.SetSize(self.GetBestSize())
  1325. self.Layout()
  1326. def GetOpacity(self):
  1327. """!Button 'OK' pressed"""
  1328. # return opacity value
  1329. opacity = float(self.value.GetValue()) / 100
  1330. return opacity
  1331. def GetImageHandlers(image):
  1332. """!Get list of supported image handlers"""
  1333. lext = list()
  1334. ltype = list()
  1335. for h in image.GetHandlers():
  1336. lext.append(h.GetExtension())
  1337. filetype = ''
  1338. if 'png' in lext:
  1339. filetype += "PNG file (*.png)|*.png|"
  1340. ltype.append({ 'type' : wx.BITMAP_TYPE_PNG,
  1341. 'ext' : 'png' })
  1342. filetype += "BMP file (*.bmp)|*.bmp|"
  1343. ltype.append({ 'type' : wx.BITMAP_TYPE_BMP,
  1344. 'ext' : 'bmp' })
  1345. if 'gif' in lext:
  1346. filetype += "GIF file (*.gif)|*.gif|"
  1347. ltype.append({ 'type' : wx.BITMAP_TYPE_GIF,
  1348. 'ext' : 'gif' })
  1349. if 'jpg' in lext:
  1350. filetype += "JPG file (*.jpg)|*.jpg|"
  1351. ltype.append({ 'type' : wx.BITMAP_TYPE_JPEG,
  1352. 'ext' : 'jpg' })
  1353. if 'pcx' in lext:
  1354. filetype += "PCX file (*.pcx)|*.pcx|"
  1355. ltype.append({ 'type' : wx.BITMAP_TYPE_PCX,
  1356. 'ext' : 'pcx' })
  1357. if 'pnm' in lext:
  1358. filetype += "PNM file (*.pnm)|*.pnm|"
  1359. ltype.append({ 'type' : wx.BITMAP_TYPE_PNM,
  1360. 'ext' : 'pnm' })
  1361. if 'tif' in lext:
  1362. filetype += "TIF file (*.tif)|*.tif|"
  1363. ltype.append({ 'type' : wx.BITMAP_TYPE_TIF,
  1364. 'ext' : 'tif' })
  1365. if 'xpm' in lext:
  1366. filetype += "XPM file (*.xpm)|*.xpm"
  1367. ltype.append({ 'type' : wx.BITMAP_TYPE_XPM,
  1368. 'ext' : 'xpm' })
  1369. return filetype, ltype