#
##
## This file is part of pyFormex 2.0 (Mon Sep 14 12:29:05 CEST 2020)
## pyFormex is a tool for generating, manipulating and transforming 3D
## geometrical models by sequences of mathematical operations.
## Home page: http://pyformex.org
## Project page: http://savannah.nongnu.org/projects/pyformex/
## Copyright 2004-2020 (C) Benedict Verhegghe (benedict.verhegghe@ugent.be)
## Distributed under the GNU General Public License version 3 or later.
##
## This program is free software: you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation, either version 3 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program. If not, see http://www.gnu.org/licenses/.
##
"""OpenGL camera handling
Python OpenGL framework for pyFormex
This OpenGL framework is intended to replace (in due time)
the current OpenGL framework in pyFormex.
(C) 2013 Benedict Verhegghe and the pyFormex project.
"""
import numpy as np
import OpenGL.GL as GL
from pyformex.opengl.matrix import Matrix4, Vector4
from pyformex import arraytools as at
from pyformex.coords import Coords
from pyformex.plugins.nurbs import Coords4
from pyformex.config import Config
#
# DEVS: collect all GL calls here, so we can easily move them somewhere else
#
[docs]def gl_projection():
"""Get the OpenGL projection matrix"""
return GL.glGetDoublev(GL.GL_PROJECTION_MATRIX)
[docs]def gl_modelview():
"""Get the OpenGL modelview matrix"""
return GL.glGetDoublev(GL.GL_MODELVIEW_MATRIX)
[docs]def gl_viewport():
"""Get the OpenGL viewport"""
return GL.glGetIntegerv(GL.GL_VIEWPORT)
[docs]def gl_loadmodelview(m):
"""Load the OpenGL modelview matrix"""
GL.glMatrixMode(GL.GL_MODELVIEW)
GL.glLoadMatrixf(m)
[docs]def gl_loadprojection(m):
"""Load the OpenGL projection matrix"""
GL.glMatrixMode(GL.GL_PROJECTION)
GL.glLoadMatrixf(m)
# we reset OpenGL engine to default MODELVIEW
GL.glMatrixMode(GL.GL_MODELVIEW)
[docs]def gl_depth(x, y):
"""Read the depth value of the pixel at (x,y)"""
return GL.glReadPixels(x, y, 1, 1, GL.GL_DEPTH_COMPONENT, GL.GL_FLOAT)
#
# Normalize and denormalize should probably be moved to arraytools.
#
[docs]def normalize(x, w):
"""Normalized coordinates inside a window.
Parameters:
- `x`: an (np,nc) array with coordinates.
- `w`: a (2,nc) array with minimal and width of the window that will
be mapped to the range -1..1.
Returns an array with the x values linearly remapped thus that values w[0]
become -1 and values w[0]+w[1] become +1.
"""
x = at.checkArray(x, (-1, -1), 'f')
np, nc = x.shape
w = at.checkArray(w, (2, nc), 'f', 'i')
return (x-w[0]) * 2 / w[1] - 1
[docs]def denormalize(x, w):
"""Map normalized coordinates to fit a window
Parameters:
- `x`: an (np,nc) array with normalized coordinates.
- `w`: a (2,nc) array with minimal and width values of the window.
Returns an array with the x values linearly remapped thus that values -1
coincide with the minimum window values and +1 with the minimum+width
values.
"""
x = at.checkArray(x, (-1, -1), 'f')
np, nc = x.shape
w = at.checkArray(w, (2, nc), 'f', 'i')
return w[0] + (1+x) * w[1] / 2
[docs]def perspective_matrix(left, right, bottom, top, near, far):
"""Create a perspective Projection matrix.
"""
m = Matrix4()
m[0, 0] = 2 * near / (right-left)
m[1, 1] = 2 * near / (top-bottom)
m[2, 0] = (right+left) / (right-left)
m[2, 1] = (top+bottom) / (top-bottom)
m[2, 2] = - (far+near) / (far-near)
m[2, 3] = -1.
m[3, 2] = -2 * near * far / (far-near)
m[3, 3] = 0.
return m
[docs]def orthogonal_matrix(left, right, bottom, top, near, far):
"""Create an orthogonal Projection matrix.
"""
m = Matrix4()
m[0, 0] = 2 / (right-left)
m[1, 1] = 2 / (top-bottom)
m[2, 2] = -2 / (far-near)
m[3, 0] = - (right+left) / (right-left)
m[3, 1] = - (top+bottom) / (top-bottom)
m[3, 2] = - (far+near) / (far-near)
return m
[docs]def pick_matrix(x, y, w, h, viewport):
"""Create a pick Projection matrix
"""
m = Matrix4()
m[0, 0] = viewport[2] / w;
m[1, 1] = viewport[3] / h;
m[3, 0] = (viewport[2] + 2.0 * (viewport[0] - x)) / w;
m[3, 1] = (viewport[3] + 2.0 * (viewport[1] - y)) / h;
return m
[docs]class Camera(object):
"""A camera for 3D model rendering.
The Camera class holds all the camera parameters related to the
rendering of a 3D scene onto a 2D canvas. These includes parameters
related to camera position and orientation, as well as lens related
parameters (opening angle, front and back clipping planes).
The class provides the required matrices to transform the 3D world
coordinates to 2D canvas coordinates, as well as a wealth of methods
to change the camera settings in a convenient way so as to simulate
smooth camera manipulation.
The basic theory of camera handling and 3D rendering can be found in
a lot of places on the internet, especially in OpenGL related places.
However, while the pyFormex rendering engine is based on OpenGL,
the way it stores and handles the camera parameters is more sophisticated
than what is usually found in popular tutorials on OpenGL rendering.
Therefore we give here a extensive description of how the pyFormex
camera handling and 3D to 2D coordinate transformation works.
.. note: The remainder below is obsolete and needs to be rewritten.
Camera position and orientation:
The camera viewing line is defined by two points: the position of
the camera and the center of the scene the camera is looking at.
We use the center of the scene as the origin of a local coordinate
system to define the camera position. For convenience, this could be
stored in spherical coordinates, as a distance value and two angles:
longitude and latitude. Furthermore, the camera can also rotate around
its viewing line. We can define this by a third angle, the twist.
From these four values, the needed translation vector and rotation
matrix for the scene rendering may be calculated.
Inversely however, we can not compute a unique set of angles from
a given rotation matrix (this is known as 'gimball lock').
As a result, continuous (smooth) camera rotation by e.g. mouse control
requires that the camera orientation be stored as the full rotation
matrix, rather than as three angles. Therefore we store the camera
position and orientation as follows:
- `ctr`: `[ x,y,z ]` : the reference point of the camera:
this is always a point on the viewing axis. Usually, it is set to
the center of the scene you are looking at.
- `dist`: distance of the camera to the reference point.
- `rot`: a 3x3 rotation matrix, rotating the global coordinate system
thus that the z-direction is oriented from center to camera.
These values have influence on the Modelview matrix.
Camera lens settings:
The lens parameters define the volume that is seen by the camera.
It is described by the following parameters:
- `fovy`: the vertical lens opening angle (Field Of View Y),
- `aspect`: the aspect ratio (width/height) of the lens. The product
`fovy * aspect` is the horizontal field of view.
- `near, far`: the position of the front and back clipping planes.
They are given as distances from the camera and should both be
strictly positive. Anything that is closer to the camera than
the `near` plane or further away than the `far` plane, will not be
shown on the canvas.
Camera methods that change these values will not directly change
the Modelview matrix. The :meth:`loadModelview` method has to be called
explicitely to make the settings active.
These values have influence on the Projection matrix.
Methods that change the camera position, orientation or lens parameters
will not directly change the related Modelview or Projection matrix.
They will just flag a change in the camera settings. The changes are
only activated by a call to the :meth:`loadModelview` or
:meth:`loadProjection` method, which will test the flags to see whether
the corresponding matrix needs a rebuild.
The default camera is at distance 1.0 of the center point [0.,0.,0.] and
looking in the -z direction.
Near and far clipping planes are by default set to 0.1, resp 10 times
the camera distance.
Properties:
- `modelview`: Matrix4: the OpenGL Modelview transformation matrix
- `projection`: Matrix4: the OpenGL Projection transformation matrix
"""
# DEVELOPERS:
# The camera class assumes that matrixmode is always Modelview on entry.
# For operations in other modes, an explicit switch before the operations
# and afterwards back to Modelview should be performed.
def __init__(self, focus=(0., 0., 0.), angles=(0., 0., 0.), dist=1.,
fovy=45., aspect=4./3., clip=(0.01, 100.), perspective=True,
area=(0., 0., 1., 1.),
locked=False, keep_aspect=True, tracking=False,
):
"""Create a new camera.
The default camera is positioned at (0.,0.,1.) looking along the -z
axis in the direction of the point (0.,0.,0.) and with the upvector
in the direction of the y-axis.
"""
self.locked = locked # this needs to be before focus setter!
self.focus = focus
self.dist = dist
self.keep_aspect = True
self.tracking = False
self._modelview = Matrix4()
self._projection = Matrix4()
self.p = self.v = None
self.viewChanged = True
self.lensChanged = True
self._perspective = perspective
self.setAngles(angles)
self.setLens(fovy, aspect)
self.setClip(*clip)
self.area = None
self.setArea(*area, relative=False)
@property
def settings(self):
"""Dict to allow save/load of camera
This dict contains all data that allow save and restore
of the camera to exactly the same settings (on the same
size of Canvas).
"""
return {
'focus': self.focus.tolist(),
'angles': self.angles,
'dist': self.dist,
'fovy': self.fovy,
'aspect': self.aspect,
'clip': (self.near, self.far),
'area': self.area.ravel().tolist(),
'perspective': self.perspective,
'locked': self.locked,
'tracking': self.tracking,
'keep_aspect': self.keep_aspect,
}
@property
def modelview(self):
"""Return the current modelview matrix.
This will recompute the modelview matrix if any camera position
parameters have changed.
"""
if self.viewChanged:
self.setModelview()
return self._modelview
@modelview.setter
def modelview(self, value):
"""Set the modelview matrix to the specified matrix.
value should be a proper modelview matrix
"""
self._modelview = Matrix4(value)
@property
def projection(self):
"""Return the current projection matrix.
This will recompute the projection matrix if any camera lens
parameters have changed.
"""
if self.lensChanged:
self.setProjection()
return self._projection
@projection.setter
def projection(self, value):
"""Set the projection matrix to the specified matrix.
value should be a proper projection matrix
"""
self._projection = Matrix4(value)
@property
def viewport(self):
"""Return the camera viewport.
This property can not be changed directly.
It should be changed by resizing the parent canvas.
"""
return gl_viewport()
@property
def focus(self):
"""Return the camera reference point (the focus point)."""
return self._focus
@focus.setter
def focus(self, vector):
"""Set the camera reference point (the focus point).
The focus is the point the camer is looking at. It is a point on
the camera's optical axis.
- `vector`: (3,) float array: the global coordinates of the focus.
"""
if not self.locked:
self._focus = at.checkArray(vector, (3,), 'f')
self.viewChanged = True
@property
def dist(self):
"""Return the camera distance.
The camera distance is the distance between the camera eye and
the camera focus point.
"""
return self._dist
@dist.setter
def dist(self, dist):
"""Set the camera distance.
- `dist`: a strictly positive float value. Invalid values are
silently ignored.
"""
if not self.locked:
if dist > 0.0 and dist != np.inf:
self._dist = dist
self.viewChanged = True
@property
def perspective(self, on=True):
"""Return the perspecive flag.
If the perspective flag is True, the camera uses a perspective
projection. If it is False, the camera uses orthogonal projection.
"""
return self._perspective
@perspective.setter
def perspective(self, flag):
"""Set the perspecive flag.
- `flag`: bool, the value to set the perspective flag to.
If changed, this forces the recalculation of the projection matrix.
"""
if not self.locked:
if flag != self._perspective:
self._perspective = bool(flag)
self.lensChanged = True
@property
def rot(self):
"""Return the camera rotation matrix."""
return self.modelview.rot
@property
def angles(self):
"""Return the camera angles.
Returns a tuple (longitude, latitude, twist) in local camera axes.
"""
R = self.modelview.rot.T
R = R[[2, 0, 1]]
R = R[:, [2, 0, 1]]
a = [at.Float(angle) for angle in at.cardanAngles(R)]
return (a[2], -a[1], a[0])
@property
def upvector(self):
"""Return the camera up vector"""
return self.modelview.rot[:, 1].reshape((3,))
@property
def axis(self):
"""Return a unit vector along the camera axis.
The camera axis points from the focus towards the camera.
"""
# this is the same as: at.normalize(self.eye-self.focus)
return self.modelview.rot[:, 2].reshape((3,))
[docs] def setAngles(self, angles, axes=None):
"""Set the rotation angles.
Parameters
----------
angles: tuple of floats
A tuple of three angles (long,lat,twist) in degrees.
A value None is also accepted, but has no effect.
axes: if specified, any number of rotations can be applied.
"""
if not self.locked:
if angles is None:
return
if isinstance(angles, str):
raise ValueError("Invalid value for camera angles: %s" % angles)
#angles = view_angles.get(angles)
if axes is None:
axes = ((0., -1., 0.), (1., 0., 0.), (0., 0., -1.))
self.setModelview(angles=list(zip(angles, axes))[::-1])
@property
def eye(self):
"""Return the position of the camera."""
return self.toWorld([0., 0., 0.])
@eye.setter
def eye(self, eye):
"""Set the position of the camera."""
return self.lookAt(eye=eye)
[docs] def lock(self, onoff=True):
"""Lock/unlock a camera.
When a camera is locked, its position and lens parameters can not be
changed.
This can e.g. be used in multiple viewports layouts to create fixed
views from different angles.
"""
self.locked = bool(onoff)
def unlock(self):
self.lock(False)
[docs] def report(self):
"""Return a report of the current camera settings."""
return """Camera Settings:
Eye: %s
Focus: %s
Distance: %s
Angles: %s
Axis: %s
UpVector: %s
Rotation Matrix:
%s
Field of View y: %s
Aspect Ratio: %s
Area: %s, %s
Near/Far Clip: %s, %s
""" % (self.eye, self.focus, self.dist, self.angles, self.axis, self.upvector, self.rot, self.fovy, self.aspect, self.area[0], self.area[1], self.near, self.far)
[docs] def dolly(self, val):
"""Move the camera eye towards/away from the scene center.
This has the effect of zooming. A value > 1 zooms out,
a value < 1 zooms in. The resulting enlargement of the view
will approximately be 1/val.
A zero value will move the camera to the center of the scene.
The front and back clipping planes may need adjustment after
a dolly operation.
"""
if not self.locked:
self.dist *= val
self.viewChanged = True
# TODO: This is broken!
[docs] def pan(self, val, axis=0):
"""Rotate the camera around axis through its eye.
The camera is rotated around an axis through the eye point.
For axes 0 and 1, this will move the focus, creating a panning
effect. The default axis is parallel to the y-axis, resulting in
horizontal panning. For vertical panning (axis=1) a convenience
alias tilt is created.
For axis = 2 the operation is equivalent to the rotate operation.
"""
if not self.locked:
if axis==0 or axis ==1:
pos = self.eye
self.eye[axis] = (self.eye[axis] + val) % 360
print(self.report())
self.focus = diff(pos, sphericalToCartesian(self.eye))
print(self.report())
elif axis==2:
print(self.report())
self.twist = (self.twist + val) % 360
self.viewChanged = True
# TODO: THis depends on the broken pan!
[docs] def tilt(self, val):
"""Rotate the camera up/down around its own horizontal axis.
The camera is rotated around and perpendicular to the plane of the
y-axis and the viewing axis. This has the effect of a vertical pan.
A positive value tilts the camera up, shifting the scene down.
The value is specified in degrees.
"""
if not self.locked:
self.pan(val, 1)
self.viewChanged = True
[docs] def move(self, dx, dy, dz):
"""Move the camera over translation (dx,dy,dz) in global coordinates.
The focus of the camera is moved over the specified translation
vector. This has the effect of moving the scene in opposite direction.
"""
if not self.locked:
x, y, z = self.ctr
self.focus += [dx, dy, dz]
## def truck(self,dx,dy,dz):
## """Move the camera translation vector in local coordinates.
## This has the effect of moving the scene in opposite direction.
## Positive coordinates mean:
## first coordinate : truck right,
## second coordinate : pedestal up,
## third coordinate : dolly out.
## """
## #pos = self.position
## ang = self.getAngles()
## tr = [dx,dy,dz]
## for i in [1,0,2]:
## r = rotationMatrix(i,ang[i])
## tr = multiply(tr, r)
## self.move(*tr)
## self.viewChanged = True
# TODO
def translate(self, vx, vy, vz, local=True):
if not self.locked:
if local:
vx, vy, vz = self.toWorld([vx, vy, vz, 1])
self.move(-vx, -vy, -vz)
[docs] def setLens(self, fovy=None, aspect=None):
"""Set the field of view of the camera.
We set the field of view by the vertical opening angle fovy
and the aspect ratio (width/height) of the viewing volume.
A parameter that is not specified is left unchanged.
"""
if fovy:
self.fovy = min(abs(fovy), 180)
if aspect:
self.aspect = abs(aspect)
self.lensChanged = True
[docs] def resetArea(self):
"""Set maximal camera area.
Resets the camera window area to its maximum values corresponding
to the fovy setting, symmetrical about the camera axes.
"""
self.setArea(0., 0., 1., 1., False)
[docs] def setArea(self, hmin, vmin, hmax, vmax, relative=True, focus=False, center=False, clip=True):
"""Set the viewable area of the camera.
Note: Use relative=False and clip=False if you want to set the zoom
exactly as in previously recorded values.
"""
area = np.array([hmin, vmin, hmax, vmax])
if clip:
area = area.clip(0., 1.)
if area[0] < area[2] and area[1] < area[3]:
area = area.reshape(2, 2)
mean = (area[1]+area[0]) * 0.5
diff = (area[1]-area[0]) * 0.5
if relative:
if self.keep_aspect:
aspect = diff[0] / diff[1]
if aspect > 1.0:
diff[1] = diff[0] # / self.aspect
# no aspect factor: this is relative!!!
else:
diff[0] = diff[1] # * self.aspect
if focus:
mean = np.zeros(2)
area[0] = mean-diff
area[1] = mean+diff
#print("RELATIVE AREA %s" % (area))
area = (1.-area) * self.area[0] + area * self.area[1]
if center:
# make center of area equal to 0.5,0.5
mean = (area[1]+area[0]) * 0.5
area += 0.5-mean
#print("OLD ZOOM AREA %s (aspect %s)" % (self.area,self.aspect))
#print("NEW ZOOM AREA %s" % (area))
self.area = area
self.lensChanged = True
[docs] def zoomArea(self, val=0.5, area=None):
"""Zoom in/out by shrinking/enlarging the camera view area.
The zoom factor is relative to the current setting.
Values smaller than 1.0 zoom in, larger values zoom out.
"""
if val>0:
#val = (1.-val) * 0.5
#self.setArea(val,val,1.-val,1.-val,focus=focus)
if area is None:
area = self.area
#print("ZOOM AREA %s (%s)" % (area.tolist(),val))
mean = (area[1]+area[0]) * 0.5
diff = (area[1]-area[0]) * 0.5 * val
area[0] = mean-diff
area[1] = mean+diff
self.area = area
#print("CAMERA AREA %s" % self.area.tolist())
self.lensChanged = True
[docs] def transArea(self, dx, dy):
"""Pan by moving the vamera area.
dx and dy are relative movements in fractions of the
current area size.
"""
#print("TRANSAREA %s,%s" % (dx,dy))
area = self.area
diff = (area[1]-area[0]) * np.array([dx, dy])
area += diff
self.area = area
self.lensChanged = True
[docs] def setClip(self, near, far):
"""Set the near and far clipping planes"""
if near > 0 and near < far:
self.near, self.far = near, far
self.lensChanged = True
else:
print("Error: Invalid Near/Far clipping values")
## def setClipRel(self,near,far):
## """Set the near and far clipping planes"""
## if near > 0 and near < far:
## self.near,self.far = near,far
## self.lensChanged = True
## else:
## print("Error: Invalid Near/Far clipping values")
## def zoom(self,val=0.5):
## """Zoom in/out by shrinking/enlarging the camera view angle.
## The zoom factor is relative to the current setting.
## Use setFovy() to specify an absolute setting.
## """
## if val>0:
## self.fovy *= val
## self.lensChanged = True
#### global manipulation ###################
[docs] def setTracking(self, onoff=True):
"""Enable/disable coordinate tracking using the camera"""
if onoff:
self.tracking = True
#self.set3Datrices()
else:
self.tracking = False
#################################################################
## Operations on modelview matrix ##
[docs] def setProjection(self):
"""Set the projection matrix.
This computes and sets the camera's projection matrix, depending
on the current camera settings. The projection can either be
an orthogonal or a perspective one.
The computed matrix is saved as the camera's projection matrix,
and the lensChanged attribute is set to False.
The matrix can be retrieved from the projection attribute,
and can be loaded in the GL context with loadProjection().
This function does nothing if the camera is locked.
"""
if self.locked:
return
fv = at.tand(self.fovy*0.5)
if self._perspective:
fv *= self.near
else:
fv *= self.dist
fh = fv * self.aspect
x0, x1 = 2*self.area - 1.0
frustum = (fh*x0[0], fh*x1[0], fv*x0[1], fv*x1[1], self.near, self.far)
if self._perspective:
func = perspective_matrix
else:
func = orthogonal_matrix
self.projection = func(*frustum)
try:
self.projection_callback(self)
except Exception:
pass
self.lensChanged = False
[docs] def loadProjection(self):
"""Load the Projection matrix.
If lens parameters of the camera have been changed, the current
Projection matrix is rebuild.
Then, the current Projection matrix of the camera is loaded into the
OpenGL engine.
"""
gl_loadprojection(self.projection.gl())
[docs] def pickMatrix(self, rect, viewport=None):
"""Return a picking matrix.
The picking matrix confines the scope of the normalized device
coordinates to a rectangular subregion of the camera viewport.
This means that values in the range -1 to +1 are inside the
rectangle.
Parameters:
- `rect`: a tuple of 4 floats (x,y,w,h) defining the picking region
center (x,y) and size (w,h)
- `viewport`: a tuple of 4 int values (xmin,ymin,xmax,ymax) defining
the size of the viewport. This is normally left unspecified and
set to the camera viewport.
"""
if viewport is None:
viewport = self.viewport
return pick_matrix(rect[0], rect[1], rect[2], rect[3], viewport)
[docs] def eyeToClip(self, x):
"""Transform a vertex from eye to clip coordinates.
This transforms the vertex using the current Projection matrix.
It is equivalent with multiplying the homogeneous
coordinates with the Projection matrix, but is done here
in an optimized way.
"""
return self.projection.transform(x)
[docs] def clipToEye(self, x):
"""Transform a vertex from clip to eye coordinates.
This transforms the vertex using the inverse of the
current Projection matrix.
It is equivalent with multiplying the homogeneous
coordinates with the inverse Projection matrix, but is done
here in an optimized way.
"""
return self.projection.invtransform(x)
#################################################################
## Operations on modelview matrix ##
[docs] def lookAt(self, focus=None, eye=None, up=None):
"""Set the Modelview matrix to look at the specified focus point.
The Modelview matrix is set with the camera positioned at eye
and looking at the focus points, while the camera up vector is
in the plane of the camera axis (focus-eye) and the specified
up vector.
If any of the arguments is left unspecified, the current value
will be used.
"""
if not self.locked:
if focus is None:
focus = self.focus
else:
focus = at.checkArray(focus, (3,), 'f')
if eye is None:
eye = self.eye
else:
eye = at.checkArray(eye, (3,), 'f')
if up is None:
up = self.upvector
else:
up = at.normalize(at.checkArray(up, (3,), 'f'))
vector = eye-focus
self.focus = focus
self.dist = at.length(vector)
axis2 = at.normalize(vector)
axis0 = at.normalize(np.cross(up, axis2))
axis1 = at.normalize(np.cross(axis2, axis0))
m = Matrix4()
m.rotate(np.column_stack([axis0, axis1, axis2]))
m.translate(-eye)
self.setModelview(m)
[docs] def rotate(self, val, vx, vy, vz):
"""Rotate the camera around current camera axes."""
if not self.locked:
rot = self._modelview.rot
m = Matrix4()
m.translate([0, 0, -self.dist])
m.rotate(val % 360, [vx, vy, vz])
m.rotate(rot)
m.translate(-self.focus)
self.setModelview(m)
[docs] def setModelview(self, m=None, angles=None):
"""Set the Modelview matrix.
The Modelview matrix can be set from one of the following sources:
- if `mat` is specified, it is a 4x4 matrix with a valuable
Modelview transformation. It will be set as the current camera
Modelview matrix.
- else, if `angles` is specified, it is a sequence of tuples
(angle, axis) each of which define a rotation of the camera
around an axis through the focus point.
The camera Modelview matrix is set from the current camera focus,
the current camera distance, and the specified angles/axes.
This option is typically used to change the viewing direction
of the camera, while keeping the focus point and camera distance.
- else, if the viewChanged flags is set, the camera Modelview
matrix is set from the current camera focus, the current camera
distance, and the current camera rotation matrix.
This option is typically used after changing the camera focus
point and/or distance, while keeping the current viewing angles.
- else, the current Modelview matrix remains unchanged.
In all cases, if a modelview callback was set, it is called,
and the viewChanged flag is cleared.
"""
if self.locked:
return
if m is None and angles is not None:
m = Matrix4()
m.translate([0, 0, -self.dist])
#print(list(angles))
for angle, axis in angles:
m.rotate(angle, axis)
m.translate(-self.focus)
elif m is None and self.viewChanged:
m = Matrix4()
m.translate([0, 0, -self.dist])
m.rotate(self._modelview.rot)
m.translate(-self.focus)
if m is not None:
self.modelview = m
try:
self.modelview_callback(self)
except Exception:
pass
self.viewChanged = False
[docs] def loadModelview(self):
"""Load the Modelview matrix.
If camera positioning parameters have been changed, the current
Modelview matrix is rebuild.
Then, the current Modelview matrix of the camera is loaded into the
OpenGL engine.
"""
gl_loadmodelview(self.modelview.gl())
[docs] def toEye(self, x):
"""Transform a vertex from world to eye coordinates.
This transforms the vertex using the current Modelview matrix.
It is equivalent with multiplying the homogeneous
coordinates with the Modelview matrix, but is done here
in an optimized way.
"""
x = at.checkArray(x, (-1, 3), 'f')
return np.dot(x, self.modelview[:3, :3]) + self.modelview[3, :3]
[docs] def toWorld(self, x):
"""Transform a vertex from eye to world coordinates.
This transforms the vertex using the inverse of the
current Modelview matrix.
It is equivalent with multiplying the homogeneous
coordinates with the inverse Modelview matrix, but is done
here in an optimized way.
"""
x = at.checkArray(x, (3,), 'f') + [0., 0., self.dist]
return np.dot(x, self.rot.T) + self.focus
#################################################################
## Transform vertices with modelview and projection matrix ##
[docs] def toWindow(self, x):
"""Convert normalized device coordinates to window coordinates"""
# This is only correct when glDepthRange(0.0, 1.0)
# We should not change the depth range
vp = gl_viewport()
#print([[vp[0], vp[1], 0], [vp[2], vp[3], 1]])
return denormalize(x[:, :3], [[vp[0], vp[1], 0], [vp[2], vp[3], 1]])
[docs] def fromWindow(self, x):
"""Convert window coordinates to normalized device coordinates"""
# This is only correct when glDepthRange(0.0, 1.0)
# We should not change the depth range
vp = gl_viewport()
x = at.checkArray(x, (-1, 3), 'f')
return normalize(x[:, :3], [[vp[0], vp[1], 0], [vp[2], vp[3], 1]])
[docs] def toNDC(self, x, rect=None):
"""Convert world coordinates to normalized device coordinates.
The normalized device coordinates (NDC) have x and y values
in the range -1 to +1 for points that are falling within the
visible region of the camera.
Parameters:
- `x`: Coords with the world coordinates to be converted
- `rect`: optional, a tuple of 4 values (x,y,w,h) specifying
a rectangular subregion of the camera's viewport. The default
is the full camera viewport.
The return value is a Coords. The z-coordinate provides
depth information.
"""
m = self.modelview*self.projection
if rect is not None:
m = m*self.pickMatrix(rect)
x = Coords4(x)
x = Coords4(np.dot(x, m))
if self._perspective:
x = x.toCoords() # This performs the perspective divide
else:
# Orthogonal projection
# This is not tested yet!!!
x = Coords(x[..., :3])
return x
[docs] def toNDC1(self, x, rect=None):
"""This is like toNDC without the perspective divide
This function is useful to compute the vertex position of a
3D point as computed by the vertex shader.
"""
m = self.modelview*self.projection
if rect is not None:
m = m*self.pickMatrix(rect)
x = Coords4(x)
x = Coords4(np.dot(x, m))
x = Coords(x[..., :3])
return x
[docs] def project(self, x):
"""Map the world coordinates (x,y,z) to window coordinates."""
#m = self.modelview*self.projection
# Modelview transform
e = Vector4(x)*self.modelview
#print("EYE COORDINATES:",e)
w = -e[:, 2]
# Projection
x = e*self.projection
#print("CLIP COORDINATES:",x)
# Perspective division
if self._perspective:
## if (w == 0.0):
## return [np.inf,np.inf,np.inf]
x = x/w
#print("NORMALIZED DEVICE COORDINATES:",x)
# Map to window
return self.toWindow(x)
[docs] def unproject(self, x):
"""Map the window coordinates x to object coordinates."""
m = self.modelview*self.projection
#print("M*P",m)
m1 = np.linalg.inv(m)
#print("M*P -1",m1)
x = self.fromWindow(x)
#print("NORMALIZED DEVICE COORDINATES:",x)
x = Vector4(x)*m1
return Coords(x[:, :3] / x[:, 3])
[docs] def inside(self, x, rect=None, return_depth=False):
"""Test if points are visible inside camera.
Parameters:
- `rect`: optional, a tuple of 4 values (x,y,w,h) specifying
a rectangular subregion of the camera's viewport. The default
is the full camera viewport.
- `return_depth`: if True, also returns the the z-depth of the
points.
Returns a boolean array with value 1 (True) for the points that
are projected inside the rectangular are of the camera.
If `return_depth` is True, a second array with the z-depth value
of all the points is returned.
"""
ndc = self.toNDC(x, rect)
# TODO: WHY THE abs ?????
xy = ndc[:, :2]
inside = (xy >= -1).all(axis=-1) * (xy <= 1).all(axis=-1)
if return_depth:
return inside, ndc[:, 2]
else:
return inside
[docs] def config(self):
"""Return a Config with the settings to be saved for a restore"""
from pyformex.plugins.saveload import dict2Config
C = dict2Config(self.settings)
return C
[docs] def save(self, filename):
"""Save the camera settings to file"""
C = self.config()
C.write(filename, header="#Camera settings saved from pyFormex\n", trailer="#End\n")
[docs] def loadConfig(self, config):
"""Load the camera settings from a Config or dict"""
from pyformex.gui import toolbar
self.__init__(**config)
# Since this might have changed the perspective state
# of the current camera, updat the gui perspective button
toolbar.updatePerspectiveButton()
apply = loadConfig
[docs] def load(self, filename):
"""Load the camera settings from file"""
config = Config()
try:
config.read(filename)
except Exception:
raise ValueError("Invalid Camera save file: %s" % filename)
self.loadConfig(config)
#################################
# Compatibility: should be removed after complete conversion
loadModelView = loadModelview
def saveModelView(self):
pass
def set3DMatrices(self):
self.loadProjection()
self.loadModelView()
def setPerspective(self, on=True):
self.perspective = on
# End
