gselect.py 60 KB

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