dialogs.py 116 KB

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