gselect.py 76 KB

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