12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678 |
- """
- @package nviz.mapwindow
- @brief wxGUI 3D view mode (map canvas)
- This module implements 3D visualization mode for map display.
- List of classes:
- - mapwindow::NvizThread
- - mapwindow::GLWindow
- (C) 2008-2011 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- @author Martin Landa <landa.martin gmail.com> (Google SoC 2008/2010)
- @author Anna Kratochvilova <kratochanna gmail.com> (Google SoC 2011)
- """
- import os
- import sys
- import time
- import copy
- import math
- import types
- from threading import Thread
- import wx
- from wx.lib.newevent import NewEvent
- from wx import glcanvas
- from wx.glcanvas import WX_GL_RGBA, WX_GL_DOUBLEBUFFER, WX_GL_DEPTH_SIZE
- import grass.script as grass
- from grass.pydispatch.signal import Signal
- from core.gcmd import GMessage, GException, GError
- from core.debug import Debug
- from mapwin.base import MapWindowBase
- from core.settings import UserSettings
- from nviz.workspace import NvizSettings
- from nviz.animation import Animation
- from nviz import wxnviz
- from core.globalvar import CheckWxVersion
- from core.utils import str2rgb, _
- from core.giface import Notification
- wxUpdateProperties, EVT_UPDATE_PROP = NewEvent()
- wxUpdateView, EVT_UPDATE_VIEW = NewEvent()
- wxUpdateLight, EVT_UPDATE_LIGHT = NewEvent()
- wxUpdateCPlane, EVT_UPDATE_CPLANE = NewEvent()
- class NvizThread(Thread):
- def __init__(self, log, progressbar, window):
- Thread.__init__(self)
- Debug.msg(5, "NvizThread.__init__():")
- self.log = log
- self.progressbar = progressbar
- self.window = window
- self._display = None
- self.setDaemon(True)
- def run(self):
- self._display = wxnviz.Nviz(self.log, self.progressbar)
- def GetDisplay(self):
- """Get display instance"""
- return self._display
- class GLWindow(MapWindowBase, glcanvas.GLCanvas):
- """OpenGL canvas for Map Display Window"""
- def __init__(self, parent, giface, frame, Map, tree, lmgr, id=wx.ID_ANY):
- """All parameters except for id are mandatory. The todo is to remove
- them completely."""
- self.parent = parent
- self.tree = tree
- self.lmgr = lmgr
- self.frame = frame
- # for wxGTK we need to set WX_GL_DEPTH_SIZE to draw vectors correctly
- # but we don't know the right value
- # in wxpython 2.9, there is IsDisplaySupported
- if CheckWxVersion(version=[2, 8, 11]) and \
- sys.platform not in ('win32', 'darwin'):
- depthBuffer = int(
- UserSettings.Get(
- group='display',
- key='nvizDepthBuffer',
- subkey='value'))
- attribs = [
- WX_GL_RGBA,
- WX_GL_DOUBLEBUFFER,
- WX_GL_DEPTH_SIZE,
- depthBuffer,
- 0]
- glcanvas.GLCanvas.__init__(self, parent, id, attribList=attribs)
- else:
- glcanvas.GLCanvas.__init__(self, parent, id)
- MapWindowBase.__init__(self, parent=parent, giface=giface, Map=Map)
- self.Hide()
- # TODO: same signals as in BufferedMapWindow
- # same interface is good, but how to ensure same names
- # or avoid duplication, define in map window base class?
- # Emitted when mouse us moving (mouse motion event)
- # Parametres are x and y of the mouse position in map (cell) units
- self.mouseMoving = Signal('GLWindow.mouseMoving')
- # Emitted when the zoom history stack is emptied
- self.zoomHistoryUnavailable = Signal('GLWindow.zoomHistoryUnavailable')
- # Emitted when the zoom history stack is not empty
- self.zoomHistoryAvailable = Signal('GLWindow.zoomHistoryAvailable')
- # Emitted when map was queried, parameters x, y are mouse coordinates
- # TODO: change pixel coordinates to map coordinates (using Pixel2Cell)
- self.mapQueried = Signal('GLWindow.mapQueried')
- self.init = False
- self.initView = False
- self.context = None
- if CheckWxVersion(version=[2, 9]):
- self.context = glcanvas.GLContext(self)
- # render mode
- self.render = {'quick': False,
- # do not render vector lines in quick mode
- 'vlines': False,
- 'vpoints': False,
- 'overlays': False}
- self.mouse = {
- 'use': 'pointer'
- }
- # list of loaded map layers (layer tree items)
- self.layers = list()
- # list of constant surfaces
- self.constants = list()
- # id of base surface (when vector is loaded and no surface exist)
- self.baseId = -1
- # list of cutting planes
- self.cplanes = list()
- # list of query points
- self.qpoints = list()
- # list of past views
- self.viewhistory = []
- self.saveHistory = False
- # offset for dialog (e.g. DisplayAttributesDialog)
- self.dialogOffset = 5
- # overlays
- self.overlays = {}
- self.imagelist = []
- self.overlay = wx.Overlay()
- #self.pdc = wx.PseudoDC()
- self.dragid = -1
- self.hitradius = 5
- # layer manager toolwindow
- self.toolWin = None
- if self.lmgr:
- self.log = self.lmgr._gconsole
- logerr = self.lmgr._gconsole.GetLog(err=True)
- logmsg = self.lmgr._gconsole.GetLog()
- else:
- self.log = logmsg = sys.stdout
- logerr = sys.stderr
- # create nviz instance - use display region instead of computational
- os.environ['GRASS_REGION'] = self.Map.SetRegion(
- windres=True, windres3=True)
- self.nvizThread = NvizThread(logerr,
- self.parent.GetProgressBar(),
- logmsg)
- self.nvizThread.start()
- time.sleep(.1)
- self._display = self.nvizThread.GetDisplay()
- # GRASS_REGION needed only for initialization
- del os.environ['GRASS_REGION']
- self.img = wx.Image(self.Map.mapfile, wx.BITMAP_TYPE_ANY)
- # size of MapWindow, to avoid resizing if size is the same
- self.size = (0, 0)
- # default values
- self.nvizDefault = NvizSettings()
- self.view = copy.deepcopy(
- UserSettings.Get(
- group='nviz',
- key='view')) # copy
- self.iview = UserSettings.Get(
- group='nviz', key='view', settings_type='internal')
- self.light = copy.deepcopy(
- UserSettings.Get(
- group='nviz',
- key='light')) # copy
- self.decoration = self.nvizDefault.SetDecorDefaultProp(type='arrow')
- self.decoration['scalebar'] = []
- self.decoration['arrow']['size'] = self._getDecorationSize()
- self.fly = self.InitFly()
- # timer for flythrough
- self.timerFly = wx.Timer(self, id=wx.NewId())
- # timer for animations
- self.timerAnim = wx.Timer(self, id=wx.NewId())
- self.animation = Animation(mapWindow=self, timer=self.timerAnim)
- self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
- self.Bind(wx.EVT_SIZE, self.OnSize)
- self.Bind(wx.EVT_PAINT, self.OnPaint)
- self._bindMouseEvents()
- self.Bind(EVT_UPDATE_PROP, self.UpdateMapObjProperties)
- self.Bind(EVT_UPDATE_VIEW, self.OnUpdateView)
- self.Bind(EVT_UPDATE_LIGHT, self.UpdateLight)
- self.Bind(EVT_UPDATE_CPLANE, self.OnUpdateCPlane)
- self.Bind(wx.EVT_TIMER, self.OnTimerAnim, self.timerAnim)
- self.Bind(wx.EVT_TIMER, self.OnTimerFly, self.timerFly)
- self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
- self.Bind(wx.EVT_KEY_UP, self.OnKeyUp)
- self.Bind(wx.EVT_CLOSE, self.OnClose)
- if CheckWxVersion(version=[2, 8, 11]) and \
- sys.platform not in ('win32', 'darwin'):
- wx.CallLater(3000, self._warningDepthBuffer)
- # cplanes cannot be initialized now
- wx.CallAfter(self.InitCPlanes)
- def _warningDepthBuffer(self):
- if not self.initView:
- message = _("Opening 3D view was not successful. "
- "Please try to change the value of depth buffer "
- "in GUI Settings dialog > tab Map Display > Advanced "
- "and restart GUI.")
- GMessage(message)
- def InitFly(self):
- """Initialize fly through dictionary"""
- fly = {'interval': 10, # interval for timerFly
- 'value': [0, 0, 0], # calculated values for navigation
- 'mode': 0, # fly through mode (0, 1)
- 'exag': { # sensitivity
- 'move': UserSettings.Get(group='nviz', key='fly', subkey=['exag', 'move']),
- 'turn': UserSettings.Get(group='nviz', key='fly', subkey=['exag', 'turn'])},
- 'exagMultiplier': 3, # speed up by Shift
- 'flySpeed': 4, # speed of flying
- 'mouseControl': None, # if mouse or keys are used
- # virtual mouse position when using arrows
- 'pos': {'x': 0, 'y': 0},
- 'arrowStep': 50, # step in pixels (when using arrows)
- 'flySpeedStep': 2,
- }
- return fly
- def OnTimerFly(self, event):
- """Fly event was emitted, move the scene"""
- if self.mouse['use'] != 'fly':
- return
- if self.fly['mouseControl']:
- mx, my = self.ComputeMxMy(*self.mouse['tmp'])
- else:
- mx, my = self.ComputeMxMy(
- self.fly['pos']['x'],
- self.fly['pos']['y'])
- self.ComputeFlyValues(mx=mx, my=my)
- self._display.FlyThrough(
- flyInfo=self.fly['value'],
- mode=self.fly['mode'],
- exagInfo=self.fly['exag'])
- self.ChangeInnerView()
- self.render['quick'] = True
- self.Refresh(False)
- def ComputeMxMy(self, x, y):
- """Compute values for flythrough navigation
- (ComputeFlyValues should follow).
- Based on visualization/nviz/src/togl_flythrough.c.
- :param x,y: screen coordinates
- """
- sx, sy = self.GetClientSize()
- dx = dy = 0.01
- mx = 2 * (float(x) / sx) - 1
- my = 2 * (float(y) / sy) - 1
- if mx < - dx:
- mx += dx
- elif mx > dx:
- mx -= dx
- else:
- mx = 0.0 # ?
- if my < - dy:
- my += dy
- elif my > dy:
- my -= dy
- else:
- my = 0.0
- mx = mx / (1.0 - dx)
- my = my / (1.0 - dy)
- # Quadratic seems smoother
- mx *= abs(mx)
- my *= abs(my)
- return mx, my
- def ComputeFlyValues(self, mx, my):
- """Compute parameters for fly-through navigation
- :param mx,my: results from ComputeMxMy method
- """
- self.fly['value'] = [0, 0, 0]
- if self.fly['mode'] == 0:
- self.fly['value'][0] = self.fly[
- 'flySpeed'] * self.fly['interval'] / 1000. # forward */
- self.fly['value'][
- 1] = mx * 0.1 * self.fly['interval'] / 1000. # heading
- self.fly['value'][
- 2] = my * 0.1 * self.fly['interval'] / 1000. # pitch
- else:
- self.fly['value'][0] = mx * 100.0 * self.fly['interval'] / 1000.
- self.fly['value'][2] = - my * 100.0 * self.fly['interval'] / 1000.
- def ChangeFlySpeed(self, increase):
- """Increase/decrease flight spped"""
- if increase:
- self.fly['flySpeed'] += self.fly['flySpeedStep']
- else:
- self.fly['flySpeed'] -= self.fly['flySpeedStep']
- def __del__(self):
- """Stop timers if running, unload data"""
- self.StopTimer(self.timerAnim)
- self.StopTimer(self.timerFly)
- self.UnloadDataLayers(force=True)
- def StopTimer(self, timer):
- """Stop timer if running"""
- if timer.IsRunning():
- timer.Stop()
- def _bindMouseEvents(self):
- self.Bind(wx.EVT_MOUSE_EVENTS, self.OnMouseAction)
- self.Bind(wx.EVT_MOTION, self.OnMotion)
- def InitCPlanes(self):
- """Initialize cutting planes list"""
- for i in range(self._display.GetCPlanesCount()):
- cplane = copy.deepcopy(
- UserSettings.Get(
- group='nviz',
- key='cplane'))
- cplane['on'] = False
- self.cplanes.append(cplane)
- def SetToolWin(self, toolWin):
- """Sets reference to nviz toolwindow in layer manager"""
- self.toolWin = toolWin
- def GetToolWin(self):
- """Returns reference to nviz toolwindow in layer manager"""
- return self.toolWin
- def OnClose(self, event):
- self.StopTimer(self.timerAnim)
- self.StopTimer(self.timerFly)
- # cleanup when window actually closes (on quit) and not just is hidden
- self.UnloadDataLayers(force=True)
- def OnEraseBackground(self, event):
- pass # do nothing, to avoid flashing on MSW
- def OnSize(self, event):
- size = self.GetClientSize()
- if CheckWxVersion(version=[2, 9]):
- context = self.context
- else:
- context = self.GetContext()
- if self.size != size \
- and context:
- Debug.msg(3, "GLCanvas.OnSize(): w = %d, h = %d" %
- (size.width, size.height))
- if CheckWxVersion(version=[2, 9]):
- self.SetCurrent(self.context)
- else:
- self.SetCurrent()
- self._display.ResizeWindow(size.width,
- size.height)
- # reposition checkbox in statusbar
- self.parent.StatusbarReposition()
- # update statusbar
- self.parent.StatusbarUpdate()
- self.size = size
- event.Skip()
- def OnPaint(self, event):
- Debug.msg(1, "GLCanvas.OnPaint()")
- self.render['overlays'] = True
- dc = wx.PaintDC(self)
- self.DoPaint()
- def DoPaint(self):
- if CheckWxVersion(version=[2, 9]):
- self.SetCurrent(self.context)
- else:
- self.SetCurrent()
- if not self.initView:
- self._display.InitView()
- self.initView = True
- self.LoadDataLayers()
- self.UnloadDataLayers()
- if not self.init:
- self.ResetView()
- if hasattr(self.lmgr, "nviz"):
- self.lmgr.nviz.UpdatePage('view')
- self.lmgr.nviz.UpdatePage('light')
- self.lmgr.nviz.UpdatePage('cplane')
- self.lmgr.nviz.UpdatePage('decoration')
- self.lmgr.nviz.UpdatePage('animation')
- layer = self.tree.GetSelectedLayer(
- multi=False, checkedOnly=True)
- if layer:
- layer = self.tree.GetLayerInfo(layer, key='maplayer')
- if layer.type == 'raster':
- self.lmgr.nviz.UpdatePage('surface')
- self.lmgr.nviz.UpdatePage('fringe')
- elif layer.type == 'vector':
- self.lmgr.nviz.UpdatePage('vector')
- self.lmgr.nviz.UpdateSettings()
- # update widgets
- win = self.lmgr.nviz.FindWindowById(
- self.lmgr.nviz.win['vector']['lines']['surface'])
- win.SetItems(self.GetLayerNames('raster'))
- self.init = True
- self.UpdateMap()
- def DrawImages(self):
- """Draw overlay image"""
- for texture in self.imagelist:
- if texture.IsActive():
- texture.Draw()
- def UpdateOverlays(self):
- """Renders overlays (legend, text).
- Once this is done _onUpdateOverlays is called"""
- self.Map.ChangeMapSize(self.GetClientSize())
- self.Map.RenderOverlays(force=True)
- def _onUpdateOverlays(self):
- """Converts rendered overlay files and text labels to wx.Image
- and then to textures so that they can be rendered by OpenGL.
- Updates self.imagelist"""
- # update images (legend and text)
- for oid, overlay in self.overlays.iteritems():
- if not overlay.IsShown() or overlay.name in ('barscale', 'northarrow'):
- continue
- if oid not in [t.GetId() for t in self.imagelist]: # new
- self.CreateTexture(overlay=overlay)
- else:
- for t in self.imagelist:
- if t.GetId() == oid: # check if it is the same
- if not t.Corresponds(overlay):
- self.imagelist.remove(t)
- t = self.CreateTexture(overlay=overlay)
- self.Refresh()
- def CreateTexture(self, overlay):
- """Create texture from overlay image"""
- texture = wxnviz.ImageTexture(
- filepath=overlay.layer.mapfile, overlayId=overlay.id,
- coords=list(overlay.coords), cmd=overlay.GetCmd())
- if not texture.textureId: # texture too big
- GMessage(
- parent=self, message=_(
- "Image is too large, your OpenGL implementation "
- "supports maximum texture size %d px.") %
- texture.maxSize)
- return texture
- self.imagelist.append(texture)
- return texture
- def ClearTextures(self):
- self.imagelist = []
- def FindObjects(self, mouseX, mouseY, radius):
- """Find object which was clicked on"""
- for texture in self.imagelist:
- if texture.HitTest(mouseX, mouseY, radius):
- return texture.id
- return -1
- def OnTimerAnim(self, event):
- self.animation.Update()
- def GetAnimation(self):
- return self.animation
- def OnKeyDown(self, event):
- """Key was pressed.
- Used for fly-through mode.
- """
- if not self.mouse['use'] == 'fly':
- return
- key = event.GetKeyCode()
- if key == wx.WXK_CONTROL: # Mac ?
- self.fly['mode'] = 1
- elif key == wx.WXK_SHIFT:
- self.fly['exag']['move'] *= self.fly['exagMultiplier']
- self.fly['exag']['turn'] *= self.fly['exagMultiplier']
- elif key == wx.WXK_ESCAPE and self.timerFly.IsRunning() and not self.fly['mouseControl']:
- self.StopTimer(self.timerFly)
- self.fly['mouseControl'] = None
- self.render['quick'] = False
- self.Refresh(False)
- elif key in (wx.WXK_UP, wx.WXK_DOWN, wx.WXK_LEFT, wx.WXK_RIGHT):
- if not self.fly['mouseControl']:
- if not self.timerFly.IsRunning():
- sx, sy = self.GetClientSize()
- self.fly['pos']['x'] = sx / 2
- self.fly['pos']['y'] = sy / 2
- self.fly['mouseControl'] = False # controlled by keyboard
- self.timerFly.Start(self.fly['interval'])
- self.ProcessFlyByArrows(keyCode=key)
- # change speed of flight when using mouse
- else:
- if key == wx.WXK_UP:
- self.ChangeFlySpeed(increase=True)
- elif key == wx.WXK_DOWN:
- self.ChangeFlySpeed(increase=False)
- elif key in (wx.WXK_HOME, wx.WXK_PAGEUP) and self.timerFly.IsRunning():
- self.ChangeFlySpeed(increase=True)
- elif key in (wx.WXK_END, wx.WXK_PAGEDOWN) and self.timerFly.IsRunning():
- self.ChangeFlySpeed(increase=False)
- event.Skip()
- def ProcessFlyByArrows(self, keyCode):
- """Process arrow key during fly-through"""
- step = self.fly['arrowStep']
- if keyCode == wx.WXK_UP:
- self.fly['pos']['y'] -= step
- elif keyCode == wx.WXK_DOWN:
- self.fly['pos']['y'] += step
- elif keyCode == wx.WXK_LEFT:
- self.fly['pos']['x'] -= step
- elif keyCode == wx.WXK_RIGHT:
- self.fly['pos']['x'] += step
- def OnKeyUp(self, event):
- """Key was released.
- Used for fly-through mode.
- """
- if not self.mouse['use'] == 'fly':
- return
- key = event.GetKeyCode()
- if key == wx.WXK_CONTROL: # Mac ?
- self.fly['mode'] = 0
- elif key == wx.WXK_SHIFT:
- self.fly['exag']['move'] = math.floor(
- self.fly['exag']['move'] / self.fly['exagMultiplier'])
- self.fly['exag']['turn'] = math.floor(
- self.fly['exag']['turn'] / self.fly['exagMultiplier'])
- event.Skip()
- def OnMouseAction(self, event):
- """Handle mouse events"""
- # zoom with mouse wheel
- if event.GetWheelRotation() != 0:
- self.OnMouseWheel(event)
- # left mouse button pressed
- elif event.LeftDown():
- self.OnLeftDown(event)
- # left mouse button released
- elif event.LeftUp():
- self.OnLeftUp(event)
- # dragging
- elif event.Dragging():
- self.OnDragging(event)
- # double click
- elif event.ButtonDClick():
- self.OnDClick(event)
- elif event.Moving():
- pixelCoordinates = event.GetPosition()
- coordinates = self.Pixel2Cell(pixelCoordinates)
- # coordinates are none when no map is loaded
- # TODO: handle in more clever way: check the state
- if coordinates is not None:
- self.mouseMoving.emit(x=coordinates[0], y=coordinates[1])
- event.Skip()
- def OnMouseWheel(self, event):
- """Change perspective"""
- if UserSettings.Get(group='display',
- key='mouseWheelZoom',
- subkey='selection') == 2:
- event.Skip()
- return
- wheel = event.GetWheelRotation()
- Debug.msg(5, "GLWindow.OnMouseWheel(): wheel = %d" % wheel)
- if self.timerFly.IsRunning() and self.fly['mouseControl']:
- if wheel > 0:
- self.ChangeFlySpeed(increase=True)
- else:
- self.ChangeFlySpeed(increase=False)
- else:
- if UserSettings.Get(group='display',
- key='scrollDirection',
- subkey='selection'):
- wheel *= -1
- self.DoZoom(zoomtype=wheel, pos=event.GetPosition())
- # update statusbar
- # self.parent.StatusbarUpdate()
- def OnLeftDown(self, event):
- """On left mouse down"""
- self.mouse['begin'] = event.GetPosition()
- self.mouse['tmp'] = event.GetPosition()
- if self.mouse['use'] == "lookHere":
- size = self.GetClientSize()
- self._display.LookHere(
- self.mouse['begin'][0],
- size[1] - self.mouse['begin'][1])
- focus = self._display.GetFocus()
- for i, coord in enumerate(('x', 'y', 'z')):
- self.iview['focus'][coord] = focus[i]
- self.saveHistory = True
- self.Refresh(False)
- toggle = self.lmgr.nviz.FindWindowByName('here')
- toggle.SetValue(False)
- self.mouse['use'] = 'pointer'
- self.SetNamedCursor('default')
- if self.mouse['use'] == 'arrow':
- pos = event.GetPosition()
- size = self.GetClientSize()
- self.SetDrawArrow((pos[0], size[1] - pos[1]))
- if self.mouse['use'] == 'scalebar':
- pos = event.GetPosition()
- size = self.GetClientSize()
- self.SetDrawScalebar((pos[0], size[1] - pos[1]))
- if self.mouse['use'] == 'pointer':
- # get decoration id
- self.dragid = self.FindObjects(
- self.mouse['tmp'][0],
- self.mouse['tmp'][1],
- self.hitradius)
- if self.mouse['use'] == 'fly':
- if not self.timerFly.IsRunning():
- self.timerFly.Start(self.fly['interval'])
- self.fly['mouseControl'] = True
- event.Skip()
- def OnDragging(self, event):
- if self.mouse['use'] == 'pointer':
- if self.dragid >= 0:
- self.DragItem(self.dragid, event.GetPosition())
- if self.mouse['use'] == 'rotate':
- dx, dy = event.GetX(
- ) - self.mouse['tmp'][0], event.GetY() - self.mouse['tmp'][1]
- angle, x, y, z = self._display.GetRotationParameters(dx, dy)
- self._display.Rotate(angle, x, y, z)
- self.render['quick'] = True
- self.Refresh(False)
- if self.mouse['use'] == 'pan':
- self.FocusPanning(event)
- self.mouse['tmp'] = event.GetPosition()
- event.Skip()
- def Pixel2Cell(self, xyCoords):
- """Convert image coordinates to real word coordinates
- :param xyCoords: image coordinates
- :return: easting, northing
- :return: None on error
- """
- size = self.GetClientSize()
- # UL -> LL
- x, y = xyCoords
- sid, x, y, z = self._display.GetPointOnSurface(x, size[1] - y)
- if not sid:
- return None
- return (x, y)
- def DoZoom(self, zoomtype, pos):
- """Change perspective and focus"""
- prev_value = self.view['persp']['value']
- if zoomtype > 0:
- value = -1 * self.view['persp']['step']
- else:
- value = self.view['persp']['step']
- self.view['persp']['value'] += value
- if self.view['persp']['value'] < 1:
- self.view['persp']['value'] = 1
- elif self.view['persp']['value'] > 180:
- self.view['persp']['value'] = 180
- if prev_value != self.view['persp']['value']:
- if hasattr(self.lmgr, "nviz"):
- self.lmgr.nviz.UpdateSettings()
- x, y = pos[0], self.GetClientSize()[1] - pos[1]
- result = self._display.GetPointOnSurface(x, y)
- if result[0]:
- self._display.LookHere(x, y)
- focus = self._display.GetFocus()
- for i, coord in enumerate(('x', 'y', 'z')):
- self.iview['focus'][coord] = focus[i]
- self._display.SetView(
- self.view['position']['x'],
- self.view['position']['y'],
- self.iview['height']['value'],
- self.view['persp']['value'],
- self.view['twist']['value'])
- self.saveHistory = True
- # redraw map
- self.DoPaint()
- def OnLeftUp(self, event):
- self.mouse['end'] = event.GetPosition()
- if self.mouse["use"] == "query":
- # here changed from 'begin' to 'end' because it is more common
- # behavior used also in 2d map window
- # and moreover we are in left up
- self.mapQueried.emit(x=self.mouse['end'][0],
- y=self.mouse['end'][1])
- elif self.mouse["use"] in ('arrow', 'scalebar'):
- self.lmgr.nviz.FindWindowById(
- self.lmgr.nviz.win['decoration'][
- self.mouse["use"]]['place']).SetValue(False)
- if self.mouse["use"] == 'scalebar':
- scalebarNum = len(self.decoration['scalebar'])
- self.lmgr.nviz.AddScalebar(scalebarNum - 1)
- else:
- self.lmgr.nviz.AddArrow()
- self.mouse['use'] = 'pointer'
- self.SetNamedCursor('default')
- elif self.mouse['use'] == 'pointer':
- if self.dragid >= 0:
- dx = self.mouse['end'][0] - self.mouse['begin'][0]
- dy = self.mouse['end'][1] - self.mouse['begin'][1]
- if self.dragid in self.overlays:
- coords = self.overlays[self.dragid].coords
- self.overlays[
- self.dragid].coords = [
- coords[0] + dx, coords[1] + dy]
- self.dragid = -1
- self.render['quick'] = False
- self.Refresh(False)
- elif self.mouse['use'] == 'rotate':
- self._display.UnsetRotation()
- self.iview['rotation'] = self._display.GetRotationMatrix()
- self.saveHistory = True
- self.render['quick'] = False
- self.Refresh(False)
- elif self.mouse['use'] == 'pan':
- self.saveHistory = True
- self.render['quick'] = False
- self.Refresh(False)
- elif self.mouse['use'] == 'fly':
- if self.fly['mouseControl']:
- self.StopTimer(self.timerFly)
- self.fly['mouseControl'] = None
- # for key in self.iview['dir'].keys():
- #self.iview[''][key] = -1
- # this causes sudden change, but it should be there
- # if hasattr(self.lmgr, "nviz"):
- # self.lmgr.nviz.UpdateSettings()
- self.render['quick'] = False
- self.Refresh(False)
- elif self.mouse['use'] == 'zoom':
- self.DoZoom(zoomtype=self.zoomtype, pos=self.mouse['end'])
- event.Skip()
- def OnDClick(self, event):
- """On mouse double click"""
- if self.mouse['use'] != 'pointer':
- return
- pos = event.GetPosition()
- self.dragid = self.FindObjects(pos[0], pos[1], self.hitradius)
- self.overlayActivated.emit(overlayId=self.dragid)
- def FocusPanning(self, event):
- """Simulation of panning using focus"""
- size = self.GetClientSize()
- id1, x1, y1, z1 = self._display.GetPointOnSurface(
- self.mouse['tmp'][0], size[1] - self.mouse['tmp'][1])
- id2, x2, y2, z2 = self._display.GetPointOnSurface(
- event.GetX(), size[1] - event.GetY())
- if id1 and id1 == id2:
- dx, dy, dz = x2 - x1, y2 - y1, z2 - z1
- focus = self.iview['focus']
- focus['x'], focus['y'], focus['z'] = self._display.GetFocus()
- focus['x'] -= dx
- focus['y'] -= dy
- focus['z'] -= dz
- # update properties
- self.PostViewEvent()
- self.mouse['tmp'] = event.GetPosition()
- self.render['quick'] = True
- self.Refresh(False)
- def HorizontalPanning(self, event):
- """Move all layers in horizontal (x, y) direction.
- Currently not used.
- """
- size = self.GetClientSize()
- id1, x1, y1, z1 = self._display.GetPointOnSurface(
- self.mouse['tmp'][0], size[1] - self.mouse['tmp'][1])
- id2, x2, y2, z2 = self._display.GetPointOnSurface(
- event.GetX(), size[1] - event.GetY())
- if id1 and id1 == id2:
- dx, dy = x2 - x1, y2 - y1
- # find raster and volume
- for item in self.layers:
- mapLayer = self.tree.GetLayerInfo(item, key='maplayer')
- data = self.tree.GetLayerInfo(item, key='nviz')
- if mapLayer.GetType() == 'raster':
- data['surface']['position']['x'] += dx
- data['surface']['position']['y'] += dy
- data['surface']['position']['update'] = None
- # update properties
- evt = wxUpdateProperties(data=data)
- wx.PostEvent(self, evt)
- if event.CmdDown() and id1 == data[
- 'surface']['object']['id']:
- break
- elif mapLayer.GetType() == 'raster_3d':
- if 'x' not in data['volume']['position']:
- data['volume']['position']['x'] = 0
- data['volume']['position']['y'] = 0
- data['volume']['position']['z'] = 0
- data['volume']['position']['x'] += dx
- data['volume']['position']['y'] += dy
- data['volume']['position']['update'] = None
- # update properties
- evt = wxUpdateProperties(data=data)
- wx.PostEvent(self, evt)
- self.mouse['tmp'] = event.GetPosition()
- self.render['quick'] = True
- self.Refresh(False)
- def DragItem(self, id, coords):
- """Drag an overlay decoration item
- """
- if id is None:
- return
- Debug.msg(5, "GLWindow.DragItem(): id=%d" % id)
- x, y = self.mouse['tmp']
- dx = coords[0] - x
- dy = coords[1] - y
- for texture in self.imagelist:
- if texture.id == id:
- texture.MoveTexture(dx, dy)
- self.render['quick'] = True
- self.Refresh(False)
- self.mouse['tmp'] = coords
- def ZoomBack(self):
- """Set previous view in history list
- """
- view = {}
- if len(self.viewhistory) > 1:
- self.viewhistory.pop()
- view = copy.deepcopy(self.viewhistory[-1])
- # disable tool if stack is empty
- if len(self.viewhistory) < 2: # disable tool
- self.zoomHistoryUnavailable.emit()
- # set view and update nviz view page
- self.lmgr.nviz.UpdateState(view=view[0], iview=view[1])
- self.lmgr.nviz.UpdatePage('view')
- # update map
- self.Refresh(False)
- def ViewHistory(self, view, iview):
- """Manages a list of last 10 views
- :param view: view dictionary
- :param iview: view dictionary (internal)
- :return: removed history item if exists (or None)
- """
- removed = None
- hview = copy.deepcopy(view)
- hiview = copy.deepcopy(iview)
- if not (self.viewhistory and self.viewhistory[-1] == (hview, hiview)):
- self.viewhistory.append((hview, hiview))
- if len(self.viewhistory) > 10:
- removed = self.viewhistory.pop(0)
- if removed:
- Debug.msg(4, "GLWindow.ViewHistory(): hist=%s, removed=%s" %
- (self.viewhistory, removed))
- else:
- Debug.msg(4, "GLWindow.ViewHistory(): hist=%s" %
- (self.viewhistory))
- # update toolbar
- if len(self.viewhistory) > 1:
- self.zoomHistoryAvailable.emit()
- else:
- self.zoomHistoryUnavailable.emit()
- return removed
- def ResetViewHistory(self):
- """Reset view history"""
- self.viewhistory = list()
- def GoTo(self, e, n):
- """Focus on given point"""
- w = self.Map.region['w']
- s = self.Map.region['s']
- e -= w
- n -= s
- focus = self.iview['focus']
- focus['x'], focus['y'] = e, n
- self.saveHistory = True
- # update properties
- self.PostViewEvent()
- self.render['quick'] = False
- self.Refresh(False)
- def QuerySurface(self, x, y):
- """Query surface on given position"""
- size = self.GetClientSize()
- result = self._display.QueryMap(x, size[1] - y)
- if result:
- self.qpoints.append((result['x'], result['y'], result['z']))
- self.log.WriteLog("%-30s: %.3f" % (_("Easting"), result['x']))
- self.log.WriteLog("%-30s: %.3f" % (_("Northing"), result['y']))
- self.log.WriteLog("%-30s: %.3f" % (_("Elevation"), result['z']))
- name = ''
- for item in self.layers:
- if self.tree.GetLayerInfo(item, key='maplayer').type == 'raster' and self.tree.GetLayerInfo(
- item, key='nviz')['surface']['object']['id'] == result['id']:
- name = self.tree.GetLayerInfo(item, key='maplayer').name
- self.log.WriteLog("%-30s: %s" % (_("Surface map name"), name))
- self.log.WriteLog(
- "%-30s: %s" %
- (_("Surface map elevation"), result['elevation']))
- self.log.WriteLog("%-30s: %s" %
- (_("Surface map color"), result['color']))
- if len(self.qpoints) > 1:
- prev = self.qpoints[-2]
- curr = self.qpoints[-1]
- dxy = math.sqrt(pow(prev[0] - curr[0], 2) +
- pow(prev[1] - curr[1], 2))
- dxyz = math.sqrt(pow(prev[0] - curr[0], 2) +
- pow(prev[1] - curr[1], 2) +
- pow(prev[2] - curr[2], 2))
- self.log.WriteLog(
- "%-30s: %.3f" %
- (_("XY distance from previous"), dxy))
- self.log.WriteLog(
- "%-30s: %.3f" %
- (_("XYZ distance from previous"), dxyz))
- self.log.WriteLog(
- "%-30s: %.3f" %
- (_("Distance along surface"), self._display.GetDistanceAlongSurface(
- result['id'], (curr[0], curr[1]), (prev[0], prev[1]), useExag=False)))
- self.log.WriteLog(
- "%-30s: %.3f" %
- (_("Distance along exag. surface"), self._display.GetDistanceAlongSurface(
- result['id'], (curr[0], curr[1]), (prev[0], prev[1]), useExag=True)))
- self.log.WriteCmdLog('-' * 80)
- else:
- self.log.WriteLog(_("No point on surface"))
- self.log.WriteCmdLog('-' * 80)
- def PostViewEvent(self, zExag=False):
- """Change view settings"""
- event = wxUpdateView(zExag=zExag)
- wx.PostEvent(self, event)
- def OnQueryVector(self, event):
- """Query vector on given position"""
- self.parent.QueryVector(*event.GetPosition())
- def ChangeInnerView(self):
- """Get current viewdir and viewpoint and set view"""
- view = self.view
- iview = self.iview
- (view['position']['x'], view['position']['y'],
- iview['height']['value']) = self._display.GetViewpointPosition()
- for key, val in zip(('x', 'y', 'z'), self._display.GetViewdir()):
- iview['dir'][key] = val
- iview['dir']['use'] = True
- def OnUpdateView(self, event):
- """Change view settings"""
- if event:
- self.UpdateView(zexag=event.zExag)
- self.saveHistory = True
- if event:
- event.Skip()
- def UpdateView(self, zexag=False):
- """Change view settings"""
- view = self.view
- iview = self.iview
- if zexag and 'value' in view['z-exag']:
- self._display.SetZExag(
- view['z-exag']['value'] /
- iview['z-exag']['llRatio'])
- self._display.SetView(view['position']['x'], view['position']['y'],
- iview['height']['value'],
- view['persp']['value'],
- view['twist']['value'])
- if iview['dir']['use']:
- self._display.SetViewdir(
- iview['dir']['x'],
- iview['dir']['y'],
- iview['dir']['z'])
- elif iview['focus']['x'] != -1:
- self._display.SetFocus(
- self.iview['focus']['x'],
- self.iview['focus']['y'],
- self.iview['focus']['z'])
- if 'rotation' in iview:
- if iview['rotation']:
- self._display.SetRotationMatrix(iview['rotation'])
- else:
- self._display.ResetRotation()
- def UpdateLight(self, event):
- """Change light settings"""
- data = self.light
- self._display.SetLight(
- x=data['position']['x'],
- y=data['position']['y'],
- z=data['position']['z'] / 100.,
- color=data['color'],
- bright=data['bright'] / 100.,
- ambient=data['ambient'] / 100.)
- self._display.DrawLightingModel()
- if event.refresh:
- self.Refresh(False)
- def UpdateMap(self, render=True):
- """Updates the canvas anytime there is a change to the
- underlaying images or to the geometry of the canvas.
- :param render: re-render map composition
- :type render: bool
- """
- start = time.clock()
- self.resize = False
- if self.render['quick'] is False:
- if sys.platform != 'darwin': # causes recursion for some reason on Mac
- self.parent.GetProgressBar().Show()
- self.parent.GetProgressBar().SetRange(2)
- self.parent.GetProgressBar().SetValue(0)
- if self.render['quick'] is False:
- if sys.platform != 'darwin':
- self.parent.GetProgressBar().SetValue(1)
- self._display.Draw(False, -1)
- if self.saveHistory:
- self.ViewHistory(view=self.view, iview=self.iview)
- self.saveHistory = False
- elif self.render['quick'] is True:
- # quick
- mode = wxnviz.DRAW_QUICK_SURFACE | wxnviz.DRAW_QUICK_VOLUME
- if self.render['vlines']:
- mode |= wxnviz.DRAW_QUICK_VLINES
- if self.render['vpoints']:
- mode |= wxnviz.DRAW_QUICK_VPOINTS
- self._display.Draw(True, mode)
- else: # None -> reuse last rendered image
- pass # TODO
- self.SwapBuffers()
- # draw fringe after SwapBuffers, otherwise it don't have to be visible
- # on some computers
- if self.render['quick'] is False:
- self._display.DrawFringe()
- if self.decoration['arrow']['show']:
- self._display.DrawArrow()
- if self.decoration['scalebar']:
- self._display.DrawScalebar()
- if self.imagelist:
- if ((self.render['quick'] and self.dragid > -1) or # during dragging
- (not self.render['quick'] and self.dragid < 0)): # redraw
- self._display.Start2D()
- self.DrawImages()
- stop = time.clock()
- if self.render['quick'] is False:
- if sys.platform != 'darwin':
- self.parent.GetProgressBar().SetValue(2)
- # hide process bar
- self.parent.GetProgressBar().Hide()
- Debug.msg(3, "GLWindow.UpdateMap(): quick = %d, -> time = %g" %
- (self.render['quick'], (stop - start)))
- def EraseMap(self):
- """Erase the canvas
- """
- self._display.EraseMap()
- self.SwapBuffers()
- def _getDecorationSize(self):
- """Get initial size of north arrow/scalebar"""
- size = self._display.GetLongDim() / 8.
- coef = 0.01
- if size < 1:
- coef = 100.
- return int(size * coef) / coef
- def SetDrawArrow(self, pos):
- """North arrow drawing.
- Also, opens Appearance page of nviz notebook (needs refactoring).
- """
- if self._display.SetArrow(pos[0], pos[1],
- self.decoration['arrow']['size'],
- self.decoration['arrow']['color']):
- self._display.DrawArrow()
- # update
- self.decoration['arrow']['show'] = True
- self.decoration['arrow']['position']['x'] = pos[0]
- self.decoration['arrow']['position']['y'] = pos[1]
- self.Refresh(False)
- # this was in mapdisp/frame.py but moved here to be with similar calls
- # such as self.lmgr.nviz.UpdatePage
- # anyway, it need to be handled in some another way
- self.lmgr.nviz.SetPage('decoration')
- def SetDrawScalebar(self, pos):
- """Add scale bar, sets properties and draw"""
- if len(self.decoration['scalebar']) == 0:
- self.decoration['scalebar'].append(
- self.nvizDefault.SetDecorDefaultProp(
- type='scalebar')['scalebar'])
- self.decoration['scalebar'][0]['size'] = self._getDecorationSize()
- else:
- self.decoration['scalebar'].append(
- copy.deepcopy(self.decoration['scalebar'][-1]))
- self.decoration['scalebar'][-1]['id'] += 1
- ret = self._display.SetScalebar(
- self.decoration['scalebar'][-1]['id'],
- pos[0],
- pos[1],
- self.decoration['scalebar'][-1]['size'],
- self.decoration['scalebar'][-1]['color'])
- if ret:
- self._display.DrawScalebar()
- # update
- self.decoration['scalebar'][-1]['position']['x'] = pos[0]
- self.decoration['scalebar'][-1]['position']['y'] = pos[1]
- self.Refresh(False)
- self.lmgr.nviz.SetPage('decoration')
- def IsLoaded(self, item):
- """Check if layer (item) is already loaded
- :param item: layer item
- """
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- data = self.tree.GetLayerInfo(item, key='nviz')
- if not data:
- return 0
- if layer.type == 'raster':
- if 'object' not in data['surface']:
- return 0
- elif layer.type == 'vector':
- if 'object' not in data['vector']['lines'] and \
- 'object' not in data['vector']['points']:
- return 0
- return 1
- def _GetDataLayers(self, item, litems):
- """Return get list of enabled map layers"""
- # load raster & vector maps
- while item and item.IsOk():
- type = self.tree.GetLayerInfo(item, key='type')
- if type == 'group':
- item = self.tree.GetNextItem(item)
- continue
- if not item.IsChecked() or \
- type not in ('raster', 'vector', 'raster_3d'):
- item = self.tree.GetNextItem(item)
- continue
- litems.append(item)
- item = self.tree.GetNextItem(item)
- def LoadDataLayers(self):
- """Load raster/vector from current layer tree
- .. todo::
- volumes
- """
- if not self.tree:
- return
- listOfItems = []
- item = self.tree.GetFirstChild(self.tree.root)[0]
- self._GetDataLayers(item, listOfItems)
- start = time.time()
- while(len(listOfItems) > 0):
- item = listOfItems.pop()
- type = self.tree.GetLayerInfo(item, key='type')
- if item in self.layers:
- continue
- # "raster (double click to set properties)" - tries to load this
- # layer - no idea how to fix it
- if ' ' in self.tree.GetLayerInfo(item, key='maplayer').name:
- return
- try:
- if type == 'raster':
- self.LoadRaster(item)
- elif type == 'raster_3d':
- self.LoadRaster3d(item)
- elif type == 'vector':
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- vInfo = grass.vector_info_topo(layer.GetName())
- if (vInfo['points']) > 0:
- # include vInfo['centroids'] to initially load
- # centroids
- self.LoadVector(item, points=True)
- if (vInfo['lines'] + vInfo['boundaries']) > 0:
- self.LoadVector(item, points=False)
- if vInfo['map3d'] and(
- vInfo['kernels'] + vInfo['faces']) > 0:
- self.LoadVector(item, points=None)
- except GException as e:
- GError(parent=self,
- message=e.value)
- stop = time.time()
- Debug.msg(1, "GLWindow.LoadDataLayers(): time = %f" % (stop - start))
- def UnloadDataLayers(self, force=False):
- """Unload any layers that have been deleted from layer tree
- :param bool force: True to unload all data layers
- """
- if not self.tree:
- return
- listOfItems = []
- if not force:
- item = self.tree.GetFirstChild(self.tree.root)[0]
- self._GetDataLayers(item, listOfItems)
- start = time.time()
- update = False
- layersTmp = self.layers[:]
- for layer in layersTmp:
- if layer in listOfItems:
- continue
- ltype = self.tree.GetLayerInfo(layer, key='type')
- try:
- if ltype == 'raster':
- self.UnloadRaster(layer)
- elif ltype == 'raster_3d':
- self.UnloadRaster3d(layer)
- elif ltype == 'vector':
- maplayer = self.tree.GetLayerInfo(layer, key='maplayer')
- vInfo = grass.vector_info_topo(maplayer.GetName())
- if (vInfo['points'] + vInfo['centroids']) > 0:
- self.UnloadVector(layer, points=True)
- if (vInfo['lines'] + vInfo['boundaries']
- ) > 0 or vInfo['map3d']:
- self.UnloadVector(layer, points=False)
- except GException as e:
- GError(parent=self,
- message=e.value)
- if force and self.baseId > 0: # unload base surface when quitting
- ret = self._display.UnloadSurface(self.baseId)
- self.baseId = -1
- if update:
- self.lmgr.nviz.UpdateSettings()
- self.UpdateView(None)
- stop = time.time()
- Debug.msg(1, "GLWindow.UnloadDataLayers(): time = %f" % (stop - start))
- def SetVectorSurface(self, data):
- """Set reference surfaces of vector"""
- data['mode']['surface'] = {}
- data['mode']['surface']['value'] = list()
- data['mode']['surface']['show'] = list()
- for name in self.GetLayerNames('raster'):
- data['mode']['surface']['value'].append(name)
- data['mode']['surface']['show'].append(True)
- def SetVectorFromCmd(self, item, data):
- """Set 3D view properties from cmd (d.vect)
- :param item: Layer Tree item
- :param nviz: data
- """
- cmd = self.tree.GetLayerInfo(item, key='cmd')
- if cmd[0] != 'd.vect':
- return
- for opt in cmd[1:]:
- try:
- key, value = opt.split('=')
- except ValueError:
- continue
- if key == 'color':
- if not ':' in value:
- value = ':'.join(map(str, str2rgb[value]))
- data['lines']['color']['value'] = value
- data['points']['color']['value'] = value
- def SetMapObjProperties(self, item, id, nvizType):
- """Set map object properties
- Properties must be afterwards updated by
- UpdateMapObjProperties().
- :param item: layer item
- :param id: nviz layer id (or -1)
- :param nvizType: nviz data type (surface, points, vector)
- """
- if nvizType != 'constant':
- mapType = self.tree.GetLayerInfo(item, key='maplayer').type
- # reference to original layer properties (can be None)
- data = self.tree.GetLayerInfo(item, key='nviz')
- else:
- mapType = nvizType
- data = self.constants[item]
- if not data:
- # init data structure
- if nvizType != 'constant':
- self.tree.SetLayerInfo(item, key='nviz', value={})
- data = self.tree.GetLayerInfo(item, key='nviz')
- if mapType == 'raster':
- # reset to default properties
- data[nvizType] = self.nvizDefault.SetSurfaceDefaultProp()
- elif mapType == 'vector':
- # reset to default properties (lines/points)
- data['vector'] = self.nvizDefault.SetVectorDefaultProp(longDim=self._display.GetLongDim())
- self.SetVectorFromCmd(item, data['vector'])
- self.SetVectorSurface(data['vector']['points'])
- self.SetVectorSurface(data['vector']['lines'])
- elif mapType == 'raster_3d':
- # reset to default properties
- data[nvizType] = self.nvizDefault.SetVolumeDefaultProp()
- elif mapType == 'constant':
- data['constant'] = self.nvizDefault.SetConstantDefaultProp()
- else:
- # complete data (use default values), not sure if this is necessary
- if mapType == 'raster':
- if not data['surface']:
- data['surface'] = self.nvizDefault.SetSurfaceDefaultProp()
- if mapType == 'vector':
- if not data['vector']['lines']:
- self.nvizDefault.SetVectorLinesDefaultProp(
- data
- ['vector']
- ['lines'])
- if not data['vector']['points']:
- self.nvizDefault.SetVectorPointsDefaultProp(
- data['vector']['points'], self._display.GetLongDim())
- # set updates
- for sec in data.keys():
- for sec1 in data[sec].keys():
- if sec1 == 'position':
- data[sec][sec1]['update'] = None
- continue
- if isinstance(data[sec][sec1], types.DictType):
- for sec2 in data[sec][sec1].keys():
- if sec2 not in ('all', 'init', 'id'):
- data[sec][sec1][sec2]['update'] = None
- elif isinstance(data[sec][sec1], types.ListType):
- for i in range(len(data[sec][sec1])):
- for sec2 in data[sec][sec1][i].keys():
- data[sec][sec1][i][sec2]['update'] = None
- event = wxUpdateProperties(data=data)
- wx.PostEvent(self, event)
- # set id
- if id > 0:
- if mapType in ('raster', 'raster_3d'):
- data[nvizType]['object'] = {'id': id,
- 'init': False}
- elif mapType == 'vector':
- data['vector'][nvizType]['object'] = {'id': id,
- 'init': False}
- elif mapType == 'constant':
- data[nvizType]['object'] = {'id': id,
- 'init': False}
- return data
- def LoadRaster(self, item):
- """Load 2d raster map and set surface attributes
- :param layer: item
- """
- return self._loadRaster(item)
- def LoadRaster3d(self, item):
- """Load 3d raster map and set surface attributes
- :param layer: item
- """
- return self._loadRaster(item)
- def _loadRaster(self, item):
- """Load 2d/3d raster map and set its attributes
- :param layer: item
- """
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- if layer.type not in ('raster', 'raster_3d'):
- return
- if layer.type == 'raster':
- id = self._display.LoadSurface(str(layer.name), None, None)
- nvizType = 'surface'
- errorMsg = _("Loading raster map")
- elif layer.type == 'raster_3d':
- id = self._display.LoadVolume(str(layer.name), None, None)
- nvizType = 'volume'
- errorMsg = _("Loading 3d raster map")
- else:
- id = -1
- if id < 0:
- if layer.type in ('raster', 'raster_3d'):
- self.log.WriteError(
- "%s <%s> %s" %
- (errorMsg, layer.name, _("failed")))
- else:
- self.log.WriteError(
- _("Unsupported layer type '%s'") %
- layer.type)
- self.layers.append(item)
- # set default/workspace layer properties
- data = self.SetMapObjProperties(item, id, nvizType)
- # update properties
- event = wxUpdateProperties(data=data)
- wx.PostEvent(self, event)
- # update tools window
- if hasattr(
- self.lmgr, "nviz") and item == self.tree.GetSelectedLayer(
- multi=False, checkedOnly=True):
- toolWin = self.lmgr.nviz
- if layer.type == 'raster':
- win = toolWin.FindWindowById(
- toolWin.win['vector']['lines']['surface'])
- win.SetItems(self.GetLayerNames(layer.type))
- # toolWin.UpdatePage(nvizType)
- # toolWin.SetPage(nvizType)
- return id
- def NewConstant(self):
- """Create new constant"""
- index = len(self.constants)
- try:
- name = self.constants[-1]['constant']['object']['name'] + 1
- except IndexError:
- name = 1
- data = dict()
- self.constants.append(data)
- data = self.SetMapObjProperties(item=index, id=-1, nvizType='constant')
- self.AddConstant(data, name)
- return name
- def AddConstant(self, data, name):
- """Add new constant"""
- id = self._display.AddConstant(
- value=data['constant']['value'],
- color=data['constant']['color'])
- self._display.SetSurfaceRes(
- id, data['constant']['resolution'],
- data['constant']['resolution'])
- data['constant']['object'] = {'id': id,
- 'name': name,
- 'init': False}
- def DeleteConstant(self, index):
- """Delete constant layer"""
- id = self.constants[index]['constant']['object']['id']
- self._display.UnloadSurface(id)
- del self.constants[index]
- def SelectCPlane(self, index):
- """Select cutting plane"""
- for plane in range(self._display.GetCPlanesCount()):
- if plane == index:
- self._display.SelectCPlane(plane)
- self.cplanes[plane]['on'] = True
- self._display.SetFenceColor(self.cplanes[plane]['shading'])
- else:
- self._display.UnselectCPlane(plane)
- try:
- self.cplanes[plane]['on'] = False
- except IndexError:
- pass
- def OnUpdateCPlane(self, event):
- """Change cutting plane settings"""
- self.UpdateCPlane(event.current, event.update)
- def UpdateCPlane(self, index, changes):
- """Change cutting plane settings"""
- for each in changes:
- if each == 'rotation':
- self._display.SetCPlaneRotation(
- 0, self.cplanes[index]['rotation']['tilt'],
- self.cplanes[index]['rotation']['rot'])
- if each == 'position':
- self._display.SetCPlaneTranslation(
- self.cplanes[index]['position']['x'],
- self.cplanes[index]['position']['y'],
- self.cplanes[index]['position']['z'])
- if each == 'shading':
- self._display.SetFenceColor(self.cplanes[index]['shading'])
- def UnloadRaster(self, item):
- """Unload 2d raster map
- :param layer: item
- """
- return self._unloadRaster(item)
- def UnloadRaster3d(self, item):
- """Unload 3d raster map
- :param layer: item
- """
- return self._unloadRaster(item)
- def _unloadRaster(self, item):
- """Unload 2d/3d raster map
- :param item: layer item
- """
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- if layer.type not in ('raster', 'raster_3d'):
- return
- data = self.tree.GetLayerInfo(item, key='nviz')
- if layer.type == 'raster':
- nvizType = 'surface'
- unloadFn = self._display.UnloadSurface
- errorMsg = _("Unable to unload raster map")
- successMsg = _("Raster map")
- else:
- nvizType = 'volume'
- unloadFn = self._display.UnloadVolume
- errorMsg = _("Unable to unload 3d raster map")
- successMsg = _("3d raster map")
- try:
- id = data[nvizType]['object']['id']
- except KeyError:
- return
- if unloadFn(id) == 0:
- self.log.WriteError("%s <%s>" % (errorMsg, layer.name))
- else:
- self.log.WriteLog(
- "%s <%s> %s" %
- (successMsg, layer.name, _("unloaded successfully")))
- data[nvizType].pop('object')
- self.layers.remove(item)
- # update tools window
- if hasattr(self.lmgr, "nviz"):
- toolWin = self.lmgr.nviz
- if layer.type == 'raster':
- win = toolWin.FindWindowById(
- toolWin.win['vector']['lines']['surface'])
- win.SetItems(self.GetLayerNames(layer.type))
- win = toolWin.FindWindowById(toolWin.win['surface']['map'])
- win.SetValue('')
- if layer.type == 'raster_3d':
- win = toolWin.FindWindowById(toolWin.win['volume']['map'])
- win.SetValue('')
- if layer.type == 'vector':
- win = toolWin.FindWindowById(toolWin.win['vector']['map'])
- win.SetValue('')
- def LoadVector(self, item, points=None, append=True):
- """Load 2D or 3D vector map overlay
- :param item: layer item
- :param points: True to load points, False to load lines, None
- to load both
- :param bool append: append vector to layer list
- """
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- if layer.type != 'vector':
- return
- # set default properties
- if points is None:
- self.SetMapObjProperties(item, -1, 'lines')
- self.SetMapObjProperties(item, -1, 'points')
- vecTypes = ('points', 'lines')
- elif points:
- self.SetMapObjProperties(item, -1, 'points')
- vecTypes = ('points', )
- else:
- self.SetMapObjProperties(item, -1, 'lines')
- vecTypes = ('lines', )
- id = -1
- for vecType in vecTypes:
- if vecType == 'lines':
- id, baseId = self._display.LoadVector(
- str(layer.GetName()), False)
- else:
- id, baseId = self._display.LoadVector(
- str(layer.GetName()), True)
- if id < 0:
- self.log.WriteError(
- _("Loading vector map <%(name)s> (%(type)s) failed") %
- {'name': layer.name, 'type': vecType})
- # update layer properties
- self.SetMapObjProperties(item, id, vecType)
- if baseId > 0:
- # id of base surface (when no surface is loaded)
- self.baseId = baseId
- if append:
- self.layers.append(item)
- # update properties
- data = self.tree.GetLayerInfo(item, key='nviz')
- event = wxUpdateProperties(data=data)
- wx.PostEvent(self, event)
- # update tools window
- if hasattr(
- self.lmgr, "nviz") and item == self.tree.GetSelectedLayer(
- multi=False, checkedOnly=True):
- toolWin = self.lmgr.nviz
- toolWin.UpdatePage('vector')
- # toolWin.SetPage('vector')
- return id
- def UnloadVector(self, item, points=None, remove=True):
- """Unload vector map overlay
- :param item: layer item
- :param points, lines: True to unload given feature type
- :param remove: remove layer from list
- :type remove: bool
- """
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- data = self.tree.GetLayerInfo(item, key='nviz')['vector']
- # if vecType is None:
- # vecType = []
- # for v in ('lines', 'points'):
- # if UserSettings.Get(group = 'nviz', key = 'vector',
- # subkey = [v, 'show']):
- # vecType.append(v)
- if points is None:
- vecTypes = ('points', 'lines')
- elif points:
- vecTypes = ('points', )
- else:
- vecTypes = ('lines', )
- for vecType in vecTypes:
- if 'object' not in data[vecType]:
- continue
- id = data[vecType]['object']['id']
- if vecType == 'lines':
- ret = self._display.UnloadVector(id, False)
- else:
- ret = self._display.UnloadVector(id, True)
- if ret == 0:
- self.log.WriteError(
- _("Unable to unload vector map <%(name)s> (%(type)s)") %
- {'name': layer.name, 'type': vecType})
- else:
- self.log.WriteLog(
- _("Vector map <%(name)s> (%(type)s) unloaded successfully") % {
- 'name': layer.name,
- 'type': vecType})
- data[vecType].pop('object')
- if remove and item in self.layers:
- self.layers.remove(item)
- def ResetView(self):
- """Reset to default view"""
- zexagOriginal, \
- self.iview['height']['value'], \
- self.iview['height']['min'], \
- self.iview['height']['max'] = self._display.SetViewDefault()
- # hack for latlon projection
- # TODO find more precise way or better rewrite it in OGSF
- self.iview['z-exag']['llRatio'] = 1
- if grass.locn_is_latlong():
- self.iview['z-exag']['llRatio'] = math.pi / 180 * 6371000 * math.cos(
- (grass.region()['n'] + grass.region()['s']) / 2)
- self.view[
- 'z-exag']['value'] = round(zexagOriginal * self.iview['z-exag']['llRatio'])
- self.view['z-exag']['min'] = UserSettings.Get(group='nviz', key='view',
- subkey=('z-exag', 'min'))
- zexagMax = UserSettings.Get(group='nviz', key='view',
- subkey=('z-exag', 'max'))
- if zexagMax <= self.view['z-exag']['value']:
- self.view['z-exag']['max'] = self.view['z-exag']['value'] * 2
- elif self.view['z-exag']['value'] < 1:
- if self.view['z-exag']['value'] == 0:
- self.view['z-exag']['value'] = 1
- self.view['z-exag']['max'] = 10 * self.view['z-exag']['value']
- else:
- self.view['z-exag']['max'] = zexagMax
- self.view['position']['x'] = UserSettings.Get(group='nviz', key='view',
- subkey=('position', 'x'))
- self.view['position']['y'] = UserSettings.Get(group='nviz', key='view',
- subkey=('position', 'y'))
- self.view['persp']['value'] = UserSettings.Get(
- group='nviz', key='view', subkey=('persp', 'value'))
- self.view['twist']['value'] = UserSettings.Get(
- group='nviz', key='view', subkey=('twist', 'value'))
- self._display.ResetRotation()
- self.iview['rotation'] = None
- self._display.LookAtCenter()
- focus = self.iview['focus']
- focus['x'], focus['y'], focus['z'] = self._display.GetFocus()
- self.PostViewEvent()
- def UpdateMapObjProperties(self, event):
- """Generic method to update data layer properties"""
- data = event.data
- if 'surface' in data:
- try:
- id = data['surface']['object']['id']
- except KeyError:
- return
- self.UpdateSurfaceProperties(id, data['surface'])
- # -> initialized
- data['surface']['object']['init'] = True
- elif 'constant' in data:
- id = data['constant']['object']['id']
- self.UpdateConstantProperties(id, data['constant'])
- # -> initialized
- data['constant']['object']['init'] = True
- elif 'volume' in data:
- id = data['volume']['object']['id']
- self.UpdateVolumeProperties(id, data['volume'])
- # -> initialized
- data['volume']['object']['init'] = True
- elif 'vector' in data:
- for type in ('lines', 'points'):
- if 'object' in data['vector'][type]:
- id = data['vector'][type]['object']['id']
- self.UpdateVectorProperties(id, data['vector'], type)
- # -> initialized
- data['vector'][type]['object']['init'] = True
- def UpdateConstantProperties(self, id, data):
- """Update surface map object properties"""
- self._display.SetSurfaceColor(id=id, map=False, value=data['color'])
- self._display.SetSurfaceTopo(id=id, map=False, value=data['value'])
- self._display.SetSurfaceRes(id, data['resolution'], data['resolution'])
- if data['transp'] == 0:
- self._display.UnsetSurfaceTransp(id)
- else:
- self._display.SetSurfaceTransp(id, map=False, value=data['transp'])
- def UpdateSurfaceProperties(self, id, data):
- """Update surface map object properties"""
- # surface attributes
- for attrb in ('color', 'mask',
- 'transp', 'shine'):
- if attrb not in data['attribute'] or \
- 'update' not in data['attribute'][attrb]:
- continue
- map = data['attribute'][attrb]['map']
- value = data['attribute'][attrb]['value']
- if map is None: # unset
- # only optional attributes
- if attrb == 'mask':
- # TODO: invert mask
- # TODO: broken in NVIZ
- self._display.UnsetSurfaceMask(id)
- elif attrb == 'transp':
- self._display.UnsetSurfaceTransp(id)
- else:
- if isinstance(value, types.StringType):
- if len(value) == 0: # ignore empty values (TODO: warning)
- continue
- if map and not grass.find_file(value, element='cell')[
- 'fullname']:
- continue
- if attrb == 'color':
- self._display.SetSurfaceColor(id, map, str(value))
- elif attrb == 'mask':
- # TODO: invert mask
- # TODO: broken in NVIZ
- self._display.SetSurfaceMask(id, False, str(value))
- elif attrb == 'transp':
- self._display.SetSurfaceTransp(id, map, str(value))
- elif attrb == 'shine':
- self._display.SetSurfaceShine(id, map, str(value))
- data['attribute'][attrb].pop('update')
- # draw res
- if 'update' in data['draw']['resolution']:
- coarse = data['draw']['resolution']['coarse']
- fine = data['draw']['resolution']['fine']
- if data['draw']['all']:
- self._display.SetSurfaceRes(-1, fine, coarse)
- else:
- self._display.SetSurfaceRes(id, fine, coarse)
- data['draw']['resolution'].pop('update')
- # draw style
- if 'update' in data['draw']['mode']:
- if data['draw']['mode']['value'] < 0: # need to calculate
- data['draw']['mode']['value'] = self.nvizDefault.GetDrawMode(
- mode=data['draw']['mode']['desc']['mode'],
- style=data['draw']['mode']['desc']['style'],
- shade=data['draw']['mode']['desc']['shading'],
- string=True)
- style = data['draw']['mode']['value']
- if data['draw']['all']:
- self._display.SetSurfaceStyle(-1, style)
- else:
- self._display.SetSurfaceStyle(id, style)
- data['draw']['mode'].pop('update')
- # wire color
- if 'update' in data['draw']['wire-color']:
- color = data['draw']['wire-color']['value']
- if data['draw']['all']:
- self._display.SetWireColor(-1, str(color))
- else:
- self._display.SetWireColor(id, str(color))
- data['draw']['wire-color'].pop('update')
- # position
- if 'update' in data['position']:
- x = data['position']['x']
- y = data['position']['y']
- z = data['position']['z']
- self._display.SetSurfacePosition(id, x, y, z)
- data['position'].pop('update')
- data['draw']['all'] = False
- def UpdateVolumeProperties(self, id, data, isosurfId=None):
- """Update volume (isosurface/slice) map object properties"""
- if 'update' in data['draw']['resolution']:
- if data['draw']['mode']['value'] == 0:
- self._display.SetIsosurfaceRes(
- id, data['draw']['resolution']['isosurface']['value'])
- else:
- self._display.SetSliceRes(
- id, data['draw']['resolution']['slice']['value'])
- data['draw']['resolution'].pop('update')
- if 'update' in data['draw']['shading']:
- if data['draw']['mode']['value'] == 0:
- if data['draw']['shading']['isosurface'][
- 'value'] < 0: # need to calculate
- mode = data['draw']['shading']['isosurface']['value'] = \
- self.nvizDefault.GetDrawMode(shade=data['draw']['shading']['isosurface'],
- string=False)
- self._display.SetIsosurfaceMode(id, mode)
- else:
- if data['draw']['shading']['slice'][
- 'value'] < 0: # need to calculate
- mode = data['draw']['shading']['slice']['value'] = \
- self.nvizDefault.GetDrawMode(shade=data['draw']['shading']['slice'],
- string=False)
- self._display.SetSliceMode(id, mode)
- data['draw']['shading'].pop('update')
- #
- # isosurface attributes
- #
- isosurfId = 0
- for isosurf in data['isosurface']:
- self._display.AddIsosurface(id, 0, isosurf_id=isosurfId)
- for attrb in ('topo', 'color', 'mask',
- 'transp', 'shine'):
- if attrb not in isosurf or \
- 'update' not in isosurf[attrb]:
- continue
- map = isosurf[attrb]['map']
- value = isosurf[attrb]['value']
- if map is None: # unset
- # only optional attributes
- if attrb == 'topo':
- self._display.SetIsosurfaceTopo(
- id, isosurfId, map, str(value))
- elif attrb == 'mask':
- # TODO: invert mask
- # TODO: broken in NVIZ
- self._display.UnsetIsosurfaceMask(id, isosurfId)
- elif attrb == 'transp':
- self._display.UnsetIsosurfaceTransp(id, isosurfId)
- else:
- if isinstance(value, types.StringType):
- if len(value) == 0: # ignore empty values (TODO: warning)
- continue
- if map and not grass.find_file(value, element='grid3')[
- 'fullname']:
- continue
- if attrb == 'color':
- self._display.SetIsosurfaceColor(
- id, isosurfId, map, str(value))
- elif attrb == 'mask':
- # TODO: invert mask
- # TODO: broken in NVIZ
- self._display.SetIsosurfaceMask(
- id, isosurfId, False, str(value))
- elif attrb == 'transp':
- self._display.SetIsosurfaceTransp(
- id, isosurfId, map, str(value))
- elif attrb == 'shine':
- self._display.SetIsosurfaceShine(
- id, isosurfId, map, str(value))
- isosurf[attrb].pop('update')
- isosurfId += 1
- #
- # slice attributes
- #
- sliceId = 0
- for slice in data['slice']:
- ret = self._display.AddSlice(id, slice_id=sliceId)
- if 'update' in slice['position']:
- pos = slice['position']
- ret = self._display.SetSlicePosition(
- id, sliceId, pos['x1'],
- pos['x2'],
- pos['y1'],
- pos['y2'],
- pos['z1'],
- pos['z2'],
- pos['axis'])
- slice['position'].pop('update')
- if 'update' in slice['transp']:
- tr = slice['transp']['value']
- self._display.SetSliceTransp(id, sliceId, tr)
- sliceId += 1
- # position
- if 'update' in data['position'] and 'x' in data['position']:
- x = data['position']['x']
- y = data['position']['y']
- z = data['position']['z']
- self._display.SetVolumePosition(id, x, y, z)
- data['position'].pop('update')
- def UpdateVectorProperties(self, id, data, type):
- """Update vector layer properties
- :param id: layer id
- :param data: properties
- :param type: lines/points
- """
- if type == 'points':
- self.UpdateVectorPointsProperties(id, data[type])
- else:
- self.UpdateVectorLinesProperties(id, data[type])
- def UpdateVectorLinesProperties(self, id, data):
- """Update vector line map object properties"""
- # mode
- if 'update' in data['color'] or \
- 'update' in data['width'] or \
- 'update' in data['mode']:
- width = data['width']['value']
- color = data['color']['value']
- if data['mode']['type'] == '3d':
- use_3D = True
- if 'surface' in data['mode']:
- data['mode'].pop('surface')
- else:
- use_3D = False
- self._display.SetVectorLineMode(id, color,
- width, use_3D)
- if 'update' in data['color']:
- data['color'].pop('update')
- if 'update' in data['width']:
- data['width'].pop('update')
- # height
- if 'update' in data['height']:
- self._display.SetVectorLineHeight(id,
- data['height']['value'])
- data['height'].pop('update')
- # thematic
- if 'update' in data['thematic']:
- color = width = None
- colorTable = False
- if data['thematic']['usecolor'] or data['thematic']['usewidth']:
- if data['thematic']['usecolor']:
- color = data['thematic']['rgbcolumn']
- if self._display.CheckColorTable(id=id, type='lines'):
- colorTable = True
- if data['thematic']['usewidth']:
- width = data['thematic']['sizecolumn']
- self._display.SetLinesStyleThematic(
- id=id, layer=data['thematic']['layer'],
- color=color, colorTable=colorTable, width=width)
- else:
- self._display.UnsetLinesStyleThematic(id=id)
- data['thematic'].pop('update')
- # surface
- if 'surface' in data['mode'] and 'update' in data['mode']:
- for item in range(len(data['mode']['surface']['value'])):
- for type in ('raster', 'constant'):
- sid = self.GetLayerId(
- type=type, name=data['mode']['surface']['value'][item])
- if sid > -1:
- if data['mode']['surface']['show'][item]:
- self._display.SetVectorLineSurface(id, sid)
- else:
- self._display.UnsetVectorLineSurface(id, sid)
- break
- if 'update' in data['mode']:
- data['mode'].pop('update')
- def UpdateVectorPointsProperties(self, id, data):
- """Update vector point map object properties"""
- if 'update' in data['size'] or \
- 'update' in data['width'] or \
- 'update' in data['marker'] or \
- 'update' in data['color']:
- ret = self._display.SetVectorPointMode(
- id, data['color']['value'],
- data['width']['value'],
- float(data['size']['value']),
- data['marker']['value'] + 1)
- error = None
- if ret == -1:
- error = _("Vector point layer not found (id = %d)") % id
- elif ret == -2:
- error = _("Unable to set data layer properties (id = %d)") % id
- if error:
- raise GException(
- _("Setting data layer properties failed.\n\n%s") %
- error)
- for prop in ('size', 'width', 'marker', 'color'):
- if 'update' in data[prop]:
- data[prop].pop('update')
- # height
- if 'update' in data['height']:
- self._display.SetVectorPointHeight(id,
- data['height']['value'])
- data['height'].pop('update')
- # thematic
- if 'update' in data['thematic']:
- color = size = None
- colorTable = False
- if data['thematic']['usecolor'] or data['thematic']['usesize']:
- if data['thematic']['usecolor']:
- color = data['thematic']['rgbcolumn']
- if self._display.CheckColorTable(id=id, type='points'):
- colorTable = True
- if data['thematic']['usesize']:
- size = data['thematic']['sizecolumn']
- self._display.SetPointsStyleThematic(
- id=id, layer=data['thematic']['layer'],
- color=color, colorTable=colorTable, size=size)
- else:
- self._display.UnsetPointsStyleThematic(id=id)
- data['thematic'].pop('update')
- # surface
- if 'update' in data['mode']:
- if data['mode'].get('3d', False):
- self._display.SetVectorPointZMode(id, True)
- elif 'surface' in data['mode']:
- self._display.SetVectorPointZMode(id, False)
- for item in range(len(data['mode']['surface']['value'])):
- for type in ('raster', 'constant'):
- sid = self.GetLayerId(
- type=type, name=data['mode']['surface']['value'][item])
- if sid > -1:
- if data['mode']['surface']['show'][item]:
- self._display.SetVectorPointSurface(id, sid)
- else:
- self._display.UnsetVectorPointSurface(id, sid)
- break
- data['mode'].pop('update')
- def GetLayerNames(self, type):
- """Return list of map layer names of given type"""
- layerName = []
- if type == 'constant':
- for item in self.constants:
- layerName.append(_("constant#") +
- str(item['constant']['object']['name']))
- else:
- for item in self.layers:
- mapLayer = self.tree.GetLayerInfo(item, key='maplayer')
- if type != mapLayer.GetType():
- continue
- layerName.append(mapLayer.GetName())
- return layerName
- def GetLayerId(self, type, name, vsubtyp=None):
- """Get layer object id or -1"""
- if len(name) < 1:
- return -1
- if type == 'constant':
- for item in self.constants:
- if _("constant#") + str(item['constant']
- ['object']['name']) == name:
- return item['constant']['object']['id']
- for item in self.layers:
- mapLayer = self.tree.GetLayerInfo(item, key='maplayer')
- if type != mapLayer.GetType() or \
- name != mapLayer.GetName():
- continue
- data = self.tree.GetLayerInfo(item, key='nviz')
- try:
- if type == 'raster':
- return data['surface']['object']['id']
- elif type == 'vector':
- if vsubtyp == 'vpoint':
- return data['vector']['points']['object']['id']
- elif vsubtyp == 'vline':
- return data['vector']['lines']['object']['id']
- elif type == 'raster_3d':
- return data['volume']['object']['id']
- except KeyError:
- return -1
- return -1
- def ReloadLayersData(self):
- """Delete nviz data of all loaded layers and reload them from current settings"""
- for item in self.layers:
- type = self.tree.GetLayerInfo(item, key='type')
- layer = self.tree.GetLayerInfo(item, key='maplayer')
- data = self.tree.GetLayerInfo(item, key='nviz')
- if type == 'raster':
- self.nvizDefault.SetSurfaceDefaultProp(data['surface'])
- if type == 'vector':
- vInfo = grass.vector_info_topo(layer.GetName())
- if (vInfo['points'] + vInfo['centroids']) > 0:
- self.nvizDefault.SetVectorPointsDefaultProp(
- data['vector']['points'], self._display.GetLongDim())
- if (vInfo['lines'] + vInfo['boundaries']) > 0:
- self.nvizDefault.SetVectorLinesDefaultProp(
- data
- ['vector']
- ['lines'])
- def NvizCmdCommand(self):
- """Generate command for m.nviz.image according to current state"""
- cmd = 'm.nviz.image '
- rasters = []
- vectors = []
- volumes = []
- for item in self.layers:
- if self.tree.GetLayerInfo(item, key='type') == 'raster':
- rasters.append(item)
- elif self.tree.GetLayerInfo(item, key='type') == 'raster_3d':
- volumes.append(item)
- elif self.tree.GetLayerInfo(item, key='type') == 'vector':
- vectors.append(item)
- # if not rasters and not self.constants:
- # return _("At least one raster map required")
- # elevation_map/elevation_value
- if self.constants:
- subcmd = "elevation_value="
- for constant in self.constants:
- subcmd += "%d," % constant['constant']['value']
- subcmd = subcmd.strip(', ') + ' '
- cmd += subcmd
- if rasters:
- subcmd = "elevation_map="
- for item in rasters:
- subcmd += "%s," % self.tree.GetLayerInfo(
- item, key='maplayer').GetName()
- subcmd = subcmd.strip(', ') + ' '
- cmd += subcmd
- #
- # draw mode
- #
- cmdMode = "mode="
- cmdFine = "resolution_fine="
- cmdCoarse = "resolution_coarse="
- cmdShading = "shading="
- cmdStyle = "style="
- cmdWire = "wire_color="
- # test -a flag
- flag_a = "-a "
- nvizDataFirst = self.tree.GetLayerInfo(
- rasters[0], key='nviz')['surface']['draw']
- for item in rasters:
- nvizData = self.tree.GetLayerInfo(item, key='nviz')[
- 'surface']['draw']
- if nvizDataFirst != nvizData:
- flag_a = ""
- cmd += flag_a
- for item in rasters:
- nvizData = self.tree.GetLayerInfo(item, key='nviz')[
- 'surface']['draw']
- cmdMode += "%s," % nvizData['mode']['desc']['mode']
- cmdFine += "%s," % nvizData['resolution']['fine']
- cmdCoarse += "%s," % nvizData['resolution']['coarse']
- cmdShading += "%s," % nvizData['mode']['desc']['shading']
- cmdStyle += "%s," % nvizData['mode']['desc']['style']
- cmdWire += "%s," % nvizData['wire-color']['value']
- for item in self.constants:
- cmdMode += "fine,"
- cmdFine += "%s," % item['constant']['resolution']
- cmdCoarse += "%s," % item['constant']['resolution']
- cmdShading += "gouraud,"
- cmdStyle += "surface,"
- cmdWire += "0:0:0,"
- mode = []
- for subcmd in (cmdMode, cmdFine, cmdCoarse,
- cmdShading, cmdStyle, cmdWire):
- if flag_a:
- mode.append(subcmd.split(',')[0] + ' ')
- else:
- subcmd = subcmd.strip(', ') + ' '
- cmd += subcmd
- if flag_a: # write only meaningful possibilities
- cmd += mode[0]
- if 'fine' in mode[0]:
- cmd += mode[1]
- elif 'coarse' in mode[0]:
- cmd += mode[2]
- elif 'both' in mode[0]:
- cmd += mode[2]
- cmd += mode[1]
- if 'flat' in mode[3]:
- cmd += mode[3]
- if 'wire' in mode[4]:
- cmd += mode[4]
- if 'coarse' in mode[0] or 'both' in mode[
- 0] and 'wire' in mode[3]:
- cmd += mode[5]
- #
- # attributes
- #
- cmdColorMap = "color_map="
- cmdColorVal = "color="
- for item in rasters:
- nvizData = self.tree.GetLayerInfo(
- item, key='nviz')['surface']['attribute']
- if 'color' not in nvizData:
- cmdColorMap += "%s," % self.tree.GetLayerInfo(
- item, key='maplayer').GetName()
- else:
- if nvizData['color']['map']:
- cmdColorMap += "%s," % nvizData['color']['value']
- else:
- cmdColorVal += "%s," % nvizData['color']['value']
- # TODO
- # transparency, shine, mask
- for item in self.constants:
- cmdColorVal += "%s," % item['constant']['color']
- if cmdColorMap.split("=")[1]:
- cmd += cmdColorMap.strip(', ') + ' '
- if cmdColorVal.split("=")[1]:
- cmd += cmdColorVal.strip(', ') + ' '
- cmd += "\\\n"
- #
- # vlines
- #
- if vectors:
- cmdLines = cmdLWidth = cmdLHeight = cmdLColor = cmdLMode = cmdLPos = \
- cmdPoints = cmdPWidth = cmdPSize = cmdPColor = cmdPMarker = cmdPPos = cmdPLayer = ""
- markers = ['x', 'box', 'sphere', 'cube', 'diamond',
- 'dec_tree', 'con_tree', 'aster', 'gyro', 'histogram']
- for vector in vectors:
- layerName = self.tree.GetLayerInfo(
- vector, key='maplayer').GetName()
- vInfo = grass.vector_info_topo(layerName)
- nvizData = self.tree.GetLayerInfo(vector, key='nviz')['vector']
- if (vInfo['lines'] + vInfo['boundaries']) > 0:
- cmdLines += "%s," % self.tree.GetLayerInfo(
- vector, key='maplayer').GetName()
- cmdLWidth += "%d," % nvizData['lines']['width']['value']
- cmdLHeight += "%d," % nvizData['lines']['height']['value']
- cmdLColor += "%s," % nvizData['lines']['color']['value']
- cmdLMode += "%s," % nvizData['lines']['mode']['type']
- cmdLPos += "0,0,%d," % nvizData['lines']['height']['value']
- if (vInfo['points'] + vInfo['centroids']) > 0:
- cmdPoints += "%s," % self.tree.GetLayerInfo(
- vector, key='maplayer').GetName()
- cmdPWidth += "%d," % nvizData['points']['width']['value']
- cmdPSize += "%d," % nvizData['points']['size']['value']
- cmdPColor += "%s," % nvizData['points']['color']['value']
- cmdPMarker += "%s," % markers[
- nvizData['points']['marker']['value']]
- cmdPPos += "0,0,%d," % nvizData[
- 'points']['height']['value']
- cmdPLayer += "1,1,"
- if cmdLines:
- cmd += "vline=" + cmdLines.strip(',') + ' '
- cmd += "vline_width=" + cmdLWidth.strip(',') + ' '
- cmd += "vline_color=" + cmdLColor.strip(',') + ' '
- cmd += "vline_height=" + cmdLHeight.strip(',') + ' '
- cmd += "vline_mode=" + cmdLMode.strip(',') + ' '
- cmd += "vline_position=" + cmdLPos.strip(',') + ' '
- if cmdPoints:
- cmd += "vpoint=" + cmdPoints.strip(',') + ' '
- cmd += "vpoint_width=" + cmdPWidth.strip(',') + ' '
- cmd += "vpoint_color=" + cmdPColor.strip(',') + ' '
- cmd += "vpoint_size=" + cmdPSize.strip(',') + ' '
- cmd += "vpoint_marker=" + cmdPMarker.strip(',') + ' '
- cmd += "vpoint_position=" + cmdPPos.strip(',') + ' '
- cmd += "vpoint_layer=" + cmdPLayer.strip(',') + ' '
- cmd += "\\\n"
- #
- # volumes
- #
- if volumes:
- cmdName = cmdShade = cmdRes = cmdPos = cmdIso = ""
- cmdIsoColorMap = cmdIsoColorVal = cmdIsoTrMap = cmdIsoTrVal = ""
- cmdSlice = cmdSliceTransp = cmdSlicePos = ""
- for i, volume in enumerate(volumes):
- nvizData = self.tree.GetLayerInfo(volume, key='nviz')['volume']
- cmdName += "%s," % self.tree.GetLayerInfo(
- volume, key='maplayer').GetName()
- cmdShade += "%s," % nvizData['draw'][
- 'shading']['isosurface']['desc']
- cmdRes += "%d," % nvizData['draw'][
- 'resolution']['isosurface']['value']
- if nvizData['position']:
- cmdPos += "%d,%d,%d," % (
- nvizData['position']['x'],
- nvizData['position']['y'],
- nvizData['position']['z'])
- for iso in nvizData['isosurface']:
- level = iso['topo']['value']
- cmdIso += "%d:%s," % (i + 1, level)
- if iso['color']['map']:
- cmdIsoColorMap += "%s," % iso['color']['value']
- else:
- cmdIsoColorVal += "%s," % iso['color']['value']
- if 'transp' in iso:
- if iso['transp']['map']:
- cmdIsoTrMap += "%s," % iso['transp']['value']
- else:
- cmdIsoTrVal += "%s," % iso['transp']['value']
- for slice in nvizData['slice']:
- axis = ('x', 'y', 'z')[slice['position']['axis']]
- cmdSlice += "%d:%s," % (i + 1, axis)
- for coord in ('x1', 'x2', 'y1', 'y2', 'z1', 'z2'):
- cmdSlicePos += "%f," % slice['position'][coord]
- cmdSliceTransp += "%s," % slice['transp']['value']
- cmd += "volume=" + cmdName.strip(',') + ' '
- cmd += "volume_shading=" + cmdShade.strip(',') + ' '
- cmd += "volume_resolution=" + cmdRes.strip(',') + ' '
- if nvizData['position']:
- cmd += "volume_position=" + cmdPos.strip(',') + ' '
- if cmdIso:
- cmd += "isosurf_level=" + cmdIso.strip(',') + ' '
- if cmdIsoColorMap:
- cmd += "isosurf_color_map=" + \
- cmdIsoColorMap.strip(',') + ' '
- if cmdIsoColorVal:
- cmd += "isosurf_color_value=" + \
- cmdIsoColorVal.strip(',') + ' '
- if cmdIsoTrMap:
- cmd += "isosurf_transp_map=" + cmdIsoTrMap.strip(',') + ' '
- if cmdIsoTrVal:
- cmd += "isosurf_transp_value=" + \
- cmdIsoTrVal.strip(',') + ' '
- if cmdSlice:
- cmd += "slice=" + cmdSlice.strip(',') + ' '
- cmd += "slice_position=" + cmdSlicePos.strip(',') + ' '
- cmd += "slice_transparency=" + cmdSliceTransp.strip(',') + ' '
- #
- # cutting planes
- #
- cplane = self.lmgr.nviz.FindWindowById(
- self.lmgr.nviz.win['cplane']['planes']).GetStringSelection()
- try:
- planeIndex = int(cplane.split()[-1]) - 1
- except (IndexError, ValueError):
- planeIndex = None
- if planeIndex is not None:
- shading = ['clear', 'top', 'bottom', 'blend', 'shaded']
- cmd += "cplane=%d " % planeIndex
- cmd += "cplane_rotation=%d " % self.cplanes[
- planeIndex]['rotation']['rot']
- cmd += "cplane_tilt=%d " % self.cplanes[
- planeIndex]['rotation']['tilt']
- cmd += "cplane_position=%d,%d,%d " % (
- self.cplanes[planeIndex]['position']['x'],
- self.cplanes[planeIndex]['position']['y'],
- self.cplanes[planeIndex]['position']['z'])
- cmd += "cplane_shading=%s " % shading[
- self.cplanes[planeIndex]['shading']]
- cmd += "\\\n"
- #
- # viewpoint
- #
- subcmd = "position=%.2f,%.2f " % (
- self.view['position']['x'],
- self.view['position']['y'])
- subcmd += "height=%d " % (self.iview['height']['value'])
- subcmd += "perspective=%d " % (self.view['persp']['value'])
- subcmd += "twist=%d " % (self.view['twist']['value'])
- subcmd += "zexag=%f " % (self.view['z-exag']
- ['value'] / self.iview['z-exag']['llRatio'])
- subcmd += "focus=%d,%d,%d " % (
- self.iview['focus']['x'],
- self.iview['focus']['y'],
- self.iview['focus']['z'])
- cmd += subcmd
- # background
- subcmd = "bgcolor=%d:%d:%d " % (self.view['background']['color'][:3])
- if self.view['background']['color'] != (255, 255, 255):
- cmd += subcmd
- cmd += "\\\n"
- # light
- subcmd = "light_position=%.2f,%.2f,%.2f " % (
- self.light['position']['x'],
- self.light['position']['y'],
- self.light['position']['z'] / 100.)
- subcmd += "light_brightness=%d " % (self.light['bright'])
- subcmd += "light_ambient=%d " % (self.light['ambient'])
- subcmd += "light_color=%d:%d:%d " % (self.light['color'][:3])
- cmd += subcmd
- cmd += "\\\n"
- # fringe
- toolWindow = self.lmgr.nviz
- direction = ''
- for dir in ('nw', 'ne', 'sw', 'se'):
- if toolWindow.FindWindowById(
- toolWindow.win['fringe'][dir]).IsChecked():
- direction += "%s," % dir
- if direction:
- subcmd = "fringe=%s " % (direction.strip(','))
- color = toolWindow.FindWindowById(
- toolWindow.win['fringe']['color']).GetValue()
- subcmd += "fringe_color=%d:%d:%d " % (color[0], color[1], color[2])
- subcmd += "fringe_elevation=%d " % (toolWindow.FindWindowById(
- toolWindow.win['fringe']['elev']).GetValue())
- cmd += subcmd
- cmd += "\\\n"
- # north arrow
- if self.decoration['arrow']['show']:
- subcmd = "arrow_position=%d,%d " % (
- self.decoration['arrow']['position']['x'],
- self.decoration['arrow']['position']['y'])
- subcmd += "arrow_color=%s " % self.decoration['arrow']['color']
- subcmd += "arrow_size=%d " % self.decoration['arrow']['size']
- cmd += subcmd
- # output
- subcmd = 'output=nviz_output '
- subcmd += 'format=ppm '
- subcmd += 'size=%d,%d ' % self.GetClientSize()
- cmd += subcmd
- return cmd
- def OnNvizCmd(self):
- """Generate and write command to command output"""
- self.log.WriteLog(
- self.NvizCmdCommand(),
- notification=Notification.RAISE_WINDOW)
- def SaveToFile(self, FileName, FileType, width, height):
- """This draws the DC to a buffer that can be saved to a file.
- .. todo::
- fix BufferedPaintDC
- :param filename: file name
- :param FileType: type of bitmap
- :param width: image width
- :param height: image height
- """
- self._display.SaveToFile(FileName, width, height, FileType)
- # pbuffer = wx.EmptyBitmap(max(1, self.Map.width), max(1, self.Map.height))
- # dc = wx.BufferedPaintDC(self, pbuffer)
- # dc.Clear()
- # self.SetCurrent()
- # self._display.Draw(False, -1)
- # pbuffer.SaveFile(FileName, FileType)
- # self.SwapBuffers()
- def GetDisplay(self):
- """Get display instance"""
- return self._display
- def ZoomToMap(self, layers):
- """Reset view
- :param layers: so far unused
- """
- self.lmgr.nviz.OnResetView(None)
- def DisactivateWin(self):
- """Use when the class instance is hidden in MapFrame."""
- pass
- def ActivateWin(self):
- """Used when the class instance is activated in MapFrame."""
- pass
|