gselect.py 76 KB

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