gselect.py 77 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024
  1. """!
  2. @package gui_core.gselect
  3. @brief Custom control that selects elements
  4. Classes:
  5. - gselect::Select
  6. - gselect::VectorSelect
  7. - gselect::TreeCrtlComboPopup
  8. - gselect::VectorDBInfo
  9. - gselect::LayerSelect
  10. - gselect::DriverSelect
  11. - gselect::DatabaseSelect
  12. - gselect::TableSelect
  13. - gselect::ColumnSelect
  14. - gselect::DbaseSelect
  15. - gselect::LocationSelect
  16. - gselect::MapsetSelect
  17. - gselect::SubGroupSelect
  18. - gselect::FormatSelect
  19. - gselect::GdalSelect
  20. - gselect::ProjSelect
  21. - gselect::ElementSelect
  22. - gselect::OgrTypeSelect
  23. - gselect::CoordinatesSelect
  24. (C) 2007-2012 by the GRASS Development Team
  25. This program is free software under the GNU General Public License
  26. (>=v2). Read the file COPYING that comes with GRASS for details.
  27. @author Michael Barton
  28. @author Martin Landa <landa.martin gmail.com>
  29. @author Vaclav Petras <wenzeslaus gmail.com> (menu customization)
  30. @author Stepan Turek <stepan.turek seznam.cz> (CoordinatesSelect)
  31. """
  32. import os
  33. import sys
  34. import glob
  35. import wx
  36. import wx.combo
  37. import wx.lib.buttons as buttons
  38. import wx.lib.filebrowsebutton as filebrowse
  39. from wx.lib.newevent import NewEvent
  40. from core import globalvar
  41. import grass.script as grass
  42. import grass.temporal as tgis
  43. from grass.script import task as gtask
  44. from core.gcmd import RunCommand, GError, GMessage
  45. from core.utils import GetListOfLocations, GetListOfMapsets, GetFormats
  46. from core.utils import GetSettingsPath, GetValidLayerName, ListSortLower
  47. from core.settings import UserSettings
  48. from core.debug import Debug
  49. wxGdalSelect, EVT_GDALSELECT = NewEvent()
  50. class Select(wx.combo.ComboCtrl):
  51. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
  52. type = None, multiple = False, nmaps = 1,
  53. mapsets = None, updateOnPopup = True, onPopup = None,
  54. fullyQualified = True):
  55. """!Custom control to create a ComboBox with a tree control to
  56. display and select GIS elements within acessible mapsets.
  57. Elements can be selected with mouse. Can allow multiple
  58. selections, when argument <em>multiple</em> is True. Multiple
  59. selections are separated by commas.
  60. @param type type of GIS elements ('raster, 'vector', ...)
  61. @param multiple True for multiple input
  62. @param nmaps number of maps to be entered
  63. @param mapsets force list of mapsets (otherwise search path)
  64. @param updateOnPopup True for updating list of elements on popup
  65. @param onPopup function to be called on Popup
  66. @param fullyQualified True to provide fully qualified names (map@mapset)
  67. """
  68. wx.combo.ComboCtrl.__init__(self, parent=parent, id=id, size=size)
  69. self.GetChildren()[0].SetName("Select")
  70. self.GetChildren()[0].type = type
  71. self.tcp = TreeCtrlComboPopup()
  72. self.SetPopupControl(self.tcp)
  73. self.SetPopupExtents(0, 100)
  74. if type:
  75. self.tcp.SetData(type = type, mapsets = mapsets,
  76. multiple = multiple, nmaps = nmaps,
  77. updateOnPopup = updateOnPopup, onPopup = onPopup,
  78. fullyQualified = fullyQualified)
  79. self.GetChildren()[0].Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  80. def OnKeyUp(self, event):
  81. """!Shows popupwindow if down arrow key is released"""
  82. if event.GetKeyCode() == wx.WXK_DOWN and not self.IsPopupShown():
  83. self.ShowPopup()
  84. else:
  85. event.Skip()
  86. def SetElementList(self, type, mapsets = None):
  87. """!Set element list
  88. @param type GIS element type
  89. @param mapsets list of acceptable mapsets (None for all in search path)
  90. """
  91. self.tcp.SetData(type = type, mapsets = mapsets)
  92. def GetElementList(self):
  93. """!Load elements"""
  94. self.tcp.GetElementList()
  95. def SetType(self, etype, multiple = False, nmaps = 1,
  96. mapsets = None, updateOnPopup = True, onPopup = None):
  97. """!Param set element type for widget
  98. @param etype element type, see gselect.ElementSelect
  99. """
  100. self.tcp.SetData(type = etype, mapsets = mapsets,
  101. multiple = multiple, nmaps = nmaps,
  102. updateOnPopup = updateOnPopup, onPopup = onPopup)
  103. class VectorSelect(Select):
  104. def __init__(self, parent, ftype, **kwargs):
  105. """!Custom to create a ComboBox with a tree control to display and
  106. select vector maps. Control allows to filter vector maps. If you
  107. don't need this feature use Select class instead
  108. @param ftype filter vector maps based on feature type
  109. """
  110. Select.__init__(self, parent = parent, id = wx.ID_ANY,
  111. type = 'vector', **kwargs)
  112. self.ftype = ftype
  113. # remove vector maps which do not contain given feature type
  114. self.tcp.SetFilter(self._isElement)
  115. def _isElement(self, vectorName):
  116. """!Check if element should be filtered out"""
  117. try:
  118. if int(grass.vector_info_topo(vectorName)[self.ftype]) < 1:
  119. return False
  120. except KeyError:
  121. return False
  122. return True
  123. class TreeCtrlComboPopup(wx.combo.ComboPopup):
  124. """!Create a tree ComboBox for selecting maps and other GIS elements
  125. in accessible mapsets within the current location
  126. """
  127. # overridden ComboPopup methods
  128. def Init(self):
  129. self.value = [] # for multiple is False -> len(self.value) in [0,1]
  130. self.curitem = None
  131. self.multiple = False
  132. self.nmaps = 1
  133. self.type = None
  134. self.mapsets = None
  135. self.updateOnPopup = True
  136. self.onPopup = None
  137. self.fullyQualified = True
  138. self.SetFilter(None)
  139. tgis.init()
  140. def Create(self, parent):
  141. self.seltree = wx.TreeCtrl(parent, style=wx.TR_HIDE_ROOT
  142. |wx.TR_HAS_BUTTONS
  143. |wx.TR_SINGLE
  144. |wx.TR_LINES_AT_ROOT
  145. |wx.SIMPLE_BORDER
  146. |wx.TR_FULL_ROW_HIGHLIGHT)
  147. self.seltree.Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  148. self.seltree.Bind(wx.EVT_MOTION, self.OnMotion)
  149. self.seltree.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  150. self.seltree.Bind(wx.EVT_TREE_ITEM_EXPANDING, self.mapsetExpanded)
  151. self.seltree.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.mapsetCollapsed)
  152. self.seltree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.mapsetActivated)
  153. self.seltree.Bind(wx.EVT_TREE_SEL_CHANGED, self.mapsetSelected)
  154. self.seltree.Bind(wx.EVT_TREE_DELETE_ITEM, lambda x: None)
  155. # the following dummy handler are needed to keep tree events from propagating up to
  156. # the parent GIS Manager layer tree
  157. def mapsetExpanded(self, event):
  158. pass
  159. def mapsetCollapsed(self, event):
  160. pass
  161. def mapsetActivated(self, event):
  162. pass
  163. def mapsetSelected(self, event):
  164. pass
  165. # end of dummy events
  166. def GetControl(self):
  167. return self.seltree
  168. def GetStringValue(self):
  169. """!Get value as a string separated by commas"""
  170. return ','.join(self.value)
  171. def SetFilter(self, filter):
  172. """!Set filter for GIS elements, see e.g. VectorSelect"""
  173. self.filterElements = filter
  174. def OnPopup(self, force = False):
  175. """!Limited only for first selected"""
  176. if not force and not self.updateOnPopup:
  177. return
  178. if self.onPopup:
  179. selected, exclude = self.onPopup(self.type)
  180. else:
  181. selected = None
  182. exclude = False
  183. self.GetElementList(selected, exclude)
  184. # selects map starting according to written text
  185. inputText = self.GetCombo().GetValue().strip()
  186. if inputText:
  187. root = self.seltree.GetRootItem()
  188. match = self.FindItem(root, inputText, startLetters = True)
  189. self.seltree.EnsureVisible(match)
  190. self.seltree.SelectItem(match)
  191. def GetElementList(self, elements = None, exclude = False):
  192. """!Get filtered list of GIS elements in accessible mapsets
  193. and display as tree with all relevant elements displayed
  194. beneath each mapset branch
  195. """
  196. # update list
  197. self.seltree.DeleteAllItems()
  198. self._getElementList(self.type, self.mapsets, elements, exclude)
  199. if len(self.value) > 0:
  200. root = self.seltree.GetRootItem()
  201. if not root:
  202. return
  203. item = self.FindItem(root, self.value[0])
  204. try:
  205. self.seltree.EnsureVisible(item)
  206. self.seltree.SelectItem(item)
  207. except:
  208. pass
  209. def SetStringValue(self, value):
  210. # this assumes that item strings are unique...
  211. root = self.seltree.GetRootItem()
  212. if not root:
  213. return
  214. winValue = self.GetCombo().GetValue().strip(',')
  215. self.value = []
  216. if winValue:
  217. self.value = winValue.split(',')
  218. def GetAdjustedSize(self, minWidth, prefHeight, maxHeight):
  219. """!Reads UserSettings to get height (which was 200 in old implementation).
  220. """
  221. height = UserSettings.Get(group = 'appearance', key = 'gSelectPopupHeight', subkey = 'value')
  222. return wx.Size(minWidth, min(height, maxHeight))
  223. def _getElementList(self, element, mapsets = None, elements = None, exclude = False):
  224. """!Get list of GIS elements in accessible mapsets and display as tree
  225. with all relevant elements displayed beneath each mapset branch
  226. @param element GIS element
  227. @param mapsets list of acceptable mapsets (None for all mapsets in search path)
  228. @param elements list of forced GIS elements
  229. @param exclude True to exclude, False for forcing the list (elements)
  230. """
  231. # get current mapset
  232. curr_mapset = grass.gisenv()['MAPSET']
  233. # map element types to g.mlist types
  234. elementdict = {'cell':'rast',
  235. 'raster':'rast',
  236. 'rast':'rast',
  237. 'raster files':'rast',
  238. 'grid3':'rast3d',
  239. 'rast3d':'rast3d',
  240. '3d-raster':'rast3d',
  241. 'raster3D':'rast3d',
  242. 'raster3D files':'rast3d',
  243. 'vector':'vect',
  244. 'vect':'vect',
  245. 'binary vector files':'vect',
  246. 'dig':'oldvect',
  247. 'oldvect':'oldvect',
  248. 'old vector':'oldvect',
  249. 'dig_ascii':'asciivect',
  250. 'asciivect':'asciivect',
  251. 'asciivector':'asciivect',
  252. 'ascii vector files':'asciivect',
  253. 'icons':'icon',
  254. 'icon':'icon',
  255. 'paint icon files':'icon',
  256. 'paint/labels':'labels',
  257. 'labels':'labels',
  258. 'label':'labels',
  259. 'paint label files':'labels',
  260. 'site_lists':'sites',
  261. 'sites':'sites',
  262. 'site list':'sites',
  263. 'site list files':'sites',
  264. 'windows':'region',
  265. 'region':'region',
  266. 'region definition':'region',
  267. 'region definition files':'region',
  268. 'windows3d':'region3d',
  269. 'region3d':'region3d',
  270. 'region3D definition':'region3d',
  271. 'region3D definition files':'region3d',
  272. 'group':'group',
  273. 'imagery group':'group',
  274. 'imagery group files':'group',
  275. '3d.view':'3dview',
  276. '3dview':'3dview',
  277. '3D viewing parameters':'3dview',
  278. '3D view parameters':'3dview',
  279. 'stds':'stds',
  280. 'strds':'strds',
  281. 'str3ds':'str3ds',
  282. 'stvds':'stvds'}
  283. if element not in elementdict:
  284. self.AddItem(_('Not selectable element'))
  285. return
  286. if element in ('stds', 'strds', 'str3ds', 'stvds'):
  287. filesdict = tgis.tlist_grouped(elementdict[element])
  288. else:
  289. if globalvar.have_mlist:
  290. filesdict = grass.mlist_grouped(elementdict[element],
  291. check_search_path = False)
  292. else:
  293. filesdict = grass.list_grouped(elementdict[element],
  294. check_search_path = False)
  295. # list of mapsets in current location
  296. if mapsets is None:
  297. mapsets = grass.mapsets(search_path = True)
  298. # current mapset first
  299. if curr_mapset in mapsets and mapsets[0] != curr_mapset:
  300. mapsets.remove(curr_mapset)
  301. mapsets.insert(0, curr_mapset)
  302. first_mapset = None
  303. for mapset in mapsets:
  304. mapset_node = self.AddItem(_('Mapset') + ': ' + mapset)
  305. if not first_mapset:
  306. first_mapset = mapset_node
  307. self.seltree.SetItemTextColour(mapset_node, wx.Colour(50, 50, 200))
  308. if mapset not in filesdict:
  309. continue
  310. try:
  311. elem_list = filesdict[mapset]
  312. elem_list.sort()
  313. for elem in elem_list:
  314. if elem != '':
  315. fullqElem = elem + '@' + mapset
  316. if elements is not None:
  317. if (exclude and fullqElem in elements) or \
  318. (not exclude and fullqElem not in elements):
  319. continue
  320. if self.filterElements:
  321. if self.filterElements(fullqElem):
  322. self.AddItem(elem, parent = mapset_node)
  323. else:
  324. self.AddItem(elem, parent = mapset_node)
  325. except StandardError, e:
  326. sys.stderr.write(_("GSelect: invalid item: %s") % e)
  327. continue
  328. if self.seltree.ItemHasChildren(mapset_node):
  329. sel = UserSettings.Get(group='appearance', key='elementListExpand',
  330. subkey='selection')
  331. collapse = True
  332. if sel == 0: # collapse all except PERMANENT and current
  333. if mapset in ('PERMANENT', curr_mapset):
  334. collapse = False
  335. elif sel == 1: # collapse all except PERMANENT
  336. if mapset == 'PERMANENT':
  337. collapse = False
  338. elif sel == 2: # collapse all except current
  339. if mapset == curr_mapset:
  340. collapse = False
  341. elif sel == 3: # collapse all
  342. pass
  343. elif sel == 4: # expand all
  344. collapse = False
  345. if collapse:
  346. self.seltree.Collapse(mapset_node)
  347. else:
  348. self.seltree.Expand(mapset_node)
  349. if first_mapset:
  350. # select first mapset (MSW hack)
  351. self.seltree.SelectItem(first_mapset)
  352. # helpers
  353. def FindItem(self, parentItem, text, startLetters = False):
  354. """!Finds item with given name or starting with given text"""
  355. startletters = startLetters
  356. item, cookie = self.seltree.GetFirstChild(parentItem)
  357. while wx.TreeItemId.IsOk(item):
  358. if self.seltree.GetItemText(item) == text:
  359. return item
  360. if self.seltree.ItemHasChildren(item):
  361. item = self.FindItem(item, text, startLetters = startletters)
  362. if wx.TreeItemId.IsOk(item):
  363. return item
  364. elif startletters and self.seltree.GetItemText(item).startswith(text.split('@', 1)[0]):
  365. return item
  366. item, cookie = self.seltree.GetNextChild(parentItem, cookie)
  367. return wx.TreeItemId()
  368. def AddItem(self, value, parent=None):
  369. if not parent:
  370. root = self.seltree.GetRootItem()
  371. if not root:
  372. root = self.seltree.AddRoot("<hidden root>")
  373. parent = root
  374. item = self.seltree.AppendItem(parent, text=value)
  375. return item
  376. # able to recieve only wx.EVT_KEY_UP
  377. def OnKeyUp(self, event):
  378. """!Enables to select items using keyboard"""
  379. item = self.seltree.GetSelection()
  380. if event.GetKeyCode() == wx.WXK_DOWN:
  381. self.seltree.SelectItem(self.seltree.GetNextVisible(item))
  382. # problem with GetPrevVisible
  383. elif event.GetKeyCode() == wx.WXK_UP:
  384. if self.seltree.ItemHasChildren(item) and self.seltree.IsExpanded(self.seltree.GetPrevSibling(item)):
  385. itemPrev = self.seltree.GetLastChild(self.seltree.GetPrevSibling(item))
  386. else:
  387. itemPrev = self.seltree.GetPrevSibling(item)
  388. if not wx.TreeItemId.IsOk(itemPrev):
  389. itemPrev = self.seltree.GetItemParent(item)
  390. if item == self.seltree.GetFirstChild(self.seltree.GetRootItem())[0]:
  391. itemPrev = item
  392. self.seltree.SelectItem(itemPrev)
  393. # selects first item starting with the written text in next mapset
  394. elif event.GetKeyCode() == wx.WXK_TAB:
  395. selected = self.seltree.GetSelection()
  396. if self.seltree.ItemHasChildren(selected):
  397. parent = selected
  398. else:
  399. parent = self.seltree.GetItemParent(selected)
  400. nextSibling = self.seltree.GetNextSibling(parent)
  401. if wx.TreeItemId.IsOk(nextSibling):
  402. match = self.FindItem(nextSibling, self.GetCombo().GetValue().strip(), True)
  403. else:
  404. match = self.FindItem(self.seltree.GetFirstChild(self.seltree.GetItemParent(parent))[0],
  405. self.GetCombo().GetValue().strip(), True)
  406. self.seltree.SelectItem(match)
  407. elif event.GetKeyCode() == wx.WXK_RIGHT:
  408. if self.seltree.ItemHasChildren(item):
  409. self.seltree.Expand(item)
  410. elif event.GetKeyCode() == wx.WXK_LEFT:
  411. if self.seltree.ItemHasChildren(item):
  412. self.seltree.Collapse(item)
  413. elif event.GetKeyCode() == wx.WXK_ESCAPE:
  414. self.Dismiss()
  415. elif event.GetKeyCode() == wx.WXK_RETURN:
  416. if self.seltree.GetRootItem() == self.seltree.GetItemParent(item):
  417. self.value = []
  418. else:
  419. mapsetItem = self.seltree.GetItemParent(item)
  420. fullName = self.seltree.GetItemText(item)
  421. if self.fullyQualified:
  422. fullName += '@' + self.seltree.GetItemText(mapsetItem).split(':', -1)[1].strip()
  423. if self.multiple:
  424. self.value.append(fullName)
  425. else:
  426. if self.nmaps > 1: # see key_desc
  427. if len(self.value) >= self.nmaps:
  428. self.value = [fullName]
  429. else:
  430. self.value.append(fullName)
  431. else:
  432. self.value = [fullName]
  433. self.Dismiss()
  434. def OnMotion(self, evt):
  435. """!Have the selection follow the mouse, like in a real combobox
  436. """
  437. item, flags = self.seltree.HitTest(evt.GetPosition())
  438. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  439. self.seltree.SelectItem(item)
  440. self.curitem = item
  441. evt.Skip()
  442. def OnLeftDown(self, evt):
  443. """!Do the combobox selection
  444. """
  445. item, flags = self.seltree.HitTest(evt.GetPosition())
  446. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  447. self.curitem = item
  448. if self.seltree.GetRootItem() == self.seltree.GetItemParent(item):
  449. self.value = [] # cannot select mapset item
  450. evt.Skip()
  451. return
  452. mapsetItem = self.seltree.GetItemParent(item)
  453. fullName = self.seltree.GetItemText(item)
  454. if self.fullyQualified:
  455. fullName += '@' + self.seltree.GetItemText(mapsetItem).split(':', -1)[1].strip()
  456. if self.multiple:
  457. self.value.append(fullName)
  458. else:
  459. if self.nmaps > 1: # see key_desc
  460. if len(self.value) >= self.nmaps:
  461. self.value = [fullName]
  462. else:
  463. self.value.append(fullName)
  464. else:
  465. self.value = [fullName]
  466. self.Dismiss()
  467. evt.Skip()
  468. def SetData(self, **kargs):
  469. """!Set object properties"""
  470. if 'type' in kargs:
  471. self.type = kargs['type']
  472. if 'mapsets' in kargs:
  473. self.mapsets = kargs['mapsets']
  474. if 'multiple' in kargs:
  475. self.multiple = kargs['multiple']
  476. if 'nmaps' in kargs:
  477. self.nmaps = kargs['nmaps']
  478. if 'updateOnPopup' in kargs:
  479. self.updateOnPopup = kargs['updateOnPopup']
  480. if 'onPopup' in kargs:
  481. self.onPopup = kargs['onPopup']
  482. if 'fullyQualified' in kargs:
  483. self.fullyQualified = kargs['fullyQualified']
  484. def GetType(self):
  485. """!Get element type
  486. """
  487. return self.type
  488. class VectorDBInfo:
  489. """!Class providing information about attribute tables
  490. linked to a vector map"""
  491. def __init__(self, map):
  492. self.map = map
  493. # dictionary of layer number and associated (driver, database, table)
  494. self.layers = {}
  495. # dictionary of table and associated columns (type, length, values, ids)
  496. self.tables = {}
  497. if not self._CheckDBConnection(): # -> self.layers
  498. return
  499. self._DescribeTables() # -> self.tables
  500. def _CheckDBConnection(self):
  501. """!Check DB connection"""
  502. nuldev = file(os.devnull, 'w+')
  503. self.layers = grass.vector_db(map = self.map, stderr = nuldev)
  504. nuldev.close()
  505. return bool(len(self.layers.keys()) > 0)
  506. def _DescribeTables(self):
  507. """!Describe linked tables"""
  508. for layer in self.layers.keys():
  509. # determine column names and types
  510. table = self.layers[layer]["table"]
  511. columns = {} # {name: {type, length, [values], [ids]}}
  512. i = 0
  513. Debug.msg(1, "gselect.VectorDBInfo._DescribeTables(): table=%s driver=%s database=%s" % \
  514. (self.layers[layer]["table"], self.layers[layer]["driver"],
  515. self.layers[layer]["database"]))
  516. for item in grass.db_describe(table = self.layers[layer]["table"],
  517. driver = self.layers[layer]["driver"],
  518. database = self.layers[layer]["database"])['cols']:
  519. name, type, length = item
  520. # FIXME: support more datatypes
  521. if type.lower() == "integer":
  522. ctype = int
  523. elif type.lower() == "double precision":
  524. ctype = float
  525. else:
  526. ctype = str
  527. columns[name.strip()] = { 'index' : i,
  528. 'type' : type.lower(),
  529. 'ctype' : ctype,
  530. 'length' : int(length),
  531. 'values' : [],
  532. 'ids' : []}
  533. i += 1
  534. # check for key column
  535. # v.db.connect -g/p returns always key column name lowercase
  536. if self.layers[layer]["key"] not in columns.keys():
  537. for col in columns.keys():
  538. if col.lower() == self.layers[layer]["key"]:
  539. self.layers[layer]["key"] = col.upper()
  540. break
  541. self.tables[table] = columns
  542. return True
  543. def Reset(self):
  544. """!Reset"""
  545. for layer in self.layers:
  546. table = self.layers[layer]["table"] # get table desc
  547. columns = self.tables[table]
  548. for name in self.tables[table].keys():
  549. self.tables[table][name]['values'] = []
  550. self.tables[table][name]['ids'] = []
  551. def GetName(self):
  552. """!Get vector name"""
  553. return self.map
  554. def GetKeyColumn(self, layer):
  555. """!Get key column of given layer
  556. @param layer vector layer number
  557. """
  558. return str(self.layers[layer]['key'])
  559. def GetTable(self, layer):
  560. """!Get table name of given layer
  561. @param layer vector layer number
  562. """
  563. return self.layers[layer]['table']
  564. def GetDbSettings(self, layer):
  565. """!Get database settins
  566. @param layer layer number
  567. @return (driver, database)
  568. """
  569. return self.layers[layer]['driver'], self.layers[layer]['database']
  570. def GetTableDesc(self, table):
  571. """!Get table columns
  572. @param table table name
  573. """
  574. return self.tables[table]
  575. class LayerSelect(wx.ComboBox):
  576. def __init__(self, parent, id = wx.ID_ANY,
  577. size = globalvar.DIALOG_COMBOBOX_SIZE,
  578. vector = None, dsn = None, choices = [], all = False, default = None):
  579. """!Creates combo box for selecting vector map layer names
  580. @param vector vector map name (native or connected via v.external)
  581. @param dsn OGR data source name
  582. """
  583. super(LayerSelect, self).__init__(parent, id, size = size, choices = choices)
  584. self.all = all
  585. self.SetName("LayerSelect")
  586. # default value
  587. self.default = default
  588. self.InsertLayers(vector = vector, dsn = dsn)
  589. def InsertLayers(self, vector = None, dsn = None):
  590. """!Insert layers for a vector into the layer combobox
  591. @param vector vector map name (native or connected via v.external)
  592. @param dsn OGR data source name
  593. """
  594. layers = list()
  595. if self.all:
  596. layers.append('-1')
  597. if vector:
  598. # TODO: use Vect_get_field2() in C modules where possible
  599. # currently the following is identical to
  600. # layers = utils.GetVectorNumberOfLayers(self, vector)
  601. ret = RunCommand('v.db.connect',
  602. read = True,
  603. quiet = True,
  604. sep = '|',
  605. flags = 'g',
  606. map = vector)
  607. if ret:
  608. for line in ret.splitlines():
  609. layerinfo = line.split('|')
  610. layername = layerinfo[0].split('/')
  611. # use this to get layer names
  612. # but only when all modules use Vect_get_field2()
  613. # which is not the case right now
  614. ### layers.append(layername[len(layername) - 1])
  615. layers.append(layername[0])
  616. elif dsn:
  617. ret = RunCommand('v.in.ogr',
  618. read = True,
  619. quiet = True,
  620. flags = 'l',
  621. dsn = dsn)
  622. if ret:
  623. layers = ret.splitlines()
  624. if self.default:
  625. if len(layers) == 0:
  626. layers.insert(0, str(self.default))
  627. elif self.default not in layers:
  628. layers.append(self.default)
  629. if len(layers) > 0:
  630. self.SetItems(layers)
  631. class DriverSelect(wx.ComboBox):
  632. """!Creates combo box for selecting database driver.
  633. """
  634. def __init__(self, parent, choices, value,
  635. id=wx.ID_ANY, pos=wx.DefaultPosition,
  636. size=globalvar.DIALOG_LAYER_SIZE, **kargs):
  637. super(DriverSelect, self).__init__(parent, id, value, pos, size,
  638. choices, style=wx.CB_READONLY)
  639. self.SetName("DriverSelect")
  640. self.SetStringSelection(value)
  641. class DatabaseSelect(wx.TextCtrl):
  642. """!Creates combo box for selecting database driver.
  643. """
  644. def __init__(self, parent, value = '', id = wx.ID_ANY,
  645. size = globalvar.DIALOG_TEXTCTRL_SIZE, **kargs):
  646. super(DatabaseSelect, self).__init__(parent, id, value, size = size, **kargs)
  647. self.SetName("DatabaseSelect")
  648. class TableSelect(wx.ComboBox):
  649. """!Creates combo box for selecting attribute tables from the database
  650. """
  651. def __init__(self, parent,
  652. id = wx.ID_ANY, value = '',
  653. size = globalvar.DIALOG_COMBOBOX_SIZE, choices = [], **kargs):
  654. super(TableSelect, self).__init__(parent, id, value, size = size, choices = choices,
  655. style = wx.CB_READONLY, **kargs)
  656. self.SetName("TableSelect")
  657. if not choices:
  658. self.InsertTables()
  659. def InsertTables(self, driver = None, database = None):
  660. """!Insert attribute tables into combobox"""
  661. items = []
  662. if not driver or not database:
  663. connect = grass.db_connection()
  664. driver = connect['driver']
  665. database = connect['database']
  666. ret = RunCommand('db.tables',
  667. flags = 'p',
  668. read = True,
  669. driver = driver,
  670. database = database)
  671. if ret:
  672. for table in ret.splitlines():
  673. items.append(table)
  674. self.SetItems(items)
  675. self.SetValue('')
  676. class ColumnSelect(wx.ComboBox):
  677. """!Creates combo box for selecting columns in the attribute table
  678. for a vector map.
  679. @param parent window parent
  680. @param id window id
  681. @param value default value
  682. @param size window size
  683. @param vector vector map name
  684. @param layer layer number
  685. @param param parameters list (see menuform.py)
  686. @param **kwags wx.ComboBox parameters
  687. """
  688. def __init__(self, parent, id = wx.ID_ANY, value = '',
  689. size = globalvar.DIALOG_COMBOBOX_SIZE,
  690. vector = None, layer = 1, param = None, **kwargs):
  691. self.defaultValue = value
  692. self.param = param
  693. super(ColumnSelect, self).__init__(parent, id, value, size = size, **kwargs)
  694. self.SetName("ColumnSelect")
  695. if vector:
  696. self.InsertColumns(vector, layer)
  697. def InsertColumns(self, vector, layer, excludeKey = False, excludeCols = None, type = None, dbInfo = None):
  698. """!Insert columns for a vector attribute table into the columns combobox
  699. @param vector vector name
  700. @param layer vector layer number
  701. @param excludeKey exclude key column from the list?
  702. @param excludeCols list of columns to be removed from the list
  703. @param type only columns of given type (given as list)
  704. """
  705. if not dbInfo:
  706. dbInfo = VectorDBInfo(vector)
  707. try:
  708. table = dbInfo.GetTable(int(layer))
  709. columnchoices = dbInfo.GetTableDesc(table)
  710. keyColumn = dbInfo.GetKeyColumn(int(layer))
  711. columns = len(columnchoices.keys()) * ['']
  712. for key, val in columnchoices.iteritems():
  713. columns[val['index']] = key
  714. if excludeKey: # exclude key column
  715. columns.remove(keyColumn)
  716. if excludeCols: # exclude key column
  717. for key in columnchoices.iterkeys():
  718. if key in excludeCols:
  719. columns.remove(key)
  720. if type: # only selected column types
  721. for key, value in columnchoices.iteritems():
  722. if value['type'] not in type:
  723. try:
  724. columns.remove(key)
  725. except ValueError:
  726. pass
  727. except (KeyError, ValueError):
  728. columns = list()
  729. self.SetItems(columns)
  730. self.SetValue(self.defaultValue)
  731. if self.param:
  732. value = self.param.get('value', '')
  733. if value != '' and value in columns:
  734. self.SetValue(value)
  735. def InsertTableColumns(self, table, driver=None, database=None):
  736. """!Insert table columns
  737. @param table table name
  738. @param driver driver name
  739. @param database database name
  740. """
  741. columns = list()
  742. ret = RunCommand('db.columns',
  743. read = True,
  744. driver = driver,
  745. database = database,
  746. table = table)
  747. if ret:
  748. columns = ret.splitlines()
  749. self.SetItems(columns)
  750. self.SetValue(self.defaultValue)
  751. if self.param:
  752. value = self.param.get('value', '')
  753. if value != '' and value in columns:
  754. self.SetValue(value)
  755. class DbaseSelect(wx.lib.filebrowsebutton.DirBrowseButton):
  756. """!Widget for selecting GRASS Database"""
  757. def __init__(self, parent, **kwargs):
  758. super(DbaseSelect, self).__init__(parent, id = wx.ID_ANY,
  759. size = globalvar.DIALOG_GSELECT_SIZE, labelText = '',
  760. dialogTitle = _('Choose GIS Data Directory'),
  761. buttonText = _('Browse'),
  762. startDirectory = grass.gisenv()['GISDBASE'],
  763. **kwargs)
  764. class LocationSelect(wx.ComboBox):
  765. """!Widget for selecting GRASS location"""
  766. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  767. gisdbase = None, **kwargs):
  768. super(LocationSelect, self).__init__(parent, id, size = size,
  769. style = wx.CB_READONLY, **kwargs)
  770. self.SetName("LocationSelect")
  771. if not gisdbase:
  772. self.gisdbase = grass.gisenv()['GISDBASE']
  773. else:
  774. self.gisdbase = gisdbase
  775. self.SetItems(GetListOfLocations(self.gisdbase))
  776. def UpdateItems(self, dbase):
  777. """!Update list of locations
  778. @param dbase path to GIS database
  779. """
  780. self.gisdbase = dbase
  781. if dbase:
  782. self.SetItems(GetListOfLocations(self.gisdbase))
  783. else:
  784. self.SetItems([])
  785. class MapsetSelect(wx.ComboBox):
  786. """!Widget for selecting GRASS mapset"""
  787. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  788. gisdbase = None, location = None, setItems = True,
  789. searchPath = False, new = False, skipCurrent = False, **kwargs):
  790. style = 0
  791. if not new:
  792. style = wx.CB_READONLY
  793. super(MapsetSelect, self).__init__(parent, id, size = size,
  794. style = style, **kwargs)
  795. self.searchPath = searchPath
  796. self.skipCurrent = skipCurrent
  797. self.SetName("MapsetSelect")
  798. if not gisdbase:
  799. self.gisdbase = grass.gisenv()['GISDBASE']
  800. else:
  801. self.gisdbase = gisdbase
  802. if not location:
  803. self.location = grass.gisenv()['LOCATION_NAME']
  804. else:
  805. self.location = location
  806. if setItems:
  807. self.SetItems(self._getMapsets())
  808. def UpdateItems(self, location, dbase = None):
  809. """!Update list of mapsets for given location
  810. @param dbase path to GIS database (None to use currently selected)
  811. @param location name of location
  812. """
  813. if dbase:
  814. self.gisdbase = dbase
  815. self.location = location
  816. if location:
  817. self.SetItems(self._getMapsets())
  818. else:
  819. self.SetItems([])
  820. def _getMapsets(self):
  821. if self.searchPath:
  822. mlist = RunCommand('g.mapsets',
  823. read = True, flags = 'p',
  824. sep = 'newline').splitlines()
  825. else:
  826. mlist = GetListOfMapsets(self.gisdbase, self.location,
  827. selectable = False)
  828. gisenv = grass.gisenv()
  829. if self.skipCurrent and \
  830. gisenv['LOCATION_NAME'] == self.location and \
  831. gisenv['MAPSET'] in mlist:
  832. mlist.remove(gisenv['MAPSET'])
  833. return mlist
  834. class SubGroupSelect(wx.ComboBox):
  835. """!Widget for selecting subgroups"""
  836. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
  837. **kwargs):
  838. super(SubGroupSelect, self).__init__(parent, id, size = size,
  839. **kwargs)
  840. self.SetName("SubGroupSelect")
  841. def Insert(self, group):
  842. """!Insert subgroups for defined group"""
  843. if not group:
  844. return
  845. gisenv = grass.gisenv()
  846. try:
  847. name, mapset = group.split('@', 1)
  848. except ValueError:
  849. name = group
  850. mapset = gisenv['MAPSET']
  851. path = os.path.join(gisenv['GISDBASE'], gisenv['LOCATION_NAME'], mapset,
  852. 'group', name, 'subgroup')
  853. try:
  854. self.SetItems(os.listdir(path))
  855. except OSError:
  856. self.SetItems([])
  857. self.SetValue('')
  858. class FormatSelect(wx.Choice):
  859. def __init__(self, parent, ogr = False,
  860. sourceType = None, id = wx.ID_ANY, size = globalvar.DIALOG_SPIN_SIZE,
  861. **kwargs):
  862. """!Widget for selecting external (GDAL/OGR) format
  863. @param parent parent window
  864. @param sourceType source type ('file', 'directory', 'database', 'protocol') or None
  865. @param ogr True for OGR otherwise GDAL
  866. """
  867. super(FormatSelect, self).__init__(parent, id, size = size,
  868. **kwargs)
  869. self.SetName("FormatSelect")
  870. if ogr:
  871. ftype = 'ogr'
  872. else:
  873. ftype = 'gdal'
  874. formats = list()
  875. for f in GetFormats()[ftype].values():
  876. formats += f
  877. self.SetItems(formats)
  878. def GetExtension(self, name):
  879. """!Get file extension by format name"""
  880. formatToExt = {
  881. # raster
  882. 'GeoTIFF' : 'tif',
  883. 'Erdas Imagine Images (.img)' : 'img',
  884. 'Ground-based SAR Applications Testbed File Format (.gff)' : 'gff',
  885. 'Arc/Info Binary Grid' : 'adf',
  886. 'Portable Network Graphics' : 'png',
  887. 'JPEG JFIF' : 'jpg',
  888. 'Japanese DEM (.mem)' : 'mem',
  889. 'Graphics Interchange Format (.gif)' : 'gif',
  890. 'X11 PixMap Format' : 'xpm',
  891. 'MS Windows Device Independent Bitmap' : 'bmp',
  892. 'SPOT DIMAP' : 'dim',
  893. 'RadarSat 2 XML Product' : 'xml',
  894. 'EarthWatch .TIL' : 'til',
  895. 'ERMapper .ers Labelled' : 'ers',
  896. 'ERMapper Compressed Wavelets' : 'ecw',
  897. 'GRIdded Binary (.grb)' : 'grb',
  898. 'EUMETSAT Archive native (.nat)' : 'nat',
  899. 'Idrisi Raster A.1' : 'rst',
  900. 'Golden Software ASCII Grid (.grd)' : 'grd',
  901. 'Golden Software Binary Grid (.grd)' : 'grd',
  902. 'Golden Software 7 Binary Grid (.grd)' : 'grd',
  903. 'R Object Data Store' : 'r',
  904. 'USGS DOQ (Old Style)' : 'doq',
  905. 'USGS DOQ (New Style)' : 'doq',
  906. 'ENVI .hdr Labelled' : 'hdr',
  907. 'ESRI .hdr Labelled' : 'hdr',
  908. 'Generic Binary (.hdr Labelled)' : 'hdr',
  909. 'PCI .aux Labelled' : 'aux',
  910. 'EOSAT FAST Format' : 'fst',
  911. 'VTP .bt (Binary Terrain) 1.3 Format' : 'bt',
  912. 'FARSITE v.4 Landscape File (.lcp)' : 'lcp',
  913. 'Swedish Grid RIK (.rik)' : 'rik',
  914. 'USGS Optional ASCII DEM (and CDED)' : 'dem',
  915. 'Northwood Numeric Grid Format .grd/.tab' : '',
  916. 'Northwood Classified Grid Format .grc/.tab' : '',
  917. 'ARC Digitized Raster Graphics' : 'arc',
  918. 'Magellan topo (.blx)' : 'blx',
  919. 'SAGA GIS Binary Grid (.sdat)' : 'sdat',
  920. # vector
  921. 'ESRI Shapefile' : 'shp',
  922. 'UK .NTF' : 'ntf',
  923. 'SDTS' : 'ddf',
  924. 'DGN' : 'dgn',
  925. 'VRT' : 'vrt',
  926. 'REC' : 'rec',
  927. 'BNA' : 'bna',
  928. 'CSV' : 'csv',
  929. 'GML' : 'gml',
  930. 'GPX' : 'gpx',
  931. 'KML' : 'kml',
  932. 'GMT' : 'gmt',
  933. 'PGeo' : 'mdb',
  934. 'XPlane' : 'dat',
  935. 'AVCBin' : 'adf',
  936. 'AVCE00' : 'e00',
  937. 'DXF' : 'dxf',
  938. 'Geoconcept' : 'gxt',
  939. 'GeoRSS' : 'xml',
  940. 'GPSTrackMaker' : 'gtm',
  941. 'VFK' : 'vfk'
  942. }
  943. try:
  944. return formatToExt[name]
  945. except KeyError:
  946. return ''
  947. class GdalSelect(wx.Panel):
  948. def __init__(self, parent, panel, ogr = False, link = False, dest = False,
  949. default = 'file', exclude = [], envHandler = None):
  950. """!Widget for selecting GDAL/OGR datasource, format
  951. @param parent parent window
  952. @param ogr use OGR selector instead of GDAL
  953. @param dest True for output (destination)
  954. @param default deafult type (ignored when dest == True)
  955. @param exclude list of types to be excluded
  956. """
  957. self.parent = parent
  958. self.ogr = ogr
  959. self.dest = dest
  960. wx.Panel.__init__(self, parent = panel, id = wx.ID_ANY)
  961. self.settingsBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  962. label = " %s " % _("Settings"))
  963. self.inputBox = wx.StaticBox(parent = self, id = wx.ID_ANY)
  964. if dest:
  965. self.inputBox.SetLabel(" %s " % _("Output settings"))
  966. else:
  967. self.inputBox.SetLabel(" %s " % _("Source settings"))
  968. # source type
  969. sources = list()
  970. self.sourceMap = { 'file' : -1,
  971. 'dir' : -1,
  972. 'db' : -1,
  973. 'db-pg' : -1,
  974. 'pro' : -1,
  975. 'native' : -1 }
  976. idx = 0
  977. if ogr and (link or dest):
  978. extraLabel = " (OGR)"
  979. else:
  980. extraLabel = ""
  981. if dest:
  982. sources.append(_("Native"))
  983. self.sourceMap['native'] = idx
  984. idx += 1
  985. if 'file' not in exclude:
  986. sources.append(_("File") + extraLabel)
  987. self.sourceMap['file'] = idx
  988. idx += 1
  989. if 'directory' not in exclude:
  990. sources.append(_("Directory") + extraLabel)
  991. self.sourceMap['dir'] = idx
  992. idx += 1
  993. if 'database' not in exclude:
  994. sources.append(_("Database") + extraLabel)
  995. self.sourceMap['db'] = idx
  996. idx += 1
  997. if 'protocol' not in exclude:
  998. sources.append(_("Protocol") + extraLabel)
  999. self.sourceMap['pro'] = idx
  1000. idx += 1
  1001. if 'database' not in exclude and ogr and (link or dest):
  1002. sources.append(_("PostGIS (PG)"))
  1003. self.sourceMap['db-pg'] = idx
  1004. if self.ogr:
  1005. self.settingsFile = os.path.join(GetSettingsPath(), 'wxOGR')
  1006. else:
  1007. self.settingsFile = os.path.join(GetSettingsPath(), 'wxGDAL')
  1008. self.settingsChoice = wx.Choice(parent = self, id = wx.ID_ANY)
  1009. self.settingsChoice.Bind(wx.EVT_CHOICE, self.OnSettingsLoad)
  1010. self._settings = self._loadSettings() # -> self.settingsChoice.SetItems()
  1011. self.btnSettingsSave = wx.Button(parent = self, id = wx.ID_SAVE)
  1012. self.btnSettingsSave.Bind(wx.EVT_BUTTON, self.OnSettingsSave)
  1013. self.btnSettingsSave.SetToolTipString(_("Save current settings"))
  1014. self.btnSettingsDel = wx.Button(parent = self, id = wx.ID_REMOVE)
  1015. self.btnSettingsDel.Bind(wx.EVT_BUTTON, self.OnSettingsDelete)
  1016. self.btnSettingsSave.SetToolTipString(_("Delete currently selected settings"))
  1017. self.source = wx.RadioBox(parent = self, id = wx.ID_ANY,
  1018. style = wx.RA_SPECIFY_COLS,
  1019. choices = sources)
  1020. if dest:
  1021. self.source.SetLabel(" %s " % _('Output type'))
  1022. else:
  1023. self.source.SetLabel(" %s " % _('Source type'))
  1024. self.source.SetSelection(0)
  1025. self.source.Bind(wx.EVT_RADIOBOX, self.OnSetType)
  1026. # dsn widgets
  1027. if not ogr:
  1028. filemask = 'GeoTIFF (%s)|%s|%s (*.*)|*.*' % \
  1029. (self._getExtPattern('tif'), self._getExtPattern('tif'), _('All files'))
  1030. else:
  1031. filemask = 'ESRI Shapefile (%s)|%s|%s (*.*)|*.*' % \
  1032. (self._getExtPattern('shp'), self._getExtPattern('shp'), _('All files'))
  1033. dsnFile = filebrowse.FileBrowseButton(parent=self, id=wx.ID_ANY,
  1034. size=globalvar.DIALOG_GSELECT_SIZE, labelText = '',
  1035. dialogTitle=_('Choose file to import'),
  1036. buttonText=_('Browse'),
  1037. startDirectory=os.getcwd(),
  1038. changeCallback=self.OnSetDsn,
  1039. fileMask=filemask)
  1040. dsnFile.Hide()
  1041. dsnDir = filebrowse.DirBrowseButton(parent=self, id=wx.ID_ANY,
  1042. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1043. dialogTitle=_('Choose input directory'),
  1044. buttonText=_('Browse'),
  1045. startDirectory=os.getcwd(),
  1046. changeCallback=self.OnSetDsn)
  1047. dsnDir.SetName('GdalSelect')
  1048. dsnDir.Hide()
  1049. dsnDbFile = filebrowse.FileBrowseButton(parent=self, id=wx.ID_ANY,
  1050. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1051. dialogTitle=_('Choose file'),
  1052. buttonText=_('Browse'),
  1053. startDirectory=os.getcwd(),
  1054. changeCallback=self.OnSetDsn)
  1055. dsnDbFile.Hide()
  1056. dsnDbFile.SetName('GdalSelect')
  1057. dsnDbText = wx.TextCtrl(parent = self, id = wx.ID_ANY)
  1058. dsnDbText.Hide()
  1059. dsnDbText.Bind(wx.EVT_TEXT, self.OnSetDsn)
  1060. dsnDbText.SetName('GdalSelect')
  1061. dsnDbChoice = wx.Choice(parent = self, id = wx.ID_ANY)
  1062. dsnDbChoice.Hide()
  1063. dsnDbChoice.Bind(wx.EVT_CHOICE, self.OnSetDsn)
  1064. dsnDbChoice.SetName('GdalSelect')
  1065. dsnPro = wx.TextCtrl(parent = self, id = wx.ID_ANY)
  1066. dsnPro.Hide()
  1067. dsnPro.Bind(wx.EVT_TEXT, self.OnSetDsn)
  1068. dsnPro.SetName('GdalSelect')
  1069. # format
  1070. self.format = FormatSelect(parent = self,
  1071. ogr = ogr)
  1072. self.format.Bind(wx.EVT_CHOICE, self.OnSetFormat)
  1073. self.extension = wx.TextCtrl(parent = self, id = wx.ID_ANY)
  1074. self.extension.Bind(wx.EVT_TEXT, self.OnSetExtension)
  1075. self.extension.Hide()
  1076. if ogr:
  1077. fType = 'ogr'
  1078. else:
  1079. fType = 'gdal'
  1080. self.input = { 'file' : [_("File:"),
  1081. dsnFile,
  1082. GetFormats(writableOnly = dest)[fType]['file']],
  1083. 'dir' : [_("Name:"),
  1084. dsnDir,
  1085. GetFormats(writableOnly = dest)[fType]['file']],
  1086. 'db' : [_("Name:"),
  1087. dsnDbText,
  1088. GetFormats(writableOnly = dest)[fType]['database']],
  1089. 'pro' : [_("Protocol:"),
  1090. dsnPro,
  1091. GetFormats(writableOnly = dest)[fType]['protocol']],
  1092. 'db-win' : { 'file' : dsnDbFile,
  1093. 'text' : dsnDbText,
  1094. 'choice' : dsnDbChoice },
  1095. 'native' : [_("Name:"), dsnDir, ''],
  1096. 'db-pg' : [_("Name:"),
  1097. dsnDbChoice,
  1098. ["PostgreSQL"]],
  1099. }
  1100. if self.dest:
  1101. current = RunCommand('v.external.out',
  1102. parent = self,
  1103. read = True, parse = grass.parse_key_val,
  1104. flags = 'g')
  1105. if current['format'] == 'native':
  1106. self.dsnType = 'native'
  1107. elif current['format'] in GetFormats()['ogr']['database']:
  1108. self.dsnType = 'db'
  1109. else:
  1110. self.dsnType = 'dir'
  1111. else:
  1112. self.dsnType = default
  1113. self.dsnText = wx.StaticText(parent = self, id = wx.ID_ANY,
  1114. label = self.input[self.dsnType][0],
  1115. size = (75, -1))
  1116. self.extensionText = wx.StaticText(parent = self, id = wx.ID_ANY,
  1117. label = _("Extension:"))
  1118. self.extensionText.Hide()
  1119. self.creationOpt = wx.TextCtrl(parent = self, id = wx.ID_ANY)
  1120. if not self.dest:
  1121. self.creationOpt.Hide()
  1122. self._layout()
  1123. self.OnSetType(event = None, sel = self.sourceMap[self.dsnType])
  1124. if self.dest:
  1125. if current['format'] != 'native':
  1126. self.OnSetFormat(event = None, format = current['format'])
  1127. self.OnSetDsn(event = None, path = current['dsn'])
  1128. self.creationOpt.SetValue(current.get('options', ''))
  1129. else:
  1130. if not ogr:
  1131. self.OnSetFormat(event = None, format = 'GeoTIFF')
  1132. else:
  1133. self.OnSetFormat(event = None, format = 'ESRI Shapefile')
  1134. def _layout(self):
  1135. """!Layout"""
  1136. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1137. settingsSizer = wx.StaticBoxSizer(self.settingsBox, wx.HORIZONTAL)
  1138. settingsSizer.Add(item = wx.StaticText(parent = self,
  1139. id = wx.ID_ANY,
  1140. label = _("Load settings:")),
  1141. flag = wx.ALIGN_CENTER_VERTICAL | wx.RIGHT,
  1142. border = 5)
  1143. settingsSizer.Add(item = self.settingsChoice,
  1144. proportion = 1,
  1145. flag = wx.EXPAND)
  1146. settingsSizer.Add(item = self.btnSettingsSave,
  1147. flag = wx.LEFT | wx.RIGHT,
  1148. border = 5)
  1149. settingsSizer.Add(item = self.btnSettingsDel,
  1150. flag = wx.RIGHT,
  1151. border = 5)
  1152. inputSizer = wx.StaticBoxSizer(self.inputBox, wx.HORIZONTAL)
  1153. self.dsnSizer = wx.GridBagSizer(vgap = 3, hgap = 3)
  1154. #self.dsnSizer.AddGrowableRow(0)
  1155. self.dsnSizer.AddGrowableCol(3)
  1156. row = 0
  1157. self.dsnSizer.Add(item = wx.StaticText(parent = self, id = wx.ID_ANY,
  1158. label = _("Format:")),
  1159. flag = wx.ALIGN_CENTER_VERTICAL,
  1160. pos = (row, 0))
  1161. self.dsnSizer.Add(item=self.format,
  1162. flag = wx.ALIGN_CENTER_VERTICAL,
  1163. pos = (row, 1))
  1164. self.dsnSizer.Add(item = self.extensionText,
  1165. flag=wx.ALIGN_CENTER_VERTICAL,
  1166. pos = (row, 2))
  1167. self.dsnSizer.Add(item=self.extension,
  1168. flag = wx.ALIGN_CENTER_VERTICAL,
  1169. pos = (row, 3))
  1170. row += 1
  1171. self.dsnSizer.Add(item = self.dsnText,
  1172. flag = wx.ALIGN_CENTER_VERTICAL,
  1173. pos = (row, 0))
  1174. self.dsnSizer.Add(item = self.input[self.dsnType][1],
  1175. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1176. pos = (row, 1), span = (1, 3))
  1177. row += 1
  1178. if self.creationOpt.IsShown():
  1179. self.dsnSizer.Add(item = wx.StaticText(parent = self, id = wx.ID_ANY,
  1180. label = _("Creation options:")),
  1181. flag = wx.ALIGN_CENTER_VERTICAL,
  1182. pos = (row, 0))
  1183. self.dsnSizer.Add(item = self.creationOpt,
  1184. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1185. pos = (row, 1), span = (1, 3))
  1186. row += 1
  1187. inputSizer.Add(item=self.dsnSizer, proportion = 1,
  1188. flag=wx.EXPAND | wx.BOTTOM, border = 10)
  1189. mainSizer.Add(item=settingsSizer, proportion=0,
  1190. flag=wx.ALL | wx.EXPAND, border=5)
  1191. mainSizer.Add(item=self.source, proportion=0,
  1192. flag=wx.LEFT | wx.RIGHT | wx.EXPAND, border=5)
  1193. mainSizer.Add(item=inputSizer, proportion=0,
  1194. flag=wx.ALL | wx.EXPAND, border=5)
  1195. self.SetSizer(mainSizer)
  1196. mainSizer.Fit(self)
  1197. def _getExtPatternGlob(self, ext):
  1198. """!Get pattern for case-insensitive globing"""
  1199. pattern = '*.'
  1200. for c in ext:
  1201. pattern += '[%s%s]' % (c.lower(), c.upper())
  1202. return pattern
  1203. def _getExtPattern(self, ext):
  1204. """!Get pattern for case-insensitive file mask"""
  1205. return '*.%s;*.%s' % (ext.lower(), ext.upper())
  1206. def OnSettingsLoad(self, event):
  1207. """!Load named settings"""
  1208. name = event.GetString()
  1209. if name not in self._settings:
  1210. GError(parent = self,
  1211. message = _("Settings <%s> not found") % name)
  1212. return
  1213. data = self._settings[name]
  1214. self.OnSetType(event = None, sel = self.sourceMap[data[0]])
  1215. self.OnSetFormat(event = None, format = data[2])
  1216. self.OnSetDsn(event = None, path = data[1])
  1217. self.creationOpt.SetValue(data[3])
  1218. def OnSettingsSave(self, event):
  1219. """!Save settings"""
  1220. dlg = wx.TextEntryDialog(parent = self,
  1221. message = _("Name:"),
  1222. caption = _("Save settings"))
  1223. if dlg.ShowModal() != wx.ID_OK:
  1224. return
  1225. # check required params
  1226. if not dlg.GetValue():
  1227. GMessage(parent = self,
  1228. message = _("Name not given, settings is not saved."))
  1229. return
  1230. if not self.GetDsn():
  1231. GMessage(parent = self,
  1232. message = _("No data source defined, settings is not saved."))
  1233. return
  1234. name = dlg.GetValue()
  1235. # check if settings item already exists
  1236. if name in self._settings:
  1237. dlgOwt = wx.MessageDialog(self, message = _("Settings <%s> already exists. "
  1238. "Do you want to overwrite the settings?") % name,
  1239. caption = _("Save settings"), style = wx.YES_NO | wx.YES_DEFAULT | wx.ICON_QUESTION)
  1240. if dlgOwt.ShowModal() != wx.ID_YES:
  1241. dlgOwt.Destroy()
  1242. return
  1243. self._settings[name] = (self.dsnType, self.GetDsn(),
  1244. self.format.GetStringSelection(),
  1245. self.creationOpt.GetValue())
  1246. if self._saveSettings() == 0:
  1247. self._settings = self._loadSettings()
  1248. self.settingsChoice.SetStringSelection(name)
  1249. dlg.Destroy()
  1250. def OnSettingsDelete(self, event):
  1251. """!Save settings"""
  1252. name = self.settingsChoice.GetStringSelection()
  1253. if not name:
  1254. GMessage(parent = self,
  1255. message = _("No settings is defined. Operation canceled."))
  1256. return
  1257. self._settings.pop(name)
  1258. if self._saveSettings() == 0:
  1259. self._settings = self._loadSettings()
  1260. def _saveSettings(self):
  1261. """!Save settings into the file
  1262. @return 0 on success
  1263. @return -1 on failure
  1264. """
  1265. try:
  1266. fd = open(self.settingsFile, 'w')
  1267. for key, value in self._settings.iteritems():
  1268. fd.write('%s;%s;%s;%s\n' % (key, value[0], value[1], value[2]))
  1269. except IOError:
  1270. GError(parent = self,
  1271. message = _("Unable to save settings"))
  1272. return -1
  1273. else:
  1274. fd.close()
  1275. return 0
  1276. def _loadSettings(self):
  1277. """!Load settings from the file
  1278. The file is defined by self.SettingsFile.
  1279. @return parsed dict
  1280. @return empty dict on error
  1281. """
  1282. data = dict()
  1283. if not os.path.exists(self.settingsFile):
  1284. return data
  1285. try:
  1286. fd = open(self.settingsFile, 'r')
  1287. for line in fd.readlines():
  1288. try:
  1289. lineData = line.rstrip('\n').split(';')
  1290. if len(lineData) > 4:
  1291. # type, dsn, format, options
  1292. data[lineData[0]] = (lineData[1], lineData[2], lineData[3], lineData[4])
  1293. else:
  1294. data[lineData[0]] = (lineData[1], lineData[2], lineData[3], '')
  1295. except ValueError:
  1296. pass
  1297. except IOError:
  1298. return data
  1299. else:
  1300. fd.close()
  1301. self.settingsChoice.SetItems(sorted(data.keys()))
  1302. return data
  1303. def OnSetType(self, event, sel = None):
  1304. """!Datasource type changed"""
  1305. if event:
  1306. sel = event.GetSelection()
  1307. else:
  1308. self.source.SetSelection(sel)
  1309. win = self.input[self.dsnType][1]
  1310. if win:
  1311. self.dsnSizer.Remove(win)
  1312. win.Hide()
  1313. if sel == self.sourceMap['file']: # file
  1314. self.dsnType = 'file'
  1315. format = self.input[self.dsnType][2][0]
  1316. try:
  1317. ext = self.format.GetExtension(format)
  1318. if not ext:
  1319. raise KeyError
  1320. format += ' (%s)|%s|%s (*.*)|*.*' % \
  1321. (self._getExtPattern(ext), self._getExtPattern(ext), _('All files'))
  1322. except KeyError:
  1323. format += '%s (*.*)|*.*' % _('All files')
  1324. win = filebrowse.FileBrowseButton(parent=self, id=wx.ID_ANY,
  1325. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1326. dialogTitle=_('Choose file to import'),
  1327. buttonText=_('Browse'),
  1328. startDirectory=os.getcwd(),
  1329. changeCallback=self.OnSetDsn,
  1330. fileMask = format)
  1331. self.input[self.dsnType][1] = win
  1332. elif sel == self.sourceMap['dir']: # directory
  1333. self.dsnType = 'dir'
  1334. elif sel == self.sourceMap['db']: # database
  1335. self.dsnType = 'db'
  1336. elif sel == self.sourceMap['pro']: # protocol
  1337. self.dsnType = 'pro'
  1338. elif sel == self.sourceMap['native']:
  1339. self.dsnType = 'native'
  1340. elif sel == self.sourceMap['db-pg']: # PostGIS database (PG data provider)
  1341. self.dsnType = 'db-pg'
  1342. if self.dsnType == 'db':
  1343. self.input[self.dsnType][1] = self.input['db-win']['text']
  1344. win = self.input[self.dsnType][1]
  1345. self.dsnSizer.Add(item = self.input[self.dsnType][1],
  1346. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1347. pos = (1, 1), span = (1, 3))
  1348. if self.dsnType != 'db-pg':
  1349. win.SetValue('')
  1350. win.Show()
  1351. if sel == self.sourceMap['native']: # native
  1352. win.Enable(False)
  1353. self.format.Enable(False)
  1354. self.creationOpt.Enable(False)
  1355. self.parent.btnOk.Enable(True)
  1356. else:
  1357. if not win.IsEnabled():
  1358. win.Enable(True)
  1359. if sel == self.sourceMap['db-pg']:
  1360. if self.format.IsEnabled():
  1361. self.format.Enable(False)
  1362. self.creationOpt.Enable(True)
  1363. else:
  1364. if not self.format.IsEnabled():
  1365. self.format.Enable(True)
  1366. self.creationOpt.Enable(True)
  1367. self.dsnText.SetLabel(self.input[self.dsnType][0])
  1368. self.format.SetItems(self.input[self.dsnType][2])
  1369. if self.parent.GetName() == 'MultiImportDialog':
  1370. self.parent.list.DeleteAllItems()
  1371. if sel in (self.sourceMap['file'], self.sourceMap['dir']):
  1372. if not self.ogr:
  1373. self.OnSetFormat(event = None, format = 'GeoTIFF')
  1374. else:
  1375. self.OnSetFormat(event = None, format = 'ESRI Shapefile')
  1376. elif sel == self.sourceMap['db-pg']:
  1377. self.OnSetFormat(event = None, format = 'PostgreSQL')
  1378. if sel == self.sourceMap['dir'] and not self.dest:
  1379. if not self.extension.IsShown():
  1380. self.extensionText.Show()
  1381. self.extension.Show()
  1382. else:
  1383. if self.extension.IsShown():
  1384. self.extensionText.Hide()
  1385. self.extension.Hide()
  1386. self.dsnSizer.Layout()
  1387. def GetDsn(self):
  1388. """!Get datasource name
  1389. """
  1390. if self.format.GetStringSelection() == 'PostgreSQL':
  1391. dsn = 'PG:dbname=%s' % self.input[self.dsnType][1].GetStringSelection()
  1392. else:
  1393. dsn = self.input[self.dsnType][1].GetValue()
  1394. return dsn
  1395. def OnSetDsn(self, event, path = None):
  1396. """!Input DXF file/OGR dsn defined, update list of layer
  1397. widget"""
  1398. if event:
  1399. path = event.GetString()
  1400. else:
  1401. if self.format.GetStringSelection() == 'PostgreSQL':
  1402. for item in path.split(':', 1)[1].split(','):
  1403. key, value = item.split('=', 1)
  1404. if key == 'dbname':
  1405. if not self.input[self.dsnType][1].SetStringSelection(value):
  1406. GMessage(_("Database <%s> not accessible.") % value,
  1407. parent = self)
  1408. break
  1409. else:
  1410. self.input[self.dsnType][1].SetValue(path)
  1411. if not path:
  1412. if self.dest:
  1413. self.parent.btnOk.Enable(False)
  1414. return
  1415. if self.dest:
  1416. self.parent.btnOk.Enable(True)
  1417. else:
  1418. self._reloadLayers()
  1419. if event:
  1420. event.Skip()
  1421. def _reloadLayers(self):
  1422. """!Reload list of layers"""
  1423. dsn = self.GetDsn()
  1424. if not dsn:
  1425. return
  1426. data = list()
  1427. layerId = 1
  1428. if self.ogr:
  1429. ret = RunCommand('v.external',
  1430. quiet = True,
  1431. read = True,
  1432. flags = 't',
  1433. dsn = dsn)
  1434. if not ret:
  1435. self.parent.list.LoadData()
  1436. if hasattr(self, "btn_run"):
  1437. self.btn_run.Enable(False)
  1438. return
  1439. layerId = 1
  1440. for line in ret.splitlines():
  1441. layerName, featureType = map(lambda x: x.strip(), line.split(' ', 1))
  1442. grassName = GetValidLayerName(layerName)
  1443. data.append((layerId, layerName, featureType, grassName))
  1444. layerId += 1
  1445. else:
  1446. if self.dsnType == 'file':
  1447. baseName = os.path.basename(dsn)
  1448. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1449. data.append((layerId, baseName, grassName))
  1450. elif self.dsnType == 'dir':
  1451. ext = self.extension.GetValue()
  1452. for filename in glob.glob(os.path.join(dsn, "%s") % self._getExtPatternGlob(ext)):
  1453. baseName = os.path.basename(filename)
  1454. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1455. data.append((layerId, baseName, grassName))
  1456. layerId += 1
  1457. if self.ogr:
  1458. dsn += '@OGR'
  1459. evt = wxGdalSelect(dsn = dsn)
  1460. evt.SetId(self.input[self.dsnType][1].GetId())
  1461. wx.PostEvent(self.parent, evt)
  1462. if self.parent.GetName() == 'MultiImportDialog':
  1463. self.parent.list.LoadData(data)
  1464. if len(data) > 0:
  1465. self.parent.btn_run.Enable(True)
  1466. else:
  1467. self.parent.btn_run.Enable(False)
  1468. def OnSetExtension(self, event):
  1469. """!Extension changed"""
  1470. if not self.dest:
  1471. # reload layers
  1472. self._reloadLayers()
  1473. def OnSetFormat(self, event, format = None):
  1474. """!Format changed"""
  1475. if self.dsnType not in ['file', 'dir', 'db', 'db-pg']:
  1476. return
  1477. win = self.input[self.dsnType][1]
  1478. self.dsnSizer.Remove(win)
  1479. if self.dsnType == 'file':
  1480. win.Destroy()
  1481. else: # database
  1482. win.Hide()
  1483. if event:
  1484. format = event.GetString()
  1485. else:
  1486. self.format.SetStringSelection(format)
  1487. if self.dsnType == 'file':
  1488. try:
  1489. ext = self.format.GetExtension(format)
  1490. if not ext:
  1491. raise KeyError
  1492. format += ' (%s)|%s|%s (*.*)|*.*' % \
  1493. (self._getExtPattern(ext), self._getExtPattern(ext), _('All files'))
  1494. except KeyError:
  1495. format += '%s (*.*)|*.*' % _('All files')
  1496. win = filebrowse.FileBrowseButton(parent=self, id=wx.ID_ANY,
  1497. size=globalvar.DIALOG_GSELECT_SIZE, labelText='',
  1498. dialogTitle=_('Choose file'),
  1499. buttonText=_('Browse'),
  1500. startDirectory=os.getcwd(),
  1501. changeCallback=self.OnSetDsn,
  1502. fileMask = format)
  1503. elif self.dsnType == 'dir':
  1504. pass
  1505. else: # database
  1506. if format == 'SQLite' or format == 'Rasterlite':
  1507. win = self.input['db-win']['file']
  1508. elif format == 'PostgreSQL' or format == 'PostGIS WKT Raster driver':
  1509. if grass.find_program('psql', ['--help']):
  1510. win = self.input['db-win']['choice']
  1511. if not win.GetItems():
  1512. p = grass.Popen(['psql', '-ltA'], stdout = grass.PIPE)
  1513. ret = p.communicate()[0]
  1514. if ret:
  1515. db = list()
  1516. for line in ret.splitlines():
  1517. sline = line.split('|')
  1518. if len(sline) < 2:
  1519. continue
  1520. dbname = sline[0]
  1521. if dbname:
  1522. db.append(dbname)
  1523. win.SetItems(db)
  1524. if self.dest and win.GetStringSelection():
  1525. self.parent.btnOk.Enable(True)
  1526. else:
  1527. win = self.input['db-win']['text']
  1528. else:
  1529. win = self.input['db-win']['text']
  1530. self.input[self.dsnType][1] = win
  1531. if not win.IsShown():
  1532. win.Show()
  1533. self.dsnSizer.Add(item = self.input[self.dsnType][1],
  1534. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1535. pos = (1, 1), span = (1, 3))
  1536. self.dsnSizer.Layout()
  1537. # update extension
  1538. self.extension.SetValue(self.GetFormatExt())
  1539. if not self.dest:
  1540. # reload layers
  1541. self._reloadLayers()
  1542. def GetType(self):
  1543. """!Get source type"""
  1544. return self.dsnType
  1545. def GetDsnWin(self):
  1546. """!Get list of DSN windows"""
  1547. win = list()
  1548. for stype in ('file', 'dir', 'pro'):
  1549. win.append(self.input[stype][1])
  1550. for stype in ('file', 'text', 'choice'):
  1551. win.append(self.input['db-win'][stype])
  1552. return win
  1553. def GetFormat(self):
  1554. """!Get format as string"""
  1555. return self.format.GetStringSelection().replace(' ', '_')
  1556. def GetFormatExt(self):
  1557. """!Get format extension"""
  1558. return self.format.GetExtension(self.format.GetStringSelection())
  1559. def GetOptions(self):
  1560. """!Get creation options"""
  1561. if not self.creationOpt.IsShown():
  1562. return ''
  1563. return self.creationOpt.GetValue()
  1564. class ProjSelect(wx.ComboBox):
  1565. """!Widget for selecting input raster/vector map used by
  1566. r.proj/v.proj modules."""
  1567. def __init__(self, parent, isRaster, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  1568. **kwargs):
  1569. super(ProjSelect, self).__init__(parent, id, size = size,
  1570. style = wx.CB_READONLY, **kwargs)
  1571. self.SetName("ProjSelect")
  1572. self.isRaster = isRaster
  1573. def UpdateItems(self, dbase, location, mapset):
  1574. """!Update list of maps
  1575. """
  1576. if not dbase:
  1577. dbase = grass.gisenv()['GISDBASE']
  1578. if not mapset:
  1579. mapset = grass.gisenv()['MAPSET']
  1580. if self.isRaster:
  1581. ret = RunCommand('r.proj',
  1582. quiet = True,
  1583. read = True,
  1584. flags = 'l',
  1585. dbase = dbase,
  1586. location = location,
  1587. mapset = mapset)
  1588. else:
  1589. ret = RunCommand('v.proj',
  1590. quiet = True,
  1591. read = True,
  1592. flags = 'l',
  1593. dbase = dbase,
  1594. location = location,
  1595. mapset = mapset)
  1596. listMaps = list()
  1597. if ret:
  1598. for line in ret.splitlines():
  1599. listMaps.append(line.strip())
  1600. ListSortLower(listMaps)
  1601. self.SetItems(listMaps)
  1602. self.SetValue('')
  1603. class ElementSelect(wx.Choice):
  1604. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  1605. **kwargs):
  1606. """!Widget for selecting GIS element
  1607. @param parent parent window
  1608. """
  1609. super(ElementSelect, self).__init__(parent, id, size = size,
  1610. **kwargs)
  1611. self.SetName("ElementSelect")
  1612. task = gtask.parse_interface('g.list')
  1613. p = task.get_param(value = 'type')
  1614. self.values = p.get('values', [])
  1615. self.valuesDesc = p.get('values_desc', [])
  1616. self.SetItems(self.valuesDesc)
  1617. def GetValue(self, name):
  1618. """!Translate value
  1619. @param name element name
  1620. """
  1621. idx = self.valuesDesc.index(name)
  1622. if idx > -1:
  1623. return self.values[idx]
  1624. return ''
  1625. class OgrTypeSelect(wx.Panel):
  1626. def __init__(self, parent, panel, **kwargs):
  1627. """!Widget to choose OGR feature type
  1628. @param parent parent window
  1629. @param panel wx.Panel instance used as parent window
  1630. """
  1631. wx.Panel.__init__(self, parent = panel, id = wx.ID_ANY)
  1632. self.ftype = wx.Choice(parent = self, id = wx.ID_ANY,
  1633. size = (200, -1),
  1634. choices = (_("Point"), _("LineString"), _("Polygon")))
  1635. self._layout()
  1636. def _layout(self):
  1637. """!Do layout"""
  1638. sizer = wx.BoxSizer(wx.HORIZONTAL)
  1639. sizer.Add(item = wx.StaticText(parent = self,
  1640. id = wx.ID_ANY,
  1641. label = _("Feature type:")),
  1642. proportion = 1,
  1643. flag = wx.ALIGN_CENTER_VERTICAL,
  1644. border = 5)
  1645. sizer.Add(item = self.ftype,
  1646. proportion = 0,
  1647. flag = wx.EXPAND | wx.ALIGN_RIGHT)
  1648. self.SetSizer(sizer)
  1649. sizer.Fit(self)
  1650. def GetType(self):
  1651. """!Get selected type as string
  1652. @return feature type as string
  1653. """
  1654. sel = self.ftype.GetSelection()
  1655. if sel == 0:
  1656. return 'point'
  1657. elif sel == 1:
  1658. return 'line'
  1659. elif sel == 2:
  1660. return 'boundary'
  1661. class CoordinatesSelect(wx.Panel):
  1662. def __init__(self, parent, lmgr = None, multiple = False, **kwargs):
  1663. """!Widget to get coordinates from map window by mouse click
  1664. @param parent parent window
  1665. @param lmgr layer manager
  1666. @param multiple - True if it is possible to insert more coordinates
  1667. """
  1668. self.lmgr = lmgr
  1669. self.multiple = multiple
  1670. self.mapWin = None
  1671. super(CoordinatesSelect, self).__init__(parent = parent, id = wx.ID_ANY)
  1672. self.coordsField = wx.TextCtrl(parent = self, id = wx.ID_ANY,
  1673. size = globalvar.DIALOG_TEXTCTRL_SIZE)
  1674. icon = wx.Bitmap(os.path.join(globalvar.ETCICONDIR, "grass", "pointer.png"))
  1675. self.buttonInsCoords = buttons.ThemedGenBitmapToggleButton(parent = self, id = wx.ID_ANY,
  1676. bitmap = icon,
  1677. size = globalvar.DIALOG_COLOR_SIZE)
  1678. self.registered = False
  1679. self.buttonInsCoords.Bind(wx.EVT_BUTTON, self._onClick)
  1680. self._doLayout()
  1681. def _doLayout(self):
  1682. self.dialogSizer = wx.BoxSizer(wx.HORIZONTAL)
  1683. self.dialogSizer.Add(item = self.coordsField,
  1684. proportion = 1,
  1685. flag = wx.EXPAND)
  1686. self.dialogSizer.Add(item = self.buttonInsCoords)
  1687. self.SetSizer(self.dialogSizer)
  1688. def _onClick(self, event):
  1689. """!Button for interacitve inserting of coordinates clicked"""
  1690. if self.buttonInsCoords.GetToggle() and self.lmgr.GetLayerTree():
  1691. self.mapWin = self.lmgr.GetLayerTree().GetMapDisplay().GetWindow()
  1692. if self.mapWin.RegisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1693. self._onMapClickHandler,
  1694. wx.StockCursor(wx.CURSOR_CROSS)) == False:
  1695. self.buttonInsCoords.SetToggle(False)
  1696. return
  1697. self.registered = True
  1698. self.lmgr.GetLayerTree().GetMapDisplay().Raise()
  1699. else:
  1700. if self.mapWin and \
  1701. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1702. self._onMapClickHandler):
  1703. self.registered = False
  1704. return
  1705. self.buttonInsCoords.SetToggle(False)
  1706. def _onMapClickHandler(self, event):
  1707. """!Gets coordinates from mapwindow"""
  1708. if event == "unregistered":
  1709. if self.buttonInsCoords:
  1710. self.buttonInsCoords.SetToggle(False)
  1711. return
  1712. e, n = self.mapWin.GetLastEN()
  1713. prevCoords = ""
  1714. if self.multiple:
  1715. prevCoords = self.coordsField.GetValue().strip()
  1716. if prevCoords != "":
  1717. prevCoords += ","
  1718. value = prevCoords + str(e) + "," + str(n)
  1719. self.coordsField.SetValue(value)
  1720. def __del__(self):
  1721. """!Unregistrates _onMapClickHandler from mapWin"""
  1722. if self.mapWin and self.registered:
  1723. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1724. self._onMapClickHandler)
  1725. def GetTextWin(self):
  1726. """!Get TextCtrl widget"""
  1727. return self.coordsField