gselect.py 85 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220
  1. """!
  2. @package gui_core.gselect
  3. @brief Custom control that selects elements
  4. Classes:
  5. - gselect::Select
  6. - gselect::VectorSelect
  7. - gselect::ListCtrlComboPopup
  8. - gselect::TreeCrtlComboPopup
  9. - gselect::VectorDBInfo
  10. - gselect::LayerSelect
  11. - gselect::DriverSelect
  12. - gselect::DatabaseSelect
  13. - gselect::TableSelect
  14. - gselect::ColumnSelect
  15. - gselect::DbaseSelect
  16. - gselect::LocationSelect
  17. - gselect::MapsetSelect
  18. - gselect::SubGroupSelect
  19. - gselect::FormatSelect
  20. - gselect::GdalSelect
  21. - gselect::ProjSelect
  22. - gselect::ElementSelect
  23. - gselect::OgrTypeSelect
  24. - gselect::CoordinatesSelect
  25. - gselect::SignatureSelect
  26. (C) 2007-2014 by the GRASS Development Team
  27. This program is free software under the GNU General Public License
  28. (>=v2). Read the file COPYING that comes with GRASS for details.
  29. @author Michael Barton
  30. @author Martin Landa <landa.martin gmail.com>
  31. @author Vaclav Petras <wenzeslaus gmail.com> (menu customization)
  32. @author Stepan Turek <stepan.turek seznam.cz> (CoordinatesSelect, ListCtrlComboPopup)
  33. """
  34. import os
  35. import sys
  36. import glob
  37. import copy
  38. import wx
  39. import wx.combo
  40. import wx.lib.buttons as buttons
  41. import wx.lib.filebrowsebutton as filebrowse
  42. from core import globalvar
  43. import grass.script as grass
  44. from grass.script import task as gtask
  45. from grass.pygrass import messages
  46. import grass.temporal as tgis
  47. from gui_core.widgets import ManageSettingsWidget
  48. from core.gcmd import RunCommand, GError, GMessage
  49. from core.utils import GetListOfLocations, GetListOfMapsets, \
  50. GetFormats, rasterFormatExtension, vectorFormatExtension
  51. from core.utils import GetSettingsPath, GetValidLayerName, ListSortLower
  52. from core.utils import GetVectorNumberOfLayers, _
  53. from core.settings import UserSettings
  54. from core.debug import Debug
  55. from grass.pydispatch.signal import Signal
  56. class Select(wx.combo.ComboCtrl):
  57. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
  58. type = None, multiple = False, nmaps = 1,
  59. mapsets = None, updateOnPopup = True, onPopup = None,
  60. fullyQualified = True, extraItems = {},
  61. validator = wx.DefaultValidator):
  62. """!Custom control to create a ComboBox with a tree control to
  63. display and select GIS elements within acessible mapsets.
  64. Elements can be selected with mouse. Can allow multiple
  65. selections, when argument <em>multiple</em> is True. Multiple
  66. selections are separated by commas.
  67. @param type type of GIS elements ('raster, 'vector', ...)
  68. @param multiple True for multiple input
  69. @param nmaps number of maps to be entered
  70. @param mapsets force list of mapsets (otherwise search path)
  71. @param updateOnPopup True for updating list of elements on popup
  72. @param onPopup function to be called on Popup
  73. @param fullyQualified True to provide fully qualified names (map@mapset)
  74. @param extraItems extra items to add (given as dictionary) - see gmodeler for usage
  75. @param validator validator for TextCtrl
  76. """
  77. wx.combo.ComboCtrl.__init__(self, parent=parent, id=id, size=size, validator=validator)
  78. self.GetChildren()[0].SetName("Select")
  79. self.GetChildren()[0].type = type
  80. self.tcp = TreeCtrlComboPopup()
  81. self.SetPopupControl(self.tcp)
  82. self.SetPopupExtents(0, 100)
  83. if type:
  84. self.tcp.SetData(type = type, mapsets = mapsets,
  85. multiple = multiple, nmaps = nmaps,
  86. updateOnPopup = updateOnPopup, onPopup = onPopup,
  87. fullyQualified = fullyQualified, extraItems = extraItems)
  88. self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
  89. def OnKeyDown(self, event):
  90. """!Open popup and send key events to the tree."""
  91. if self.IsPopupShown():
  92. self.tcp.OnKeyDown(event)
  93. else:
  94. if event.GetKeyCode() == wx.WXK_DOWN:
  95. self.ShowPopup()
  96. event.Skip()
  97. def SetElementList(self, type, mapsets = None):
  98. """!Set element list
  99. @param type GIS element type
  100. @param mapsets list of acceptable mapsets (None for all in search path)
  101. """
  102. self.tcp.SetData(type = type, mapsets = mapsets)
  103. def GetElementList(self):
  104. """!Load elements"""
  105. self.tcp.GetElementList()
  106. def SetType(self, etype, multiple = False, nmaps = 1,
  107. mapsets = None, updateOnPopup = True, onPopup = None):
  108. """!Param set element type for widget
  109. @param etype element type, see gselect.ElementSelect
  110. """
  111. self.tcp.SetData(type = etype, mapsets = mapsets,
  112. multiple = multiple, nmaps = nmaps,
  113. updateOnPopup = updateOnPopup, onPopup = onPopup)
  114. class VectorSelect(Select):
  115. def __init__(self, parent, ftype, **kwargs):
  116. """!Custom to create a ComboBox with a tree control to display and
  117. select vector maps. You can filter the vector maps. If you
  118. don't need this feature use Select class instead
  119. @param ftype filter vector maps based on feature type
  120. """
  121. Select.__init__(self, parent = parent, id = wx.ID_ANY,
  122. type = 'vector', **kwargs)
  123. self.ftype = ftype
  124. # remove vector maps which do not contain given feature type
  125. self.tcp.SetFilter(self._isElement)
  126. def _isElement(self, vectorName):
  127. """!Check if element should be filtered out"""
  128. try:
  129. if int(grass.vector_info_topo(vectorName)[self.ftype]) < 1:
  130. return False
  131. except KeyError:
  132. return False
  133. return True
  134. class ListCtrlComboPopup(wx.combo.ComboPopup):
  135. """!Create a list ComboBox using TreeCtrl with hidden root.
  136. @todo: use event.EventObject instead of hardcoding (see forms.py)
  137. https://groups.google.com/forum/#!topic/wxpython-users/pRz6bi0k0XY
  138. """
  139. # overridden ComboPopup methods
  140. def Init(self):
  141. self.value = [] # for multiple is False -> len(self.value) in [0,1]
  142. self.curitem = None
  143. self.multiple = False
  144. self.updateOnPopup = True
  145. def Create(self, parent):
  146. self.seltree = wx.TreeCtrl(parent, style=wx.TR_HIDE_ROOT
  147. |wx.TR_HAS_BUTTONS
  148. |wx.TR_SINGLE
  149. |wx.TR_LINES_AT_ROOT
  150. |wx.SIMPLE_BORDER
  151. |wx.TR_FULL_ROW_HIGHLIGHT)
  152. self.seltree.Bind(wx.EVT_MOTION, self.OnMotion)
  153. self.seltree.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  154. # the following dummy handler are needed to keep tree events
  155. # from propagating up to the parent GIS Manager layer tree
  156. self.seltree.Bind(wx.EVT_TREE_ITEM_EXPANDING, lambda x: None)
  157. self.seltree.Bind(wx.EVT_TREE_ITEM_COLLAPSED, lambda x: None)
  158. self.seltree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, lambda x: None)
  159. self.seltree.Bind(wx.EVT_TREE_SEL_CHANGED, lambda x: None)
  160. self.seltree.Bind(wx.EVT_TREE_DELETE_ITEM, lambda x: None)
  161. self.seltree.Bind(wx.EVT_TREE_BEGIN_DRAG, lambda x: None)
  162. self.seltree.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK, lambda x: None)
  163. def GetControl(self):
  164. return self.seltree
  165. def GetStringValue(self):
  166. """!Get value as a string separated by commas
  167. """
  168. return ','.join(self.value)
  169. def SetStringValue(self, value):
  170. # this assumes that item strings are unique...
  171. root = self.seltree.GetRootItem()
  172. if not root:
  173. return
  174. winValue = self.GetCombo().GetValue().strip(',')
  175. self.value = []
  176. if winValue:
  177. self.value = winValue.split(',')
  178. def OnPopup(self, force = False):
  179. """!Limited only for first selected
  180. """
  181. if not force and not self.updateOnPopup:
  182. return
  183. # selects map starting according to written text
  184. inputText = self.GetCombo().GetValue().strip()
  185. if inputText:
  186. root = self.seltree.GetRootItem()
  187. match = self.FindItem(root, inputText, startLetters = True)
  188. self.seltree.EnsureVisible(match)
  189. self.seltree.SelectItem(match)
  190. def GetAdjustedSize(self, minWidth, prefHeight, maxHeight):
  191. """!Reads UserSettings to get height (which was 200 in old implementation).
  192. """
  193. height = UserSettings.Get(group = 'appearance', key = 'gSelectPopupHeight', subkey = 'value')
  194. return wx.Size(minWidth, min(height, maxHeight))
  195. def FindItem(self, parentItem, text, startLetters = False):
  196. """!Finds item with given name or starting with given text
  197. """
  198. startletters = startLetters
  199. item, cookie = self.seltree.GetFirstChild(parentItem)
  200. while wx.TreeItemId.IsOk(item):
  201. if self.seltree.GetItemText(item) == text:
  202. return item
  203. if self.seltree.ItemHasChildren(item):
  204. item = self.FindItem(item, text, startLetters = startletters)
  205. if wx.TreeItemId.IsOk(item):
  206. return item
  207. elif startletters and self.seltree.GetItemText(item).startswith(text.split('@', 1)[0]):
  208. return item
  209. item, cookie = self.seltree.GetNextChild(parentItem, cookie)
  210. return wx.TreeItemId()
  211. def AddItem(self, value):
  212. root = self.seltree.GetRootItem()
  213. if not root:
  214. root = self.seltree.AddRoot("<hidden root>")
  215. self.seltree.AppendItem(root, text = value)
  216. def SetItems(self, items):
  217. root = self.seltree.GetRootItem()
  218. if not root:
  219. root = self.seltree.AddRoot("<hidden root>")
  220. for item in items:
  221. self.seltree.AppendItem(root, text = item)
  222. def OnKeyUp(self, event):
  223. """!Enable to select items using keyboard
  224. """
  225. item = self.seltree.GetSelection()
  226. if event.GetKeyCode() == wx.WXK_DOWN:
  227. self.seltree.SelectItem(self.seltree.GetNextVisible(item))
  228. elif event.GetKeyCode() == wx.WXK_UP:
  229. itemPrev = self.seltree.GetPrevSibling(item)
  230. self.seltree.SelectItem(itemPrev)
  231. elif event.GetKeyCode() == wx.WXK_ESCAPE:
  232. self.Dismiss()
  233. elif event.GetKeyCode() == wx.WXK_RETURN:
  234. self.seltree.SelectItem(item)
  235. self.curitem = item
  236. item_str = self.seltree.GetItemText(self.curitem)
  237. if self.multiple:
  238. if item_str not in self.value:
  239. self.value.append(item_str)
  240. else:
  241. self.value = [item_str]
  242. self.Dismiss()
  243. def OnMotion(self, evt):
  244. """!Have the selection follow the mouse, like in a real combobox
  245. """
  246. item, flags = self.seltree.HitTest(evt.GetPosition())
  247. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  248. self.seltree.SelectItem(item)
  249. self.curitem = item
  250. evt.Skip()
  251. def OnLeftDown(self, evt):
  252. """!Do the combobox selection
  253. """
  254. if self.curitem is None:
  255. return
  256. item_str = self.seltree.GetItemText(self.curitem)
  257. if self.multiple:
  258. if item_str not in self.value:
  259. self.value.append(item_str)
  260. else:
  261. self.value = [item_str]
  262. self.Dismiss()
  263. evt.Skip()
  264. def SetData(self, **kargs):
  265. """!Set object properties"""
  266. if 'multiple' in kargs:
  267. self.multiple = kargs['multiple']
  268. if 'onPopup' in kargs:
  269. self.onPopup = kargs['onPopup']
  270. def DeleteAllItems(self):
  271. """!Delete all items in popup"""
  272. self.seltree.DeleteAllItems()
  273. class TreeCtrlComboPopup(ListCtrlComboPopup):
  274. """!Create a tree ComboBox for selecting maps and other GIS elements
  275. in accessible mapsets within the current location
  276. """
  277. # overridden ComboPopup methods
  278. def Init(self):
  279. ListCtrlComboPopup.Init(self)
  280. self.nmaps = 1
  281. self.type = None
  282. self.mapsets = None
  283. self.onPopup = None
  284. self.fullyQualified = True
  285. self.extraItems = dict()
  286. self.SetFilter(None)
  287. self.tgis_error = False
  288. def SetFilter(self, filter):
  289. """!Set filter for GIS elements, see e.g. VectorSelect"""
  290. self.filterElements = filter
  291. def OnPopup(self, force = False):
  292. """!Limited only for first selected"""
  293. if not force and not self.updateOnPopup:
  294. return
  295. if self.onPopup:
  296. selected, exclude = self.onPopup(self.type)
  297. else:
  298. selected = None
  299. exclude = False
  300. self.GetElementList(selected, exclude)
  301. ListCtrlComboPopup.OnPopup(self, force)
  302. def GetElementList(self, elements = None, exclude = False):
  303. """!Get filtered list of GIS elements in accessible mapsets
  304. and display as tree with all relevant elements displayed
  305. beneath each mapset branch
  306. """
  307. # update list
  308. self.seltree.DeleteAllItems()
  309. self._getElementList(self.type, self.mapsets, elements, exclude)
  310. if len(self.value) > 0:
  311. root = self.seltree.GetRootItem()
  312. if not root:
  313. return
  314. item = self.FindItem(root, self.value[0])
  315. try:
  316. self.seltree.EnsureVisible(item)
  317. self.seltree.SelectItem(item)
  318. except:
  319. pass
  320. def _getElementList(self, element, mapsets = None, elements = None, exclude = False):
  321. """!Get list of GIS elements in accessible mapsets and display as tree
  322. with all relevant elements displayed beneath each mapset branch
  323. @param element GIS element
  324. @param mapsets list of acceptable mapsets (None for all mapsets in search path)
  325. @param elements list of forced GIS elements
  326. @param exclude True to exclude, False for forcing the list (elements)
  327. """
  328. # get current mapset
  329. curr_mapset = grass.gisenv()['MAPSET']
  330. # map element types to g.mlist types
  331. elementdict = {'cell':'rast',
  332. 'raster':'rast',
  333. 'rast':'rast',
  334. 'raster files':'rast',
  335. 'grid3':'rast3d',
  336. 'rast3d':'rast3d',
  337. '3d-raster':'rast3d',
  338. 'raster3D':'rast3d',
  339. 'raster3D files':'rast3d',
  340. 'vector':'vect',
  341. 'vect':'vect',
  342. 'binary vector files':'vect',
  343. 'dig':'oldvect',
  344. 'oldvect':'oldvect',
  345. 'old vector':'oldvect',
  346. 'dig_ascii':'asciivect',
  347. 'asciivect':'asciivect',
  348. 'asciivector':'asciivect',
  349. 'ascii vector files':'asciivect',
  350. 'icons':'icon',
  351. 'icon':'icon',
  352. 'paint icon files':'icon',
  353. 'paint/labels':'labels',
  354. 'labels':'labels',
  355. 'label':'labels',
  356. 'paint label files':'labels',
  357. 'site_lists':'sites',
  358. 'sites':'sites',
  359. 'site list':'sites',
  360. 'site list files':'sites',
  361. 'windows':'region',
  362. 'region':'region',
  363. 'region definition':'region',
  364. 'region definition files':'region',
  365. 'windows3d':'region3d',
  366. 'region3d':'region3d',
  367. 'region3D definition':'region3d',
  368. 'region3D definition files':'region3d',
  369. 'group':'group',
  370. 'imagery group':'group',
  371. 'imagery group files':'group',
  372. '3d.view':'3dview',
  373. '3dview':'3dview',
  374. '3D viewing parameters':'3dview',
  375. '3D view parameters':'3dview',
  376. 'stds':'stds',
  377. 'strds':'strds',
  378. 'str3ds':'str3ds',
  379. 'stvds':'stvds'}
  380. if element not in elementdict:
  381. self.AddItem(_('Not selectable element'), node = False)
  382. return
  383. if element in ('stds', 'strds', 'str3ds', 'stvds'):
  384. if self.tgis_error is False:
  385. filesdict = tgis.tlist_grouped(elementdict[element], element == 'stds')
  386. else:
  387. filesdict = None
  388. else:
  389. if globalvar.have_mlist:
  390. filesdict = grass.mlist_grouped(elementdict[element],
  391. check_search_path = False)
  392. else:
  393. filesdict = grass.list_grouped(elementdict[element],
  394. check_search_path = False)
  395. # add extra items first
  396. if self.extraItems:
  397. for group, items in self.extraItems.iteritems():
  398. node = self.AddItem(group, node = True)
  399. self.seltree.SetItemTextColour(node, wx.Colour(50, 50, 200))
  400. for item in items:
  401. self.AddItem(item, node = False, parent = node)
  402. self.seltree.ExpandAllChildren(node)
  403. # list of mapsets in current location
  404. if mapsets is None:
  405. mapsets = grass.mapsets(search_path = True)
  406. # current mapset first
  407. if curr_mapset in mapsets and mapsets[0] != curr_mapset:
  408. mapsets.remove(curr_mapset)
  409. mapsets.insert(0, curr_mapset)
  410. first_mapset = None
  411. for mapset in mapsets:
  412. mapset_node = self.AddItem(_('Mapset') + ': ' + mapset, node = True, mapset = mapset)
  413. node = mapset_node
  414. if not first_mapset:
  415. first_mapset = mapset_node
  416. self.seltree.SetItemTextColour(mapset_node, wx.Colour(50, 50, 200))
  417. if mapset not in filesdict:
  418. continue
  419. try:
  420. if type(filesdict[mapset]) == dict:
  421. for elementType in filesdict[mapset].keys():
  422. node = self.AddItem(_('Type: ') + elementType, mapset = mapset,
  423. node = True, parent = mapset_node)
  424. self.seltree.SetItemTextColour(node, wx.Colour(50, 50, 200))
  425. elem_list = filesdict[mapset][elementType]
  426. self._addItems(elist = elem_list, elements = elements,
  427. mapset = mapset, exclude = exclude, node = node)
  428. else:
  429. elem_list = filesdict[mapset]
  430. self._addItems(elist = elem_list, elements = elements,
  431. mapset = mapset, exclude = exclude, node = node)
  432. except StandardError, e:
  433. sys.stderr.write(_("GSelect: invalid item: %s") % e)
  434. continue
  435. if self.seltree.ItemHasChildren(mapset_node):
  436. sel = UserSettings.Get(group='appearance', key='elementListExpand',
  437. subkey='selection')
  438. collapse = True
  439. if sel == 0: # collapse all except PERMANENT and current
  440. if mapset in ('PERMANENT', curr_mapset):
  441. collapse = False
  442. elif sel == 1: # collapse all except PERMANENT
  443. if mapset == 'PERMANENT':
  444. collapse = False
  445. elif sel == 2: # collapse all except current
  446. if mapset == curr_mapset:
  447. collapse = False
  448. elif sel == 3: # collapse all
  449. pass
  450. elif sel == 4: # expand all
  451. collapse = False
  452. if collapse:
  453. self.seltree.CollapseAllChildren(mapset_node)
  454. else:
  455. self.seltree.ExpandAllChildren(mapset_node)
  456. if first_mapset:
  457. # select first mapset (MSW hack)
  458. self.seltree.SelectItem(first_mapset)
  459. # helpers
  460. def _addItems(self, elist, elements, mapset, exclude, node):
  461. """!Helper function for adding multiple items (maps, stds).
  462. @param elist list of map/stds names
  463. @param elements list of forced elements
  464. @param mapset mapset name
  465. @param exclude True to exclude, False for forcing the list
  466. @param node parent node
  467. """
  468. elist.sort()
  469. for elem in elist:
  470. if elem != '':
  471. fullqElem = elem + '@' + mapset
  472. if elements is not None:
  473. if (exclude and fullqElem in elements) or \
  474. (not exclude and fullqElem not in elements):
  475. continue
  476. if self.filterElements:
  477. if self.filterElements(fullqElem):
  478. self.AddItem(elem, mapset = mapset, node = False, parent = node)
  479. else:
  480. self.AddItem(elem, mapset = mapset, node = False, parent = node)
  481. def AddItem(self, value, mapset = None, node = True, parent = None):
  482. if not parent:
  483. root = self.seltree.GetRootItem()
  484. if not root:
  485. root = self.seltree.AddRoot("<hidden root>")
  486. parent = root
  487. data = {'node': node, 'mapset': mapset}
  488. item = self.seltree.AppendItem(parent, text = value, data = wx.TreeItemData(data))
  489. return item
  490. def OnKeyDown(self, event):
  491. """!Enables to select items using keyboard"""
  492. item = self.seltree.GetSelection()
  493. if event.GetKeyCode() == wx.WXK_DOWN:
  494. self.seltree.SelectItem(self.seltree.GetNextVisible(item))
  495. # problem with GetPrevVisible
  496. elif event.GetKeyCode() == wx.WXK_UP:
  497. if self.seltree.ItemHasChildren(item) and self.seltree.IsExpanded(self.seltree.GetPrevSibling(item)):
  498. itemPrev = self.seltree.GetLastChild(self.seltree.GetPrevSibling(item))
  499. else:
  500. itemPrev = self.seltree.GetPrevSibling(item)
  501. if not wx.TreeItemId.IsOk(itemPrev):
  502. itemPrev = self.seltree.GetItemParent(item)
  503. if item == self.seltree.GetFirstChild(self.seltree.GetRootItem())[0]:
  504. itemPrev = item
  505. self.seltree.SelectItem(itemPrev)
  506. # selects first item starting with the written text in next mapset
  507. elif event.GetKeyCode() == wx.WXK_TAB:
  508. selected = self.seltree.GetSelection()
  509. if self.seltree.ItemHasChildren(selected):
  510. parent = selected
  511. else:
  512. parent = self.seltree.GetItemParent(selected)
  513. nextSibling = self.seltree.GetNextSibling(parent)
  514. if wx.TreeItemId.IsOk(nextSibling):
  515. match = self.FindItem(nextSibling, self.GetCombo().GetValue().strip(), True)
  516. else:
  517. match = self.FindItem(self.seltree.GetFirstChild(self.seltree.GetItemParent(parent))[0],
  518. self.GetCombo().GetValue().strip(), True)
  519. self.seltree.SelectItem(match)
  520. elif event.GetKeyCode() == wx.WXK_RIGHT:
  521. if self.seltree.ItemHasChildren(item):
  522. self.seltree.Expand(item)
  523. elif event.GetKeyCode() == wx.WXK_LEFT:
  524. if self.seltree.ItemHasChildren(item):
  525. self.seltree.Collapse(item)
  526. elif event.GetKeyCode() == wx.WXK_ESCAPE:
  527. self.Dismiss()
  528. elif event.GetKeyCode() == wx.WXK_RETURN:
  529. if self.seltree.GetPyData(item)['node']:
  530. self.value = []
  531. else:
  532. fullName = self.seltree.GetItemText(item)
  533. if self.fullyQualified and self.seltree.GetPyData(item)['mapset']:
  534. fullName += '@' + self.seltree.GetPyData(item)['mapset']
  535. if self.multiple:
  536. self.value.append(fullName)
  537. else:
  538. if self.nmaps > 1: # see key_desc
  539. if len(self.value) >= self.nmaps:
  540. self.value = [fullName]
  541. else:
  542. self.value.append(fullName)
  543. else:
  544. self.value = [fullName]
  545. self.Dismiss()
  546. def OnLeftDown(self, evt):
  547. """!Do the combobox selection
  548. """
  549. item, flags = self.seltree.HitTest(evt.GetPosition())
  550. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  551. self.curitem = item
  552. if self.seltree.GetPyData(item)['node']:
  553. evt.Skip()
  554. return
  555. fullName = self.seltree.GetItemText(item)
  556. if self.fullyQualified and self.seltree.GetPyData(item)['mapset']:
  557. fullName += '@' + self.seltree.GetPyData(item)['mapset']
  558. if self.multiple:
  559. self.value.append(fullName)
  560. else:
  561. if self.nmaps > 1: # see key_desc
  562. if len(self.value) >= self.nmaps:
  563. self.value = [fullName]
  564. else:
  565. self.value.append(fullName)
  566. else:
  567. self.value = [fullName]
  568. self.Dismiss()
  569. evt.Skip()
  570. def SetData(self, **kargs):
  571. """!Set object properties"""
  572. ListCtrlComboPopup.SetData(self, **kargs)
  573. if 'type' in kargs:
  574. self.type = kargs['type']
  575. if self.type in ('stds', 'strds', 'str3ds', 'stvds'):
  576. # Initiate the temporal framework. Catch database error
  577. # and set the error flag for the stds listing.
  578. try:
  579. tgis.init(True)
  580. except messages.FatalError, e:
  581. sys.stderr.write("Temporal GIS error:\n%s" % e)
  582. self.tgis_error = True
  583. if 'mapsets' in kargs:
  584. self.mapsets = kargs['mapsets']
  585. if 'nmaps' in kargs:
  586. self.nmaps = kargs['nmaps']
  587. if 'updateOnPopup' in kargs:
  588. self.updateOnPopup = kargs['updateOnPopup']
  589. if 'fullyQualified' in kargs:
  590. self.fullyQualified = kargs['fullyQualified']
  591. if 'extraItems' in kargs:
  592. self.extraItems = kargs['extraItems']
  593. def GetType(self):
  594. """!Get element type
  595. """
  596. return self.type
  597. class VectorDBInfo:
  598. """!Class providing information about attribute tables
  599. linked to a vector map"""
  600. def __init__(self, map):
  601. self.map = map
  602. # dictionary of layer number and associated (driver, database, table)
  603. self.layers = {}
  604. # dictionary of table and associated columns (type, length, values, ids)
  605. self.tables = {}
  606. if not self._CheckDBConnection(): # -> self.layers
  607. return
  608. self._DescribeTables() # -> self.tables
  609. def _CheckDBConnection(self):
  610. """!Check DB connection"""
  611. nuldev = file(os.devnull, 'w+')
  612. self.layers = grass.vector_db(map = self.map, stderr = nuldev)
  613. nuldev.close()
  614. return bool(len(self.layers.keys()) > 0)
  615. def _DescribeTables(self):
  616. """!Describe linked tables"""
  617. for layer in self.layers.keys():
  618. # determine column names and types
  619. table = self.layers[layer]["table"]
  620. columns = {} # {name: {type, length, [values], [ids]}}
  621. i = 0
  622. Debug.msg(1, "gselect.VectorDBInfo._DescribeTables(): table=%s driver=%s database=%s" % \
  623. (self.layers[layer]["table"], self.layers[layer]["driver"],
  624. self.layers[layer]["database"]))
  625. for item in grass.db_describe(table = self.layers[layer]["table"],
  626. driver = self.layers[layer]["driver"],
  627. database = self.layers[layer]["database"])['cols']:
  628. name, type, length = item
  629. # FIXME: support more datatypes
  630. if type.lower() == "integer":
  631. ctype = int
  632. elif type.lower() == "double precision":
  633. ctype = float
  634. else:
  635. ctype = str
  636. columns[name.strip()] = { 'index' : i,
  637. 'type' : type.lower(),
  638. 'ctype' : ctype,
  639. 'length' : int(length),
  640. 'values' : [],
  641. 'ids' : []}
  642. i += 1
  643. # check for key column
  644. # v.db.connect -g/p returns always key column name lowercase
  645. if self.layers[layer]["key"] not in columns.keys():
  646. for col in columns.keys():
  647. if col.lower() == self.layers[layer]["key"]:
  648. self.layers[layer]["key"] = col.upper()
  649. break
  650. self.tables[table] = columns
  651. return True
  652. def Reset(self):
  653. """!Reset"""
  654. for layer in self.layers:
  655. table = self.layers[layer]["table"] # get table desc
  656. for name in self.tables[table].keys():
  657. self.tables[table][name]['values'] = []
  658. self.tables[table][name]['ids'] = []
  659. def GetName(self):
  660. """!Get vector name"""
  661. return self.map
  662. def GetKeyColumn(self, layer):
  663. """!Get key column of given layer
  664. @param layer vector layer number
  665. """
  666. return str(self.layers[layer]['key'])
  667. def GetTable(self, layer):
  668. """!Get table name of given layer
  669. @param layer vector layer number
  670. """
  671. return self.layers[layer]['table']
  672. def GetDbSettings(self, layer):
  673. """!Get database settins
  674. @param layer layer number
  675. @return (driver, database)
  676. """
  677. return self.layers[layer]['driver'], self.layers[layer]['database']
  678. def GetTableDesc(self, table):
  679. """!Get table columns
  680. @param table table name
  681. """
  682. return self.tables[table]
  683. class LayerSelect(wx.ComboBox):
  684. def __init__(self, parent, id = wx.ID_ANY,
  685. size = globalvar.DIALOG_COMBOBOX_SIZE,
  686. vector = None, dsn = None, choices = [], all = False, default = None):
  687. """!Creates combo box for selecting vector map layer names
  688. @param vector vector map name (native or connected via v.external)
  689. @param dsn OGR data source name
  690. """
  691. super(LayerSelect, self).__init__(parent, id, size = size, choices = choices)
  692. self.all = all
  693. self.SetName("LayerSelect")
  694. # default value
  695. self.default = default
  696. self.InsertLayers(vector = vector, dsn = dsn)
  697. def InsertLayers(self, vector = None, dsn = None):
  698. """!Insert layers for a vector into the layer combobox
  699. @param vector vector map name (native or connected via v.external)
  700. @param dsn OGR data source name
  701. """
  702. layers = list()
  703. if vector:
  704. layers = GetVectorNumberOfLayers(vector)
  705. elif dsn:
  706. ret = RunCommand('v.in.ogr',
  707. read = True,
  708. quiet = True,
  709. flags = 'l',
  710. dsn = dsn)
  711. if ret:
  712. layers = ret.splitlines()
  713. if self.default:
  714. if len(layers) == 0:
  715. layers.insert(0, str(self.default))
  716. elif self.default not in layers:
  717. layers.append(self.default)
  718. if self.all:
  719. layers.insert(0, '-1')
  720. if len(layers) > 0:
  721. self.SetItems(layers)
  722. else:
  723. self.Clear()
  724. self.SetValue("")
  725. if self.default and self.default in layers:
  726. self.SetValue(self.default)
  727. class DriverSelect(wx.ComboBox):
  728. """!Creates combo box for selecting database driver.
  729. """
  730. def __init__(self, parent, choices, value,
  731. id=wx.ID_ANY, pos=wx.DefaultPosition,
  732. size=globalvar.DIALOG_LAYER_SIZE, **kargs):
  733. super(DriverSelect, self).__init__(parent, id, value, pos, size,
  734. choices, style=wx.CB_READONLY)
  735. self.SetName("DriverSelect")
  736. self.SetStringSelection(value)
  737. class DatabaseSelect(wx.TextCtrl):
  738. """!Creates combo box for selecting database driver.
  739. """
  740. def __init__(self, parent, value = '', id = wx.ID_ANY,
  741. size = globalvar.DIALOG_TEXTCTRL_SIZE, **kargs):
  742. super(DatabaseSelect, self).__init__(parent, id, value, size = size, **kargs)
  743. self.SetName("DatabaseSelect")
  744. class TableSelect(wx.ComboBox):
  745. """!Creates combo box for selecting attribute tables from the database
  746. """
  747. def __init__(self, parent,
  748. id = wx.ID_ANY, value = '',
  749. size = globalvar.DIALOG_COMBOBOX_SIZE, choices = [], **kargs):
  750. super(TableSelect, self).__init__(parent, id, value, size = size, choices = choices,
  751. style = wx.CB_READONLY, **kargs)
  752. self.SetName("TableSelect")
  753. if not choices:
  754. self.InsertTables()
  755. def InsertTables(self, driver = None, database = None):
  756. """!Insert attribute tables into combobox"""
  757. items = []
  758. if not driver or not database:
  759. connect = grass.db_connection()
  760. driver = connect['driver']
  761. database = connect['database']
  762. ret = RunCommand('db.tables',
  763. flags = 'p',
  764. read = True,
  765. driver = driver,
  766. database = database)
  767. if ret:
  768. for table in ret.splitlines():
  769. items.append(table)
  770. self.SetItems(items)
  771. self.SetValue('')
  772. class ColumnSelect(wx.combo.ComboCtrl):
  773. """!Creates combo box for selecting columns in the attribute table
  774. for a vector map.
  775. @param parent window parent
  776. @param id window id
  777. @param value default value
  778. @param size window size
  779. @param vector vector map name
  780. @param layer layer number
  781. @param multiple - True if it is possible to add multiple columns
  782. @param param parameters list (see menuform.py)
  783. @param **kwags wx.ComboBox parameters
  784. """
  785. def __init__(self, parent, id = wx.ID_ANY, value = '',
  786. size = globalvar.DIALOG_COMBOBOX_SIZE,
  787. vector = None, layer = 1, multiple = False,
  788. param = None, **kwargs):
  789. self.defaultValue = value
  790. self.param = param
  791. self.columns = []
  792. wx.combo.ComboCtrl.__init__(self, parent, id, size = size, **kwargs)
  793. self.GetChildren()[0].SetName("ColumnSelect")
  794. self.GetChildren()[0].type = type
  795. self.tcp = ListCtrlComboPopup()
  796. self.SetPopupControl(self.tcp)
  797. self.tcp.SetData(multiple = multiple)
  798. if vector:
  799. self.InsertColumns(vector, layer)
  800. self.GetChildren()[0].Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  801. def GetColumns(self):
  802. return self.columns
  803. def OnKeyUp(self, event):
  804. """!Shows popupwindow if down arrow key is released"""
  805. if event.GetKeyCode() == wx.WXK_DOWN and not self.IsPopupShown():
  806. self.ShowPopup()
  807. else:
  808. event.Skip()
  809. def Clear(self):
  810. self.tcp.DeleteAllItems()
  811. self.SetValue('')
  812. def InsertColumns(self, vector, layer, excludeKey = False, excludeCols = None, type = None, dbInfo = None):
  813. """!Insert columns for a vector attribute table into the columns combobox
  814. @param vector vector name
  815. @param layer vector layer number
  816. @param excludeKey exclude key column from the list?
  817. @param excludeCols list of columns to be removed from the list
  818. @param type only columns of given type (given as list)
  819. """
  820. if not dbInfo:
  821. dbInfo = VectorDBInfo(vector)
  822. try:
  823. try:
  824. layer = int(layer)
  825. except TypeError:
  826. # assuming layer 1
  827. layer = 1
  828. table = dbInfo.GetTable(layer)
  829. columnchoices = dbInfo.GetTableDesc(table)
  830. keyColumn = dbInfo.GetKeyColumn(layer)
  831. self.columns = len(columnchoices.keys()) * ['']
  832. for key, val in columnchoices.iteritems():
  833. self.columns[val['index']] = key
  834. if excludeKey: # exclude key column
  835. self.columns.remove(keyColumn)
  836. if excludeCols: # exclude key column
  837. for key in columnchoices.iterkeys():
  838. if key in excludeCols:
  839. self.columns.remove(key)
  840. if type: # only selected column types
  841. for key, value in columnchoices.iteritems():
  842. if value['type'] not in type:
  843. try:
  844. self.columns.remove(key)
  845. except ValueError:
  846. pass
  847. except (KeyError, ValueError):
  848. self.columns[:] = []
  849. # update list
  850. self.tcp.DeleteAllItems()
  851. for col in self.columns:
  852. self.tcp.AddItem(col)
  853. self.SetValue(self.defaultValue)
  854. if self.param:
  855. value = self.param.get('value', '')
  856. if value != '' and value in self.columns:
  857. self.SetValue(value)
  858. def InsertTableColumns(self, table, driver=None, database=None):
  859. """!Insert table columns
  860. @param table table name
  861. @param driver driver name
  862. @param database database name
  863. """
  864. self.columns[:] = []
  865. ret = RunCommand('db.columns',
  866. read = True,
  867. driver = driver,
  868. database = database,
  869. table = table)
  870. if ret:
  871. self.columns = ret.splitlines()
  872. # update list
  873. self.tcp.DeleteAllItems()
  874. self.SetValue(self.defaultValue)
  875. for col in self.columns:
  876. self.tcp.AddItem(col)
  877. if self.param:
  878. value = self.param.get('value', '')
  879. if value != '' and value in self.columns:
  880. self.SetValue(value)
  881. class DbaseSelect(wx.lib.filebrowsebutton.DirBrowseButton):
  882. """!Widget for selecting GRASS Database"""
  883. def __init__(self, parent, **kwargs):
  884. super(DbaseSelect, self).__init__(parent, id = wx.ID_ANY,
  885. size = globalvar.DIALOG_GSELECT_SIZE, labelText = '',
  886. dialogTitle = _('Choose GIS Data Directory'),
  887. buttonText = _('Browse'),
  888. startDirectory = grass.gisenv()['GISDBASE'],
  889. **kwargs)
  890. class LocationSelect(wx.ComboBox):
  891. """!Widget for selecting GRASS location"""
  892. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  893. gisdbase = None, **kwargs):
  894. super(LocationSelect, self).__init__(parent, id, size = size,
  895. style = wx.CB_READONLY, **kwargs)
  896. self.SetName("LocationSelect")
  897. if not gisdbase:
  898. self.gisdbase = grass.gisenv()['GISDBASE']
  899. else:
  900. self.gisdbase = gisdbase
  901. self.SetItems(GetListOfLocations(self.gisdbase))
  902. def UpdateItems(self, dbase):
  903. """!Update list of locations
  904. @param dbase path to GIS database
  905. """
  906. self.gisdbase = dbase
  907. if dbase:
  908. self.SetItems(GetListOfLocations(self.gisdbase))
  909. else:
  910. self.SetItems([])
  911. class MapsetSelect(wx.combo.ComboCtrl):
  912. """!Widget for selecting GRASS mapset"""
  913. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  914. gisdbase = None, location = None, setItems = True,
  915. searchPath = False, new = False, skipCurrent = False, multiple = False, **kwargs):
  916. style = 0
  917. ### disabled, read-only widget has no TextCtrl children (TODO: rewrite)
  918. ### if not new and not multiple:
  919. ### style = wx.CB_READONLY
  920. wx.combo.ComboCtrl.__init__(self, parent, id, size = size,
  921. style = style, **kwargs)
  922. self.searchPath = searchPath
  923. self.skipCurrent = skipCurrent
  924. self.SetName("MapsetSelect")
  925. if not gisdbase:
  926. self.gisdbase = grass.gisenv()['GISDBASE']
  927. else:
  928. self.gisdbase = gisdbase
  929. if not location:
  930. self.location = grass.gisenv()['LOCATION_NAME']
  931. else:
  932. self.location = location
  933. self.tcp = ListCtrlComboPopup()
  934. self.SetPopupControl(self.tcp)
  935. self.tcp.SetData(multiple = multiple)
  936. if setItems:
  937. self.tcp.SetItems(self._getMapsets())
  938. def UpdateItems(self, location, dbase = None):
  939. """!Update list of mapsets for given location
  940. @param dbase path to GIS database (None to use currently selected)
  941. @param location name of location
  942. """
  943. if dbase:
  944. self.gisdbase = dbase
  945. self.location = location
  946. self.tcp.DeleteAllItems()
  947. if location:
  948. self.tcp.SetItems(self._getMapsets())
  949. else:
  950. self.tcp.SetItems([])
  951. def _getMapsets(self):
  952. if self.searchPath:
  953. mlist = RunCommand('g.mapsets',
  954. read = True, flags = 'p',
  955. sep = 'newline').splitlines()
  956. else:
  957. mlist = GetListOfMapsets(self.gisdbase, self.location,
  958. selectable = False)
  959. gisenv = grass.gisenv()
  960. if self.skipCurrent and \
  961. gisenv['LOCATION_NAME'] == self.location and \
  962. gisenv['MAPSET'] in mlist:
  963. mlist.remove(gisenv['MAPSET'])
  964. return mlist
  965. def GetStringSelection(self):
  966. """!For backward compatibility. MapsetSelect changed to allow
  967. multiple selection, this required to change super-class from
  968. wx.ComboBox to wx.combo.ComboCtrl"""
  969. return self.GetValue()
  970. def SetStringSelection(self, text):
  971. """!For backward compatibility. MapsetSelect changed to allow
  972. multiple selection, this required to change super-class from
  973. wx.ComboBox to wx.combo.ComboCtrl"""
  974. return self.SetValue(text)
  975. def SetItems(self, items):
  976. """!For backward compatibility. MapsetSelect changed to allow
  977. multiple selection, this required to change super-class from
  978. wx.ComboBox to wx.combo.ComboCtrl"""
  979. self.tcp.DeleteAllItems()
  980. self.tcp.SetItems(items)
  981. class SubGroupSelect(wx.ComboBox):
  982. """!Widget for selecting subgroups"""
  983. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
  984. **kwargs):
  985. super(SubGroupSelect, self).__init__(parent, id, size = size,
  986. **kwargs)
  987. self.SetName("SubGroupSelect")
  988. def Insert(self, group):
  989. """!Insert subgroups for defined group"""
  990. if not group:
  991. return
  992. gisenv = grass.gisenv()
  993. try:
  994. name, mapset = group.split('@', 1)
  995. except ValueError:
  996. name = group
  997. mapset = gisenv['MAPSET']
  998. mlist = RunCommand('i.group', group=group,
  999. read=True, flags='sg').splitlines()
  1000. try:
  1001. self.SetItems(mlist)
  1002. except OSError:
  1003. self.SetItems([])
  1004. class FormatSelect(wx.Choice):
  1005. def __init__(self, parent, srcType, ogr=False,
  1006. size=globalvar.DIALOG_SPIN_SIZE,
  1007. **kwargs):
  1008. """!Widget for selecting external (GDAL/OGR) format
  1009. @param parent parent window
  1010. @param srcType source type ('file', 'database', 'protocol')
  1011. @param ogr True for OGR otherwise GDAL
  1012. """
  1013. super(FormatSelect, self).__init__(parent, id=wx.ID_ANY, size=size,
  1014. **kwargs)
  1015. self.SetName("FormatSelect")
  1016. if ogr:
  1017. ftype = 'ogr'
  1018. else:
  1019. ftype = 'gdal'
  1020. formats = list()
  1021. for f in GetFormats()[ftype][srcType].values():
  1022. formats += f
  1023. self.SetItems(formats)
  1024. def GetExtension(self, name):
  1025. """!Get file extension by format name"""
  1026. formatToExt = dict()
  1027. formatToExt.update(rasterFormatExtension)
  1028. formatToExt.update(vectorFormatExtension)
  1029. return formatToExt.get(name, '')
  1030. # unused code since r47938
  1031. # wxGdalSelect, EVT_GDALSELECT = NewEvent()
  1032. class GdalSelect(wx.Panel):
  1033. def __init__(self, parent, panel, ogr=False, link=False, dest=False,
  1034. exclude=None):
  1035. """!Widget for selecting GDAL/OGR datasource, format
  1036. @todo Split into GdalSelect and OgrSelect and optionally to
  1037. GdalSelectOutput, OgrSelectOutput
  1038. @param parent parent window
  1039. @param ogr use OGR selector instead of GDAL
  1040. @param dest True for output (destination)
  1041. @param default deafult type (ignored when dest == True)
  1042. @param exclude list of types to be excluded
  1043. """
  1044. self.parent = parent
  1045. self.ogr = ogr
  1046. self.link = link
  1047. self.dest = dest
  1048. self._sourceType = None
  1049. wx.Panel.__init__(self, parent=panel)
  1050. self.reloadDataRequired = Signal('GdalSelect.reloadDataRequired')
  1051. if self.ogr:
  1052. settingsFile = os.path.join(GetSettingsPath(), 'wxOGR')
  1053. else:
  1054. settingsFile = os.path.join(GetSettingsPath(), 'wxGDAL')
  1055. self.settsManager = ManageSettingsWidget(parent=self,
  1056. settingsFile=settingsFile)
  1057. self.settsManager.settingsChanged.connect(self.OnSettingsChanged)
  1058. self.settsManager.settingsSaving.connect(self.OnSettingsSaving)
  1059. self.inputBox = wx.StaticBox(parent=self)
  1060. if dest:
  1061. self.inputBox.SetLabel(" %s " % _("Output settings"))
  1062. else:
  1063. self.inputBox.SetLabel(" %s " % _("Source settings"))
  1064. # source type
  1065. sources = list()
  1066. self.sourceMap = { 'file' : -1,
  1067. 'dir' : -1,
  1068. 'db' : -1,
  1069. 'pro' : -1,
  1070. 'native' : -1 }
  1071. idx = 0
  1072. if dest:
  1073. sources.append(_("Native"))
  1074. self.sourceMap['native'] = idx
  1075. idx += 1
  1076. if exclude is None:
  1077. exclude = []
  1078. if 'file' not in exclude:
  1079. sources.append(_("File"))
  1080. self.sourceMap['file'] = idx
  1081. idx += 1
  1082. if 'directory' not in exclude:
  1083. sources.append(_("Directory"))
  1084. self.sourceMap['dir'] = idx
  1085. idx += 1
  1086. if 'database' not in exclude:
  1087. sources.append(_("Database"))
  1088. self.sourceMap['db'] = idx
  1089. idx += 1
  1090. if 'protocol' not in exclude:
  1091. sources.append(_("Protocol"))
  1092. self.sourceMap['pro'] = idx
  1093. idx += 1
  1094. self.sourceMapByIdx = {}
  1095. for name, idx in self.sourceMap.items():
  1096. self.sourceMapByIdx[ idx ] = name
  1097. self.source = wx.RadioBox(parent=self, id=wx.ID_ANY,
  1098. style=wx.RA_SPECIFY_COLS,
  1099. choices=sources)
  1100. if dest:
  1101. self.source.SetLabel(" %s " % _('Output type'))
  1102. else:
  1103. self.source.SetLabel(" %s " % _('Source type'))
  1104. self.source.SetSelection(0)
  1105. self.source.Bind(wx.EVT_RADIOBOX,
  1106. lambda evt: self.SetSourceType(self.sourceMapByIdx[evt.GetInt()]))
  1107. self.nativeWidgets = {}
  1108. self.fileWidgets = {}
  1109. self.dirWidgets = {}
  1110. self.dbWidgets = {}
  1111. self.protocolWidgets = {}
  1112. self.pgWidgets = {}
  1113. if ogr:
  1114. fType = 'ogr'
  1115. else:
  1116. fType = 'gdal'
  1117. # file
  1118. fileMask = '%(all)s (*.*)|*.*|' % {'all': _('All files')}
  1119. for name, ext in sorted(rasterFormatExtension.items()):
  1120. fileMask += '%(name)s (*.%(low)s;*.%(up)s)|*.%(low)s;*.%(up)s|' % {'name': name,
  1121. 'low': ext.lower(),
  1122. 'up': ext.upper()}
  1123. fileMask += '%s (*.zip;*.ZIP)|*.zip;*.ZIP|' % _('ZIP files')
  1124. fileMask += '%s (*.gz;*.GZ)|*.gz;*.GZ|' % _('GZIP files')
  1125. fileMask += '%s (*.tar;*.TAR)|*.tar;*.TAR|' % _('TAR files')
  1126. fileMask += '%s (*.tar.gz;*.TAR.GZ;*.tgz;*.TGZ)|*.tar.gz;*.TAR.GZ;*.tgz;*.TGZ' % _('TARGZ files')
  1127. # only contains formats with extensions hardcoded
  1128. self.filePanel = wx.Panel(parent=self)
  1129. browse = filebrowse.FileBrowseButton(parent=self.filePanel, id=wx.ID_ANY,
  1130. size=globalvar.DIALOG_GSELECT_SIZE,
  1131. labelText = _('File:'),
  1132. dialogTitle=_('Choose file to import'),
  1133. buttonText=_('Browse'),
  1134. startDirectory=os.getcwd(),
  1135. changeCallback=self.OnUpdate,
  1136. fileMask=fileMask)
  1137. self.fileWidgets['browse'] = browse
  1138. self.fileWidgets['options'] = wx.TextCtrl(parent=self.filePanel)
  1139. # directory
  1140. self.dirPanel = wx.Panel(parent=self)
  1141. browse = filebrowse.DirBrowseButton(parent=self.dirPanel, id=wx.ID_ANY,
  1142. size=globalvar.DIALOG_GSELECT_SIZE,
  1143. labelText=_('Directory:'),
  1144. dialogTitle=_('Choose input directory'),
  1145. buttonText=_('Browse'),
  1146. startDirectory=os.getcwd(),
  1147. changeCallback=self.OnUpdate)
  1148. self.dirWidgets['browse'] = browse
  1149. formatSelect = wx.Choice(parent=self.dirPanel, size=(300, -1))
  1150. self.dirWidgets['format'] = formatSelect
  1151. fileFormats = GetFormats(writableOnly=dest)[fType]['file']
  1152. formatSelect.SetItems(sorted(list(fileFormats)))
  1153. formatSelect.Bind(wx.EVT_CHOICE, lambda evt: self.SetExtension(self.dirWidgets['format'].GetStringSelection()))
  1154. formatSelect.Bind(wx.EVT_CHOICE, self.OnUpdate)
  1155. self.dirWidgets['extensionLabel'] = wx.StaticText(parent=self.dirPanel,
  1156. label = _("Extension:"))
  1157. self.dirWidgets['extension'] = wx.TextCtrl(parent=self.dirPanel)
  1158. self.dirWidgets['extension'].Bind(wx.EVT_TEXT, self.ExtensionChanged)
  1159. self.dirWidgets['options'] = wx.TextCtrl(parent=self.dirPanel)
  1160. if self.ogr:
  1161. shapefile = 'ESRI Shapefile'
  1162. if shapefile in fileFormats:
  1163. formatSelect.SetStringSelection(shapefile)
  1164. self.SetExtension(shapefile)
  1165. else:
  1166. tiff = 'GeoTIFF'
  1167. if tiff in fileFormats:
  1168. formatSelect.SetStringSelection(tiff)
  1169. self.SetExtension(tiff)
  1170. # database
  1171. self.dbPanel = wx.Panel(parent=self)
  1172. self.dbFormats = GetFormats(writableOnly=dest)[fType]['database']
  1173. dbChoice = wx.Choice(parent=self.dbPanel, choices=self.dbFormats)
  1174. dbChoice.Bind(wx.EVT_CHOICE, lambda evt: self.SetDatabase(db=dbChoice.GetStringSelection()))
  1175. self.dbWidgets['format'] = dbChoice
  1176. browse = filebrowse.FileBrowseButton(parent=self.dbPanel, id=wx.ID_ANY,
  1177. size=globalvar.DIALOG_GSELECT_SIZE,
  1178. labelText=_("Name:"),
  1179. dialogTitle=_('Choose file'),
  1180. buttonText=_('Browse'),
  1181. startDirectory=os.getcwd(),
  1182. changeCallback=self.OnUpdate)
  1183. self.dbWidgets['browse'] = browse
  1184. self.dbWidgets['choice'] = wx.Choice(parent=self.dbPanel)
  1185. self.dbWidgets['choice'].Bind(wx.EVT_CHOICE, self.OnUpdate)
  1186. self.dbWidgets['text'] = wx.TextCtrl(parent=self.dbPanel)
  1187. self.dbWidgets['text'].Bind(wx.EVT_TEXT, self.OnUpdate)
  1188. self.dbWidgets['textLabel1'] = wx.StaticText(parent=self.dbPanel, label=_("Name:"))
  1189. self.dbWidgets['textLabel2'] = wx.StaticText(parent=self.dbPanel, label=_("Name:"))
  1190. self.dbWidgets['featType'] = wx.RadioBox(parent=self.dbPanel, id=wx.ID_ANY,
  1191. label = " %s " % _("Feature type:"),
  1192. choices = [_("simple features"), _("topological")],
  1193. majorDimension=2,
  1194. style = wx.RA_SPECIFY_COLS)
  1195. if dest:
  1196. self.dbWidgets['featType'].Disable()
  1197. else:
  1198. self.dbWidgets['featType'].Hide()
  1199. browse = filebrowse.DirBrowseButton(parent=self.dbPanel, id=wx.ID_ANY,
  1200. size=globalvar.DIALOG_GSELECT_SIZE,
  1201. labelText=_('Directory:'),
  1202. dialogTitle=_('Choose input directory'),
  1203. buttonText=_('Browse'),
  1204. startDirectory=os.getcwd(),
  1205. changeCallback=self.OnUpdate)
  1206. self.dbWidgets['dirbrowse'] = browse
  1207. self.dbWidgets['options'] = wx.TextCtrl(parent=self.dbPanel)
  1208. # protocol
  1209. self.protocolPanel = wx.Panel(parent=self)
  1210. protocolFormats = GetFormats(writableOnly=self.dest)[fType]['protocol']
  1211. protocolChoice = wx.Choice(parent=self.protocolPanel, choices=protocolFormats)
  1212. self.protocolWidgets['format'] = protocolChoice
  1213. self.protocolWidgets['text'] = wx.TextCtrl(parent=self.protocolPanel)
  1214. self.protocolWidgets['text'].Bind(wx.EVT_TEXT, self.OnUpdate)
  1215. self.protocolWidgets['options'] = wx.TextCtrl(parent=self.protocolPanel)
  1216. # native
  1217. self.nativePanel = wx.Panel(parent=self)
  1218. self._layout()
  1219. sourceType = 'file'
  1220. self.SetSourceType(sourceType) # needed always to fit dialog size
  1221. if self.dest:
  1222. current = RunCommand('v.external.out',
  1223. parent=self,
  1224. read=True, parse=grass.parse_key_val,
  1225. flags='g')
  1226. if current['format'] == 'native':
  1227. sourceType = 'native'
  1228. elif current['format'] in GetFormats()['ogr']['database']:
  1229. sourceType = 'db'
  1230. else:
  1231. sourceType = 'dir'
  1232. if self.dest:
  1233. wx.CallAfter(self._postInit, sourceType, current)
  1234. def _postInit(self, sourceType, data):
  1235. """!Fill in default values."""
  1236. format = data.get('format', '')
  1237. pg = 'conninfo' in data.keys()
  1238. if pg:
  1239. dsn=''
  1240. for item in data.get('conninfo').split(' '):
  1241. k, v = item.split('=')
  1242. if k == 'dbname':
  1243. dsn=v
  1244. break
  1245. optList = list()
  1246. for k, v in data.iteritems():
  1247. if k in ('format', 'conninfo', 'topology'):
  1248. continue
  1249. optList.append('%s=%s' % (k, v))
  1250. options = ','.join(optList)
  1251. else:
  1252. dsn = data.get('dsn')
  1253. options = data.get('options', '')
  1254. self.SetSourceType(sourceType)
  1255. self.source.SetSelection(self.sourceMap[sourceType])
  1256. dsn = os.path.expandvars(dsn) # v.external.out uses $HOME
  1257. # fill in default values
  1258. if sourceType == 'dir':
  1259. self.dirWidgets['format'].SetStringSelection(format)
  1260. self.dirWidgets['browse'].SetValue(dsn)
  1261. self.dirWidgets['options'].SetValue(options)
  1262. elif sourceType == 'db':
  1263. self.dbWidgets['format'].SetStringSelection(format)
  1264. self.dbWidgets['options'].SetValue(options)
  1265. name = self._getCurrentDbWidgetName()
  1266. if name == 'choice':
  1267. if dsn in self.dbWidgets[name].GetItems():
  1268. self.dbWidgets[name].SetStringSelection(dsn)
  1269. if 'topology' in data.keys():
  1270. self.dbWidgets['featType'].SetSelection(1)
  1271. else:
  1272. self.dbWidgets[name].SetValue(dsn)
  1273. def _layout(self):
  1274. """!Layout"""
  1275. mainSizer = wx.BoxSizer(wx.VERTICAL)
  1276. self.changingSizer = wx.StaticBoxSizer(self.inputBox, wx.VERTICAL)
  1277. # file
  1278. paddingSizer = wx.BoxSizer(wx.VERTICAL)
  1279. sizer = wx.GridBagSizer(vgap=5, hgap=10)
  1280. paddingSizer.Add(item=self.fileWidgets['browse'],
  1281. flag=wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1282. border=30)
  1283. sizer.Add(item=paddingSizer, flag=wx.EXPAND, pos=(0, 0), span=(1, 2))
  1284. sizer.AddGrowableCol(0)
  1285. if self.dest:
  1286. sizer.Add(item=wx.StaticText(parent=self.filePanel,
  1287. label = _("Creation options:")),
  1288. flag = wx.ALIGN_CENTER_VERTICAL,
  1289. pos = (1, 0))
  1290. sizer.Add(item=self.fileWidgets['options'],
  1291. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1292. pos=(1, 1))
  1293. else:
  1294. self.fileWidgets['options'].Hide()
  1295. self.filePanel.SetSizer(sizer)
  1296. # directory
  1297. sizer = wx.GridBagSizer(vgap=3, hgap=10)
  1298. sizer.Add(item=wx.StaticText(parent=self.dirPanel,
  1299. label = _("Format:")),
  1300. flag=wx.ALIGN_CENTER_VERTICAL,
  1301. pos=(0, 0))
  1302. sizer.Add(item=self.dirWidgets['format'],
  1303. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1304. pos=(0, 1))
  1305. sizer.Add(item=self.dirWidgets['extensionLabel'],
  1306. flag=wx.ALIGN_CENTER_VERTICAL,
  1307. pos=(0, 2))
  1308. sizer.Add(item=self.dirWidgets['extension'],
  1309. flag=wx.ALIGN_CENTER_VERTICAL,
  1310. pos=(0, 3))
  1311. sizer.Add(item=self.dirWidgets['browse'],
  1312. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1313. pos=(1, 0), span=(1, 4))
  1314. if self.dest:
  1315. sizer.Add(item=wx.StaticText(parent=self.dirPanel,
  1316. label = _("Creation options:")),
  1317. flag = wx.ALIGN_CENTER_VERTICAL,
  1318. pos = (2, 0))
  1319. sizer.Add(item=self.dirWidgets['options'],
  1320. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1321. pos=(2, 1))
  1322. helpBtn = wx.Button(parent=self.dirPanel, id=wx.ID_HELP)
  1323. helpBtn.Bind(wx.EVT_BUTTON, self.OnHelp)
  1324. sizer.Add(item=helpBtn,
  1325. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1326. pos=(2, 2))
  1327. self.dirWidgets['extensionLabel'].Hide()
  1328. self.dirWidgets['extension'].Hide()
  1329. else:
  1330. self.dirWidgets['options'].Hide()
  1331. sizer.AddGrowableCol(1)
  1332. self.dirPanel.SetSizer(sizer)
  1333. # database
  1334. sizer = wx.GridBagSizer(vgap=1, hgap=5)
  1335. sizer.Add(item=wx.StaticText(parent=self.dbPanel,
  1336. label = _("Format:")),
  1337. flag=wx.ALIGN_CENTER_VERTICAL,
  1338. pos=(0, 0))
  1339. sizer.Add(item=self.dbWidgets['format'],
  1340. flag=wx.ALIGN_CENTER_VERTICAL,
  1341. pos=(0, 1))
  1342. sizer.Add(item=self.dbWidgets['textLabel1'],
  1343. flag=wx.ALIGN_CENTER_VERTICAL,
  1344. pos=(1, 0))
  1345. sizer.Add(item=self.dbWidgets['text'],
  1346. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1347. pos=(1, 1), span=(1, 2))
  1348. sizer.Add(item=self.dbWidgets['browse'],
  1349. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1350. pos=(2, 0), span=(1, 3))
  1351. sizer.Add(item=self.dbWidgets['dirbrowse'],
  1352. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1353. pos=(3, 0), span=(1, 2))
  1354. sizer.Add(item=self.dbWidgets['textLabel2'],
  1355. flag=wx.ALIGN_CENTER_VERTICAL,
  1356. pos=(4, 0))
  1357. sizer.Add(item=self.dbWidgets['choice'],
  1358. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1359. pos=(4, 1), span=(1, 2))
  1360. if self.dest:
  1361. sizer.Add(item=self.dbWidgets['featType'],
  1362. pos=(0, 2), flag=wx.EXPAND)
  1363. sizer.Add(item=wx.StaticText(parent=self.dbPanel,
  1364. label = _("Creation options:")),
  1365. flag = wx.ALIGN_CENTER_VERTICAL,
  1366. pos = (5, 0))
  1367. sizer.Add(item=self.dbWidgets['options'],
  1368. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1369. pos=(5, 1), span=(1, 2))
  1370. # help button
  1371. helpBtn = wx.Button(parent=self.dbPanel, id=wx.ID_HELP)
  1372. helpBtn.Bind(wx.EVT_BUTTON, self.OnHelp)
  1373. sizer.Add(item=helpBtn,
  1374. pos=(5, 3))
  1375. else:
  1376. self.dbWidgets['options'].Hide()
  1377. self.dbPanel.SetSizer(sizer)
  1378. sizer.SetEmptyCellSize((0, 0))
  1379. sizer.AddGrowableCol(1)
  1380. # protocol
  1381. sizer = wx.GridBagSizer(vgap=3, hgap=3)
  1382. sizer.Add(item=wx.StaticText(parent=self.protocolPanel,
  1383. label = _("Format:")),
  1384. flag=wx.ALIGN_CENTER_VERTICAL,
  1385. pos=(0, 0))
  1386. sizer.Add(item=self.protocolWidgets['format'],
  1387. flag=wx.ALIGN_CENTER_VERTICAL,
  1388. pos=(0, 1))
  1389. sizer.Add(item=wx.StaticText(parent=self.protocolPanel,
  1390. label = _("Protocol:")),
  1391. flag=wx.ALIGN_CENTER_VERTICAL,
  1392. pos=(1, 0))
  1393. sizer.Add(item=self.protocolWidgets['text'],
  1394. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1395. pos=(1, 1))
  1396. if self.dest:
  1397. sizer.Add(item=wx.StaticText(parent=self.protocolPanel,
  1398. label = _("Creation options:")),
  1399. flag = wx.ALIGN_CENTER_VERTICAL,
  1400. pos = (2, 0))
  1401. sizer.Add(item=self.protocolWidgets['options'],
  1402. flag=wx.ALIGN_CENTER_VERTICAL|wx.EXPAND,
  1403. pos=(2, 1))
  1404. else:
  1405. self.protocolWidgets['options'].Hide()
  1406. sizer.AddGrowableCol(1)
  1407. self.protocolPanel.SetSizer(sizer)
  1408. # native
  1409. sizer = wx.BoxSizer(wx.VERTICAL)
  1410. sizer.Add(item=wx.StaticText(parent=self.nativePanel,
  1411. label = _("No settings available")),
  1412. flag=wx.ALIGN_CENTER_VERTICAL|wx.ALL|wx.EXPAND, border=5)
  1413. self.nativePanel.SetSizer(sizer)
  1414. for panel in (self.nativePanel, self.filePanel,
  1415. self.dirPanel, self.dbPanel,
  1416. self.protocolPanel):
  1417. self.changingSizer.Add(item=panel, proportion=1,
  1418. flag=wx.EXPAND)
  1419. mainSizer.Add(item=self.settsManager, proportion=0,
  1420. flag=wx.ALL | wx.EXPAND, border=5)
  1421. mainSizer.Add(item=self.source, proportion=0,
  1422. flag=wx.LEFT | wx.RIGHT | wx.EXPAND, border=5)
  1423. mainSizer.Add(item=self.changingSizer, proportion=1,
  1424. flag=wx.ALL | wx.EXPAND, border=5)
  1425. self.mainSizer = mainSizer
  1426. self.SetSizer(mainSizer)
  1427. mainSizer.Fit(self)
  1428. def _getExtension(self, name):
  1429. """!Get file extension by format name"""
  1430. formatToExt = dict()
  1431. formatToExt.update(rasterFormatExtension)
  1432. formatToExt.update(vectorFormatExtension)
  1433. return formatToExt.get(name, '')
  1434. def SetSourceType(self, sourceType):
  1435. """!Set source type (db, file, dir, ...).
  1436. Does not switch radioboxes."""
  1437. self._sourceType = sourceType
  1438. self.changingSizer.Show(item=self.filePanel, show=(sourceType == 'file'))
  1439. self.changingSizer.Show(item=self.nativePanel, show=(sourceType == 'native'))
  1440. self.changingSizer.Show(item=self.dirPanel, show=(sourceType == 'dir'))
  1441. self.changingSizer.Show(item=self.protocolPanel, show=(sourceType == 'pro'))
  1442. self.changingSizer.Show(item=self.dbPanel, show=(sourceType is 'db'))
  1443. self.changingSizer.Layout()
  1444. if sourceType == 'db':
  1445. self.dbWidgets['format'].SetItems(self.dbFormats)
  1446. if self.dbFormats:
  1447. if 'PostgreSQL' in self.dbFormats:
  1448. self.dbWidgets['format'].SetStringSelection('PostgreSQL')
  1449. else:
  1450. self.dbWidgets['format'].SetSelection(0)
  1451. self.dbWidgets['format'].Enable()
  1452. if sourceType == 'db':
  1453. db = self.dbWidgets['format'].GetStringSelection()
  1454. self.SetDatabase(db)
  1455. if not self.dest:
  1456. self.reloadDataRequired.emit(data=None)
  1457. self._reloadLayers()
  1458. def OnSettingsChanged(self, data):
  1459. """!User changed setting"""
  1460. # data list: [type, dsn, format, options]
  1461. if len(data) == 3:
  1462. data.append('')
  1463. elif len < 3:
  1464. return
  1465. self.source.SetSelection(self.sourceMap[data[0]])
  1466. self.SetSourceType(data[0])
  1467. if data[0] == 'file':
  1468. self.fileWidgets['browse'].SetValue(data[1])
  1469. self.fileWidgets['options'].SetValue(data[3])
  1470. elif data[0] == 'dir':
  1471. self.dirWidgets['browse'].SetValue(data[1])
  1472. self.dirWidgets['format'].SetStringSelection(data[2])
  1473. self.dirWidgets['options'].SetValue(data[3])
  1474. self.SetExtension(data[2])
  1475. elif data[0] == 'pro':
  1476. self.protocolWidgets['text'].SetValue(data[2])
  1477. self.protocolWidgets['options'].SetValue(data[3])
  1478. elif data[0] == 'db':
  1479. name = self._getCurrentDbWidgetName()
  1480. if name =='choice':
  1481. if len(data[1].split(':', 1)) > 1:
  1482. for item in data[1].split(':', 1)[1].split(','):
  1483. key, value = item.split('=', 1)
  1484. if key == 'dbname':
  1485. self.dbWidgets[name].SetStringSelection(value)
  1486. break
  1487. else:
  1488. self.dbWidgets[name].SetStringSelection(data[1])
  1489. else:
  1490. self.dbWidgets[name].SetValue(data[1])
  1491. self.dbWidgets['options'].SetValue(data[3])
  1492. if not self.dest:
  1493. self.reloadDataRequired.emit(data=None)
  1494. self._reloadLayers()
  1495. def OnSettingsSaving(self, name):
  1496. """!Saving data"""
  1497. if not self.GetDsn():
  1498. GMessage(parent = self,
  1499. message = _("No data source defined, settings are not saved."))
  1500. return
  1501. self.settsManager.SetDataToSave((self._sourceType, self.GetDsn(),
  1502. self.GetFormat(), self.GetOptions()))
  1503. self.settsManager.SaveSettings(name)
  1504. def _getExtPatternGlob(self, ext):
  1505. """!Get pattern for case-insensitive globing"""
  1506. pattern = '*.'
  1507. for c in ext:
  1508. pattern += '[%s%s]' % (c.lower(), c.upper())
  1509. return pattern
  1510. def _getCurrentDbWidgetName(self):
  1511. """!Returns active dns database widget name."""
  1512. for widget in ('browse', 'dirbrowse', 'text', 'choice'):
  1513. if self.dbWidgets[widget].IsShown():
  1514. return widget
  1515. def GetDsn(self):
  1516. """!Get datasource name
  1517. """
  1518. if self._sourceType == 'db':
  1519. if self.dbWidgets['format'].GetStringSelection() in ('PostgreSQL',
  1520. 'PostGIS Raster driver'):
  1521. dsn = 'PG:dbname=%s' % self.dbWidgets['choice'].GetStringSelection()
  1522. else:
  1523. name = self._getCurrentDbWidgetName()
  1524. if name == 'choice':
  1525. dsn = self.dbWidgets[name].GetStringSelection()
  1526. else:
  1527. dsn = self.dbWidgets[name].GetValue()
  1528. else:
  1529. if self._sourceType == 'file':
  1530. dsn = self.fileWidgets['browse'].GetValue()
  1531. elif self._sourceType == 'dir':
  1532. dsn = self.dirWidgets['browse'].GetValue()
  1533. elif self._sourceType == 'pro':
  1534. dsn = self.protocolWidgets['text'].GetValue()
  1535. else:
  1536. dsn = ''
  1537. # check compressed files
  1538. try:
  1539. ext = os.path.splitext(dsn)[1].lower()
  1540. except KeyError:
  1541. ext = None
  1542. if ext == '.zip':
  1543. dsn = '/vsizip/' + dsn
  1544. elif ext == '.gzip':
  1545. dsn = '/vsigzip/' + dsn
  1546. elif ext in ('.tar', '.tar.gz', '.tgz'):
  1547. dsn = '/vsitar/' + dsn
  1548. return dsn
  1549. def SetDatabase(self, db):
  1550. """!Update database panel."""
  1551. sizer = self.dbPanel.GetSizer()
  1552. showBrowse = db in ('SQLite', 'Rasterlite')
  1553. showDirbrowse = db in ('FileGDB')
  1554. showChoice = db in ('PostgreSQL','PostGIS WKT Raster driver',
  1555. 'PostGIS Raster driver')
  1556. enableFeatType = self.dest and self.ogr and db in ('PostgreSQL')
  1557. showText = not(showBrowse or showChoice or showDirbrowse)
  1558. sizer.Show(self.dbWidgets['browse'], show=showBrowse)
  1559. sizer.Show(self.dbWidgets['dirbrowse'], show=showDirbrowse)
  1560. sizer.Show(self.dbWidgets['choice'], show=showChoice)
  1561. sizer.Show(self.dbWidgets['textLabel2'], show=showChoice)
  1562. sizer.Show(self.dbWidgets['text'], show=showText)
  1563. sizer.Show(self.dbWidgets['textLabel1'], show=showText)
  1564. self.dbWidgets['featType'].Enable(enableFeatType)
  1565. if showChoice:
  1566. # try to get list of PG databases
  1567. dbNames = RunCommand('db.databases', quiet=True, read=True,
  1568. driver='pg').splitlines()
  1569. if dbNames is not None:
  1570. self.dbWidgets['choice'].SetItems(sorted(dbNames))
  1571. self.dbWidgets['choice'].SetSelection(0)
  1572. elif grass.find_program('psql', '--help'):
  1573. if not self.dbWidgets['choice'].GetItems():
  1574. p = grass.Popen(['psql', '-ltA'], stdout = grass.PIPE)
  1575. ret = p.communicate()[0]
  1576. if ret:
  1577. dbNames = list()
  1578. for line in ret.splitlines():
  1579. sline = line.split('|')
  1580. if len(sline) < 2:
  1581. continue
  1582. dbname = sline[0]
  1583. if dbname:
  1584. dbNames.append(dbname)
  1585. self.dbWidgets['choice'].SetItems(db)
  1586. self.dbWidgets['choice'].SetSelection(0)
  1587. else:
  1588. sizer.Show(self.dbWidgets['text'])
  1589. sizer.Show(self.dbWidgets['choice'], False)
  1590. sizer.Layout()
  1591. def OnUpdate(self, event):
  1592. """!Update required - load layers."""
  1593. if not self.dest:
  1594. self._reloadLayers()
  1595. event.Skip()
  1596. def _reloadLayers(self):
  1597. """!Reload list of layers"""
  1598. dsn = self.GetDsn()
  1599. if not dsn:
  1600. return
  1601. data = list()
  1602. layerId = 1
  1603. if self.ogr:
  1604. ret = RunCommand('v.external',
  1605. quiet = True,
  1606. read = True,
  1607. flags = 't',
  1608. dsn = dsn)
  1609. if not ret:
  1610. self.reloadDataRequired.emit(data=None)
  1611. return
  1612. layerId = 1
  1613. for line in ret.splitlines():
  1614. layerName, featureType, projection = map(lambda x: x.strip(), line.split(','))
  1615. if projection == '0':
  1616. projectionMatch = _("No")
  1617. else:
  1618. projectionMatch = _("Yes")
  1619. grassName = GetValidLayerName(layerName)
  1620. data.append((layerId, layerName, featureType, projectionMatch, grassName))
  1621. layerId += 1
  1622. else:
  1623. if self._sourceType == 'file':
  1624. baseName = os.path.basename(dsn)
  1625. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1626. data.append((layerId, baseName, grassName))
  1627. elif self._sourceType == 'dir':
  1628. ext = self.dirWidgets['extension'].GetValue()
  1629. for filename in glob.glob(os.path.join(dsn, "%s") % self._getExtPatternGlob(ext)):
  1630. baseName = os.path.basename(filename)
  1631. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1632. data.append((layerId, baseName, grassName))
  1633. layerId += 1
  1634. # unused code since r47938
  1635. # if self.ogr:
  1636. # dsn += '@OGR'
  1637. #
  1638. # evt = wxGdalSelect(dsn = dsn)
  1639. # evt.SetId(self.input[self.dsnType][1].GetId())
  1640. # wx.PostEvent(self.parent, evt)
  1641. if self.parent.GetName() == 'MultiImportDialog':
  1642. self.reloadDataRequired.emit(data=data)
  1643. def ExtensionChanged(self, event):
  1644. if not self.dest:
  1645. # reload layers
  1646. self._reloadLayers()
  1647. def SetExtension(self, name):
  1648. """!Extension changed"""
  1649. ext = self._getExtension(name)
  1650. self.dirWidgets['extension'].SetValue(ext)
  1651. def GetType(self):
  1652. """!Get source type"""
  1653. return self._sourceType
  1654. def GetFormat(self):
  1655. """!Get format as string"""
  1656. if self._sourceType == 'dir':
  1657. format = self.dirWidgets['format'].GetStringSelection()
  1658. elif self._sourceType == 'pro':
  1659. format = self.protocolWidgets['format'].GetStringSelection()
  1660. elif self._sourceType == 'db':
  1661. format = self.dbWidgets['format'].GetStringSelection()
  1662. else:
  1663. format = ''
  1664. return format.replace(' ', '_')
  1665. def GetFormatExt(self):
  1666. """!Get format extension"""
  1667. return self._getExtension(self.GetFormat())
  1668. def GetOptions(self):
  1669. """!Get creation options"""
  1670. if self._sourceType == 'file':
  1671. options = self.fileWidgets['options'].GetValue()
  1672. elif self._sourceType == 'dir':
  1673. options = self.dirWidgets['options'].GetValue()
  1674. elif self._sourceType == 'pro':
  1675. options = self.protocolWidgets['options'].GetValue()
  1676. elif self._sourceType == 'db':
  1677. if self.dbWidgets['featType'].GetSelection() == 1:
  1678. options = 'topology=yes '
  1679. else:
  1680. options = ''
  1681. options += self.dbWidgets['options'].GetValue()
  1682. return options.strip()
  1683. def OnHelp(self, event):
  1684. """!Show related manual page"""
  1685. cmd = ''
  1686. if self.dest:
  1687. if self.ogr:
  1688. cmd = 'v.external.out'
  1689. else:
  1690. cmd = 'r.external.out'
  1691. else:
  1692. if self.link:
  1693. if self.ogr:
  1694. cmd = 'v.external'
  1695. else:
  1696. cmd = 'r.external'
  1697. else:
  1698. if self.ogr:
  1699. cmd = 'v.in.ogr'
  1700. else:
  1701. cmd = 'r.in.gdal'
  1702. RunCommand('g.manual', entry = cmd)
  1703. class ProjSelect(wx.ComboBox):
  1704. """!Widget for selecting input raster/vector map used by
  1705. r.proj/v.proj modules."""
  1706. def __init__(self, parent, isRaster, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  1707. **kwargs):
  1708. super(ProjSelect, self).__init__(parent, id, size = size,
  1709. style = wx.CB_READONLY, **kwargs)
  1710. self.SetName("ProjSelect")
  1711. self.isRaster = isRaster
  1712. def UpdateItems(self, dbase, location, mapset):
  1713. """!Update list of maps
  1714. """
  1715. if not dbase:
  1716. dbase = grass.gisenv()['GISDBASE']
  1717. if not mapset:
  1718. mapset = grass.gisenv()['MAPSET']
  1719. if self.isRaster:
  1720. ret = RunCommand('r.proj',
  1721. quiet = True,
  1722. read = True,
  1723. flags = 'l',
  1724. dbase = dbase,
  1725. location = location,
  1726. mapset = mapset)
  1727. else:
  1728. ret = RunCommand('v.proj',
  1729. quiet = True,
  1730. read = True,
  1731. flags = 'l',
  1732. dbase = dbase,
  1733. location = location,
  1734. mapset = mapset)
  1735. listMaps = list()
  1736. if ret:
  1737. for line in ret.splitlines():
  1738. listMaps.append(line.strip())
  1739. ListSortLower(listMaps)
  1740. self.SetItems(listMaps)
  1741. self.SetValue('')
  1742. class ElementSelect(wx.Choice):
  1743. def __init__(self, parent, id = wx.ID_ANY, size = globalvar.DIALOG_COMBOBOX_SIZE,
  1744. **kwargs):
  1745. """!Widget for selecting GIS element
  1746. @param parent parent window
  1747. """
  1748. super(ElementSelect, self).__init__(parent, id, size = size,
  1749. **kwargs)
  1750. self.SetName("ElementSelect")
  1751. task = gtask.parse_interface('g.list')
  1752. p = task.get_param(value = 'type')
  1753. self.values = p.get('values', [])
  1754. self.valuesDesc = p.get('values_desc', [])
  1755. self.SetItems(self.valuesDesc)
  1756. def GetValue(self, name):
  1757. """!Translate value
  1758. @param name element name
  1759. """
  1760. idx = self.valuesDesc.index(name)
  1761. if idx > -1:
  1762. return self.values[idx]
  1763. return ''
  1764. class OgrTypeSelect(wx.Panel):
  1765. def __init__(self, parent, panel, **kwargs):
  1766. """!Widget to choose OGR feature type
  1767. @param parent parent window
  1768. @param panel wx.Panel instance used as parent window
  1769. """
  1770. wx.Panel.__init__(self, parent = panel, id = wx.ID_ANY)
  1771. self.ftype = wx.Choice(parent = self, id = wx.ID_ANY,
  1772. size = (200, -1),
  1773. choices = (_("Point"), _("LineString"), _("Polygon")))
  1774. self._layout()
  1775. def _layout(self):
  1776. """!Do layout"""
  1777. sizer = wx.BoxSizer(wx.HORIZONTAL)
  1778. sizer.Add(item = wx.StaticText(parent = self,
  1779. id = wx.ID_ANY,
  1780. label = _("Feature type:")),
  1781. proportion = 1,
  1782. flag = wx.ALIGN_CENTER_VERTICAL,
  1783. border = 5)
  1784. sizer.Add(item = self.ftype,
  1785. proportion = 0,
  1786. flag = wx.EXPAND | wx.ALIGN_RIGHT)
  1787. self.SetSizer(sizer)
  1788. sizer.Fit(self)
  1789. def GetType(self):
  1790. """!Get selected type as string
  1791. @return feature type as string
  1792. """
  1793. sel = self.ftype.GetSelection()
  1794. if sel == 0:
  1795. return 'point'
  1796. elif sel == 1:
  1797. return 'line'
  1798. elif sel == 2:
  1799. return 'boundary'
  1800. class CoordinatesSelect(wx.Panel):
  1801. def __init__(self, parent, giface, multiple = False, **kwargs):
  1802. """!Widget to get coordinates from map window by mouse click
  1803. @param parent parent window
  1804. @param giface GRASS interface
  1805. @param multiple - True if it is possible to insert more coordinates
  1806. """
  1807. self._giface = giface
  1808. self.multiple = multiple
  1809. self.mapWin = None
  1810. super(CoordinatesSelect, self).__init__(parent = parent, id = wx.ID_ANY)
  1811. self.coordsField = wx.TextCtrl(parent = self, id = wx.ID_ANY,
  1812. size = globalvar.DIALOG_TEXTCTRL_SIZE)
  1813. icon = wx.Bitmap(os.path.join(globalvar.ETCICONDIR, "grass", "pointer.png"))
  1814. self.buttonInsCoords = buttons.ThemedGenBitmapToggleButton(parent = self, id = wx.ID_ANY,
  1815. bitmap = icon,
  1816. size = globalvar.DIALOG_COLOR_SIZE)
  1817. self.registered = False
  1818. self.buttonInsCoords.Bind(wx.EVT_BUTTON, self._onClick)
  1819. switcher = self._giface.GetMapDisplay().GetToolSwitcher()
  1820. switcher.AddCustomToolToGroup(group='mouseUse',
  1821. btnId=self.buttonInsCoords.GetId(),
  1822. toggleHandler=self.buttonInsCoords.SetValue)
  1823. self._doLayout()
  1824. def _doLayout(self):
  1825. self.dialogSizer = wx.BoxSizer(wx.HORIZONTAL)
  1826. self.dialogSizer.Add(item = self.coordsField,
  1827. proportion = 1,
  1828. flag = wx.EXPAND)
  1829. self.dialogSizer.Add(item = self.buttonInsCoords)
  1830. self.SetSizer(self.dialogSizer)
  1831. def _onClick(self, event):
  1832. """!Button for interacitve inserting of coordinates clicked"""
  1833. switcher = self._giface.GetMapDisplay().GetToolSwitcher()
  1834. switcher.ToolChanged(self.buttonInsCoords.GetId())
  1835. self.mapWin = self._giface.GetMapWindow()
  1836. if self.buttonInsCoords.GetToggle() and self.mapWin:
  1837. if self.mapWin.RegisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1838. self._onMapClickHandler,
  1839. 'cross') == False:
  1840. return
  1841. self.registered = True
  1842. self._giface.GetMapDisplay().Raise()
  1843. else:
  1844. if self.mapWin and \
  1845. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1846. self._onMapClickHandler):
  1847. self.registered = False
  1848. return
  1849. def _onMapClickHandler(self, event):
  1850. """!Gets coordinates from mapwindow"""
  1851. if event == "unregistered":
  1852. return
  1853. e, n = self.mapWin.GetLastEN()
  1854. prevCoords = ""
  1855. if self.multiple:
  1856. prevCoords = self.coordsField.GetValue().strip()
  1857. if prevCoords != "":
  1858. prevCoords += ","
  1859. value = prevCoords + str(e) + "," + str(n)
  1860. self.coordsField.SetValue(value)
  1861. def OnClose(self):
  1862. """!Unregistrates _onMapClickHandler from mapWin"""
  1863. switcher = self._giface.GetMapDisplay().GetToolSwitcher()
  1864. switcher.RemoveCustomToolFromGroup(self.buttonInsCoords.GetId())
  1865. if self.mapWin and self.registered:
  1866. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1867. self._onMapClickHandler)
  1868. def GetTextWin(self):
  1869. """!Get TextCtrl widget"""
  1870. return self.coordsField
  1871. class SignatureSelect(wx.ComboBox):
  1872. """!Widget for selecting signatures"""
  1873. def __init__(self, parent, element, id = wx.ID_ANY, size = globalvar.DIALOG_GSELECT_SIZE,
  1874. **kwargs):
  1875. super(SignatureSelect, self).__init__(parent, id, size = size,
  1876. **kwargs)
  1877. self.element = element
  1878. self.SetName("SignatureSelect")
  1879. def Insert(self, group, subgroup = None):
  1880. """!Insert signatures for defined group/subgroup
  1881. @param group group name (can be fully-qualified)
  1882. @param subgroup non fully-qualified name of subgroup
  1883. """
  1884. if not group:
  1885. return
  1886. gisenv = grass.gisenv()
  1887. try:
  1888. name, mapset = group.split('@', 1)
  1889. except ValueError:
  1890. name = group
  1891. mapset = gisenv['MAPSET']
  1892. path = os.path.join(gisenv['GISDBASE'], gisenv['LOCATION_NAME'], mapset,
  1893. 'group', name)
  1894. if subgroup:
  1895. path = os.path.join(path, 'subgroup', subgroup)
  1896. try:
  1897. items = list()
  1898. for element in os.listdir(os.path.join(path, self.element)):
  1899. items.append(element)
  1900. self.SetItems(items)
  1901. except OSError:
  1902. self.SetItems([])
  1903. self.SetValue('')