Fixed img2py to work correctly with Python 2.1.

Added enhanced wxVTKRenderWindow by Prabhu Ramachandran


git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@10855 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
Robin Dunn
2001-07-06 04:11:34 +00:00
parent 68cb30154e
commit 286e2db69c
5 changed files with 526 additions and 112 deletions

View File

@@ -2,7 +2,7 @@ CHANGES.txt for wxPython
---------------------------------------------------------------------- ----------------------------------------------------------------------
????? 2.3.1
----- -----
Added EVT_GRID_EDITOR_CREATED and wxGridEditorCreatedEvent so the user Added EVT_GRID_EDITOR_CREATED and wxGridEditorCreatedEvent so the user
code can get access to the edit control when it is created, (to push code can get access to the edit control when it is created, (to push
@@ -18,6 +18,10 @@ Added wxGridAutoEditMixin to the mixins library package.
Made ColourSelect be derived from wxButton. Made ColourSelect be derived from wxButton.
Fixed img2py to work correctly with Python 2.1.
Added enhanced wxVTKRenderWindow by Prabhu Ramachandran
2.3.0 2.3.0

View File

@@ -1957,15 +1957,9 @@ static struct { char *n1; char *n2; void *(*pcnv)(void *); } _swig_mapping[] = {
{ "_wxPrintQuality","_EBool",0}, { "_wxPrintQuality","_EBool",0},
{ "_wxPrintQuality","_size_t",0}, { "_wxPrintQuality","_size_t",0},
{ "_wxPrintQuality","_time_t",0}, { "_wxPrintQuality","_time_t",0},
{ "_class_wxEvtHandler","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxEvtHandler},
{ "_class_wxEvtHandler","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxEvtHandler},
{ "_byte","_unsigned_char",0}, { "_byte","_unsigned_char",0},
{ "_long","_unsigned_long",0}, { "_long","_unsigned_long",0},
{ "_long","_signed_long",0}, { "_long","_signed_long",0},
{ "_class_wxObject","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxObject},
{ "_class_wxObject","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxObject},
{ "_class_wxObject","_class_wxDiagram",SwigwxDiagramTowxObject},
{ "_class_wxObject","_wxDiagram",SwigwxDiagramTowxObject},
{ "_size_t","_wxCoord",0}, { "_size_t","_wxCoord",0},
{ "_size_t","_wxPrintQuality",0}, { "_size_t","_wxPrintQuality",0},
{ "_size_t","_time_t",0}, { "_size_t","_time_t",0},
@@ -1973,7 +1967,6 @@ static struct { char *n1; char *n2; void *(*pcnv)(void *); } _swig_mapping[] = {
{ "_size_t","_int",0}, { "_size_t","_int",0},
{ "_size_t","_wxWindowID",0}, { "_size_t","_wxWindowID",0},
{ "_size_t","_uint",0}, { "_size_t","_uint",0},
{ "_wxPanel","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxPanel},
{ "_wxPanel","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxPanel}, { "_wxPanel","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxPanel},
{ "_uint","_wxCoord",0}, { "_uint","_wxCoord",0},
{ "_uint","_wxPrintQuality",0}, { "_uint","_wxPrintQuality",0},
@@ -1990,8 +1983,6 @@ static struct { char *n1; char *n2; void *(*pcnv)(void *); } _swig_mapping[] = {
{ "_EBool","_int",0}, { "_EBool","_int",0},
{ "_EBool","_wxWindowID",0}, { "_EBool","_wxWindowID",0},
{ "_unsigned_long","_long",0}, { "_unsigned_long","_long",0},
{ "_class_wxPanel","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxPanel},
{ "_class_wxPanel","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxPanel},
{ "_signed_int","_wxCoord",0}, { "_signed_int","_wxCoord",0},
{ "_signed_int","_wxPrintQuality",0}, { "_signed_int","_wxPrintQuality",0},
{ "_signed_int","_EBool",0}, { "_signed_int","_EBool",0},
@@ -2004,15 +1995,10 @@ static struct { char *n1; char *n2; void *(*pcnv)(void *); } _swig_mapping[] = {
{ "_unsigned_short","_wxDateTime_t",0}, { "_unsigned_short","_wxDateTime_t",0},
{ "_unsigned_short","_WXTYPE",0}, { "_unsigned_short","_WXTYPE",0},
{ "_unsigned_short","_short",0}, { "_unsigned_short","_short",0},
{ "_class_wxWindow","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxWindow},
{ "_class_wxWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxWindow},
{ "_wxObject","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxObject},
{ "_wxObject","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxObject}, { "_wxObject","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxObject},
{ "_wxObject","_class_wxDiagram",SwigwxDiagramTowxObject},
{ "_wxObject","_wxDiagram",SwigwxDiagramTowxObject}, { "_wxObject","_wxDiagram",SwigwxDiagramTowxObject},
{ "_signed_short","_WXTYPE",0}, { "_signed_short","_WXTYPE",0},
{ "_signed_short","_short",0}, { "_signed_short","_short",0},
{ "_wxScrolledWindow","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxScrolledWindow},
{ "_wxScrolledWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxScrolledWindow}, { "_wxScrolledWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxScrolledWindow},
{ "_unsigned_char","_byte",0}, { "_unsigned_char","_byte",0},
{ "_unsigned_int","_wxCoord",0}, { "_unsigned_int","_wxCoord",0},
@@ -2063,11 +2049,7 @@ static struct { char *n1; char *n2; void *(*pcnv)(void *); } _swig_mapping[] = {
{ "_wxCoord","_size_t",0}, { "_wxCoord","_size_t",0},
{ "_wxCoord","_time_t",0}, { "_wxCoord","_time_t",0},
{ "_wxCoord","_wxPrintQuality",0}, { "_wxCoord","_wxPrintQuality",0},
{ "_wxEvtHandler","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxEvtHandler},
{ "_wxEvtHandler","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxEvtHandler}, { "_wxEvtHandler","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxEvtHandler},
{ "_class_wxScrolledWindow","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxScrolledWindow},
{ "_class_wxScrolledWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxScrolledWindow},
{ "_wxWindow","_class_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxWindow},
{ "_wxWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxWindow}, { "_wxWindow","_wxPyShapeCanvas",SwigwxPyShapeCanvasTowxWindow},
{0,0,0}}; {0,0,0}};

View File

@@ -8,36 +8,27 @@ except ImportError:
#---------------------------------------------------------------------- #----------------------------------------------------------------------
def MakeCone(vtkwin):
cone = vtk.vtkConeSource()
cone.SetResolution(128)
coneMapper = vtk.vtkPolyDataMapper()
coneMapper.SetInput(cone.GetOutput())
coneActor = vtk.vtkActor()
coneActor.SetMapper(coneMapper)
ren = vtk.vtkRenderer()
vtkwin.GetRenderWindow().AddRenderer(ren)
ren.AddActor(coneActor)
#----------------------------------------------------------------------
def runTest(frame, nb, log): def runTest(frame, nb, log):
if haveVTK: if haveVTK:
f = wxFrame(frame, -1, "wxVTKRenderWindow", size=(450, 300)) f = wxFrame(frame, -1, "wxVTKRenderWindow", size=(450, 300),
style=wxDEFAULT_FRAME_STYLE|wxNO_FULL_REPAINT_ON_RESIZE)
win = vtk.wxVTKRenderWindow(f, -1) win = vtk.wxVTKRenderWindow(f, -1)
MakeCone(win)
# Get the render window
renWin = win.GetRenderWindow()
# Next, do the VTK stuff
ren = vtk.vtkRenderer()
renWin.AddRenderer(ren)
cone = vtk.vtkConeSource()
cone.SetResolution(80)
coneMapper = vtk.vtkPolyDataMapper()
coneMapper.SetInput(cone.GetOutput())
coneActor = vtk.vtkActor()
coneActor.SetMapper(coneMapper)
ren.AddActor(coneActor)
coneMapper.GetLookupTable().Build()
# Create a scalar bar
scalarBar = vtk.vtkScalarBarActor()
scalarBar.SetLookupTable(coneMapper.GetLookupTable())
scalarBar.SetTitle("Temperature")
scalarBar.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport()
scalarBar.GetPositionCoordinate().SetValue(0.1, 0.01)
scalarBar.SetOrientationToHorizontal()
scalarBar.SetWidth(0.8)
scalarBar.SetHeight(0.17)
ren.AddActor2D(scalarBar)
f.Show(true) f.Show(true)
frame.otherWin = f frame.otherWin = f

View File

@@ -68,6 +68,7 @@ def crunch_data(data, compressed):
c = i = 0 c = i = 0
word = "" word = ""
octdigits = "01234567" octdigits = "01234567"
hexdigits = "0123456789abcdef"
while i < len(data): while i < len(data):
if data[i] != "\\": if data[i] != "\\":
word = data[i] word = data[i]
@@ -79,9 +80,16 @@ def crunch_data(data, compressed):
break break
word = data[i:i+n] word = data[i:i+n]
i = i + n i = i + n
elif data[i+1] == 'x':
for n in range(2, 5):
if data[i+n] not in hexdigits:
break
word = data[i:i+n]
i = i + n
else: else:
word = data[i:i+2] word = data[i:i+2]
i = i + 2 i = i + 2
l = len(word) l = len(word)
if c + l >= 78-1: if c + l >= 78-1:
fp.write("\\\n") fp.write("\\\n")

View File

@@ -12,120 +12,549 @@
# Licence: wxWindows license # Licence: wxWindows license
#---------------------------------------------------------------------- #----------------------------------------------------------------------
# This class has been rewritten and improved by Prabhu Ramachandran
# <prabhu@aero.iitm.ernet.in>. It has been tested under Win32 and
# Linux. Many thanks to Eric Boix <frog@creatis.insa-lyon.fr> for
# testing it under Windows(TM) and finding and fixing many errors.
# Thanks also to Sebastien BARRE <sebastien@barre.nom.fr> for his
# suggestions.
try: # For Win32,
try:
from vtkpython import * from vtkpython import *
except ImportError: except ImportError:
try: # for Unix. Why they do it this way is anybody's guess... raise ImportError, "VTK extension module not found"
from libVTKCommonPython import *
from libVTKGraphicsPython import *
from libVTKImagingPython import *
except ImportError:
raise ImportError, "VTK extension module not found"
from wxPython.wx import * from wxPython.wx import *
import math import math
#---------------------------------------------------------------------- #----------------------------------------------------------------------
class wxVTKRenderWindow(wxScrolledWindow): DEBUG = 0
def debug(msg):
if DEBUG:
print msg
class wxVTKRenderWindowBase(wxScrolledWindow):
"""
A base class that enables one to embed a vtkRenderWindow into
a wxPython widget. This class embeds the RenderWindow correctly
under different platforms. Provided are some empty methods that
can be overloaded to provide a user defined interaction behaviour.
The event handling functions have names that are similar to the
ones in the vtkInteractorStyle class included with VTK.
"""
def __init__(self, parent, id, position=wxDefaultPosition, def __init__(self, parent, id, position=wxDefaultPosition,
size=wxDefaultSize, style=0): size=wxDefaultSize, style=0):
wxScrolledWindow.__init__(self, parent, id, position, size, style) wxScrolledWindow.__init__(self, parent, id, position, size, style)
style = style | wxWANTS_CHARS
self.renderWindow = vtkRenderWindow() # This box is necessary to eliminate flicker and enable proper
# event handling under Linux/GTK.
self.box = wxPanel(self, -1, position, size, style)
self._RenderWindow = vtkRenderWindow()
self.__InExpose = 0
self.__Created = 0
if wxPlatform != '__WXMSW__': if wxPlatform != '__WXMSW__':
# We can't get the handle in wxGTK until after the widget # We can't get the handle in wxGTK until after the widget
# is created, the window create event happens later so we'll # is created, the window create event happens later so we'll
# catch it there # catch it there
EVT_WINDOW_CREATE(self, self.OnCreateWindow) EVT_WINDOW_CREATE(self.box, self.OnCreateWindow)
EVT_PAINT (self, self.OnExpose)
else: else:
# but in MSW, the window create event happens durring the above # but in MSW, the window create event happens durring the above
# call to __init__ so we have to do it here. # call to __init__ so we have to do it here.
hdl = self.GetHandle() hdl = self.box.GetHandle()
self.renderWindow.SetWindowInfo(str(hdl)) self._RenderWindow.SetWindowInfo(str(hdl))
EVT_PAINT (self.box, self.OnExpose)
self.__Created = 1
# common for all platforms
EVT_SIZE (self, self.OnConfigure)
# setup the user defined events.
self.SetupEvents()
EVT_LEFT_DOWN (self, self.SaveClick) def SetupEvents(self):
EVT_MIDDLE_DOWN(self, self.SaveClick) "Setup the user defined event bindings."
EVT_RIGHT_DOWN (self, self.SaveClick) # Remember to bind everything to self.box and NOT self
EVT_LEFT_UP (self, self.Release) EVT_LEFT_DOWN (self.box, self.OnLeftButtonDown)
EVT_MIDDLE_UP (self, self.Release) EVT_MIDDLE_DOWN (self.box, self.OnMiddleButtonDown)
EVT_RIGHT_UP (self, self.Release) EVT_RIGHT_DOWN (self.box, self.OnRightButtonDown)
EVT_MOTION (self, self.MouseMotion) EVT_LEFT_UP (self.box, self.OnLeftButtonUp)
EVT_MIDDLE_UP (self.box, self.OnMiddleButtonUp)
EVT_ERASE_BACKGROUND(self, self.OnEraseBackground) EVT_RIGHT_UP (self.box, self.OnRightButtonUp)
EVT_PAINT(self, self.OnPaint) EVT_MOTION (self.box, self.OnMouseMove)
EVT_ENTER_WINDOW (self.box, self.OnEnter)
EVT_LEAVE_WINDOW (self.box, self.OnLeave)
EVT_CHAR (self.box, self.OnChar)
# Add your bindings if you want them in the derived class.
def GetRenderer(self): def GetRenderer(self):
self.renderWindow.GetRenderers().InitTraversal() self._RenderWindow.GetRenderers().InitTraversal()
return self.renderWindow.GetRenderers().GetNextItem() return self._RenderWindow.GetRenderers().GetNextItem()
def GetRenderWindow(self): def GetRenderWindow(self):
return self.renderWindow return self._RenderWindow
def OnPaint(self, event): def Render(self):
dc = wxPaintDC(self) if self.__Created:
self.renderWindow.Render() # if block needed because calls to render before creation
# will prevent the renderwindow from being embedded into a
# wxPython widget.
self._RenderWindow.Render()
def OnExpose(self, event):
# I need this for the MDIDemo. Somehow OnCreateWindow was
# not getting called.
if not self.__Created:
self.OnCreateWindow(event)
if (not self.__InExpose):
self.__InExpose = 1
dc = wxPaintDC(self.box)
self._RenderWindow.Render()
self.__InExpose = 0
def OnCreateWindow(self, event): def OnCreateWindow(self, event):
hdl = self.GetHandle() hdl = self.box.GetHandle()
self.renderWindow.SetWindowInfo(str(hdl)) try:
self._RenderWindow.SetParentInfo(str(hdl))
except:
self._RenderWindow.SetWindowInfo(str(hdl))
msg = "Warning:\n "\
"Unable to call vtkRenderWindow.SetParentInfo\n\n"\
"Using the SetWindowInfo method instead. This\n"\
"is likely to cause a lot of flicker when\n"\
"rendering in the vtkRenderWindow. Please\n"\
"use a recent Nightly VTK release (later than\n"\
"March 10 2001) to eliminate this problem."
dlg = wxMessageDialog(NULL, msg, "Warning!",
wxOK |wxICON_INFORMATION)
dlg.ShowModal()
dlg.Destroy()
self.__Created = 1
def OnEraseBackground(self, event):
def OnConfigure(self, event):
sz = self.GetSize()
self.box.SetSize(sz)
# Ugly hack that according to Eric Boix is necessary under
# Windows. If possible Try commenting this out and test under
# Windows.
#self._RenderWindow.GetSize()
#
self._RenderWindow.SetSize(sz.width, sz.height)
def OnLeftButtonDown(self, event):
"Left mouse button pressed."
pass
def OnMiddleButtonDown(self, event):
"Middle mouse button pressed."
pass
def OnRightButtonDown(self, event):
"Right mouse button pressed."
pass
def OnLeftButtonUp(self, event):
"Left mouse button released."
pass
def OnMiddleButtonUp(self, event):
"Middle mouse button released."
pass
def OnRightButtonUp(self, event):
"Right mouse button released."
pass
def OnMouseMove(self, event):
"Mouse has moved."
pass
def OnEnter(self, event):
"Entering the vtkRenderWindow."
pass
def OnLeave(self, event):
"Leaving the vtkRenderWindow."
pass
def OnChar(self, event):
"Process Key events."
pass
def OnKeyDown(self, event):
"Key pressed down."
pass
def OnKeyUp(self, event):
"Key released."
pass pass
def SaveClick(self, event):
self.prev_x, self.prev_y = event.GetPositionTuple()
self.CaptureMouse()
def Release(self, event): class wxVTKRenderWindow(wxVTKRenderWindowBase):
self.ReleaseMouse() """
An example of a fully functional wxVTKRenderWindow that is
based on the vtkRenderWidget.py provided with the VTK sources.
"""
def MouseMotion(self, event): def __init__(self, parent, id, position=wxDefaultPosition,
size=wxDefaultSize, style=0):
wxVTKRenderWindowBase.__init__(self, parent, id, position, size,
style)
self._CurrentRenderer = None
self._CurrentCamera = None
self._CurrentZoom = 1.0
self._CurrentLight = None
self._ViewportCenterX = 0
self._ViewportCenterY = 0
self._Picker = vtkCellPicker()
self._PickedAssembly = None
self._PickedProperty = vtkProperty()
self._PickedProperty.SetColor(1,0,0)
self._PrePickedProperty = None
self._OldFocus = None
# these record the previous mouse position
self._LastX = 0
self._LastY = 0
def OnLeftButtonDown(self, event):
"Left mouse button pressed."
self.StartMotion(event)
def OnMiddleButtonDown(self, event):
"Middle mouse button pressed."
self.StartMotion(event)
def OnRightButtonDown(self, event):
"Right mouse button pressed."
self.StartMotion(event)
def OnLeftButtonUp(self, event):
"Left mouse button released."
self.EndMotion(event)
def OnMiddleButtonUp(self, event):
"Middle mouse button released."
self.EndMotion(event)
def OnRightButtonUp(self, event):
"Right mouse button released."
self.EndMotion(event)
def OnMouseMove(self, event):
event.x, event.y = event.GetPositionTuple() event.x, event.y = event.GetPositionTuple()
size = self.GetClientSize()
if event.LeftIsDown(): if event.LeftIsDown():
# rotate if event.ShiftDown():
camera = self.GetRenderer().GetActiveCamera() self.Pan(event.x, event.y)
camera.Azimuth(float(self.prev_x - event.x) / size.width * 360) else:
camera.Elevation(float(event.y - self.prev_y) / size.width * 360) self.Rotate(event.x, event.y)
camera.OrthogonalizeViewUp()
self.MotionUpdate(event)
elif event.MiddleIsDown(): elif event.MiddleIsDown():
# pan self.Pan(event.x, event.y)
camera = self.GetRenderer().GetActiveCamera()
camera.Yaw(-float(self.prev_x - event.x) / size.width * 30)
camera.Pitch(float(event.y - self.prev_y) / size.width * 30)
camera.OrthogonalizeViewUp()
self.MotionUpdate(event)
elif event.RightIsDown(): elif event.RightIsDown():
# dolly self.Zoom(event.x, event.y)
camera = self.GetRenderer().GetActiveCamera()
camera.Dolly(math.exp(float((event.x - self.prev_x) - \
(event.y - self.prev_y))/ \
size.width))
self.MotionUpdate(event)
def MotionUpdate(self,event): def OnEnter(self, event):
renderer = self.GetRenderer() self.__OldFocus = wxWindow_FindFocus()
renderer.GetLights().InitTraversal() self.SetFocus()
light = renderer.GetLights().GetNextItem() x, y = event.GetPositionTuple()
camera = renderer.GetActiveCamera() self.UpdateRenderer(x,y)
light.SetPosition(camera.GetPosition())
light.SetFocalPoint(camera.GetFocalPoint())
self.renderWindow.Render()
self.prev_x = event.x
self.prev_y = event.y
def OnLeave(self, event):
if (self._OldFocus != None):
self.__OldFocus.SetFocus()
def OnChar(self, event):
key = event.KeyCode()
if (key == ord('r')) or (key == ord('R')):
self.Reset()
elif (key == ord('w')) or (key == ord('W')):
self.Wireframe()
elif (key == ord('s')) or (key == ord('S')):
self.Surface()
elif (key == ord('p')) or (key == ord('P')):
x, y = event.GetPositionTuple()
self.PickActor(x, y)
else:
event.Skip()
# Start of internal functions
def GetZoomFactor(self):
return self._CurrentZoom
def SetZoomFactor(self, zf):
self._CurrentZoom = zf
def GetPicker(self):
return self._Picker
def Render(self):
if (self._CurrentLight):
light = self._CurrentLight
light.SetPosition(self._CurrentCamera.GetPosition())
light.SetFocalPoint(self._CurrentCamera.GetFocalPoint())
wxVTKRenderWindowBase.Render(self)
def UpdateRenderer(self, x, y):
"""
UpdateRenderer will identify the renderer under the mouse and set
up _CurrentRenderer, _CurrentCamera, and _CurrentLight.
"""
sz = self.GetSize()
windowX = sz.width
windowY = sz.height
renderers = self._RenderWindow.GetRenderers()
numRenderers = renderers.GetNumberOfItems()
self._CurrentRenderer = None
renderers.InitTraversal()
for i in range(0,numRenderers):
renderer = renderers.GetNextItem()
vx,vy = (0,0)
if (windowX > 1):
vx = float (x)/(windowX-1)
if (windowY > 1):
vy = (windowY-float(y)-1)/(windowY-1)
(vpxmin,vpymin,vpxmax,vpymax) = renderer.GetViewport()
if (vx >= vpxmin and vx <= vpxmax and
vy >= vpymin and vy <= vpymax):
self._CurrentRenderer = renderer
self._ViewportCenterX = float(windowX)*(vpxmax-vpxmin)/2.0\
+vpxmin
self._ViewportCenterY = float(windowY)*(vpymax-vpymin)/2.0\
+vpymin
self._CurrentCamera = self._CurrentRenderer.GetActiveCamera()
lights = self._CurrentRenderer.GetLights()
lights.InitTraversal()
self._CurrentLight = lights.GetNextItem()
break
self._LastX = x
self._LastY = y
def GetCurrentRenderer(self):
return self._CurrentRenderer
def StartMotion(self, event):
x, y = event.GetPositionTuple()
self.UpdateRenderer(x,y)
self.box.CaptureMouse()
def EndMotion(self, event=None):
if self._CurrentRenderer:
self.Render()
self.box.ReleaseMouse()
def Rotate(self,x,y):
if self._CurrentRenderer:
self._CurrentCamera.Azimuth(self._LastX - x)
self._CurrentCamera.Elevation(y - self._LastY)
self._CurrentCamera.OrthogonalizeViewUp()
self._LastX = x
self._LastY = y
self._CurrentRenderer.ResetCameraClippingRange()
self.Render()
def PanAbsolute(self, x_vec, y_vec):
if self._CurrentRenderer:
renderer = self._CurrentRenderer
camera = self._CurrentCamera
(pPoint0,pPoint1,pPoint2) = camera.GetPosition()
(fPoint0,fPoint1,fPoint2) = camera.GetFocalPoint()
if (camera.GetParallelProjection()):
renderer.SetWorldPoint(fPoint0,fPoint1,fPoint2,1.0)
renderer.WorldToDisplay()
fx,fy,fz = renderer.GetDisplayPoint()
renderer.SetDisplayPoint(fx+x_vec,
fy+y_vec,
fz)
renderer.DisplayToWorld()
fx,fy,fz,fw = renderer.GetWorldPoint()
camera.SetFocalPoint(fx,fy,fz)
renderer.SetWorldPoint(pPoint0,pPoint1,pPoint2,1.0)
renderer.WorldToDisplay()
fx,fy,fz = renderer.GetDisplayPoint()
renderer.SetDisplayPoint(fx+x_vec,
fy+y_vec,
fz)
renderer.DisplayToWorld()
fx,fy,fz,fw = renderer.GetWorldPoint()
camera.SetPosition(fx,fy,fz)
else:
(fPoint0,fPoint1,fPoint2) = camera.GetFocalPoint()
# Specify a point location in world coordinates
renderer.SetWorldPoint(fPoint0,fPoint1,fPoint2,1.0)
renderer.WorldToDisplay()
# Convert world point coordinates to display coordinates
dPoint = renderer.GetDisplayPoint()
focalDepth = dPoint[2]
aPoint0 = self._ViewportCenterX + x_vec
aPoint1 = self._ViewportCenterY + y_vec
renderer.SetDisplayPoint(aPoint0,aPoint1,focalDepth)
renderer.DisplayToWorld()
(rPoint0,rPoint1,rPoint2,rPoint3) = renderer.GetWorldPoint()
if (rPoint3 != 0.0):
rPoint0 = rPoint0/rPoint3
rPoint1 = rPoint1/rPoint3
rPoint2 = rPoint2/rPoint3
camera.SetFocalPoint((fPoint0 - rPoint0) + fPoint0,
(fPoint1 - rPoint1) + fPoint1,
(fPoint2 - rPoint2) + fPoint2)
camera.SetPosition((fPoint0 - rPoint0) + pPoint0,
(fPoint1 - rPoint1) + pPoint1,
(fPoint2 - rPoint2) + pPoint2)
self.Render()
def Pan(self, x, y):
self.PanAbsolute(x - self._LastX, - y + self._LastY)
self._LastX = x
self._LastY = y
def Zoom(self,x,y):
if self._CurrentRenderer:
renderer = self._CurrentRenderer
camera = self._CurrentCamera
zoomFactor = math.pow(1.02,(0.5*(self._LastY - y)))
self._CurrentZoom = self._CurrentZoom * zoomFactor
if camera.GetParallelProjection():
parallelScale = camera.GetParallelScale()/zoomFactor
camera.SetParallelScale(parallelScale)
else:
camera.Dolly(zoomFactor)
renderer.ResetCameraClippingRange()
self._LastX = x
self._LastY = y
self.Render()
def Reset(self):
if self._CurrentRenderer:
self._CurrentRenderer.ResetCamera()
self.Render()
def Wireframe(self):
actors = self._CurrentRenderer.GetActors()
numActors = actors.GetNumberOfItems()
actors.InitTraversal()
for i in range(0,numActors):
actor = actors.GetNextItem()
actor.GetProperty().SetRepresentationToWireframe()
self.Render()
def Surface(self):
actors = self._CurrentRenderer.GetActors()
numActors = actors.GetNumberOfItems()
actors.InitTraversal()
for i in range(0,numActors):
actor = actors.GetNextItem()
actor.GetProperty().SetRepresentationToSurface()
self.Render()
def PickActor(self,x,y):
if self._CurrentRenderer:
renderer = self._CurrentRenderer
picker = self._Picker
windowY = self.GetSize().height
picker.Pick(x,(windowY - y - 1),0.0,renderer)
assembly = picker.GetAssembly()
if (self._PickedAssembly != None and
self._PrePickedProperty != None):
self._PickedAssembly.SetProperty(self._PrePickedProperty)
# release hold of the property
self._PrePickedProperty.UnRegister(self._PrePickedProperty)
self._PrePickedProperty = None
if (assembly != None):
self._PickedAssembly = assembly
self._PrePickedProperty = self._PickedAssembly.GetProperty()
# hold onto the property
self._PrePickedProperty.Register(self._PrePickedProperty)
self._PickedAssembly.SetProperty(self._PickedProperty)
self.Render()