dicom

examples/other/spherical_harmonics1.py

@@ -1,3 +1,8 @@
+from __future__ import division, print_function
+
+import numpy as np
+from vtkplotter import Plotter, Points, cos, datadir, mag, sin
+
 """
 Example on how to use the intersectWithLine() method:
  intersect an actor with lines from the origin
@@ -9,12 +14,9 @@
  and then truncate the expansion to a specific lmax and
  reconstruct the projected points in red
 """
-from __future__ import division, print_function
 
 print(__doc__)
 
-from vtkplotter import Plotter, Points, datadir, mag, sin, cos
-import numpy as np
 
 ##########################################################
 N = 100  # number of sample points on the unit sphere
@@ -25,14 +27,14 @@
 ##########################################################
 
 vp = Plotter(shape=[1, 2], verbose=0, axes=0)
-shape = vp.load(datadir+"shapes/icosahedron.vtk").normalize().pos(x0).lineWidth(2)
+shape = vp.load(datadir + "shapes/icosahedron.vtk").normalize().pos(x0).lineWidth(2)
 
 agrid, pts = [], []
 for th in np.linspace(0, np.pi, N, endpoint=True):
     lats = []
     for ph in np.linspace(0, 2 * np.pi, N, endpoint=True):
         p = np.array([sin(th) * cos(ph), sin(th) * sin(ph), cos(th)]) * rmax
-        intersections = shape.intersectWithLine([0, 0, 0], p)  ### <--
+        intersections = shape.intersectWithLine([0, 0, 0], p)  # <--
         if len(intersections):
             value = mag(intersections[0])
             lats.append(value - rbias)

examples/other/spherical_harmonics2.py

@@ -1,19 +1,21 @@
 """
 Morph one shape into another using spherical harmonics package shtools.
+
 In this example we morph a sphere into a octahedron and viceversa.
 """
 from __future__ import division, print_function
 
+import numpy as np
+from vtkplotter import Plotter, Points, Sphere, arange, cos, datadir, mag, sin
+
 try:
     import pyshtools
     print(__doc__)
-except:
+except ModuleNotFoundError:
     print("Please install pyshtools to run this example")
     print("Follow instructions at https://shtools.oca.eu/shtools")
     exit(0)
 
-from vtkplotter import Plotter, datadir, mag, arange, Points, Sphere, sin, cos
-import numpy as np
 
 ##########################################################
 N = 100  # number of sample points on the unit sphere
@@ -30,7 +32,7 @@ def makeGrid(shape, N):
         lats = []
         for ph in np.linspace(0, 2 * np.pi, N, endpoint=True):
             p = np.array([sin(th) * cos(ph), sin(th) * sin(ph), cos(th)]) * rmax
-            intersections = shape.intersectWithLine([0, 0, 0], p)  ### <--
+            intersections = shape.intersectWithLine([0, 0, 0], p)
             if len(intersections):
                 value = mag(intersections[0])
                 lats.append(value - rbias)
@@ -45,7 +47,7 @@ def makeGrid(shape, N):
 
 
 def morph(clm1, clm2, t, lmax):
-    # interpolate linearly the two sets of sph harm. coeeficients
+    """Interpolate linearly the two sets of sph harm. coeeficients."""
     clm = (1 - t) * clm1 + t * clm2
     grid_reco = clm.expand(lmax=lmax)  # cut "high frequency" components
     agrid_reco = grid_reco.to_array()
@@ -65,24 +67,27 @@ def morph(clm1, clm2, t, lmax):
 vp = Plotter(shape=[2, 2], verbose=0, axes=3, interactive=0)
 
 shape1 = Sphere(alpha=0.2)
-shape2 = vp.load(datadir+"shapes/icosahedron.vtk").normalize().lineWidth(1)
+shape2 = vp.load(datadir + "shapes/icosahedron.vtk").normalize().lineWidth(1)
 
 agrid1, actorpts1 = makeGrid(shape1, N)
+
 vp.show(shape1, actorpts1, at=0)
 
 agrid2, actorpts2 = makeGrid(shape2, N)
 vp.show(shape2, actorpts2, at=1)
+
 vp.camera.Zoom(1.2)
 vp.interactive = False
 
 clm1 = pyshtools.SHGrid.from_array(agrid1).expand()
 clm2 = pyshtools.SHGrid.from_array(agrid2).expand()
-# clm1.plot_spectrum2d() # plot the value of the sph harm. coefficients
+# clm1.plot_spectrum2d()  # plot the value of the sph harm. coefficients
 # clm2.plot_spectrum2d()
 
 for t in arange(0, 1, 0.005):
     act21 = Points(morph(clm2, clm1, t, lmax), c="r", r=4)
     act12 = Points(morph(clm1, clm2, t, lmax), c="g", r=4)
+
     vp.show(act21, at=2, resetcam=0)
     vp.show(act12, at=3)
     vp.camera.Azimuth(2)