Quick hack to get visualisation of ray trajectories working :(

Author: mattngc

demos/logging_trajectories.py

@@ -0,0 +1,77 @@
+
+# External imports
+# import matplotlib as mpl
+# from mpl_toolkits.mplot3d import Axes3D
+import matplotlib.pyplot as plt
+import numpy as np
+
+# Raysect imports
+from raysect.optical import World, translate, rotate, Point3D, d65_white, Ray, Vector3D
+from raysect.optical.material.absorber import AbsorbingSurface
+from raysect.optical.library import schott
+from raysect.primitive import Sphere, Box
+from raysect.optical.loggingray import LoggingRay
+from raysect.primitive.lens.spherical import *
+
+
+world = World()
+
+# Create a glass BiConvex lens we want to study
+lens_glass = schott("N-BK7")
+lens_glass.transmission_only = True
+lens = BiConvex(0.0254, 0.0052, 0.0506, 0.0506, parent=world, material=lens_glass)
+lens.meta['viz-color'] = (66/255, 188/255, 244/255)
+lens.meta['viz-opacity'] = 0.5
+
+# lens.meta['viz']['color'] = (66/255, 188/255, 244/255)
+# lens.meta['viz']['opacity'] = 0.5
+
+# Create a target plane behind the lens.
+target = Box(lower=Point3D(-0.05, -0.05, -0), upper=Point3D(0.05, 0.05, 0), material=AbsorbingSurface(),
+             transform=translate(0, 0, 0.1), parent=world)
+target.meta['viz-color'] = (224/255, 100/255, 17/255)
+
+
+# for each sample direction trace a logging ray and plot the ray trajectory
+plt.ion()
+fig = plt.figure()
+# ax = fig.gca(projection='3d')
+
+# for u in np.linspace(-0.006, 0.006, 5):
+for v in np.linspace(-0.012, 0.012, 11):
+
+    start = Point3D(v, 0, -0.05)
+    log_ray = LoggingRay(start, Vector3D(0, 0, 1))
+    log_ray.trace(world)
+
+    vertices = log_ray.path_vertices
+
+    p = [(v.x, v.z) for v in vertices]
+    p = np.array(p)
+
+    plt.plot(p[:, 0], p[:, 1], 'k-')
+    plt.plot(p[:, 0], p[:, 1], 'r.')
+
+
+from raysect_mayavi import visualise_scenegraph
+from mayavi import mlab
+
+
+visualise_scenegraph(world)
+
+
+for v in np.linspace(-0.012, 0.012, 11):
+
+    start = Point3D(v, 0, -0.05)
+    log_ray = LoggingRay(start, Vector3D(0, 0, 1))
+    log_ray.trace(world)
+
+    vertices = log_ray.path_vertices
+
+    p = [(v.x, v.y, v.z) for v in vertices]
+    p = np.array(p)
+
+    mlab.plot3d(p[:, 0], p[:, 1], p[:, 2], tube_radius=0.0005)
+
+plt.ioff()
+plt.show()

raysect_mayavi/primitives/to_mesh.py

@@ -4,6 +4,7 @@
 
 from raysect.core import Point3D, Vector3D
 from raysect.primitive import Mesh, Box, Sphere, Cylinder, Cone, Parabola, Intersect, Union, Subtract
+from raysect.primitive.lens.spherical import BiConvex
 
 from raysect_mayavi.primitives.mesh_tools import subdivide
 from raysect_mayavi.primitives.mesh_csg import perform_mesh_csg
@@ -419,6 +420,10 @@ def csg_to_mesh(csg_primitive):
     return vertices, triangles
 
 
+def biconvex_to_mesh(biconvex_primitive):
+    return to_mesh(biconvex_primitive._primitive)
+
+
 _object_handlers = {
     Box: box_to_mesh,
     Sphere: sphere_to_mesh,
@@ -427,7 +432,8 @@ def csg_to_mesh(csg_primitive):
     Mesh: mesh_to_mesh,
     Intersect: csg_to_mesh,
     Union: csg_to_mesh,
-    Subtract: csg_to_mesh
+    Subtract: csg_to_mesh,
+    BiConvex: biconvex_to_mesh
 }
 
 

raysect_mayavi/scenegraph_viewer.py

@@ -14,7 +14,7 @@ def _parse_nodes(node, mesh_list):
         elif isinstance(child, Observer):
             return
         else:
-            mesh_list.append(to_mesh(child))
+            mesh_list.append((to_mesh(child), child.meta))
 
 
 def visualise_scenegraph(world, show_axes=False, axes_length=1):
@@ -29,13 +29,25 @@ def visualise_scenegraph(world, show_axes=False, axes_length=1):
 
     for mesh_description in meshes:
 
-        vertices, triangles = mesh_description
+        vertices, triangles = mesh_description[0]
         dx = vertices[:, 0]
         dy = vertices[:, 1]
         dz = vertices[:, 2]
 
-        mlab.triangular_mesh(dx, dy, dz, triangles, color=(163 / 255.0, 163 / 255.0, 163 / 255.0),
-                             figure=fig)  # , transparent=True, opacity=0.3)
+        meta = mesh_description[1]
+        try:
+            color = meta['viz-color']
+        except KeyError:
+            color = (163 / 255.0, 163 / 255.0, 163 / 255.0)
+        try:
+            opacity = meta['viz-opacity']
+            transparent = True
+        except KeyError:
+            opacity = 1
+            transparent = False
+
+        mlab.triangular_mesh(dx, dy, dz, triangles, color=color,
+                             figure=fig, transparent=transparent, opacity=opacity)
 
     if show_axes:
         mlab.plot3d([0, axes_length], [0, 0], [0, 0], tube_radius=axes_length/100, color=(1, 0, 0))