import_export.py 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013
  1. """
  2. @package modules.import_export
  3. @brief Import/export dialogs used in wxGUI.
  4. List of classes:
  5. - :class:`ImportDialog`
  6. - :class:`GdalImportDialog`
  7. - :class:`OgrImportDialog`
  8. - :class:`GdalOutputDialog`
  9. - :class:`DxfImportDialog`
  10. - :class:`ReprojectionDialog`
  11. (C) 2008-2016 by the GRASS Development Team
  12. This program is free software under the GNU General Public License
  13. (>=v2). Read the file COPYING that comes with GRASS for details.
  14. @author Martin Landa <landa.martin gmail.com>
  15. @author Anna Kratochvilova <kratochanna gmail.com> (GroupDialog, SymbolDialog)
  16. """
  17. import os
  18. import sys
  19. import wx
  20. from core import globalvar
  21. if globalvar.wxPythonPhoenix:
  22. try:
  23. import agw.flatnotebook as FN
  24. except ImportError: # if it's not there locally, try the wxPython lib.
  25. import wx.lib.agw.flatnotebook as FN
  26. else:
  27. import wx.lib.flatnotebook as FN
  28. import wx.lib.filebrowsebutton as filebrowse
  29. from grass.script import core as grass
  30. from grass.script import task as gtask
  31. from core.gcmd import GError, GMessage, GWarning, RunCommand
  32. from gui_core.forms import CmdPanel
  33. from gui_core.gselect import OgrTypeSelect, GdalSelect, SubGroupSelect
  34. from gui_core.widgets import GListCtrl, GNotebook, LayersList, \
  35. LayersListValidator
  36. from gui_core.wrap import Button, CloseButton, StaticText, StaticBox
  37. from core.utils import GetValidLayerName
  38. from core.settings import UserSettings, GetDisplayVectSettings
  39. class ImportDialog(wx.Dialog):
  40. """Dialog for bulk import of various data (base class)"""
  41. def __init__(self, parent, giface, itype,
  42. id=wx.ID_ANY, title=_("Multiple import"),
  43. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  44. self.parent = parent # GMFrame
  45. self._giface = giface # used to add layers
  46. self.importType = itype
  47. self.options = dict() # list of options
  48. self.options_par = dict()
  49. self.commandId = -1 # id of running command
  50. wx.Dialog.__init__(self, parent, id, title, style=style,
  51. name="MultiImportDialog")
  52. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  53. self.layerBox = StaticBox(parent=self.panel, id=wx.ID_ANY)
  54. if self.importType == 'gdal':
  55. label = _("List of raster layers")
  56. elif self.importType == 'ogr':
  57. label = _("List of vector layers")
  58. else:
  59. label = _("List of %s layers") % self.importType.upper()
  60. self.layerBox.SetLabel(" %s - %s " %
  61. (label, _("right click to (un)select all")))
  62. # list of layers
  63. columns = [_('Layer id'),
  64. _('Layer name'),
  65. _('Name for output GRASS map (editable)')]
  66. if itype == 'ogr':
  67. columns.insert(2, _('Feature type'))
  68. columns.insert(3, _('Projection match'))
  69. elif itype == 'gdal':
  70. columns.insert(2, _('Projection match'))
  71. self.list = LayersList(parent=self.panel, columns=columns)
  72. self.list.LoadData()
  73. self.override = wx.CheckBox(
  74. parent=self.panel, id=wx.ID_ANY,
  75. label=_("Override projection check (use current location's projection)"))
  76. self.overwrite = wx.CheckBox(
  77. parent=self.panel, id=wx.ID_ANY,
  78. label=_("Allow output files to overwrite existing files"))
  79. self.overwrite.SetValue(
  80. UserSettings.Get(
  81. group='cmd',
  82. key='overwrite',
  83. subkey='enabled'))
  84. self.overwrite.Bind(wx.EVT_CHECKBOX, self.OnCheckOverwrite)
  85. if UserSettings.Get(group='cmd', key='overwrite',
  86. subkey='enabled'):
  87. self.list.validate = False
  88. self.add = wx.CheckBox(parent=self.panel, id=wx.ID_ANY)
  89. self.closeOnFinish = wx.CheckBox(parent=self.panel, id=wx.ID_ANY,
  90. label=_("Close dialog on finish"))
  91. self.closeOnFinish.SetValue(
  92. UserSettings.Get(
  93. group='cmd',
  94. key='closeDlg',
  95. subkey='enabled'))
  96. #
  97. # buttons
  98. #
  99. # cancel
  100. self.btn_close = CloseButton(parent=self.panel)
  101. self.btn_close.SetToolTip(_("Close dialog"))
  102. self.btn_close.Bind(wx.EVT_BUTTON, self.OnClose)
  103. # run
  104. self.btn_run = Button(
  105. parent=self.panel, id=wx.ID_OK, label=_("&Import"))
  106. self.btn_run.SetToolTip(_("Import selected layers"))
  107. self.btn_run.SetDefault()
  108. self.btn_run.Bind(wx.EVT_BUTTON, self.OnRun)
  109. self.Bind(wx.EVT_CLOSE, lambda evt: self.Destroy())
  110. self.notebook = GNotebook(parent=self,
  111. style=globalvar.FNPageDStyle)
  112. self.notebook.AddPage(page=self.panel,
  113. text=_('Source settings'),
  114. name='source')
  115. self.createSettingsPage()
  116. def createSettingsPage(self):
  117. self._blackList = {
  118. 'enabled': True,
  119. 'items': {
  120. self._getCommand(): {
  121. 'params': self._getBlackListedParameters(),
  122. 'flags': self._getBlackListedFlags()}}}
  123. grass_task = gtask.parse_interface(self._getCommand(),
  124. blackList=self._blackList)
  125. self.advancedPagePanel = CmdPanel(
  126. parent=self, giface=None, task=grass_task, frame=None)
  127. self.notebook.AddPage(page=self.advancedPagePanel,
  128. text=_('Import settings'),
  129. name='settings')
  130. def doLayout(self):
  131. """Do layout"""
  132. dialogSizer = wx.BoxSizer(wx.VERTICAL)
  133. # dsn input
  134. dialogSizer.Add(self.dsnInput, proportion=0,
  135. flag=wx.EXPAND)
  136. #
  137. # list of DXF layers
  138. #
  139. layerSizer = wx.StaticBoxSizer(self.layerBox, wx.HORIZONTAL)
  140. layerSizer.Add(self.list, proportion=1,
  141. flag=wx.ALL | wx.EXPAND, border=5)
  142. dialogSizer.Add(
  143. layerSizer,
  144. proportion=1,
  145. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
  146. border=5)
  147. dialogSizer.Add(self.override, proportion=0,
  148. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  149. dialogSizer.Add(self.overwrite, proportion=0,
  150. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  151. dialogSizer.Add(self.add, proportion=0,
  152. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  153. dialogSizer.Add(self.closeOnFinish, proportion=0,
  154. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM, border=5)
  155. #
  156. # buttons
  157. #
  158. btnsizer = wx.BoxSizer(orient=wx.HORIZONTAL)
  159. btnsizer.Add(self.btn_close, proportion=0,
  160. flag=wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER,
  161. border=10)
  162. btnsizer.Add(self.btn_run, proportion=0,
  163. flag=wx.RIGHT | wx.ALIGN_CENTER,
  164. border=10)
  165. dialogSizer.Add(
  166. btnsizer,
  167. proportion=0,
  168. flag=wx.BOTTOM | wx.ALIGN_RIGHT,
  169. border=10)
  170. # dialogSizer.SetSizeHints(self.panel)
  171. self.panel.SetAutoLayout(True)
  172. self.panel.SetSizer(dialogSizer)
  173. dialogSizer.Fit(self.panel)
  174. # auto-layout seems not work here - FIXME
  175. size = wx.Size(globalvar.DIALOG_GSELECT_SIZE[0] + 322, 550)
  176. self.SetMinSize(size)
  177. self.SetSize((size.width, size.height + 100))
  178. # width = self.GetSize()[0]
  179. # self.list.SetColumnWidth(col = 1, width = width / 2 - 50)
  180. self.Layout()
  181. def _getCommand(self):
  182. """Get command"""
  183. raise NotImplementedError()
  184. def _getBlackListedParameters(self):
  185. """Get parameters which will not be showed in Settings page"""
  186. raise NotImplementedError()
  187. def _getBlackListedFlags(self):
  188. """Get flags which will not be showed in Settings page"""
  189. raise NotImplementedError()
  190. def _nameValidationFailed(self, layers_list):
  191. """Output map name validation callback
  192. :param layers_list: LayersList class instance
  193. """
  194. if isinstance(layers_list.output_map, list):
  195. maps = [
  196. '<{}>'.format(m) for m in layers_list.output_map]
  197. message = _(
  198. "Output map names %(names)s exist. "
  199. ) % {
  200. 'names': ', '.join(maps)}
  201. else:
  202. message = _(
  203. "Output map name <%(name)s> exist. "
  204. ) % {
  205. 'name': layers_list.output_map}
  206. GError(parent=self, message=message, caption=_("Invalid name"))
  207. def _validateOutputMapName(self):
  208. """Enable/disable output map name validation according the
  209. overwrite state"""
  210. if not self.overwrite.IsChecked():
  211. if not self.list.GetValidator().\
  212. Validate(win=self.list, validate_all=True):
  213. return False
  214. return True
  215. def OnClose(self, event=None):
  216. """Close dialog"""
  217. self.Close()
  218. def OnRun(self, event):
  219. """Import/Link data (each layes as separate vector map)"""
  220. pass
  221. def OnCheckOverwrite(self, event):
  222. """Check/uncheck overwrite checkbox widget"""
  223. if self.overwrite.IsChecked():
  224. self.list.validate = False
  225. else:
  226. self.list.validate = True
  227. def AddLayers(self, returncode, cmd=None, userData=None):
  228. """Add imported/linked layers into layer tree"""
  229. if not self.add.IsChecked() or returncode != 0:
  230. return
  231. # TODO: if importing map creates more map the folowing does not work
  232. # * do nothing if map does not exist or
  233. # * try to determine names using regexp or
  234. # * persuade import tools to report map names
  235. self.commandId += 1
  236. layer, output = self.list.GetLayers()[self.commandId][:2]
  237. if '@' not in output:
  238. name = output + '@' + grass.gisenv()['MAPSET']
  239. else:
  240. name = output
  241. # add imported layers into layer tree
  242. # an alternative would be emit signal (mapCreated) and (optionally)
  243. # connect to this signal
  244. llist = self._giface.GetLayerList()
  245. if self.importType == 'gdal':
  246. if userData:
  247. nBands = int(userData.get('nbands', 1))
  248. else:
  249. nBands = 1
  250. if UserSettings.Get(group='rasterLayer',
  251. key='opaque', subkey='enabled'):
  252. nFlag = True
  253. else:
  254. nFlag = False
  255. for i in range(1, nBands + 1):
  256. nameOrig = name
  257. if nBands > 1:
  258. mapName, mapsetName = name.split('@')
  259. mapName += '.%d' % i
  260. name = mapName + '@' + mapsetName
  261. cmd = ['d.rast',
  262. 'map=%s' % name]
  263. if nFlag:
  264. cmd.append('-n')
  265. llist.AddLayer(ltype='raster',
  266. name=name, checked=True,
  267. cmd=cmd)
  268. name = nameOrig
  269. else:
  270. llist.AddLayer(ltype='vector',
  271. name=name, checked=True,
  272. cmd=['d.vect',
  273. 'map=%s' % name] + GetDisplayVectSettings())
  274. self._giface.GetMapWindow().ZoomToMap()
  275. def OnAbort(self, event):
  276. """Abort running import
  277. .. todo::
  278. not yet implemented
  279. """
  280. pass
  281. def OnCmdDone(self, event):
  282. """Do what has to be done after importing"""
  283. pass
  284. def _getLayersToReprojetion(self, projMatch_idx, grassName_idx):
  285. """If there are layers with different projection from loation projection,
  286. show dialog to user to explicitly select layers which will be reprojected..."""
  287. differentProjLayers = []
  288. data = self.list.GetData(checked=True)
  289. for itm in data:
  290. layerId = itm[-1]
  291. # select only layers with different projetion
  292. if self.layersData[layerId][projMatch_idx] == 0:
  293. dt = [itm[0], itm[grassName_idx]]
  294. differentProjLayers.append(tuple(dt))
  295. layers = self.list.GetLayers()
  296. if not self.link and \
  297. differentProjLayers and \
  298. not self.override.IsChecked(): # '-o' not in self.getSettingsPageCmd():
  299. dlg = ReprojectionDialog(
  300. parent=self,
  301. giface=self._giface,
  302. data=differentProjLayers)
  303. ret = dlg.ShowModal()
  304. if ret == wx.ID_OK:
  305. # do not import unchecked layers
  306. for itm in reversed(list(dlg.GetData(checked=False))):
  307. idx = itm[-1]
  308. layers.pop(idx)
  309. else:
  310. return None
  311. return layers
  312. def getSettingsPageCmd(self):
  313. return self.advancedPagePanel.createCmd(
  314. ignoreErrors=True, ignoreRequired=True)
  315. class GdalImportDialog(ImportDialog):
  316. def __init__(self, parent, giface, link=False):
  317. """Dialog for bulk import of various raster data
  318. .. todo::
  319. split importing logic from gui code
  320. :param parent: parent window
  321. :param link: True for linking data otherwise importing data
  322. """
  323. self._giface = giface
  324. self.link = link
  325. self.layersData = []
  326. ImportDialog.__init__(self, parent, giface=giface, itype='gdal')
  327. self.list.SetValidator(
  328. LayersListValidator(
  329. condition='raster',
  330. callback=self._nameValidationFailed))
  331. if link:
  332. self.SetTitle(_("Link external raster data"))
  333. else:
  334. self.SetTitle(_("Import raster data"))
  335. self.dsnInput = GdalSelect(parent=self, panel=self.panel,
  336. ogr=False, link=link)
  337. self.dsnInput.AttachSettings()
  338. self.dsnInput.reloadDataRequired.connect(self.reload)
  339. if link:
  340. self.add.SetLabel(_("Add linked layers into layer tree"))
  341. else:
  342. self.add.SetLabel(_("Add imported layers into layer tree"))
  343. self.add.SetValue(
  344. UserSettings.Get(
  345. group='cmd',
  346. key='addNewLayer',
  347. subkey='enabled'))
  348. if link:
  349. self.btn_run.SetLabel(_("&Link"))
  350. self.btn_run.SetToolTip(_("Link selected layers"))
  351. else:
  352. self.btn_run.SetLabel(_("&Import"))
  353. self.btn_run.SetToolTip(_("Import selected layers"))
  354. self.doLayout()
  355. def reload(self, data, listData):
  356. self.list.LoadData(listData)
  357. self.list.SelectAll(select=True)
  358. self.layersData = data
  359. def OnRun(self, event):
  360. """Import/Link data (each layes as separate vector map)"""
  361. self.commandId = -1
  362. data = self.list.GetLayers()
  363. data = self._getLayersToReprojetion(2, 3)
  364. if data is None:
  365. return
  366. if not data:
  367. GMessage(_("No layers selected. Operation canceled."),
  368. parent=self)
  369. return
  370. if not self._validateOutputMapName():
  371. return
  372. dsn = self.dsnInput.GetDsn()
  373. ext = self.dsnInput.GetFormatExt()
  374. for layer, output, listId in data:
  375. userData = {}
  376. if self.dsnInput.GetType() == 'dir':
  377. idsn = os.path.join(dsn, layer)
  378. else:
  379. idsn = dsn
  380. # check number of bands
  381. nBandsStr = RunCommand('r.in.gdal',
  382. flags='p',
  383. input=idsn, read=True)
  384. nBands = -1
  385. if nBandsStr:
  386. try:
  387. nBands = int(nBandsStr.rstrip('\n'))
  388. except:
  389. pass
  390. if nBands < 0:
  391. GWarning(_("Unable to determine number of raster bands"),
  392. parent=self)
  393. nBands = 1
  394. userData['nbands'] = nBands
  395. cmd = self.getSettingsPageCmd()
  396. cmd.append('input=%s' % idsn)
  397. cmd.append('output=%s' % output)
  398. if self.override.IsChecked():
  399. cmd.append('-o')
  400. if self.overwrite.IsChecked():
  401. cmd.append('--overwrite')
  402. if UserSettings.Get(group='cmd', key='overwrite',
  403. subkey='enabled') and '--overwrite' not in cmd:
  404. cmd.append('--overwrite')
  405. # run in Layer Manager
  406. self._giface.RunCmd(
  407. cmd,
  408. onDone=self.OnCmdDone,
  409. userData=userData,
  410. addLayer=False)
  411. def OnCmdDone(self, event):
  412. """Load layers and close if required"""
  413. if not hasattr(self, 'AddLayers'):
  414. return
  415. self.AddLayers(event.returncode, event.cmd, event.userData)
  416. if event.returncode == 0 and self.closeOnFinish.IsChecked():
  417. self.Close()
  418. def _getCommand(self):
  419. """Get command"""
  420. if self.link:
  421. return 'r.external'
  422. else:
  423. return 'r.import'
  424. def _getBlackListedParameters(self):
  425. """Get flags which will not be showed in Settings page"""
  426. return ['input', 'output']
  427. def _getBlackListedFlags(self):
  428. """Get flags which will not be showed in Settings page"""
  429. return ['overwrite', 'o']
  430. class OgrImportDialog(ImportDialog):
  431. def __init__(self, parent, giface, link=False):
  432. """Dialog for bulk import of various vector data
  433. .. todo::
  434. split importing logic from gui code
  435. :param parent: parent window
  436. :param link: True for linking data otherwise importing data
  437. """
  438. self._giface = giface
  439. self.link = link
  440. self.layersData = []
  441. ImportDialog.__init__(self, parent, giface=giface, itype='ogr')
  442. self.list.SetValidator(
  443. LayersListValidator(
  444. condition='vector',
  445. callback=self._nameValidationFailed))
  446. if link:
  447. self.SetTitle(_("Link external vector data"))
  448. else:
  449. self.SetTitle(_("Import vector data"))
  450. self.dsnInput = GdalSelect(parent=self, panel=self.panel,
  451. ogr=True, link=link)
  452. self.dsnInput.AttachSettings()
  453. self.dsnInput.reloadDataRequired.connect(self.reload)
  454. if link:
  455. self.add.SetLabel(_("Add linked layers into layer tree"))
  456. else:
  457. self.add.SetLabel(_("Add imported layers into layer tree"))
  458. self.add.SetValue(
  459. UserSettings.Get(
  460. group='cmd',
  461. key='addNewLayer',
  462. subkey='enabled'))
  463. if link:
  464. self.btn_run.SetLabel(_("&Link"))
  465. self.btn_run.SetToolTip(_("Link selected layers"))
  466. else:
  467. self.btn_run.SetLabel(_("&Import"))
  468. self.btn_run.SetToolTip(_("Import selected layers"))
  469. self.doLayout()
  470. def reload(self, data, listData):
  471. self.list.LoadData(listData)
  472. self.list.SelectAll(select=True)
  473. self.layersData = data
  474. def OnRun(self, event):
  475. """Import/Link data (each layes as separate vector map)"""
  476. self.commandId = -1
  477. data = self.list.GetLayers()
  478. data = self._getLayersToReprojetion(3, 4)
  479. if data is None:
  480. return
  481. if not data:
  482. GMessage(_("No layers selected. Operation canceled."),
  483. parent=self)
  484. return
  485. if not self._validateOutputMapName():
  486. return
  487. dsn = self.dsnInput.GetDsn()
  488. ext = self.dsnInput.GetFormatExt()
  489. # determine data driver for PostGIS links
  490. self.popOGR = False
  491. if self.dsnInput.GetType() == 'db' and \
  492. self.dsnInput.GetFormat() == 'PostgreSQL' and \
  493. 'GRASS_VECTOR_OGR' not in os.environ:
  494. self.popOGR = True
  495. os.environ['GRASS_VECTOR_OGR'] = '1'
  496. for layer, output, listId in data:
  497. userData = {}
  498. if ext and layer.rfind(ext) > -1:
  499. layer = layer.replace('.' + ext, '')
  500. if '|' in layer:
  501. layer, geometry = layer.split('|', 1)
  502. else:
  503. geometry = None
  504. # TODO: v.import has no geometry option
  505. # if geometry:
  506. # cmd.append('geometry=%s' % geometry)
  507. cmd = self.getSettingsPageCmd()
  508. cmd.append('input=%s' % dsn)
  509. cmd.append('layer=%s' % layer)
  510. cmd.append('output=%s' % output)
  511. if self.override.IsChecked():
  512. cmd.append('-o')
  513. if self.overwrite.IsChecked():
  514. cmd.append('--overwrite')
  515. # TODO options
  516. if UserSettings.Get(group='cmd', key='overwrite',
  517. subkey='enabled') and '--overwrite' not in cmd:
  518. cmd.append('--overwrite')
  519. # run in Layer Manager
  520. self._giface.RunCmd(
  521. cmd,
  522. onDone=self.OnCmdDone,
  523. userData=userData,
  524. addLayer=False)
  525. def OnCmdDone(self, event):
  526. """Load layers and close if required"""
  527. if not hasattr(self, 'AddLayers'):
  528. return
  529. self.AddLayers(event.returncode, event.cmd, event.userData)
  530. if self.popOGR:
  531. os.environ.pop('GRASS_VECTOR_OGR')
  532. if event.returncode == 0 and self.closeOnFinish.IsChecked():
  533. self.Close()
  534. def _getCommand(self):
  535. """Get command"""
  536. if self.link:
  537. return 'v.external'
  538. else:
  539. return 'v.import'
  540. def _getBlackListedParameters(self):
  541. """Get parametrs which will not be showed in Settings page"""
  542. return ['input', 'output', 'layer']
  543. def _getBlackListedFlags(self):
  544. """Get flags which will not be showed in Settings page"""
  545. return ['overwrite', 'o', 'l', 'f']
  546. class GdalOutputDialog(wx.Dialog):
  547. def __init__(self, parent, id=wx.ID_ANY, ogr=False,
  548. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, *kwargs):
  549. """Dialog for setting output format for rasters/vectors
  550. .. todo::
  551. Split into GdalOutputDialog and OgrOutputDialog
  552. :param parent: parent window
  553. :param id: window id
  554. :param ogr: True for OGR (vector) otherwise GDAL (raster)
  555. :param style: window style
  556. :param *kwargs: other wx.Dialog's arguments
  557. """
  558. self.parent = parent # GMFrame
  559. self.ogr = ogr
  560. wx.Dialog.__init__(self, parent, id=id, style=style, *kwargs)
  561. if self.ogr:
  562. self.SetTitle(_("Define output format for vector data"))
  563. else:
  564. self.SetTitle(_("Define output format for raster data"))
  565. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  566. # buttons
  567. self.btnCancel = Button(parent=self.panel, id=wx.ID_CANCEL)
  568. self.btnCancel.SetToolTip(_("Close dialog"))
  569. self.btnOk = Button(parent=self.panel, id=wx.ID_OK)
  570. self.btnOk.SetToolTip(_("Set external format and close dialog"))
  571. self.btnOk.SetDefault()
  572. self.dsnInput = GdalSelect(parent=self, panel=self.panel,
  573. ogr=ogr,
  574. exclude=['file', 'protocol'], dest=True)
  575. self.dsnInput.AttachSettings()
  576. self.Bind(wx.EVT_BUTTON, self.OnCancel, self.btnCancel)
  577. self.Bind(wx.EVT_BUTTON, self.OnOK, self.btnOk)
  578. self._layout()
  579. def _layout(self):
  580. dialogSizer = wx.BoxSizer(wx.VERTICAL)
  581. dialogSizer.Add(self.dsnInput, proportion=1,
  582. flag=wx.EXPAND)
  583. btnSizer = wx.BoxSizer(orient=wx.HORIZONTAL)
  584. btnSizer.Add(self.btnCancel, proportion=0,
  585. flag=wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER,
  586. border=10)
  587. btnSizer.Add(self.btnOk, proportion=0,
  588. flag=wx.RIGHT | wx.ALIGN_CENTER,
  589. border=10)
  590. dialogSizer.Add(
  591. btnSizer,
  592. proportion=0,
  593. flag=wx.BOTTOM | wx.TOP | wx.ALIGN_RIGHT,
  594. border=10)
  595. self.panel.SetAutoLayout(True)
  596. self.panel.SetSizer(dialogSizer)
  597. dialogSizer.Fit(self.panel)
  598. size = wx.Size(
  599. globalvar.DIALOG_GSELECT_SIZE[0] + 320,
  600. self.GetBestSize()[1] + 35)
  601. self.SetMinSize(size)
  602. self.SetSize((size.width, size.height))
  603. self.Layout()
  604. def OnCancel(self, event):
  605. self.Destroy()
  606. def OnOK(self, event):
  607. if self.dsnInput.GetType() == 'native':
  608. RunCommand('v.external.out',
  609. parent=self,
  610. flags='r')
  611. else:
  612. dsn = self.dsnInput.GetDsn()
  613. frmt = self.dsnInput.GetFormat()
  614. options = self.dsnInput.GetOptions()
  615. if not dsn:
  616. GMessage(_("No data source selected."), parent=self)
  617. return
  618. RunCommand('v.external.out',
  619. parent=self,
  620. output=dsn, format=frmt,
  621. options=options)
  622. self.Close()
  623. class DxfImportDialog(ImportDialog):
  624. """Dialog for bulk import of DXF layers"""
  625. def __init__(self, parent, giface):
  626. ImportDialog.__init__(self, parent, giface=giface, itype='dxf',
  627. title=_("Import DXF layers"))
  628. self.list.SetValidator(
  629. LayersListValidator(
  630. condition='vector',
  631. callback=self._nameValidationFailed))
  632. self._giface = giface
  633. self.dsnInput = filebrowse.FileBrowseButton(
  634. parent=self.panel,
  635. id=wx.ID_ANY,
  636. size=globalvar.DIALOG_GSELECT_SIZE,
  637. labelText='',
  638. dialogTitle=_('Choose DXF file to import'),
  639. buttonText=_('Browse'),
  640. startDirectory=os.getcwd(),
  641. fileMode=0,
  642. changeCallback=self.OnSetDsn,
  643. fileMask="DXF File (*.dxf)|*.dxf")
  644. self.add.SetLabel(_("Add imported layers into layer tree"))
  645. self.add.SetValue(
  646. UserSettings.Get(
  647. group='cmd',
  648. key='addNewLayer',
  649. subkey='enabled'))
  650. self.doLayout()
  651. def OnRun(self, event):
  652. """Import/Link data (each layes as separate vector map)"""
  653. data = self.list.GetLayers()
  654. if not data:
  655. GMessage(_("No layers selected."), parent=self)
  656. return
  657. if not self._validateOutputMapName():
  658. return
  659. # hide dialog
  660. self.Hide()
  661. inputDxf = self.dsnInput.GetValue()
  662. for layer, output, itemId in data:
  663. cmd = self.getSettingsPageCmd()
  664. cmd.append('input=%s' % inputDxf)
  665. cmd.append('layer=%s' % layer)
  666. cmd.append('output=%s' % output)
  667. for key in self.options.keys():
  668. if self.options[key].IsChecked():
  669. cmd.append('-%s' % key)
  670. if self.overwrite.IsChecked() or UserSettings.Get(
  671. group='cmd', key='overwrite', subkey='enabled'):
  672. cmd.append('--overwrite')
  673. # run in Layer Manager
  674. self._giface.RunCmd(cmd, onDone=self.OnCmdDone, addLayer=False)
  675. def OnCmdDone(self, event):
  676. """Load layers and close if required"""
  677. if not hasattr(self, 'AddLayers'):
  678. return
  679. self.AddLayers(event.returncode, event.cmd)
  680. if self.closeOnFinish.IsChecked():
  681. self.Close()
  682. def OnSetDsn(self, event):
  683. """Input DXF file defined, update list of layer widget"""
  684. path = event.GetString()
  685. if not path:
  686. return
  687. data = list()
  688. ret = RunCommand('v.in.dxf',
  689. quiet=True,
  690. parent=self,
  691. read=True,
  692. flags='l',
  693. input=path)
  694. if not ret:
  695. self.list.LoadData()
  696. return
  697. for line in ret.splitlines():
  698. layerId = line.split(':')[0].split(' ')[1]
  699. layerName = line.split(':')[1].strip()
  700. grassName = GetValidLayerName(layerName)
  701. data.append((layerId, layerName.strip(), grassName.strip()))
  702. self.list.LoadData(data)
  703. def _getCommand(self):
  704. """Get command"""
  705. return 'v.in.dxf'
  706. def _getBlackListedParameters(self):
  707. """Get parametrs which will not be showed in Settings page"""
  708. return ['input', 'output', 'layers']
  709. def _getBlackListedFlags(self):
  710. """Get flags which will not be showed in Settings page"""
  711. return ['overwrite']
  712. class ReprojectionDialog(wx.Dialog):
  713. """ """
  714. def __init__(self, parent, giface, data,
  715. id=wx.ID_ANY, title=_("Reprojection"),
  716. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  717. self.parent = parent # GMFrame
  718. self._giface = giface # used to add layers
  719. wx.Dialog.__init__(self, parent, id, title, style=style,
  720. name="MultiImportDialog")
  721. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  722. # list of layers
  723. columns = [_('Layer id'),
  724. _('Name for output GRASS map')]
  725. self.layerBox = StaticBox(parent=self.panel, id=wx.ID_ANY)
  726. self.layerSizer = wx.StaticBoxSizer(self.layerBox, wx.HORIZONTAL)
  727. self.list = GListCtrl(parent=self.panel)
  728. for i in range(len(columns)):
  729. self.list.InsertColumn(i, columns[i])
  730. width = (65, 180)
  731. for i in range(len(width)):
  732. self.list.SetColumnWidth(col=i, width=width[i])
  733. self.list.LoadData(data)
  734. self.list.SelectAll(True)
  735. self.labelText = StaticText(parent=self.panel, id=wx.ID_ANY, label=_(
  736. "Projection of following layers do not match with projection of current location. "))
  737. label = _("Layers to be reprojected")
  738. self.layerBox.SetLabel(" %s - %s " %
  739. (label, _("right click to (un)select all")))
  740. #
  741. # buttons
  742. #
  743. # cancel
  744. self.btn_close = Button(parent=self.panel, id=wx.ID_CANCEL)
  745. # run
  746. self.btn_run = Button(
  747. parent=self.panel,
  748. id=wx.ID_OK,
  749. label=_("&Import && reproject"))
  750. self.btn_run.SetToolTip(_("Reproject selected layers"))
  751. self.btn_run.SetDefault()
  752. self.doLayout()
  753. def doLayout(self):
  754. """Do layout"""
  755. dialogSizer = wx.BoxSizer(wx.VERTICAL)
  756. dialogSizer.Add(self.labelText,
  757. flag=wx.ALL | wx.EXPAND, border=5)
  758. self.layerSizer.Add(self.list, proportion=1,
  759. flag=wx.ALL | wx.EXPAND, border=5)
  760. dialogSizer.Add(
  761. self.layerSizer,
  762. proportion=1,
  763. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
  764. border=5)
  765. #
  766. # buttons
  767. #
  768. btnsizer = wx.BoxSizer(orient=wx.HORIZONTAL)
  769. btnsizer.Add(self.btn_close, proportion=0,
  770. flag=wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER,
  771. border=10)
  772. btnsizer.Add(self.btn_run, proportion=0,
  773. flag=wx.RIGHT | wx.ALIGN_CENTER,
  774. border=10)
  775. dialogSizer.Add(
  776. btnsizer,
  777. proportion=0,
  778. flag=wx.BOTTOM | wx.ALIGN_RIGHT,
  779. border=10)
  780. self.panel.SetSizer(dialogSizer)
  781. dialogSizer.Fit(self.panel)
  782. self.Layout()
  783. def GetData(self, checked):
  784. return self.list.GetData(checked)