dialogs.py 118 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860
  1. """!
  2. @package vnet.dialogs
  3. @brief Dialog for vector network analysis front-end
  4. Classes:
  5. - dialogs::VNETDialog
  6. - dialogs::PtsList
  7. - dialogs::SettingsDialog
  8. - dialogs::AddLayerDialog
  9. - dialogs::VnetTmpVectMaps
  10. - dialogs::VectMap
  11. - dialogs::History
  12. - dialogs::VnetStatusbar
  13. (C) 2012 by the GRASS Development Team
  14. This program is free software under the GNU General Public License
  15. (>=v2). Read the file COPYING that comes with GRASS for details.
  16. @author Stepan Turek <stepan.turek seznam.cz> (GSoC 2012, mentor: Martin Landa)
  17. """
  18. import os
  19. import sys
  20. import types
  21. try:
  22. import grass.lib.vector as vectlib
  23. from ctypes import pointer, byref, c_char_p, c_int, c_double
  24. haveCtypes = True
  25. except ImportError:
  26. haveCtypes = False
  27. from copy import copy
  28. from grass.script import core as grass
  29. import wx
  30. import wx.aui
  31. import wx.lib.flatnotebook as FN
  32. import wx.lib.colourselect as csel
  33. from core import globalvar, utils
  34. from core.gcmd import RunCommand, GMessage
  35. from core.events import gUpdateMap
  36. from core.settings import UserSettings
  37. from dbmgr.base import DbMgrBase
  38. from gui_core.widgets import GNotebook
  39. from gui_core.goutput import GConsoleWindow
  40. from core.gconsole import CmdThread, EVT_CMD_DONE, GConsole
  41. from gui_core.gselect import Select, LayerSelect, ColumnSelect
  42. from vnet.widgets import PointsList
  43. from vnet.toolbars import MainToolbar, PointListToolbar, AnalysisToolbar
  44. #Main TODOs
  45. # - when layer tree of is changed, tmp result map is removed from render list
  46. # - optimization of map drawing
  47. # - tmp maps - add number of process
  48. # - destructor problem - when GRASS GIS is closed with open VNETDialog,
  49. # it's destructor is not called
  50. class VNETDialog(wx.Dialog):
  51. def __init__(self, parent, giface,
  52. id = wx.ID_ANY, title = _("GRASS GIS Vector Network Analysis Tool"),
  53. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  54. """!Dialog for vector network analysis"""
  55. wx.Dialog.__init__(self, parent, id, style=style, title = title, **kwargs)
  56. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass.ico'), wx.BITMAP_TYPE_ICO))
  57. self.parent = parent # mapdisp.frame MapFrame class
  58. self.mapWin = parent.MapWindow
  59. self._giface = giface
  60. # contains current analysis result (do not have to be last one, when history is browsed),
  61. # it is instance of VectMap class
  62. self.tmp_result = None
  63. # initialization of History class used for saving and reading data from file
  64. # it is used for browsing analysis results
  65. self.history = History(self)
  66. # variable, which appends unique number to every name of map, which is saved into history
  67. self.histTmpVectMapNum = 0
  68. # list of maps, which will be saved into history
  69. self.tmpVectMapsToHist = []
  70. # this class instance manages all temporary vector maps created during life of VNETDialog
  71. self.tmpMaps = VnetTmpVectMaps(parent = self)
  72. # setting initialization
  73. self._initSettings()
  74. # registration graphics for drawing
  75. self.pointsToDraw = self.mapWin.RegisterGraphicsToDraw(graphicsType = "point",
  76. setStatusFunc = self.SetPointStatus)
  77. self.SetPointDrawSettings()
  78. # information, whether mouse event handler is registered in map window
  79. self.handlerRegistered = False
  80. # get attribute table columns only with numbers (for cost columns in vnet analysis)
  81. self.columnTypes = ['integer', 'double precision']
  82. # initialization of v.net.* analysis parameters (data which characterizes particular analysis)
  83. self._initVnetParams()
  84. # toobars
  85. self.toolbars = {}
  86. self.toolbars['mainToolbar'] = MainToolbar(parent = self)
  87. self.toolbars['analysisToolbar'] = AnalysisToolbar(parent = self)
  88. #
  89. # Fancy gui
  90. #
  91. self._mgr = wx.aui.AuiManager(self)
  92. # Columns in points list
  93. # TODO init dynamically, not hard typed v.net.path
  94. self.cols = [
  95. ['type', _('type'), [_(""), _("Start point"), _("End point")], 0],
  96. ['topology', _('topology'), None, ""]
  97. ]
  98. self.mainPanel = wx.Panel(parent=self)
  99. self.notebook = GNotebook(parent = self.mainPanel,
  100. style = FN.FNB_FANCY_TABS | FN.FNB_BOTTOM |
  101. FN.FNB_NO_X_BUTTON)
  102. # statusbar
  103. self.stPriorities = {'important' : 5, 'iformation' : 1}
  104. self.stBar = VnetStatusbar(parent = self.mainPanel, style = 0)
  105. self.stBar.SetFieldsCount(number = 1)
  106. # Create tabs
  107. # Stores all data related to snapping
  108. self.snapData = {}
  109. self.snapData['snap_mode'] = False
  110. # Stores widgets which sets some of analysis parameters (e. g. v.ne.iso -> iso lines)
  111. self.anSettings = {}
  112. self._createPointsPage()
  113. # stores widgets which are shown on parameters page
  114. # they set data, on which analysis will be done
  115. self.inputData = {}
  116. self._createParametersPage()
  117. # Output console for analysis
  118. self._createOutputPage()
  119. # Stores data which are needed for attribute table browser of analysis input map layers
  120. self.inpDbMgrData = {}
  121. self._createInputDbMgrPage()
  122. # Stores data which are need for attribute table browser of analysis result map layers
  123. self.resultDbMgrData = {}
  124. self._createResultDbMgrPage()
  125. self.notebook.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnPageChanged)
  126. self.Bind(wx.EVT_CLOSE, self.OnCloseDialog)
  127. self._addPanes()
  128. self._doVnetDialogLayout()
  129. self._mgr.Update()
  130. # adds 2 points into list
  131. for i in range(2):
  132. self.list.AddItem()
  133. self.list.EditCellIndex(i, 'type', i + 1)
  134. self.list.CheckItem(i, True)
  135. # selects first point
  136. self.list.selected = 0
  137. self.list.Select(self.list.selected)
  138. dlgSize = (420, 560)
  139. self.SetMinSize(dlgSize)
  140. self.SetInitialSize(dlgSize)
  141. #fix goutput's pane size (required for Mac OSX)
  142. if self.gwindow:
  143. self.gwindow.SetSashPosition(int(self.GetSize()[1] * .75))
  144. self.OnAnalysisChanged(None)
  145. self.notebook.SetSelectionByName("parameters")
  146. self.toolbars['analysisToolbar'].SetMinSize((-1, self.toolbars['mainToolbar'].GetSize()[1]))
  147. def __del__(self):
  148. """!Removes temp layers, unregisters handlers and graphics"""
  149. update = self.tmpMaps.DeleteAllTmpMaps()
  150. self.mapWin.UnregisterGraphicsToDraw(self.pointsToDraw)
  151. if self.handlerRegistered:
  152. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  153. self.OnMapClickHandler)
  154. if update:
  155. up_map_evt = gUpdateMap(render = True, renderVector = True)
  156. wx.PostEvent(self.mapWin, up_map_evt)
  157. else:
  158. up_map_evt = gUpdateMap(render = True, renderVector = True)
  159. wx.PostEvent(self.mapWin, up_map_evt)
  160. def _addPanes(self):
  161. """!Adds toolbar pane and pane with tabs"""
  162. self._mgr.AddPane(self.toolbars['mainToolbar'],
  163. wx.aui.AuiPaneInfo().
  164. Name("pointlisttools").Caption(_("Point list toolbar")).
  165. ToolbarPane().Top().Row(0).
  166. Dockable(False).
  167. CloseButton(False).Layer(0))
  168. self._mgr.AddPane(self.toolbars['analysisToolbar'],
  169. wx.aui.AuiPaneInfo().
  170. Name("analysisTools").Caption(_("Analysis toolbar")).
  171. ToolbarPane().Top().Row(1).
  172. Dockable(False).
  173. CloseButton(False).Layer(0))
  174. self._mgr.AddPane(self.mainPanel,
  175. wx.aui.AuiPaneInfo().
  176. Name("tabs").CaptionVisible(visible = False).
  177. Center().
  178. Dockable(False).
  179. CloseButton(False).Layer(0))
  180. def _doVnetDialogLayout(self):
  181. sizer = wx.BoxSizer(wx.VERTICAL)
  182. sizer.Add(item = self.notebook, proportion = 1,
  183. flag = wx.EXPAND)
  184. sizer.Add(item = self.stBar, proportion = 0)
  185. self.mainPanel.SetSizer(sizer)
  186. sizer.Fit(self)
  187. self.Layout()
  188. def _createPointsPage(self):
  189. """!Tab with points list and analysis settings"""
  190. pointsPanel = wx.Panel(parent = self)
  191. maxValue = 1e8
  192. listBox = wx.StaticBox(parent = pointsPanel, id = wx.ID_ANY,
  193. label =" %s " % _("Points for analysis:"))
  194. self.notebook.AddPage(page = pointsPanel,
  195. text=_('Points'),
  196. name = 'points')
  197. self.list = PtsList(parent = pointsPanel, dialog = self, cols = self.cols)
  198. self.toolbars['pointsList'] = PointListToolbar(parent = pointsPanel, list = self.list)
  199. anSettingsPanel = wx.Panel(parent = pointsPanel)
  200. anSettingsBox = wx.StaticBox(parent = anSettingsPanel, id = wx.ID_ANY,
  201. label =" %s " % _("Analysis settings:"))
  202. maxDistPanel = wx.Panel(parent = anSettingsPanel)
  203. maxDistLabel = wx.StaticText(parent = maxDistPanel, id = wx.ID_ANY, label = _("Maximum distance of point to the network:"))
  204. self.anSettings["max_dist"] = wx.SpinCtrl(parent = maxDistPanel, id = wx.ID_ANY, min = 0, max = maxValue)
  205. self.anSettings["max_dist"].SetValue(100000) #TODO init val
  206. #showCutPanel = wx.Panel(parent = anSettingsPanel)
  207. #self.anSettings["show_cut"] = wx.CheckBox(parent = showCutPanel, id=wx.ID_ANY,
  208. # label = _("Show minimal cut"))
  209. #self.anSettings["show_cut"].Bind(wx.EVT_CHECKBOX, self.OnShowCut)
  210. isoLinesPanel = wx.Panel(parent = anSettingsPanel)
  211. isoLineslabel = wx.StaticText(parent = isoLinesPanel, id = wx.ID_ANY, label = _("Iso lines:"))
  212. self.anSettings["iso_lines"] = wx.TextCtrl(parent = isoLinesPanel, id = wx.ID_ANY)
  213. self.anSettings["iso_lines"].SetValue("1000,2000,3000")
  214. # Layout
  215. AnalysisSizer = wx.BoxSizer(wx.VERTICAL)
  216. listSizer = wx.StaticBoxSizer(listBox, wx.VERTICAL)
  217. listSizer.Add(item = self.toolbars['pointsList'], proportion = 0)
  218. listSizer.Add(item = self.list, proportion = 1, flag = wx.EXPAND)
  219. anSettingsSizer = wx.StaticBoxSizer(anSettingsBox, wx.VERTICAL)
  220. maxDistSizer = wx.BoxSizer(wx.HORIZONTAL)
  221. maxDistSizer.Add(item = maxDistLabel, flag = wx.ALIGN_CENTER_VERTICAL, proportion = 1)
  222. maxDistSizer.Add(item = self.anSettings["max_dist"],
  223. flag = wx.EXPAND | wx.ALL, border = 5, proportion = 0)
  224. maxDistPanel.SetSizer(maxDistSizer)
  225. anSettingsSizer.Add(item = maxDistPanel, proportion = 1, flag = wx.EXPAND)
  226. #showCutSizer = wx.BoxSizer(wx.HORIZONTAL)
  227. #showCutPanel.SetSizer(showCutSizer)
  228. #showCutSizer.Add(item = self.anSettings["show_cut"],
  229. # flag = wx.EXPAND | wx.ALL, border = 5, proportion = 0)
  230. #anSettingsSizer.Add(item = showCutPanel, proportion = 1, flag = wx.EXPAND)
  231. isoLinesSizer = wx.BoxSizer(wx.HORIZONTAL)
  232. isoLinesSizer.Add(item = isoLineslabel, flag = wx.ALIGN_CENTER_VERTICAL, proportion = 1)
  233. isoLinesSizer.Add(item = self.anSettings["iso_lines"],
  234. flag = wx.EXPAND | wx.ALL, border = 5, proportion = 1)
  235. isoLinesPanel.SetSizer(isoLinesSizer)
  236. anSettingsSizer.Add(item = isoLinesPanel, proportion = 1, flag = wx.EXPAND)
  237. AnalysisSizer.Add(item = listSizer, proportion = 1, flag = wx.EXPAND | wx.ALL, border = 5)
  238. AnalysisSizer.Add(item = anSettingsPanel, proportion = 0, flag = wx.EXPAND | wx.RIGHT | wx.LEFT | wx.BOTTOM, border = 5)
  239. anSettingsPanel.SetSizer(anSettingsSizer)
  240. pointsPanel.SetSizer(AnalysisSizer)
  241. def OnShowCut(self, event):
  242. """!Shows vector map with minimal cut (v.net.flow) - not yet implemented"""
  243. val = event.IsChecked()
  244. if val and self.tmp_result:
  245. self.tmp_result.DeleteRenderLayer()
  246. cmd = self.GetLayerStyle()
  247. self.vnetFlowTmpCut.AddRenderLayer(cmd)
  248. else:
  249. self.vnetFlowTmpCut.DeleteRenderLayer()
  250. cmd = self.GetLayerStyle()
  251. self.tmp_result.AddRenderLayer(cmd)
  252. up_map_evt = gUpdateMap(render = True, renderVector = True)
  253. wx.PostEvent(self.mapWin, up_map_evt)
  254. def _createOutputPage(self):
  255. """!Tab with output console"""
  256. outputPanel = wx.Panel(parent = self)
  257. self.notebook.AddPage(page = outputPanel,
  258. text = _("Output"),
  259. name = 'output')
  260. self.goutput = GConsole(guiparent = self)
  261. self.gwindow = GConsoleWindow(parent = outputPanel, gconsole = self.goutput)
  262. #Layout
  263. outputSizer = wx.BoxSizer(wx.VERTICAL)
  264. outputSizer.Add(item = self.gwindow, proportion = 1, flag = wx.EXPAND)
  265. self.gwindow.SetMinSize((-1,-1))
  266. outputPanel.SetSizer(outputSizer)
  267. def _createParametersPage(self):
  268. """!Tab for selection of data for analysis"""
  269. dataPanel = wx.Panel(parent=self)
  270. self.notebook.AddPage(page = dataPanel,
  271. text=_('Parameters'),
  272. name = 'parameters')
  273. label = {}
  274. dataSelects = [
  275. ['input', "Choose vector map for analysis:", Select],
  276. ['alayer', "Arc layer number or name:", LayerSelect],
  277. ['nlayer', "Node layer number or name:", LayerSelect],
  278. ['afcolumn', self.attrCols['afcolumn']['label'], ColumnSelect],
  279. ['abcolumn', self.attrCols['abcolumn']['label'], ColumnSelect],
  280. ['ncolumn', self.attrCols['ncolumn']['label'], ColumnSelect],
  281. ]
  282. selPanels = {}
  283. for dataSel in dataSelects:
  284. selPanels[dataSel[0]] = wx.Panel(parent = dataPanel)
  285. if dataSel[0] == 'input':
  286. self.inputData[dataSel[0]] = dataSel[2](parent = selPanels[dataSel[0]],
  287. size = (-1, -1),
  288. type = 'vector')
  289. icon = wx.Image(os.path.join(globalvar.ETCICONDIR, "grass", "layer-vector-add.png"))
  290. icon.Rescale(18, 18)
  291. icon = wx.BitmapFromImage(icon)
  292. self.addToTreeBtn = wx.BitmapButton(parent = selPanels[dataSel[0]],
  293. bitmap = icon,
  294. size = globalvar.DIALOG_COLOR_SIZE)
  295. self.addToTreeBtn.SetToolTipString(_("Add vector map into layer tree"))
  296. self.addToTreeBtn.Disable()
  297. self.addToTreeBtn.Bind(wx.EVT_BUTTON, self.OnToTreeBtn)
  298. elif dataSel[0] != 'input':
  299. self.inputData[dataSel[0]] = dataSel[2](parent = selPanels[dataSel[0]],
  300. size = (-1, -1))
  301. label[dataSel[0]] = wx.StaticText(parent = selPanels[dataSel[0]],
  302. name = dataSel[0])
  303. label[dataSel[0]].SetLabel(dataSel[1])
  304. self.inputData['input'].Bind(wx.EVT_TEXT, self.OnVectSel)
  305. self.inputData['alayer'].Bind(wx.EVT_TEXT, self.OnALayerSel)
  306. self.inputData['nlayer'].Bind(wx.EVT_TEXT, self.OnNLayerSel)
  307. # Layout
  308. mainSizer = wx.BoxSizer(wx.VERTICAL)
  309. box = wx.StaticBox(dataPanel, -1, "Vector map and layers for analysis")
  310. bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  311. mainSizer.Add(item = bsizer, proportion = 0,
  312. flag = wx.EXPAND | wx.TOP | wx.LEFT | wx.RIGHT, border = 5)
  313. for sel in ['input', 'alayer', 'nlayer']:
  314. if sel== 'input':
  315. btn = self.addToTreeBtn
  316. else:
  317. btn = None
  318. selPanels[sel].SetSizer(self._doSelLayout(title = label[sel],
  319. sel = self.inputData[sel],
  320. btn = btn))
  321. bsizer.Add(item = selPanels[sel], proportion = 1,
  322. flag = wx.EXPAND)
  323. box = wx.StaticBox(dataPanel, -1, "Costs")
  324. bsizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  325. mainSizer.Add(item = bsizer, proportion = 0,
  326. flag = wx.EXPAND | wx.TOP | wx.LEFT | wx.RIGHT, border = 5)
  327. for sel in ['afcolumn', 'abcolumn', 'ncolumn']:
  328. selPanels[sel].SetSizer(self._doSelLayout(title = label[sel], sel = self.inputData[sel]))
  329. bsizer.Add(item = selPanels[sel], proportion = 0,
  330. flag = wx.EXPAND)
  331. dataPanel.SetSizer(mainSizer)
  332. def _doSelLayout(self, title, sel, btn = None):
  333. # helper function for layout of self.inputData widgets initialized in _createParametersPage
  334. selSizer = wx.BoxSizer(orient = wx.VERTICAL)
  335. selTitleSizer = wx.BoxSizer(wx.HORIZONTAL)
  336. selTitleSizer.Add(item = title, proportion = 1,
  337. flag = wx.LEFT | wx.TOP | wx.EXPAND, border = 5)
  338. selSizer.Add(item = selTitleSizer, proportion = 0,
  339. flag = wx.EXPAND)
  340. if btn:
  341. selFiledSizer = wx.BoxSizer(orient = wx.HORIZONTAL)
  342. selFiledSizer.Add(item = sel, proportion = 1,
  343. flag = wx.EXPAND | wx.ALL)
  344. selFiledSizer.Add(item = btn, proportion = 0,
  345. flag = wx.EXPAND | wx.ALL)
  346. selSizer.Add(item = selFiledSizer, proportion = 0,
  347. flag = wx.EXPAND | wx.ALL| wx.ALIGN_CENTER_VERTICAL,
  348. border = 5)
  349. else:
  350. selSizer.Add(item = sel, proportion = 1,
  351. flag = wx.EXPAND | wx.ALL| wx.ALIGN_CENTER_VERTICAL,
  352. border = 5)
  353. return selSizer
  354. def _createInputDbMgrPage(self):
  355. """!Tab with attribute tables of analysis input layers"""
  356. self.inpDbMgrData['dbMgr'] = DbMgrBase()
  357. selMapName = None
  358. # if selected vector map is in layer tree then set it
  359. if self.mapWin.tree and self.mapWin.tree.layer_selected:
  360. selMapData = self.mapWin.tree.GetPyData(self.mapWin.tree.layer_selected)[0]
  361. if selMapData['type'] == 'vector': # wrap somehow in LayerTree
  362. selMapName = selMapData['maplayer'].name
  363. self.inpDbMgrData['browse'] = self.inpDbMgrData['dbMgr'].CreateDbMgrPage(parent = self.notebook,
  364. pageName = 'browse')
  365. self.inpDbMgrData['browse'].SetTabAreaColour(globalvar.FNPageColor)
  366. self.inpDbMgrData['input'] = None
  367. if selMapName:
  368. self.inputData['input'].SetValue(selMapName)
  369. self.OnVectSel(None)
  370. def _updateInputDbMgrPage(self, show):
  371. """!Show or hide input tables tab"""
  372. if show and self.notebook.GetPageIndexByName('inputDbMgr') == -1:
  373. self.notebook.AddPage(page = self.inpDbMgrData['browse'],
  374. text=_('Input tables'),
  375. name = 'inputDbMgr')
  376. elif not show:
  377. self.notebook.RemovePage(page = 'inputDbMgr')
  378. def _createResultDbMgrPage(self):
  379. """!Tab with attribute tables of analysis result layers"""
  380. self.resultDbMgrData['dbMgr'] = DbMgrBase()
  381. self.resultDbMgrData['browse'] = self.resultDbMgrData['dbMgr'].CreateDbMgrPage(parent = self.notebook,
  382. pageName = 'browse')
  383. self.resultDbMgrData['browse'].SetTabAreaColour(globalvar.FNPageColor)
  384. if self.tmp_result:
  385. self.resultDbMgrData['input'] = self.tmp_result.GetVectMapName()
  386. else:
  387. self.resultDbMgrData['input'] = None
  388. def _updateResultDbMgrPage(self):
  389. """!Show or Hide Result tables tab"""
  390. # analysis, which created result
  391. analysis = self.resultDbMgrData['analysis']
  392. #TODO maybe no need to store this information, just check it has attribute table, if so show it
  393. haveDbMgr = self.vnetParams[analysis]["resultProps"]["dbMgr"]
  394. if haveDbMgr and self.notebook.GetPageIndexByName('resultDbMgr') == -1:
  395. self.notebook.AddPage(page = self.resultDbMgrData['browse'],
  396. text=_('Result tables'),
  397. name = 'resultDbMgr')
  398. elif not haveDbMgr:
  399. self.notebook.RemovePage(page = 'resultDbMgr')
  400. def OnPageChanged(self, event):
  401. """!Tab switched"""
  402. if event.GetEventObject() == self.notebook:
  403. dbMgrIndxs = []
  404. dbMgrIndxs.append(self.notebook.GetPageIndexByName('inputDbMgr'))
  405. dbMgrIndxs.append(self.notebook.GetPageIndexByName('resultDbMgr'))
  406. if self.notebook.GetSelection() in dbMgrIndxs:
  407. self.stBar.AddStatusItem(text = _('Loading tables...'),
  408. key = 'dbMgr',
  409. priority = self.stPriorities['important'])
  410. self._updateDbMgrData()
  411. self.stBar.RemoveStatusItem(key = 'dbMgr')
  412. # update columns (when some is added in input tables browser), TODO needs optimization
  413. elif self.notebook.GetSelection() == self.notebook.GetPageIndexByName('parameters'):
  414. self.OnALayerSel(None)
  415. self.OnNLayerSel(None)
  416. def _updateDbMgrData(self):
  417. """!Updates input/result tables page """
  418. if self.notebook.GetSelection() == self.notebook.GetPageIndexByName('inputDbMgr'):
  419. self._updateInputDbMgrData()
  420. elif self.notebook.GetSelection() == self.notebook.GetPageIndexByName('resultDbMgr'):
  421. self._updateResultDbMgrData()
  422. else:
  423. self.stBar.RemoveStatusItem('manager')
  424. def _updateInputDbMgrData(self):
  425. """!Loads data according to selected layers in Parameters tab"""
  426. inpSel = self.inputData['input'].GetValue().strip()
  427. # changed vector map
  428. if self.inpDbMgrData['input'] != inpSel:
  429. wx.BeginBusyCursor()
  430. self.inpDbMgrData['dbMgr'].ChangeVectorMap(vectorName = inpSel)
  431. self.inpDbMgrData['input'] = inpSel
  432. for layerName in ['alayer', 'nlayer']:
  433. try:
  434. layer = int(self.inputData[layerName].GetValue())
  435. except ValueError:
  436. continue
  437. self.inpDbMgrData['browse'].AddLayer(layer)
  438. wx.EndBusyCursor()
  439. # same vector map
  440. else:
  441. needLayers = []
  442. browseLayers = self.inpDbMgrData['browse'].GetAddedLayers()
  443. for layerName in ['alayer', 'nlayer']:
  444. try:
  445. inpLayer = int(self.inputData[layerName].GetValue())
  446. except ValueError:
  447. continue
  448. if inpLayer in browseLayers:
  449. needLayers.append(inpLayer)
  450. continue
  451. else:
  452. wx.BeginBusyCursor()
  453. self.inpDbMgrData['browse'].AddLayer(inpLayer)
  454. wx.EndBusyCursor()
  455. needLayers.append(inpLayer)
  456. for layer in browseLayers:
  457. if layer not in needLayers:
  458. self.inpDbMgrData['browse'].DeletePage(layer)
  459. def _updateResultDbMgrData(self):
  460. """!Loads data from analysis result map"""
  461. if not self.tmp_result:
  462. return
  463. vectName = self.tmp_result.GetVectMapName()
  464. if self.resultDbMgrData['input'] != vectName:
  465. wx.BeginBusyCursor()
  466. dbMgr = self.resultDbMgrData['dbMgr']
  467. dbMgr.ChangeVectorMap(vectorName = vectName)
  468. for layer in dbMgr.GetVectorLayers():
  469. self.resultDbMgrData['browse'].AddLayer(layer)
  470. self.resultDbMgrData['input'] = vectName
  471. wx.EndBusyCursor()
  472. def OnToTreeBtn(self, event):
  473. """!Add vector map into layer tree"""
  474. vectorMap = self.inputData['input'].GetValue()
  475. vectMapName, mapSet = self._parseMapStr(vectorMap)
  476. vectorMap = vectMapName + '@' + mapSet
  477. existsMap = grass.find_file(name = vectMapName,
  478. element = 'vector',
  479. mapset = mapSet)
  480. if not existsMap["name"]:
  481. return
  482. cmd = ['d.vect',
  483. 'map=' + vectorMap]
  484. if self.mapWin.tree and \
  485. self.mapWin.tree.FindItemByData(key = 'name', value = vectorMap) is None:
  486. self.mapWin.tree.AddLayer(ltype = "vector",
  487. lcmd = cmd,
  488. lname =vectorMap,
  489. lchecked = True)
  490. #d.mon case
  491. else:
  492. self.renderLayer = self.mapWin.Map.AddLayer(ltype = "vector", command = cmd,
  493. name = vectorMap, active = True,
  494. opacity = 1.0, render = True,
  495. pos = -1)
  496. up_map_evt = gUpdateMap(render = True, renderVector = True)
  497. wx.PostEvent(self.mapWin, up_map_evt)
  498. def OnVectSel(self, event):
  499. """!When vector map is selected it populates other comboboxes in Parameters tab (layer selects, columns selects)"""
  500. if self.snapData['snap_mode']:
  501. self.OnSnapping(event = None)
  502. vectMapName, mapSet = self._parseMapStr(self.inputData['input'].GetValue())
  503. vectorMap = vectMapName + '@' + mapSet
  504. self.inputData['alayer'].Clear()
  505. self.inputData['nlayer'].Clear()
  506. self.inputData['alayer'].InsertLayers(vector = vectorMap)
  507. self.inputData['nlayer'].InsertLayers(vector = vectorMap)
  508. items = self.inputData['alayer'].GetItems()
  509. itemsLen = len(items)
  510. if itemsLen < 1:
  511. self.addToTreeBtn.Disable()
  512. if hasattr(self, 'inpDbMgrData'):
  513. self._updateInputDbMgrPage(show = False)
  514. self.inputData['alayer'].SetValue("")
  515. self.inputData['nlayer'].SetValue("")
  516. for sel in ['afcolumn', 'abcolumn', 'ncolumn']:
  517. self.inputData[sel].SetValue("")
  518. return
  519. elif itemsLen == 1:
  520. self.inputData['alayer'].SetSelection(0)
  521. self.inputData['nlayer'].SetSelection(0)
  522. elif itemsLen >= 1:
  523. if unicode("1") in items:
  524. iItem = items.index(unicode("1"))
  525. self.inputData['alayer'].SetSelection(iItem)
  526. if unicode("2") in items:
  527. iItem = items.index(unicode("2"))
  528. self.inputData['nlayer'].SetSelection(iItem)
  529. self.addToTreeBtn.Enable()
  530. if hasattr(self, 'inpDbMgrData'):
  531. self._updateInputDbMgrPage(show = True)
  532. self.OnALayerSel(event)
  533. self.OnNLayerSel(event)
  534. def OnALayerSel(self, event):
  535. """!When arc layer from vector map is selected, populates corespondent columns selects"""
  536. self.inputData['afcolumn'].InsertColumns(vector = self.inputData['input'].GetValue(),
  537. layer = self.inputData['alayer'].GetValue(),
  538. type = self.columnTypes)
  539. self.inputData['abcolumn'].InsertColumns(vector = self.inputData['input'].GetValue(),
  540. layer = self.inputData['alayer'].GetValue(),
  541. type = self.columnTypes)
  542. def OnNLayerSel(self, event):
  543. """!When node layer from vector map is selected, populates corespondent column select"""
  544. if self.snapData['snap_mode']:
  545. self.OnSnapping(event = None)
  546. self.inputData['ncolumn'].InsertColumns(vector = self.inputData['input'].GetValue(),
  547. layer = self.inputData['nlayer'].GetValue(),
  548. type = self.columnTypes)
  549. def _getInvalidInputs(self, inpToTest):
  550. """!Check of analysis input data for invalid values (Parameters tab)"""
  551. # dict of invalid values {key from self.itemData (comboboxes from Parameters tab) : invalid value}
  552. errInput = {}
  553. mapVal = self.inputData['input'].GetValue()
  554. mapName, mapSet = self._parseMapStr(mapVal)
  555. vectMaps = grass.list_grouped('vect')[mapSet]
  556. # check of vector map
  557. if not inpToTest or "input" in inpToTest:
  558. if mapName not in vectMaps:
  559. errInput['input'] = mapVal
  560. # check of arc/node layer
  561. for layerSelName in ['alayer', 'nlayer'] :
  562. if not inpToTest or layerSelName in inpToTest:
  563. layerItems = self.inputData[layerSelName].GetItems()
  564. layerVal = self.inputData[layerSelName].GetValue().strip()
  565. if layerVal not in layerItems:
  566. errInput[layerSelName] = layerVal
  567. # check of costs columns
  568. currModCols = self.vnetParams[self.currAnModule]["cmdParams"]["cols"]
  569. for col, colData in self.attrCols.iteritems():
  570. if not inpToTest or col in inpToTest:
  571. if col not in currModCols:
  572. continue
  573. if "inputField" in self.attrCols[col]:
  574. colInptF = self.attrCols[col]["inputField"]
  575. else:
  576. colInptF = col
  577. if not self.inputData[colInptF].IsShown():
  578. continue
  579. colVal = self.inputData[colInptF].GetValue().strip()
  580. if not colVal:
  581. continue
  582. if colVal != self.inputData[colInptF].GetValue():
  583. errInput[col] = colVal
  584. return errInput
  585. def _parseMapStr(self, vectMapStr):
  586. """!Create full map name (add current mapset if it is not present in name)"""
  587. mapValSpl = vectMapStr.strip().split("@")
  588. if len(mapValSpl) > 1:
  589. mapSet = mapValSpl[1]
  590. else:
  591. mapSet = grass.gisenv()['MAPSET']
  592. mapName = mapValSpl[0]
  593. return mapName, mapSet
  594. def InputsErrorMsgs(self, msg, inpToTest = None):
  595. """!Checks input data in Parameters tab and shows messages if some value is not valid
  596. @param msg (str) - message added to start of message string
  597. @param inpToTest (list) - list of keys from self.itemData map, which says which input data should be checked,
  598. if is empty or None, all input data are checked
  599. @return True - if checked inputs are OK
  600. @return False - if some of checked inputs is not ok
  601. """
  602. errInput = self._getInvalidInputs(inpToTest)
  603. errMapStr = ""
  604. if errInput.has_key('input'):
  605. self.notebook.SetSelectionByName("parameters")
  606. if errInput['input']:
  607. errMapStr = _("Vector map '%s' does not exist.") % (errInput['input'])
  608. else:
  609. errMapStr = _("Vector map was not chosen.")
  610. if errMapStr:
  611. GMessage(parent = self,
  612. message = msg + "\n" + errMapStr)
  613. return False
  614. errLayerStr = ""
  615. for layer, layerLabel in {'alayer' : _("arc layer"),
  616. 'nlayer' : _("node layer")}.iteritems():
  617. if errInput.has_key(layer):
  618. if errInput[layer]:
  619. errLayerStr += _("Chosen %s '%s' does not exist in vector map '%s'.\n") % \
  620. (layerLabel, self.inputData[layer].GetValue(), self.inputData['input'].GetValue())
  621. else:
  622. errLayerStr += _("Choose existing %s.\n") % \
  623. (layerLabel)
  624. if errLayerStr:
  625. GMessage(parent = self,
  626. message = msg + "\n" + errLayerStr)
  627. return False
  628. errColStr = ""
  629. for col, colData in self.attrCols.iteritems():
  630. if col in errInput.iterkeys():
  631. errColStr += _("Chosen column '%s' does not exist in attribute table of layer '%s' of vector map '%s'.\n") % \
  632. (errInput[col], self.inputData[layer].GetValue(), self.inputData['input'].GetValue())
  633. if errColStr:
  634. self.notebook.SetSelectionByName("parameters")
  635. GMessage(parent = self,
  636. message = msg + "\n" + errColStr)
  637. return False
  638. return True
  639. def OnCloseDialog(self, event):
  640. """!Cancel dialog"""
  641. self.parent.dialogs['vnet'] = None
  642. self.Destroy()
  643. def SetPointStatus(self, item, itemIndex):
  644. """!Before point is drawn, decides properties of drawing style"""
  645. key = self.list.GetItemData(itemIndex)
  646. cats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  647. if key == self.list.selected:
  648. wxPen = "selected"
  649. elif not self.list.IsChecked(key):
  650. wxPen = "unused"
  651. item.hide = False
  652. elif len(cats) > 1:
  653. idx = self.list.GetCellSelIdx(key, 'type')
  654. if idx == 2: #End/To/Sink point
  655. wxPen = "used2cat"
  656. else:
  657. wxPen = "used1cat"
  658. else:
  659. wxPen = "used1cat"
  660. item.SetPropertyVal('label', str(itemIndex + 1))
  661. item.SetPropertyVal('penName', wxPen)
  662. def OnMapClickHandler(self, event):
  663. """!Take coordinates from map window"""
  664. if event == 'unregistered':
  665. ptListToolbar = self.toolbars['pointsList']
  666. if ptListToolbar:
  667. ptListToolbar.ToggleTool( id = ptListToolbar.GetToolId("insertPoint"),
  668. toggle = False)
  669. self.handlerRegistered = False
  670. return
  671. self.notebook.SetSelectionByName("points")
  672. if not self.list.itemDataMap:
  673. self.list.AddItem(None)
  674. e, n = self.mapWin.GetLastEN()
  675. index = self.list.selected
  676. key = self.list.GetItemData(index)
  677. if self.snapData['snap_mode']:
  678. coords = [e, n]
  679. if self._snapPoint(coords):
  680. msg = ("snapped to node")
  681. else:
  682. msg = _("new point")
  683. e = coords[0]
  684. n = coords[1]
  685. else:
  686. msg = _("new point")
  687. self.list.EditCellKey(key = self.list.selected ,
  688. colName = 'topology',
  689. cellData = msg)
  690. self.pointsToDraw.GetItem(key).SetCoords([e, n])
  691. if self.list.selected == self.list.GetItemCount() - 1:
  692. self.list.selected = 0
  693. else:
  694. self.list.selected += 1
  695. self.list.Select(self.list.selected)
  696. def _snapPoint(self, coords):
  697. """!Find nearest node to click coordinates (within given threshold)"""
  698. e = coords[0]
  699. n = coords[1]
  700. # compute threshold
  701. snapTreshPix = int(UserSettings.Get(group ='vnet',
  702. key = 'other',
  703. subkey = 'snap_tresh'))
  704. res = max(self.mapWin.Map.region['nsres'], self.mapWin.Map.region['ewres'])
  705. snapTreshDist = snapTreshPix * res
  706. vectorMap = self.inputData['input'].GetValue()
  707. vectMapName, mapSet = self._parseMapStr(vectorMap)
  708. inpMapExists = grass.find_file(name = vectMapName,
  709. element = 'vector',
  710. mapset = mapSet)
  711. if not inpMapExists['name']:
  712. return False
  713. openedMap = pointer(vectlib.Map_info())
  714. ret = vectlib.Vect_open_old2(openedMap,
  715. c_char_p(vectMapName),
  716. c_char_p(mapSet),
  717. c_char_p(self.inputData['alayer'].GetValue()))
  718. if ret == 1:
  719. vectlib.Vect_close(openedMap)
  720. if ret != 2:
  721. return False
  722. nodeNum = vectlib.Vect_find_node(openedMap,
  723. c_double(e),
  724. c_double(n),
  725. c_double(0),
  726. c_double(snapTreshDist),
  727. vectlib.WITHOUT_Z)
  728. if nodeNum > 0:
  729. e = c_double(0)
  730. n = c_double(0)
  731. vectlib.Vect_get_node_coor(openedMap,
  732. nodeNum,
  733. byref(e),
  734. byref(n),
  735. None); # z
  736. e = e.value
  737. n = n.value
  738. else:
  739. vectlib.Vect_close(openedMap)
  740. return False
  741. coords[0] = e
  742. coords[1] = n
  743. return True
  744. def OnAnalyze(self, event):
  745. """!Called when network analysis is started"""
  746. # Check of parameters for analysis
  747. if not self.InputsErrorMsgs(msg = _("Analysis can not be done.")):
  748. return
  749. if self.tmp_result:
  750. self.tmp_result.DeleteRenderLayer()
  751. # history - delete data in buffer for hist step
  752. self.history.DeleteNewHistStepData()
  753. # empty list for maps to be saved to history
  754. self.tmpVectMapsToHist= []
  755. # create new map (included to history) for result of analysis
  756. self.tmp_result = self.NewTmpVectMapToHist('vnet_tmp_result')
  757. if not self.tmp_result:
  758. return
  759. self.stBar.AddStatusItem(text = _('Analysing...'),
  760. key = 'analyze',
  761. priority = self.stPriorities['important'])
  762. # for case there is some map with same name
  763. # (when analysis does not produce any map, this map would have been shown as result)
  764. RunCommand('g.remove',
  765. vect = self.tmp_result.GetVectMapName())
  766. # save data from
  767. self._saveAnInputToHist()
  768. self.resultDbMgrData['analysis'] = self.currAnModule
  769. # Creates part of cmd fro analysis
  770. cmdParams = [self.currAnModule]
  771. cmdParams.extend(self._getInputParams())
  772. cmdParams.append("output=" + self.tmp_result.GetVectMapName())
  773. catPts = self._getPtByCat()
  774. if self.currAnModule == "v.net.path":
  775. self._vnetPathRunAn(cmdParams, catPts)
  776. else:
  777. self._runAn(cmdParams, catPts)
  778. def _vnetPathRunAn(self, cmdParams, catPts):
  779. """!Called when analysis is run for v.net.path module"""
  780. if len(self.pointsToDraw.GetAllItems()) < 1:
  781. return
  782. cats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  783. cmdPts = []
  784. for cat in cats:
  785. if len(catPts[cat[0]]) < 1:
  786. GMessage(parent = self,
  787. message=_("Please choose '%s' and '%s' point.") % (cats[0][1], cats[1][1]))
  788. return
  789. cmdPts.append(catPts[cat[0]][0])
  790. resId = 1
  791. inpPoints = str(resId) + " " + str(cmdPts[0][0]) + " " + str(cmdPts[0][1]) + \
  792. " " + str(cmdPts[1][0]) + " " + str(cmdPts[1][1])
  793. self.coordsTmpFile = grass.tempfile()
  794. coordsTmpFileOpened = open(self.coordsTmpFile, 'w')
  795. coordsTmpFileOpened.write(inpPoints)
  796. coordsTmpFileOpened.close()
  797. cmdParams.append("file=" + self.coordsTmpFile)
  798. cmdParams.append("dmax=" + str(self.anSettings["max_dist"].GetValue()))
  799. cmdParams.append("input=" + self.inputData['input'].GetValue())
  800. cmdParams.append("--overwrite")
  801. self._prepareCmd(cmd = cmdParams)
  802. self.goutput.RunCmd(command = cmdParams, onDone = self._vnetPathRunAnDone)
  803. def _vnetPathRunAnDone(self, cmd, returncode):
  804. """!Called when v.net.path analysis is done"""
  805. grass.try_remove(self.coordsTmpFile)
  806. self._saveHistStep()
  807. self.tmp_result.SaveVectMapState()
  808. self._updateResultDbMgrPage()
  809. self._updateDbMgrData()
  810. cmd = self.GetLayerStyle()
  811. self.tmp_result.AddRenderLayer(cmd)
  812. up_map_evt = gUpdateMap(render = True, renderVector = True)
  813. wx.PostEvent(self.mapWin, up_map_evt)
  814. mainToolbar = self.toolbars['mainToolbar']
  815. id = vars(mainToolbar)['showResult']
  816. mainToolbar.ToggleTool(id =id,
  817. toggle = True)
  818. self.stBar.RemoveStatusItem(key = 'analyze')
  819. def _runAn(self, cmdParams, catPts):
  820. """!Called for all v.net.* analysis (except v.net.path)"""
  821. cats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  822. if len(cats) > 1:
  823. for cat in cats:
  824. if len(catPts[cat[0]]) < 1:
  825. GMessage(parent = self,
  826. message = _("Please choose '%s' and '%s' point.") \
  827. % (cats[0][1], cats[1][1]))
  828. return
  829. else:
  830. for cat in cats:
  831. if len(catPts[cat[0]]) < 2:
  832. GMessage(parent = self,
  833. message = _("Please choose at least two points."))
  834. return
  835. # TODO add also to thread for analysis?
  836. vcatResult = RunCommand("v.category",
  837. input = self.inputData['input'].GetValue(),
  838. option = "report",
  839. flags = "g",
  840. read = True)
  841. vcatResult = vcatResult.splitlines()
  842. for cat in vcatResult:#TODO
  843. cat = cat.split()
  844. if "all" in cat:
  845. maxCat = int(cat[4])
  846. break
  847. layerNum = self.inputData["nlayer"].GetValue().strip()
  848. pt_ascii, catsNums = self._getAsciiPts (catPts = catPts,
  849. maxCat = maxCat,
  850. layerNum = layerNum)
  851. self.tmpPtsAsciiFile = grass.tempfile()#TODO better tmp files cleanup (make class for managing tmp files)
  852. tmpPtsAsciiFileOpened = open(self.tmpPtsAsciiFile, 'w')
  853. tmpPtsAsciiFileOpened.write(pt_ascii)
  854. tmpPtsAsciiFileOpened.close()
  855. self.tmpInPts = self._addTmpMapAnalysisMsg("vnet_tmp_in_pts")
  856. if not self.tmpInPts:
  857. return
  858. self.tmpInPtsConnected = self._addTmpMapAnalysisMsg("vnet_tmp_in_pts_connected")
  859. if not self.tmpInPtsConnected:
  860. return
  861. cmdParams.append("input=" + self.tmpInPtsConnected.GetVectMapName())
  862. cmdParams.append("--overwrite")
  863. # append parameters needed for particular analysis
  864. if self.currAnModule == "v.net.distance":
  865. cmdParams.append("from_layer=1")
  866. cmdParams.append("to_layer=1")
  867. elif self.currAnModule == "v.net.flow":
  868. self.vnetFlowTmpCut = self.NewTmpVectMapToHist('vnet_tmp_flow_cut')
  869. if not self.vnetFlowTmpCut:
  870. return
  871. cmdParams.append("cut=" + self.vnetFlowTmpCut.GetVectMapName())
  872. elif self.currAnModule == "v.net.iso":
  873. costs = self.anSettings["iso_lines"].GetValue()
  874. cmdParams.append("costs=" + costs)
  875. for catName, catNum in catsNums.iteritems():
  876. if catNum[0] == catNum[1]:
  877. cmdParams.append(catName + "=" + str(catNum[0]))
  878. else:
  879. cmdParams.append(catName + "=" + str(catNum[0]) + "-" + str(catNum[1]))
  880. # create and run commands which goes to analysis thread
  881. cmdVEdit = [
  882. "v.edit",
  883. "map=" + self.tmpInPts.GetVectMapName(),
  884. "input=" + self.tmpPtsAsciiFile,
  885. "tool=create",
  886. "--overwrite",
  887. "-n"
  888. ]
  889. self._prepareCmd(cmdVEdit)
  890. self.goutput.RunCmd(command = cmdVEdit)
  891. cmdVNet = [
  892. "v.net",
  893. "points=" + self.tmpInPts.GetVectMapName(),
  894. "input=" + self.inputData['input'].GetValue(),
  895. "output=" + self.tmpInPtsConnected.GetVectMapName(),
  896. "alayer=" + self.inputData["alayer"].GetValue().strip(),
  897. "nlayer=" + self.inputData["nlayer"].GetValue().strip(),
  898. "operation=connect",
  899. "thresh=" + str(self.anSettings["max_dist"].GetValue()),
  900. "--overwrite"
  901. ] #TODO snapping to nodes optimization
  902. self._prepareCmd(cmdVNet)
  903. self.goutput.RunCmd(command = cmdVNet)
  904. self._prepareCmd(cmdParams)
  905. self.goutput.RunCmd(command = cmdParams, onDone = self._runAnDone)
  906. def _runAnDone(self, cmd, returncode):
  907. """!Called when analysis is done"""
  908. self.tmpMaps.DeleteTmpMap(self.tmpInPts) #TODO remove earlier (OnDone lambda?)
  909. self.tmpMaps.DeleteTmpMap(self.tmpInPtsConnected)
  910. grass.try_remove(self.tmpPtsAsciiFile)
  911. self._saveHistStep()
  912. self.tmp_result.SaveVectMapState()
  913. if self.currAnModule == "v.net.flow":
  914. self.vnetFlowTmpCut.SaveVectMapState()
  915. self._updateResultDbMgrPage()
  916. self._updateDbMgrData()
  917. cmd = self.GetLayerStyle()
  918. self.tmp_result.AddRenderLayer(cmd)
  919. up_map_evt = gUpdateMap(render = True, renderVector = True)
  920. wx.PostEvent(self.mapWin, up_map_evt)
  921. mainToolbar = self.toolbars['mainToolbar']
  922. id = vars(mainToolbar)['showResult']
  923. mainToolbar.ToggleTool(id =id,
  924. toggle = True)
  925. self.stBar.RemoveStatusItem(key = 'analyze')
  926. def _getInputParams(self):
  927. """!Return list of chosen values (vector map, layers) from Parameters tab.
  928. The list items are in form to be used in command for analysis e.g. 'alayer=1'.
  929. """
  930. inParams = []
  931. for col in self.vnetParams[self.currAnModule]["cmdParams"]["cols"]:
  932. if "inputField" in self.attrCols[col]:
  933. colInptF = self.attrCols[col]["inputField"]
  934. else:
  935. colInptF = col
  936. inParams.append(col + '=' + self.inputData[colInptF].GetValue())
  937. for layer in ['alayer', 'nlayer']:
  938. inParams.append(layer + "=" + self.inputData[layer].GetValue().strip())
  939. return inParams
  940. def _getPtByCat(self):
  941. """!Return points separated by theirs categories"""
  942. cats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  943. ptByCats = {}
  944. for cat in self.vnetParams[self.currAnModule]["cmdParams"]["cats"]:
  945. ptByCats[cat[0]] = []
  946. for i in range(len(self.list.itemDataMap)):
  947. key = self.list.GetItemData(i)
  948. if self.list.IsChecked(key):
  949. for cat in cats:
  950. if cat[1] == self.list.itemDataMap[key][1] or len(ptByCats) == 1:
  951. ptByCats[cat[0]].append(self.pointsToDraw.GetItem(key).GetCoords())
  952. continue
  953. return ptByCats
  954. def _getAsciiPts (self, catPts, maxCat, layerNum):
  955. """!Return points separated by categories in GRASS ASCII vector representation"""
  956. catsNums = {}
  957. pt_ascii = ""
  958. catNum = maxCat
  959. for catName, pts in catPts.iteritems():
  960. catsNums[catName] = [catNum + 1]
  961. for pt in pts:
  962. catNum += 1
  963. pt_ascii += "P 1 1\n"
  964. pt_ascii += str(pt[0]) + " " + str(pt[1]) + "\n"
  965. pt_ascii += str(layerNum) + " " + str(catNum) + "\n"
  966. catsNums[catName].append(catNum)
  967. return pt_ascii, catsNums
  968. def _prepareCmd(self, cmd):
  969. """!Helper function for preparation of cmd list into form for RunCmd method"""
  970. for c in cmd[:]:
  971. if c.find("=") == -1:
  972. continue
  973. v = c.split("=")
  974. if len(v) != 2:
  975. cmd.remove(c)
  976. elif not v[1].strip():
  977. cmd.remove(c)
  978. def GetLayerStyle(self):
  979. """!Returns cmd for d.vect, with set style for analysis result"""
  980. resProps = self.vnetParams[self.currAnModule]["resultProps"]
  981. width = UserSettings.Get(group='vnet', key='res_style', subkey= "line_width")
  982. layerStyleCmd = ["layer=1",'width=' + str(width)]
  983. if "catColor" in resProps:
  984. layerStyleCmd.append('flags=c')
  985. elif "singleColor" in resProps:
  986. col = UserSettings.Get(group='vnet', key='res_style', subkey= "line_color")
  987. layerStyleCmd.append('color=' + str(col[0]) + ':' + str(col[1]) + ':' + str(col[2]))
  988. if "attrColColor" in resProps:
  989. colorStyle = UserSettings.Get(group='vnet', key='res_style', subkey= "color_table")
  990. invert = UserSettings.Get(group='vnet', key='res_style', subkey= "invert_colors")
  991. layerStyleVnetColors = [
  992. "v.colors",
  993. "map=" + self.tmp_result.GetVectMapName(),
  994. "color=" + colorStyle,
  995. "column=" + resProps["attrColColor"],
  996. ]
  997. if invert:
  998. layerStyleVnetColors.append("-n")
  999. layerStyleVnetColors = utils.CmdToTuple(layerStyleVnetColors)
  1000. RunCommand( layerStyleVnetColors[0],
  1001. **layerStyleVnetColors[1])
  1002. return layerStyleCmd
  1003. def OnShowResult(self, event):
  1004. """!Show/hide analysis result"""
  1005. mainToolbar = self.toolbars['mainToolbar']
  1006. id = vars(mainToolbar)['showResult']
  1007. toggleState = mainToolbar.GetToolState(id)
  1008. if not self.tmp_result:
  1009. mainToolbar.ToggleTool(id =id,
  1010. toggle = False)
  1011. elif toggleState:
  1012. self._checkResultMapChanged(self.tmp_result)
  1013. cmd = self.GetLayerStyle()
  1014. self.tmp_result.AddRenderLayer(cmd)
  1015. else:
  1016. cmd = self.GetLayerStyle()
  1017. self.tmp_result.DeleteRenderLayer()
  1018. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1019. wx.PostEvent(self.mapWin, up_map_evt)
  1020. def OnInsertPoint(self, event):
  1021. """!Registers/unregisters mouse handler into map window"""
  1022. if self.handlerRegistered == False:
  1023. self.mapWin.RegisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1024. self.OnMapClickHandler,
  1025. wx.StockCursor(wx.CURSOR_CROSS))
  1026. self.handlerRegistered = True
  1027. else:
  1028. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  1029. self.OnMapClickHandler)
  1030. self.handlerRegistered = False
  1031. def OnSaveTmpLayer(self, event):
  1032. """!Permanently saves temporary map of analysis result"""
  1033. dlg = AddLayerDialog(parent = self)
  1034. msg = _("Vector map with analysis result does not exist.")
  1035. if dlg.ShowModal() == wx.ID_OK:
  1036. if not hasattr(self.tmp_result, "GetVectMapName"):
  1037. GMessage(parent = self,
  1038. message = msg)
  1039. return
  1040. mapToAdd = self.tmp_result.GetVectMapName()
  1041. mapToAddEx = grass.find_file(name = mapToAdd,
  1042. element = 'vector',
  1043. mapset = grass.gisenv()['MAPSET'])
  1044. if not mapToAddEx["name"]:
  1045. GMessage(parent = self,
  1046. message = msg)
  1047. dlg.Destroy()
  1048. return
  1049. addedMap = dlg.vectSel.GetValue()
  1050. mapName, mapSet = self._parseMapStr(addedMap)
  1051. if mapSet != grass.gisenv()['MAPSET']:
  1052. GMessage(parent = self,
  1053. message = _("Map can be saved only to currently set mapset"))
  1054. return
  1055. existsMap = grass.find_file(name = mapName,
  1056. element = 'vector',
  1057. mapset = grass.gisenv()['MAPSET'])
  1058. dlg.Destroy()
  1059. if existsMap["name"]:
  1060. dlg = wx.MessageDialog(parent = self.parent.parent,
  1061. message = _("Vector map %s already exists. " +
  1062. "Do you want to overwrite it?") %
  1063. (existsMap["fullname"]),
  1064. caption = _("Overwrite map layer"),
  1065. style = wx.YES_NO | wx.NO_DEFAULT |
  1066. wx.ICON_QUESTION | wx.CENTRE)
  1067. ret = dlg.ShowModal()
  1068. if ret == wx.ID_NO:
  1069. dlg.Destroy()
  1070. return
  1071. dlg.Destroy()
  1072. RunCommand("g.copy",
  1073. overwrite = True,
  1074. vect = [self.tmp_result.GetVectMapName(), mapName])
  1075. cmd = self.GetLayerStyle()#TODO get rid of insert
  1076. cmd.insert(0, 'd.vect')
  1077. cmd.append('map=%s' % mapName)
  1078. if not self.mapWin.tree:
  1079. return
  1080. if self.mapWin.tree.FindItemByData(key = 'name', value = mapName) is None:
  1081. self.mapWin.tree.AddLayer(ltype = "vector",
  1082. lname = mapName,
  1083. lcmd = cmd,
  1084. lchecked = True)
  1085. else:
  1086. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1087. wx.PostEvent(self.mapWin, up_map_evt)
  1088. def OnSettings(self, event):
  1089. """!Displays vnet settings dialog"""
  1090. dlg = SettingsDialog(parent=self, id=wx.ID_ANY, title=_('Settings'))
  1091. if dlg.ShowModal() == wx.ID_OK:
  1092. pass
  1093. dlg.Destroy()
  1094. def OnAnalysisChanged(self, event):
  1095. """!Updates dialog when analysis is changed"""
  1096. # set chosen analysis
  1097. iAn = self.toolbars['analysisToolbar'].anChoice.GetSelection()
  1098. self.currAnModule = self.vnetModulesOrder[iAn]
  1099. # update dialog for particular analysis
  1100. if self.currAnModule == "v.net.iso":
  1101. self.anSettings['iso_lines'].GetParent().Show()
  1102. else:
  1103. self.anSettings['iso_lines'].GetParent().Hide()
  1104. #if self.currAnModule == "v.net.flow": TODO not implemented
  1105. # self.anSettings['show_cut'].GetParent().Show()
  1106. #else:
  1107. # self.anSettings['show_cut'].GetParent().Hide()
  1108. # show only corresponding selects for chosen v.net module
  1109. skip = []
  1110. for col in self.attrCols.iterkeys():
  1111. if "inputField" in self.attrCols[col]:
  1112. colInptF = self.attrCols[col]["inputField"]
  1113. else:
  1114. colInptF = col
  1115. if col in skip:
  1116. continue
  1117. inputPanel = self.inputData[colInptF].GetParent()
  1118. if col in self.vnetParams[self.currAnModule]["cmdParams"]["cols"]:
  1119. inputPanel.Show()
  1120. inputPanel.FindWindowByName(colInptF).SetLabel(self.attrCols[col]["label"])
  1121. inputPanel.Layout()
  1122. if col != colInptF:
  1123. skip.append(colInptF)
  1124. else:
  1125. self.inputData[colInptF].GetParent().Hide()
  1126. self.Layout()
  1127. # if module has only one category -> hide type column in points list otherwise show it
  1128. if len(self.vnetParams[self.currAnModule]["cmdParams"]["cats"]) > 1:
  1129. if not self.list.IsShown('type'):
  1130. self.list.ShowColumn('type', 1)
  1131. currParamsCats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  1132. self.list.AdaptPointsList(currParamsCats)
  1133. else:
  1134. if self.list.IsShown('type'):
  1135. self.list.HideColumn('type')
  1136. # for v.net.path just one 'Start point' and one 'End point' can be checked
  1137. if self.currAnModule == "v.net.path":
  1138. self.list.UpdateCheckedItems(index = None)
  1139. def OnSnapping(self, event):
  1140. """!Start/stop snapping mode"""
  1141. ptListToolbar = self.toolbars['pointsList']
  1142. if not haveCtypes:
  1143. ptListToolbar.ToggleTool(id = ptListToolbar.GetToolId("snapping"),
  1144. toggle = False)
  1145. GMessage(parent = self,
  1146. message = _("Unable to use ctypes. \n") + \
  1147. _("Snapping mode can not be activated."))
  1148. return
  1149. if not event or not event.IsChecked():
  1150. if not event:
  1151. ptListToolbar.ToggleTool(id = ptListToolbar.GetToolId("snapping"),
  1152. toggle = False)
  1153. if self.tmpMaps.HasTmpVectMap("vnet_snap_points"):
  1154. self.snapPts.DeleteRenderLayer()
  1155. up_map_evt = gUpdateMap(render = False, renderVector = False)
  1156. wx.PostEvent(self.mapWin, up_map_evt)
  1157. if self.snapData.has_key('cmdThread'):
  1158. self.snapData['cmdThread'].abort()
  1159. self.snapData['snap_mode'] = False
  1160. return
  1161. if not self.InputsErrorMsgs(msg = _("Snapping mode can not be activated."),
  1162. inpToTest = ["input", "nlayer"]):
  1163. ptListToolbar.ToggleTool(id = ptListToolbar.GetToolId("snapping"),
  1164. toggle = False)
  1165. return
  1166. if not self.tmpMaps.HasTmpVectMap("vnet_snap_points"):
  1167. endStr = _("Do you really want to activate snapping and overwrite it?")
  1168. self.snapPts = self.tmpMaps.AddTmpVectMap("vnet_snap_points", endStr)
  1169. if not self.snapPts:
  1170. ptListToolbar.ToggleTool(id = ptListToolbar.GetToolId("snapping"),
  1171. toggle = False)
  1172. return
  1173. elif self.snapPts.VectMapState() == 0:
  1174. dlg = wx.MessageDialog(parent = self.parent,
  1175. message = _("Temporary map '%s' was changed outside " +
  1176. "vector analysis tool.\n"
  1177. "Do you really want to activate " +
  1178. "snapping and overwrite it? ") % \
  1179. self.snapPts.GetVectMapName(),
  1180. caption = _("Overwrite map"),
  1181. style = wx.YES_NO | wx.NO_DEFAULT |
  1182. wx.ICON_QUESTION | wx.CENTRE)
  1183. ret = dlg.ShowModal()
  1184. dlg.Destroy()
  1185. if ret == wx.ID_NO:
  1186. self.tmpMaps.DeleteTmpMap(self.snapPts)
  1187. ptListToolbar.ToggleTool(id = ptListToolbar.GetToolId("snapping"),
  1188. toggle = False)
  1189. return
  1190. self.snapData['snap_mode'] = True
  1191. inpName = self.inputData['input'].GetValue()
  1192. inpName, mapSet = self._parseMapStr(inpName)
  1193. inpFullName = inpName + '@' + mapSet
  1194. computeNodes = True
  1195. if not self.snapData.has_key("inputMap"):
  1196. pass
  1197. elif inpFullName != self.snapData["inputMap"].GetVectMapName():
  1198. self.snapData["inputMap"] = VectMap(self, inpFullName)
  1199. elif self.snapData["inputMapNlayer"] == self.inputData["nlayer"].GetValue():
  1200. if self.snapData["inputMap"].VectMapState() == 1:
  1201. computeNodes = False
  1202. # new map need
  1203. if computeNodes:
  1204. self.stBar.AddStatusItem(text = _('Computing nodes...'),
  1205. key = 'snap',
  1206. priority = self.stPriorities['important'])
  1207. if not self.snapData.has_key('cmdThread'):
  1208. self.snapData['cmdThread'] = CmdThread(self)
  1209. cmd = ["v.to.points", "input=" + self.inputData['input'].GetValue(),
  1210. "output=" + self.snapPts.GetVectMapName(),
  1211. "llayer=" + self.inputData["nlayer"].GetValue(),
  1212. "-n", "--overwrite"]
  1213. # process GRASS command with argument
  1214. self.snapData["inputMap"] = VectMap(self, inpFullName)
  1215. self.snapData["inputMap"].SaveVectMapState()
  1216. self.Bind(EVT_CMD_DONE, self._onToPointsDone)
  1217. self.snapData['cmdThread'].RunCmd(cmd)
  1218. self.snapData["inputMapNlayer"] = self.inputData["nlayer"].GetValue()
  1219. # map is already created and up to date for input data
  1220. else:
  1221. self.snapPts.AddRenderLayer()
  1222. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1223. wx.PostEvent(self.mapWin, up_map_evt)
  1224. def _onToPointsDone(self, event):
  1225. """!Update map window, when map with nodes to snap is created"""
  1226. self.stBar.RemoveStatusItem(key = 'snap')
  1227. if not event.aborted:
  1228. self.snapPts.SaveVectMapState()
  1229. self.snapPts.AddRenderLayer()
  1230. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1231. wx.PostEvent(self.mapWin, up_map_evt)
  1232. def OnUndo(self, event):
  1233. """!Step back in history"""
  1234. histStepData = self.history.GetPrev()
  1235. self.toolbars['mainToolbar'].UpdateUndoRedo()
  1236. if histStepData:
  1237. self._updateHistStepData(histStepData)
  1238. def OnRedo(self, event):
  1239. """!Step forward in history"""
  1240. histStepData = self.history.GetNext()
  1241. self.toolbars['mainToolbar'].UpdateUndoRedo()
  1242. if histStepData:
  1243. self._updateHistStepData(histStepData)
  1244. def _saveAnInputToHist(self):
  1245. """!Save all data needed for analysis into history buffer"""
  1246. pts = self.pointsToDraw.GetAllItems()
  1247. for iPt, pt in enumerate(pts):
  1248. ptName = "pt" + str(iPt)
  1249. coords = pt.GetCoords()
  1250. self.history.Add(key = "points",
  1251. subkey = [ptName, "coords"],
  1252. value = coords)
  1253. # save type column
  1254. # if is shown
  1255. if len(self.vnetParams[self.currAnModule]["cmdParams"]["cats"]) > 1:
  1256. cat = self.list.GetCellSelIdx(iPt, 'type')
  1257. self.history.Add(key = "points",
  1258. subkey = [ptName, "catIdx"],
  1259. value = cat)
  1260. # if is hidden
  1261. else:
  1262. self.history.Add(key = 'points_hidden_cols',
  1263. subkey = 'type',
  1264. value = self.list.GetHiddenColSelIdxs('type'))
  1265. topology = self.list.GetCellValue(iPt, 'topology')
  1266. self.history.Add(key = "points",
  1267. subkey = [ptName, "topology"],
  1268. value = topology)
  1269. self.history.Add(key = "points",
  1270. subkey = [ptName, "checked"],
  1271. value = self.list.IsChecked(iPt))
  1272. for inpName, inp in self.inputData.iteritems():
  1273. if inpName == "input":
  1274. vectMapName, mapSet = self._parseMapStr(inp.GetValue())
  1275. inpMapFullName = vectMapName + '@' + mapSet
  1276. inpMap = VectMap(self, inpMapFullName)
  1277. self.history.Add(key = "other",
  1278. subkey = "input_modified",
  1279. value = inpMap.GetLastModified())
  1280. inpVal = inpMapFullName
  1281. else:
  1282. inpVal = inp.GetValue()
  1283. self.history.Add(key = "input_data",
  1284. subkey = inpName,
  1285. value = inpVal)
  1286. self.history.Add(key = "vnet_modules",
  1287. subkey = "curr_module",
  1288. value = self.currAnModule)
  1289. for settName, sett in self.anSettings.iteritems():
  1290. self.history.Add(key = "an_settings",
  1291. subkey = settName,
  1292. value = sett.GetValue())
  1293. def _saveHistStep(self):
  1294. """!Save new step into history"""
  1295. removedHistData = self.history.SaveHistStep()
  1296. self.toolbars['mainToolbar'].UpdateUndoRedo()
  1297. if not removedHistData:
  1298. return
  1299. # delete temporary maps in history steps which were deleted
  1300. for removedStep in removedHistData.itervalues():
  1301. mapsNames = removedStep["tmp_data"]["maps"]
  1302. for vectMapName in mapsNames:
  1303. tmpMap = self.tmpMaps.GetTmpVectMap(vectMapName)
  1304. self.tmpMaps.DeleteTmpMap(tmpMap)
  1305. def _updateHistStepData(self, histStepData):
  1306. """!Updates dialog according to chosen history step"""
  1307. # set analysis module
  1308. self.currAnModule = histStepData["vnet_modules"]["curr_module"]
  1309. # optimization -map is not re-rendered when change in points list is done
  1310. self.list.SetUpdateMap(updateMap = False)
  1311. # delete points list items
  1312. while self.list.GetSelected() != wx.NOT_FOUND:
  1313. self.list.DeleteItem()
  1314. # show/hide 'type' column according to particular analysis
  1315. if len(self.vnetParams[self.currAnModule]["cmdParams"]["cats"]) > 1:
  1316. hasType = True
  1317. self.list.ShowColumn('type', 1)
  1318. else:
  1319. hasType = False
  1320. self.list.HideColumn('type')
  1321. # add points to list
  1322. for iPt in range(len(histStepData["points"])):
  1323. ptData = histStepData["points"]["pt" + str(iPt)]
  1324. coords = ptData["coords"]
  1325. self.list.AddItem()
  1326. item = self.pointsToDraw.GetItem(iPt)
  1327. item.SetCoords(coords)
  1328. if hasType:
  1329. self.list.EditCellKey(iPt, 'type', int(ptData["catIdx"]))
  1330. self.list.EditCellKey(iPt, 'topology', ptData["topology"])
  1331. if ptData["checked"]:
  1332. self.list.CheckItem(iPt, True)
  1333. if hasType:
  1334. currParamsCats = self.vnetParams[self.currAnModule]["cmdParams"]["cats"]
  1335. self.list.AdaptPointsList(currParamsCats)
  1336. else:
  1337. self.list.SetHiddenSelIdxs('type', histStepData["points_hidden_cols"]["type"])
  1338. # set analysis combobox
  1339. anChoice = self.toolbars['analysisToolbar'].anChoice
  1340. anChoice.SetSelection(self.vnetModulesOrder.index(self.currAnModule))
  1341. # update analysis result maps
  1342. mapsNames = histStepData["tmp_data"]["maps"]
  1343. for vectMapName in mapsNames:
  1344. if "vnet_tmp_result" in vectMapName:
  1345. self.tmp_result.DeleteRenderLayer()
  1346. self.tmp_result = self.tmpMaps.GetTmpVectMap(vectMapName)
  1347. self._checkResultMapChanged(self.tmp_result)
  1348. cmd = self.GetLayerStyle()
  1349. self.tmp_result.AddRenderLayer(cmd)
  1350. # update Parameters tab
  1351. histInputData = histStepData["input_data"]
  1352. for inpName, inp in histInputData.iteritems():
  1353. self.inputData[inpName].SetValue(str(inp))
  1354. if inpName == "input":
  1355. inpMap = inp
  1356. prevInpModTime = str(histStepData["other"]["input_modified"])
  1357. currInpModTime = VectMap(self, inpMap).GetLastModified()
  1358. if currInpModTime.strip()!= prevInpModTime.strip():
  1359. dlg = wx.MessageDialog(parent = self,
  1360. message = _("Input map '%s' for analysis was changed outside " +
  1361. "vector network analysis tool.\n" +
  1362. "Topology column may not " +
  1363. "correspond to changed situation.") %\
  1364. inpMap,
  1365. caption = _("Input changed outside"),
  1366. style = wx.ICON_INFORMATION| wx.CENTRE)
  1367. dlg.ShowModal()
  1368. dlg.Destroy()
  1369. # update Points tab (analysis settings)
  1370. histAnSettData = histStepData["an_settings"]
  1371. for settName, sett in histAnSettData.iteritems():
  1372. if settName == 'iso_lines':
  1373. sett = str(sett)
  1374. self.anSettings[settName].SetValue(sett)
  1375. self.resultDbMgrData['analysis'] = self.currAnModule
  1376. self._updateResultDbMgrPage()
  1377. self._updateDbMgrData()
  1378. self.OnAnalysisChanged(None)
  1379. self.list.SetUpdateMap(updateMap = True)
  1380. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1381. wx.PostEvent(self.mapWin, up_map_evt)
  1382. def _checkResultMapChanged(self, resultVectMap):
  1383. """!Check if map was modified outside"""
  1384. if resultVectMap.VectMapState() == 0:
  1385. dlg = wx.MessageDialog(parent = self,
  1386. message = _("Temporary map '%s' with result " +
  1387. "was changed outside vector network analysis tool.\n" +
  1388. "Showed result may not correspond " +
  1389. "original analysis result.") %\
  1390. resultVectMap.GetVectMapName(),
  1391. caption = _("Result changed outside"),
  1392. style = wx.ICON_INFORMATION| wx.CENTRE)
  1393. dlg.ShowModal()
  1394. dlg.Destroy()
  1395. def NewTmpVectMapToHist(self, prefMapName):
  1396. """!Add new vector map, which will be saved into history step"""
  1397. mapName = prefMapName + str(self.histTmpVectMapNum)
  1398. self.histTmpVectMapNum += 1
  1399. tmpMap = self._addTmpMapAnalysisMsg(mapName)
  1400. if not tmpMap:
  1401. return tmpMap
  1402. self.tmpVectMapsToHist.append(tmpMap.GetVectMapName())
  1403. self.history.Add(key = "tmp_data",
  1404. subkey = "maps",
  1405. value = self.tmpVectMapsToHist)
  1406. return tmpMap
  1407. def _addTmpMapAnalysisMsg(self, mapName):
  1408. """!Wraped AddTmpVectMap"""
  1409. msg = _("Temporary map %s already exists.\n" +
  1410. "Do you want to continue in analysis and overwrite it?") \
  1411. % (mapName +'@' + grass.gisenv()['MAPSET'])
  1412. tmpMap = self.tmpMaps.AddTmpVectMap(mapName, msg)
  1413. return tmpMap
  1414. def _initVnetParams(self):
  1415. """!Initializes parameters for different v.net.* modules """
  1416. self.attrCols = {
  1417. 'afcolumn' : {
  1418. "label" : _("Arc forward/both direction(s) cost column:"),
  1419. "name" : _("arc forward/both")
  1420. },
  1421. 'abcolumn' : {
  1422. "label" : _("Arc backward direction cost column:"),
  1423. "name" : _("arc backward")
  1424. },
  1425. 'acolumn' : {
  1426. "label" : _("Arcs' cost column (for both directions):"),
  1427. "name" : _("arc"),
  1428. "inputField" : 'afcolumn',
  1429. },
  1430. 'ncolumn' : {
  1431. "label" : _("Node cost column:"),
  1432. "name" : _("node")
  1433. }
  1434. }
  1435. self.vnetParams = {
  1436. "v.net.path" : {
  1437. "label" : _("Shortest path %s") % "(v.net.path)",
  1438. "cmdParams" : {
  1439. "cats" : [
  1440. ["st_pt", _("Start point")],
  1441. ["end_pt", _("End point")]
  1442. ],
  1443. "cols" : [
  1444. 'afcolumn',
  1445. 'abcolumn',
  1446. 'ncolumn'
  1447. ],
  1448. },
  1449. "resultProps" : {
  1450. "singleColor" : None,
  1451. "dbMgr" : True
  1452. }
  1453. },
  1454. "v.net.salesman" : {
  1455. "label" : _("Traveling salesman %s") % "(v.net.salesman)",
  1456. "cmdParams" : {
  1457. "cats" : [["ccats", None]],
  1458. "cols" : [
  1459. 'afcolumn',
  1460. 'abcolumn'
  1461. ],
  1462. },
  1463. "resultProps" : {
  1464. "singleColor" : None,
  1465. "dbMgr" : False
  1466. }
  1467. },
  1468. "v.net.flow" : {
  1469. "label" : _("Maximum flow %s") % "(v.net.flow)",
  1470. "cmdParams" : {
  1471. "cats" : [
  1472. ["source_cats", _("Source point")],
  1473. ["sink_cats", _("Sink point")]
  1474. ],
  1475. "cols" : [
  1476. 'afcolumn',
  1477. 'abcolumn',
  1478. 'ncolumn'
  1479. ]
  1480. },
  1481. "resultProps" : {
  1482. "attrColColor": "flow",
  1483. "dbMgr" : True
  1484. }
  1485. },
  1486. "v.net.alloc" : {
  1487. "label" : _("Subnets for nearest centers %s") % "(v.net.alloc)",
  1488. "cmdParams" : {
  1489. "cats" : [["ccats", None]],
  1490. "cols" : [
  1491. 'afcolumn',
  1492. 'abcolumn',
  1493. 'ncolumn'
  1494. ]
  1495. },
  1496. "resultProps" : {
  1497. "catColor" : None,
  1498. "dbMgr" : False
  1499. }
  1500. },
  1501. "v.net.steiner" : {
  1502. "label" : _("Steiner tree for the network and given terminals %s") % "(v.net.steiner)",
  1503. "cmdParams" : {
  1504. "cats" : [["tcats", None]],
  1505. "cols" : [
  1506. 'acolumn',
  1507. ]
  1508. },
  1509. "resultProps" : {
  1510. "singleColor" : None,
  1511. "dbMgr" : False
  1512. }
  1513. },
  1514. "v.net.distance" : {
  1515. "label" : _("Shortest distance via the network %s") % "(v.net.distance)",
  1516. "cmdParams" : {
  1517. "cats" : [
  1518. ["from_cats", "From point"],
  1519. ["to_cats", "To point"]
  1520. ],
  1521. "cols" : [
  1522. 'afcolumn',
  1523. 'abcolumn',
  1524. 'ncolumn'
  1525. ],
  1526. },
  1527. "resultProps" : {
  1528. "catColor" : None,
  1529. "dbMgr" : True
  1530. }
  1531. },
  1532. "v.net.iso" : {
  1533. "label" : _("Cost isolines %s") % "(v.net.iso)",
  1534. "cmdParams" : {
  1535. "cats" : [["ccats", None]],
  1536. "cols" : [
  1537. 'afcolumn',
  1538. 'abcolumn',
  1539. 'ncolumn'
  1540. ]
  1541. },
  1542. "resultProps" : {
  1543. "catColor" : None,
  1544. "dbMgr" : False
  1545. }
  1546. }
  1547. }
  1548. # order in combobox for choose of analysis
  1549. self.vnetModulesOrder = ["v.net.path",
  1550. "v.net.salesman",
  1551. "v.net.flow",
  1552. "v.net.alloc",
  1553. "v.net.distance",
  1554. "v.net.iso",
  1555. #"v.net.steiner"
  1556. ]
  1557. self.currAnModule = self.vnetModulesOrder[0]
  1558. def _initSettings(self):
  1559. """!Initialization of settings (if not already defined)"""
  1560. # initializes default settings
  1561. initSettings = [
  1562. ['res_style', 'line_width', 5],
  1563. ['res_style', 'line_color', (192,0,0)],
  1564. ['res_style', 'color_table', 'byr'],
  1565. ['res_style', 'invert_colors', False],
  1566. ['point_symbol', 'point_size', 10],
  1567. ['point_symbol', 'point_width', 2],
  1568. ['point_colors', "unused", (131,139,139)],
  1569. ['point_colors', "used1cat", (192,0,0)],
  1570. ['point_colors', "used2cat", (0,0,255)],
  1571. ['point_colors', "selected", (9,249,17)],
  1572. ['other', "snap_tresh", 10],
  1573. ['other', "max_hist_steps", 5]
  1574. ]
  1575. for init in initSettings:
  1576. UserSettings.ReadSettingsFile()
  1577. UserSettings.Append(dict = UserSettings.userSettings,
  1578. group ='vnet',
  1579. key = init[0],
  1580. subkey =init[1],
  1581. value = init[2],
  1582. overwrite = False)
  1583. def SetPointDrawSettings(self):
  1584. """!Set settings for drawing of points"""
  1585. ptSize = int(UserSettings.Get(group='vnet', key='point_symbol', subkey = 'point_size'))
  1586. self.pointsToDraw.SetPropertyVal("size", ptSize)
  1587. colors = UserSettings.Get(group='vnet', key='point_colors')
  1588. ptWidth = int(UserSettings.Get(group='vnet', key='point_symbol', subkey = 'point_width'))
  1589. textProp = self.pointsToDraw.GetPropertyVal("text")
  1590. textProp["font"].SetPointSize(ptSize + 2)
  1591. for colKey, col in colors.iteritems():
  1592. pen = self.pointsToDraw.GetPen(colKey)
  1593. if pen:
  1594. pen.SetColour(wx.Colour(col[0], col[1], col[2], 255))
  1595. pen.SetWidth(ptWidth)
  1596. else:
  1597. self.pointsToDraw.AddPen(colKey, wx.Pen(colour = wx.Colour(col[0], col[1], col[2], 255), width = ptWidth))
  1598. class PtsList(PointsList):
  1599. def __init__(self, parent, dialog, cols, id=wx.ID_ANY):
  1600. """! List with points for analysis"""
  1601. self.updateMap = True
  1602. self.dialog = dialog # VNETDialog class
  1603. PointsList.__init__(self, parent = parent, cols = cols, id = id)
  1604. def AddItem(self, event = None, updateMap = True):
  1605. """!Append point to list"""
  1606. self.dialog.pointsToDraw.AddItem(coords = [0,0])
  1607. PointsList.AddItem(self, event)
  1608. self.EditCellKey(key = self.selected ,
  1609. colName = 'topology',
  1610. cellData = _("new point"))
  1611. def DeleteItem(self, event = None):
  1612. """!Delete selected point in list"""
  1613. key = self.GetItemData(self.selected)
  1614. if self.selected != wx.NOT_FOUND:
  1615. item = self.dialog.pointsToDraw.GetItem(key)
  1616. self.dialog.pointsToDraw.DeleteItem(item)
  1617. PointsList.DeleteItem(self, event)
  1618. def OnItemSelected(self, event):
  1619. """!Item selected"""
  1620. PointsList.OnItemSelected(self, event)
  1621. if self.updateMap:
  1622. up_map_evt = gUpdateMap(render = False, renderVector = False)
  1623. wx.PostEvent(self.dialog.mapWin, up_map_evt)
  1624. def AdaptPointsList(self, currParamsCats):
  1625. """Rename category values when module is changed. Expample: Start point -> Sink point"""
  1626. colValues = [""]
  1627. for ptCat in currParamsCats:
  1628. colValues.append(ptCat[1])
  1629. colNum = self._getColumnNum('type')
  1630. self.ChangeColEditable('type', colValues)
  1631. for iItem, item in enumerate(self.itemDataMap):
  1632. self.EditCellKey(iItem, 'type', self.selIdxs[iItem][colNum])
  1633. if not item[1]:
  1634. self.CheckItem(iItem, False)
  1635. def OnCheckItem(self, index, flag):
  1636. """!Item is checked/unchecked"""
  1637. key = self.GetItemData(index)
  1638. checkedVal = self.itemDataMap[key][1]
  1639. currModule = self.dialog.currAnModule #TODO public func
  1640. cats = self.dialog.vnetParams[currModule]["cmdParams"]["cats"]
  1641. if self.updateMap:
  1642. up_map_evt = gUpdateMap(render = False, renderVector = False)
  1643. wx.PostEvent(self.dialog.mapWin, up_map_evt)
  1644. if len(cats) <= 1:
  1645. return
  1646. if checkedVal == "":
  1647. self.CheckItem(key, False)
  1648. return
  1649. if currModule == "v.net.path" and flag:
  1650. self.UpdateCheckedItems(index)
  1651. def UpdateCheckedItems(self, index):
  1652. """!For v.net.path - max. just one checked 'Start point' and 'End point'"""
  1653. alreadyChecked = []
  1654. colNum = self._getColumnNum('type')
  1655. if colNum == -1:
  1656. return
  1657. if index:
  1658. checkedKey = self.GetItemData(index)
  1659. checkedVal = self.selIdxs[checkedKey][colNum]
  1660. alreadyChecked.append(checkedVal)
  1661. else:
  1662. checkedKey = -1
  1663. for iKey, idx in enumerate(self.selIdxs):
  1664. index = self._findIndex(iKey)
  1665. if (idx[colNum] in alreadyChecked and checkedKey != iKey) \
  1666. or idx[colNum] == 0:
  1667. self.CheckItem(index, False)
  1668. elif self.IsChecked(index):
  1669. alreadyChecked.append(idx[colNum])
  1670. def SetUpdateMap(self, updateMap):
  1671. """!Update/Not update map window when some change in list is made"""
  1672. self.updateMap = updateMap
  1673. def GetHiddenColSelIdxs(self, colName):
  1674. """!Get indexes of chosen values in hidden 'type' column"""
  1675. if self.hiddenCols.has_key(colName):
  1676. return self.hiddenCols[colName]['selIdxs']
  1677. return None
  1678. def SetHiddenSelIdxs(self, colName, selIdxs):
  1679. """!Set indexes of chosen values in hidden 'type' column and update text in it's cells"""
  1680. if self.hiddenCols.has_key(colName):
  1681. self.hiddenCols[colName]['selIdxs'] = map(int, selIdxs)
  1682. self.hiddenCols[colName]['itemDataMap'] = []
  1683. # update text in hidden column cells
  1684. for idx in self.hiddenCols[colName]['selIdxs']:
  1685. self.hiddenCols[colName]['itemDataMap'].append(self.hiddenCols[colName]['colsData'][2][idx])
  1686. return True
  1687. return False
  1688. class SettingsDialog(wx.Dialog):
  1689. def __init__(self, parent, id, title, pos=wx.DefaultPosition, size=wx.DefaultSize,
  1690. style=wx.DEFAULT_DIALOG_STYLE):
  1691. """!Settings dialog"""
  1692. wx.Dialog.__init__(self, parent, id, title, pos, size, style)
  1693. maxValue = 1e8
  1694. self.parent = parent
  1695. self.settings = {}
  1696. rules = RunCommand('v.colors',
  1697. read = True,
  1698. flags = 'l')
  1699. settsLabels = {}
  1700. settsLabels['color_table'] = wx.StaticText(parent = self, id = wx.ID_ANY,
  1701. label = _('Color table style %s:') % '(v.net.flow)')
  1702. self.settings['color_table'] = wx.ComboBox(parent = self, id = wx.ID_ANY,
  1703. choices = rules.split(),
  1704. style = wx.CB_READONLY, size = (180, -1))
  1705. setStyle = UserSettings.Get(group ='vnet', key = "res_style", subkey = "color_table")
  1706. i = self.settings['color_table'].FindString(setStyle)
  1707. if i != wx.NOT_FOUND:
  1708. self.settings['color_table'].Select(i)
  1709. self.settings["invert_colors"] = wx.CheckBox(parent = self, id=wx.ID_ANY,
  1710. label = _('Invert colors %s:') % '(v.net.flow)')
  1711. setInvert = UserSettings.Get(group ='vnet', key = "res_style", subkey = "invert_colors")
  1712. self.settings["invert_colors"].SetValue(setInvert)
  1713. self.colorsSetts = {
  1714. "line_color" : ["res_style", _("Line color:")],
  1715. "unused" : ["point_colors", _("Color for unused point:")],
  1716. "used1cat" : ["point_colors", _("Color for Start/From/Source/Used point:")],
  1717. "used2cat" : ["point_colors", _("Color for End/To/Sink point:")],
  1718. "selected" : ["point_colors", _("Color for selected point:")]
  1719. }
  1720. for settKey, sett in self.colorsSetts.iteritems():
  1721. settsLabels[settKey] = wx.StaticText(parent = self, id = wx.ID_ANY, label = sett[1])
  1722. col = UserSettings.Get(group ='vnet', key = sett[0], subkey = settKey)
  1723. self.settings[settKey] = csel.ColourSelect(parent = self, id = wx.ID_ANY,
  1724. colour = wx.Colour(col[0],
  1725. col[1],
  1726. col[2],
  1727. 255))
  1728. self.sizeSetts = {
  1729. "line_width" : ["res_style", _("Line width:")],
  1730. "point_size" : ["point_symbol", _("Point size:")],
  1731. "point_width" : ["point_symbol", _("Point width:")],
  1732. "snap_tresh" : ["other", _("Snapping treshold in pixels:")],
  1733. "max_hist_steps" : ["other", _("Maximum number of results in history:")]
  1734. }
  1735. for settKey, sett in self.sizeSetts.iteritems():
  1736. settsLabels[settKey] = wx.StaticText(parent = self, id = wx.ID_ANY, label = sett[1])
  1737. self.settings[settKey] = wx.SpinCtrl(parent = self, id = wx.ID_ANY, min = 1, max = 50)
  1738. size = int(UserSettings.Get(group = 'vnet', key = sett[0], subkey = settKey))
  1739. self.settings[settKey].SetValue(size)
  1740. # buttons
  1741. self.btnSave = wx.Button(self, wx.ID_SAVE)
  1742. self.btnApply = wx.Button(self, wx.ID_APPLY)
  1743. self.btnClose = wx.Button(self, wx.ID_CLOSE)
  1744. self.btnApply.SetDefault()
  1745. # bindings
  1746. self.btnApply.Bind(wx.EVT_BUTTON, self.OnApply)
  1747. self.btnApply.SetToolTipString(_("Apply changes for the current session"))
  1748. self.btnSave.Bind(wx.EVT_BUTTON, self.OnSave)
  1749. self.btnSave.SetToolTipString(_("Apply and save changes to user settings file (default for next sessions)"))
  1750. self.btnClose.Bind(wx.EVT_BUTTON, self.OnClose)
  1751. self.btnClose.SetToolTipString(_("Close dialog"))
  1752. #Layout
  1753. # Analysis result style layout
  1754. self.SetMinSize(self.GetBestSize())
  1755. sizer = wx.BoxSizer(wx.VERTICAL)
  1756. styleBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  1757. label =" %s " % _("Analysis result style:"))
  1758. styleBoxSizer = wx.StaticBoxSizer(styleBox, wx.VERTICAL)
  1759. gridSizer = wx.GridBagSizer(vgap = 1, hgap = 1)
  1760. row = 0
  1761. gridSizer.Add(item = settsLabels["line_color"], flag = wx.ALIGN_CENTER_VERTICAL, pos =(row, 0))
  1762. gridSizer.Add(item = self.settings["line_color"],
  1763. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5,
  1764. pos =(row, 1))
  1765. row += 1
  1766. gridSizer.Add(item = settsLabels["line_width"], flag=wx.ALIGN_CENTER_VERTICAL, pos=(row, 0))
  1767. gridSizer.Add(item = self.settings["line_width"],
  1768. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5,
  1769. pos = (row, 1))
  1770. row += 1
  1771. gridSizer.Add(item = settsLabels['color_table'], flag=wx.ALIGN_CENTER_VERTICAL, pos=(row, 0))
  1772. gridSizer.Add(item = self.settings['color_table'],
  1773. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5,
  1774. pos = (row, 1))
  1775. row += 1
  1776. gridSizer.Add(item = self.settings["invert_colors"], flag=wx.ALIGN_CENTER_VERTICAL, pos=(row, 0))
  1777. gridSizer.AddGrowableCol(1)
  1778. styleBoxSizer.Add(item = gridSizer, flag = wx.EXPAND)
  1779. # Point style layout
  1780. ptsStyleBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  1781. label =" %s " % _("Point style:"))
  1782. ptsStyleBoxSizer = wx.StaticBoxSizer(ptsStyleBox, wx.VERTICAL)
  1783. gridSizer = wx.GridBagSizer(vgap = 1, hgap = 1)
  1784. row = 0
  1785. setts = dict(self.colorsSetts.items() + self.sizeSetts.items())
  1786. settsOrder = ["selected", "used1cat", "used2cat", "unused", "point_size", "point_width"]
  1787. for settKey in settsOrder:
  1788. sett = setts[settKey]
  1789. gridSizer.Add(item = settsLabels[settKey], flag = wx.ALIGN_CENTER_VERTICAL, pos =(row, 0))
  1790. gridSizer.Add(item = self.settings[settKey],
  1791. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5,
  1792. pos =(row, 1))
  1793. row += 1
  1794. gridSizer.AddGrowableCol(1)
  1795. ptsStyleBoxSizer.Add(item = gridSizer, flag = wx.EXPAND)
  1796. # Other settings layout
  1797. otherBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  1798. label =" %s " % _("Other settings"))
  1799. otherBoxSizer = wx.StaticBoxSizer(otherBox, wx.VERTICAL)
  1800. gridSizer = wx.GridBagSizer(vgap = 1, hgap = 1)
  1801. row = 0
  1802. for otherSettName in ["snap_tresh", "max_hist_steps"]:
  1803. gridSizer.Add(item = settsLabels[otherSettName], flag=wx.ALIGN_CENTER_VERTICAL, pos=(row, 0))
  1804. gridSizer.Add(item = self.settings[otherSettName],
  1805. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5,
  1806. pos = (row, 1))
  1807. row += 1
  1808. gridSizer.AddGrowableCol(1)
  1809. otherBoxSizer.Add(item = gridSizer, flag = wx.EXPAND)
  1810. btnSizer = wx.BoxSizer(wx.HORIZONTAL)
  1811. btnSizer.Add(self.btnApply, flag = wx.LEFT | wx.RIGHT, border = 5)
  1812. btnSizer.Add(self.btnSave, flag=wx.LEFT | wx.RIGHT, border=5)
  1813. btnSizer.Add(self.btnClose, flag = wx.LEFT | wx.RIGHT, border = 5)
  1814. sizer.Add(item = styleBoxSizer, flag = wx.EXPAND | wx.ALL, border = 5)
  1815. sizer.Add(item = ptsStyleBoxSizer, flag = wx.EXPAND | wx.ALL, border = 5)
  1816. sizer.Add(item = otherBoxSizer, flag = wx.EXPAND | wx.ALL, border = 5)
  1817. sizer.Add(item = btnSizer, flag = wx.EXPAND | wx.ALL, border = 5, proportion = 0)
  1818. self.SetSizer(sizer)
  1819. sizer.Fit(self)
  1820. def OnSave(self, event):
  1821. """!Button 'Save' pressed"""
  1822. self.UpdateSettings()
  1823. fileSettings = {}
  1824. UserSettings.ReadSettingsFile(settings=fileSettings)
  1825. fileSettings['vnet'] = UserSettings.Get(group='vnet')
  1826. UserSettings.SaveToFile(fileSettings)
  1827. self.Close()
  1828. def UpdateSettings(self):
  1829. UserSettings.Set(group ='vnet', key = "res_style", subkey ='line_width',
  1830. value = self.settings["line_width"].GetValue())
  1831. for settKey, sett in self.colorsSetts.iteritems():
  1832. col = tuple(self.settings[settKey].GetColour())
  1833. UserSettings.Set(group = 'vnet',
  1834. key = sett[0],
  1835. subkey = settKey,
  1836. value = col)
  1837. for settKey, sett in self.sizeSetts.iteritems():
  1838. UserSettings.Set(group = 'vnet', key = sett[0], subkey = settKey,
  1839. value = self.settings[settKey].GetValue())
  1840. UserSettings.Set(group = 'vnet', key = 'res_style', subkey = 'color_table',
  1841. value = self.settings['color_table'].GetStringSelection())
  1842. UserSettings.Set(group = 'vnet', key = 'res_style', subkey = 'invert_colors',
  1843. value = self.settings['invert_colors'].IsChecked())
  1844. self.parent.SetPointDrawSettings()
  1845. if not self.parent.tmp_result or \
  1846. not self.parent.tmpMaps.HasTmpVectMap(self.parent.tmp_result.GetVectMapName()):
  1847. up_map_evt = gUpdateMap(render = False, renderVector = False)
  1848. wx.PostEvent(self.parent.mapWin, up_map_evt)
  1849. elif self.parent.tmp_result.GetRenderLayer():
  1850. cmd = self.parent.GetLayerStyle()
  1851. self.parent.tmp_result.AddRenderLayer(cmd)
  1852. up_map_evt = gUpdateMap(render = True, renderVector = True)
  1853. wx.PostEvent(self.parent.mapWin, up_map_evt)#TODO optimization
  1854. else:
  1855. up_map_evt = gUpdateMap(render = False, renderVector = False)
  1856. wx.PostEvent(self.parent.mapWin, up_map_evt)
  1857. def OnApply(self, event):
  1858. """!Button 'Apply' pressed"""
  1859. self.UpdateSettings()
  1860. #self.Close()
  1861. def OnClose(self, event):
  1862. """!Button 'Cancel' pressed"""
  1863. self.Close()
  1864. class AddLayerDialog(wx.Dialog):
  1865. def __init__(self, parent,id=wx.ID_ANY,
  1866. title =_("Save analysis result"), style=wx.DEFAULT_DIALOG_STYLE):
  1867. """!Save analysis result"""
  1868. wx.Dialog.__init__(self, parent, id, title = _(title), style = style)
  1869. self.panel = wx.Panel(parent = self)
  1870. # text fields and it's captions
  1871. self.vectSel = Select(parent = self.panel, type = 'vector',
  1872. mapsets = [grass.gisenv()['MAPSET']],size = (-1, -1))
  1873. self.vectSellabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  1874. label = _("Name:"))
  1875. # buttons
  1876. self.btnCancel = wx.Button(self.panel, wx.ID_CANCEL)
  1877. self.btnOk = wx.Button(self.panel, wx.ID_OK)
  1878. self.btnOk.SetDefault()
  1879. self.SetInitialSize((400, -1))
  1880. self._layout()
  1881. def _layout(self):
  1882. sizer = wx.BoxSizer(wx.VERTICAL)
  1883. box = wx.StaticBox (parent = self.panel, id = wx.ID_ANY,
  1884. label = "Vector map")
  1885. boxSizer = wx.StaticBoxSizer(box, wx.HORIZONTAL)
  1886. boxSizer.Add(item = self.vectSellabel,
  1887. flag = wx.ALIGN_CENTER_VERTICAL,
  1888. proportion = 0)
  1889. boxSizer.Add(item = self.vectSel, proportion = 1,
  1890. flag = wx.EXPAND | wx.ALL, border = 5)
  1891. sizer.Add(item = boxSizer, proportion = 1,
  1892. flag = wx.EXPAND | wx.ALL, border = 5)
  1893. btnSizer = wx.StdDialogButtonSizer()
  1894. btnSizer.AddButton(self.btnCancel)
  1895. btnSizer.AddButton(self.btnOk)
  1896. btnSizer.Realize()
  1897. sizer.Add(item = btnSizer, proportion = 0,
  1898. flag = wx.ALIGN_RIGHT | wx.ALL, border = 5)
  1899. self.panel.SetSizer(sizer)
  1900. sizer.Fit(self)
  1901. class VnetTmpVectMaps:
  1902. """!Class which creates, stores and destroys all tmp maps created during analysis"""
  1903. def __init__(self, parent):
  1904. self.tmpMaps = [] # temporary maps
  1905. self.parent = parent
  1906. self.mapWin = self.parent.mapWin
  1907. def AddTmpVectMap(self, mapName, msg):
  1908. """!New temporary map
  1909. @return instance of VectMap representing temporary map
  1910. """
  1911. currMapSet = grass.gisenv()['MAPSET']
  1912. tmpMap = grass.find_file(name = mapName,
  1913. element = 'vector',
  1914. mapset = currMapSet)
  1915. fullName = tmpMap["fullname"]
  1916. # map already exists
  1917. if fullName:
  1918. #TODO move dialog out of class, AddTmpVectMap(self, mapName, overvrite = False)
  1919. dlg = wx.MessageDialog(parent = self.parent,
  1920. message = msg,
  1921. caption = _("Overwrite map layer"),
  1922. style = wx.YES_NO | wx.NO_DEFAULT |
  1923. wx.ICON_QUESTION | wx.CENTRE)
  1924. ret = dlg.ShowModal()
  1925. dlg.Destroy()
  1926. if ret == wx.ID_NO:
  1927. return None
  1928. else:
  1929. fullName = mapName + "@" + currMapSet
  1930. newVectMap = VectMap(self, fullName)
  1931. self.tmpMaps.append(newVectMap)
  1932. return newVectMap
  1933. def HasTmpVectMap(self, vectMapName):
  1934. """
  1935. @param vectMapName name of vector map
  1936. @return True if it contains the map
  1937. @return False if not
  1938. """
  1939. mapValSpl = vectMapName.strip().split("@")
  1940. if len(mapValSpl) > 1:
  1941. mapSet = mapValSpl[1]
  1942. else:
  1943. mapSet = grass.gisenv()['MAPSET']
  1944. mapName = mapValSpl[0]
  1945. fullName = mapName + "@" + mapSet
  1946. for vectTmpMap in self.tmpMaps:
  1947. if vectTmpMap.GetVectMapName() == fullName:
  1948. return True
  1949. return False
  1950. def GetTmpVectMap(self, vectMapName):
  1951. """ Get instance of VectMap with name vectMapName"""
  1952. for vectMap in self.tmpMaps:
  1953. if vectMap.GetVectMapName() == vectMapName.strip():
  1954. return vectMap
  1955. return None
  1956. def RemoveFromTmpMaps(self, vectMap):
  1957. """!Temporary map is removed from the class instance however it is not deleted
  1958. @param vectMap instance of VectMap class to be removed
  1959. @return True if was removed
  1960. @return False if does not contain the map
  1961. """
  1962. try:
  1963. self.tmpMaps.remove(vectMap)
  1964. return True
  1965. except ValueError:
  1966. return False
  1967. def DeleteTmpMap(self, vectMap):
  1968. """!Temporary map is removed from the class and it is deleted
  1969. @param vectMap instance of VectMap class to be deleted
  1970. @return True if was removed
  1971. @return False if does not contain the map
  1972. """
  1973. if vectMap:
  1974. vectMap.DeleteRenderLayer()
  1975. RunCommand('g.remove',
  1976. vect = vectMap.GetVectMapName())
  1977. self.RemoveFromTmpMaps(vectMap)
  1978. return True
  1979. return False
  1980. def DeleteAllTmpMaps(self):
  1981. """Delete all temporary maps in the class"""
  1982. update = False
  1983. for tmpMap in self.tmpMaps:
  1984. RunCommand('g.remove',
  1985. vect = tmpMap.GetVectMapName())
  1986. if tmpMap.DeleteRenderLayer():
  1987. update = True
  1988. return update
  1989. class VectMap:
  1990. """!Represents map
  1991. It can check if it was modified or render it
  1992. """
  1993. def __init__(self, parent, fullName):
  1994. self.fullName = fullName
  1995. self.parent = parent
  1996. self.renderLayer = None
  1997. self.modifTime = None # time, for modification check
  1998. def __del__(self):
  1999. self.DeleteRenderLayer()
  2000. def AddRenderLayer(self, cmd = None):
  2001. """!Add map from map window layers to render """
  2002. existsMap = grass.find_file(name = self.fullName,
  2003. element = 'vector',
  2004. mapset = grass.gisenv()['MAPSET'])
  2005. if not existsMap["name"]:
  2006. self.DeleteRenderLayer()
  2007. return False
  2008. if not cmd:
  2009. cmd = []
  2010. cmd.insert(0, 'd.vect')
  2011. cmd.append('map=%s' % self.fullName)
  2012. if self.renderLayer:
  2013. self.DeleteRenderLayer()
  2014. self.renderLayer = self.parent.mapWin.Map.AddLayer(ltype = "vector", command = cmd,
  2015. name = self.fullName, active = True,
  2016. opacity = 1.0, render = True,
  2017. pos = -1)
  2018. return True
  2019. def DeleteRenderLayer(self):
  2020. """!Remove map from map window layers to render"""
  2021. if self.renderLayer:
  2022. self.parent.mapWin.Map.DeleteLayer(self.renderLayer)
  2023. self.renderLayer = None
  2024. return True
  2025. return False
  2026. def GetRenderLayer(self):
  2027. return self.renderLayer
  2028. def GetVectMapName(self):
  2029. return self.fullName
  2030. def SaveVectMapState(self):
  2031. """!Save modification time for vector map"""
  2032. self.modifTime = self.GetLastModified()
  2033. def VectMapState(self):
  2034. """!Checks if map was modified
  2035. @return -1 - if no modification time was saved
  2036. @return 0 - if map was modified
  2037. @return 1 - if map was not modified
  2038. """
  2039. if self.modifTime is None:
  2040. return -1
  2041. if self.modifTime != self.GetLastModified():
  2042. return 0
  2043. return 1
  2044. def GetLastModified(self):
  2045. """!Get modification time
  2046. @return MAP DATE time string from vector map head file
  2047. """
  2048. mapValSpl = self.fullName.split("@")
  2049. mapSet = mapValSpl[1]
  2050. mapName = mapValSpl[0]
  2051. headPath = os.path.join(grass.gisenv()['GISDBASE'],
  2052. grass.gisenv()['LOCATION_NAME'],
  2053. mapSet,
  2054. "vector",
  2055. mapName,
  2056. "head")
  2057. try:
  2058. head = open(headPath, 'r')
  2059. for line in head.readlines():
  2060. i = line.find('MAP DATE:', )
  2061. if i == 0:
  2062. head.close()
  2063. return line.split(':', 1)[1].strip()
  2064. head.close()
  2065. return ""
  2066. except IOError:
  2067. return ""
  2068. class History:
  2069. """!Class which reads and saves history data (based on gui.core.settings Settings class file save/load)"""
  2070. def __init__(self, parent):
  2071. # max number of steps in history (zero based)
  2072. self.maxHistSteps = 3
  2073. # current history step
  2074. self.currHistStep = 0
  2075. # number of steps saved in history
  2076. self.histStepsNum = 0
  2077. # dict contains data saved in history for current history step
  2078. self.currHistStepData = {}
  2079. # buffer for data to be saved into history
  2080. self.newHistStepData = {}
  2081. self.histFile = grass.tempfile()
  2082. # key/value separator
  2083. self.sep = ';'
  2084. def __del__(self):
  2085. grass.try_remove(self.histFile)
  2086. def GetNext(self):
  2087. """!Go one step forward in history"""
  2088. self.currHistStep -= 1
  2089. self.currHistStepData.clear()
  2090. self.currHistStepData = self._getHistStepData(self.currHistStep)
  2091. return self.currHistStepData
  2092. def GetPrev(self):
  2093. """!Go one step back in history"""
  2094. self.currHistStep += 1
  2095. self.currHistStepData.clear()
  2096. self.currHistStepData = self._getHistStepData(self.currHistStep)
  2097. return self.currHistStepData
  2098. def GetStepsNum(self):
  2099. """!Get number of steps saved in history"""
  2100. return self.histStepsNum
  2101. def GetCurrHistStep(self):
  2102. """!Get current history step"""
  2103. return self.currHistStep
  2104. def Add(self, key, subkey, value):
  2105. """!Add new data into buffer"""
  2106. if key not in self.newHistStepData:
  2107. self.newHistStepData[key] = {}
  2108. if type(subkey) == types.ListType:
  2109. if subkey[0] not in self.newHistStepData[key]:
  2110. self.newHistStepData[key][subkey[0]] = {}
  2111. self.newHistStepData[key][subkey[0]][subkey[1]] = value
  2112. else:
  2113. self.newHistStepData[key][subkey] = value
  2114. def SaveHistStep(self):
  2115. """!Create new history step with data in buffer"""
  2116. self.maxHistSteps = UserSettings.Get(group ='vnet',
  2117. key = 'other',
  2118. subkey = 'max_hist_steps')
  2119. self.currHistStep = 0
  2120. newHistFile = grass.tempfile()
  2121. newHist = open(newHistFile, "w")
  2122. self._saveNewHistStep(newHist)
  2123. oldHist = open(self.histFile)
  2124. removedHistData = self._savePreviousHist(newHist, oldHist)
  2125. oldHist.close()
  2126. newHist.close()
  2127. grass.try_remove(self.histFile)
  2128. self.histFile = newHistFile
  2129. self.newHistStepData.clear()
  2130. return removedHistData
  2131. def _savePreviousHist(self, newHist, oldHist):
  2132. """!Save previous history into new file"""
  2133. newHistStep = False
  2134. removedHistData = {}
  2135. newHistStepsNum = self.histStepsNum
  2136. for line in oldHist.readlines():
  2137. if not line.strip():
  2138. newHistStep = True
  2139. newHistStepsNum += 1
  2140. continue
  2141. if newHistStep:
  2142. newHistStep = False
  2143. line = line.split("=")
  2144. line[1] = str(newHistStepsNum)
  2145. line = "=".join(line)
  2146. if newHistStepsNum >= self.maxHistSteps:
  2147. removedHistStep = removedHistData[line] = {}
  2148. continue
  2149. else:
  2150. newHist.write('%s%s%s' % (os.linesep, line, os.linesep))
  2151. self.histStepsNum = newHistStepsNum
  2152. else:
  2153. if newHistStepsNum >= self.maxHistSteps:
  2154. self._parseLine(line, removedHistStep)
  2155. else:
  2156. newHist.write('%s' % line)
  2157. return removedHistData
  2158. def _saveNewHistStep(self, newHist):
  2159. """!Save buffer (new step) data into file"""
  2160. newHist.write('%s%s%s' % (os.linesep, "history step=0", os.linesep))
  2161. for key in self.newHistStepData.keys():
  2162. subkeys = self.newHistStepData[key].keys()
  2163. newHist.write('%s%s' % (key, self.sep))
  2164. for idx in range(len(subkeys)):
  2165. value = self.newHistStepData[key][subkeys[idx]]
  2166. if type(value) == types.DictType:
  2167. if idx > 0:
  2168. newHist.write('%s%s%s' % (os.linesep, key, self.sep))
  2169. newHist.write('%s%s' % (subkeys[idx], self.sep))
  2170. kvalues = self.newHistStepData[key][subkeys[idx]].keys()
  2171. srange = range(len(kvalues))
  2172. for sidx in srange:
  2173. svalue = self._parseValue(self.newHistStepData[key][subkeys[idx]][kvalues[sidx]])
  2174. newHist.write('%s%s%s' % (kvalues[sidx], self.sep, svalue))
  2175. if sidx < len(kvalues) - 1:
  2176. newHist.write('%s' % self.sep)
  2177. else:
  2178. if idx > 0 and \
  2179. type( self.newHistStepData[key][subkeys[idx - 1]]) == types.DictType:
  2180. newHist.write('%s%s%s' % (os.linesep, key, self.sep))
  2181. value = self._parseValue(self.newHistStepData[key][subkeys[idx]])
  2182. newHist.write('%s%s%s' % (subkeys[idx], self.sep, value))
  2183. if idx < len(subkeys) - 1 and \
  2184. type(self.newHistStepData[key][subkeys[idx + 1]]) != types.DictType:
  2185. newHist.write('%s' % self.sep)
  2186. newHist.write(os.linesep)
  2187. self.histStepsNum = 0
  2188. def _parseValue(self, value, read = False):
  2189. """!Parse value"""
  2190. if read: # -> read data (cast values)
  2191. if value:
  2192. if value[0] == '[' and value[-1] == ']':# TODO, possible wrong interpretation
  2193. value = value[1:-1].split(',')
  2194. value = map(self._castValue, value)
  2195. return value
  2196. if value == 'True':
  2197. value = True
  2198. elif value == 'False':
  2199. value = False
  2200. elif value == 'None':
  2201. value = None
  2202. elif ':' in value: # -> color
  2203. try:
  2204. value = tuple(map(int, value.split(':')))
  2205. except ValueError: # -> string
  2206. pass
  2207. else:
  2208. try:
  2209. value = int(value)
  2210. except ValueError:
  2211. try:
  2212. value = float(value)
  2213. except ValueError:
  2214. pass
  2215. else: # -> write data
  2216. if type(value) == type(()): # -> color
  2217. value = str(value[0]) + ':' +\
  2218. str(value[1]) + ':' + \
  2219. str(value[2])
  2220. return value
  2221. def _castValue(self, value):
  2222. """!Cast value"""
  2223. try:
  2224. value = int(value)
  2225. except ValueError:
  2226. try:
  2227. value = float(value)
  2228. except ValueError:
  2229. value = value[1:-1]
  2230. return value
  2231. def _getHistStepData(self, histStep):
  2232. """!Load data saved in history step"""
  2233. hist = open(self.histFile)
  2234. histStepData = {}
  2235. newHistStep = False
  2236. isSearchedHistStep = False
  2237. for line in hist.readlines():
  2238. if not line.strip() and isSearchedHistStep:
  2239. break
  2240. elif not line.strip():
  2241. newHistStep = True
  2242. continue
  2243. elif isSearchedHistStep:
  2244. self._parseLine(line, histStepData)
  2245. if newHistStep:
  2246. line = line.split("=")
  2247. if int(line[1]) == histStep:
  2248. isSearchedHistStep = True
  2249. newHistStep = False
  2250. hist.close()
  2251. return histStepData
  2252. def _parseLine(self, line, histStepData):
  2253. """!Parse line in file with history"""
  2254. line = line.rstrip('%s' % os.linesep).split(self.sep)
  2255. key = line[0]
  2256. kv = line[1:]
  2257. idx = 0
  2258. subkeyMaster = None
  2259. if len(kv) % 2 != 0: # multiple (e.g. nviz)
  2260. subkeyMaster = kv[0]
  2261. del kv[0]
  2262. idx = 0
  2263. while idx < len(kv):
  2264. if subkeyMaster:
  2265. subkey = [subkeyMaster, kv[idx]]
  2266. else:
  2267. subkey = kv[idx]
  2268. value = kv[idx+1]
  2269. value = self._parseValue(value, read = True)
  2270. if key not in histStepData:
  2271. histStepData[key] = {}
  2272. if type(subkey) == types.ListType:
  2273. if subkey[0] not in histStepData[key]:
  2274. histStepData[key][subkey[0]] = {}
  2275. histStepData[key][subkey[0]][subkey[1]] = value
  2276. else:
  2277. histStepData[key][subkey] = value
  2278. idx += 2
  2279. def DeleteNewHistStepData(self):
  2280. """!Delete buffer data for new history step"""
  2281. self.newHistStepData.clear()
  2282. class VnetStatusbar(wx.StatusBar):
  2283. """!Extends wx.StatusBar class with functionality to show multiple messages with the highest priority"""
  2284. def __init__(self, parent, style, id = wx.ID_ANY, **kwargs):
  2285. wx.StatusBar.__init__(self, parent, id, style, **kwargs)
  2286. self.maxPriority = 0
  2287. self.statusItems = []
  2288. def AddStatusItem(self, text, key, priority):
  2289. """!Add new item to show
  2290. @param text - string to show
  2291. @param key - item identifier, if already contains
  2292. item with same identifier, overwrites this item
  2293. @param priority - only items with highest priority are showed
  2294. """
  2295. statusTextItem = {
  2296. 'text' : text,
  2297. 'key' : key,
  2298. 'priority' : priority
  2299. }
  2300. for item in self.statusItems:
  2301. if item['key'] == statusTextItem['key']:
  2302. self.statusItems.remove(item)
  2303. self.statusItems.append(statusTextItem)
  2304. if self.maxPriority < statusTextItem['priority']:
  2305. self.maxPriority = statusTextItem['priority']
  2306. self._updateStatus()
  2307. def _updateStatus(self):
  2308. currStatusText = ''
  2309. for item in reversed(self.statusItems):
  2310. if item['priority'] == self.maxPriority:
  2311. if currStatusText:
  2312. currStatusText += '; '
  2313. currStatusText += item['text']
  2314. wx.StatusBar.SetStatusText(self, currStatusText)
  2315. def RemoveStatusItem(self, key):
  2316. """!Remove item
  2317. @param key - item identifier
  2318. """
  2319. update = False
  2320. for iItem, item in enumerate(self.statusItems):
  2321. if item['key'] == key:
  2322. if item['priority'] == self.maxPriority:
  2323. update = True
  2324. self.statusItems.pop(iItem)
  2325. if update:
  2326. for item in self.statusItems:
  2327. self.maxPriority = 0
  2328. if self.maxPriority < item['priority']:
  2329. self.maxPriority = item['priority']
  2330. self._updateStatus()