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