gselect.py 89 KB

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