gdialogs.py 41 KB

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