gselect.py 71 KB

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