gselect.py 62 KB

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