""" @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 (Google SoC 2008/2010) @author Anna Kratochvilova (Google SoC 2011) """ import os import sys import six import time import copy import math 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) # timer for animations self.timerAnim = wx.Timer(self) 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.CallLater(1000, 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 GetContentScaleFactor(self): if sys.platform == "darwin" and not CheckWxVersion(version=[4, 1, 0]): return 1 return super().GetContentScaleFactor() 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, self.GetContentScaleFactor()) # 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 six.iteritems(self.overlays): 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], self.GetContentScaleFactor()) 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, self.GetContentScaleFactor()) 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, self.GetContentScaleFactor()) if result[0]: self._display.LookHere(x, y, self.GetContentScaleFactor()) 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], self.GetContentScaleFactor()) id2, x2, y2, z2 = self._display.GetPointOnSurface( event.GetX(), size[1] - event.GetY(), self.GetContentScaleFactor()) 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], self.GetContentScaleFactor()) id2, x2, y2, z2 = self._display.GetPointOnSurface( event.GetX(), size[1] - event.GetY(), self.GetContentScaleFactor()) 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, self.GetContentScaleFactor()) 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 = grass.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 = grass.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 = grass.clock() 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) # when nviz.tools is not yet ready # during opening 3D view 2nd time except: pass stop = grass.clock() 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 = grass.clock() 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 = grass.clock() 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], dict): 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], list): 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, str): 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, str): 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 width, height = self.GetClientSize() subcmd = 'output=nviz_output ' subcmd += 'format=ppm ' subcmd += 'size=%d,%d ' % (width, height) 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