Cube to Sphere

In [1]:

# set backend for vtk/pyvista in jupyter
import pyvista as pv
pv.set_jupyter_backend('static')
from pyvirtualdisplay import Display
#pv.start_xvfb()

# set backend for vtk/pyvista in jupyter import pyvista as pv pv.set_jupyter_backend('static') from pyvirtualdisplay import Display #pv.start_xvfb()

In [2]:

# cfdtools packages
import cfdtools.meshbase.simple as sm
from cfdtools.vtk import vtkMesh

# cfdtools packages import cfdtools.meshbase.simple as sm from cfdtools.vtk import vtkMesh

In [3]:

cube = sm.Cube(10, 10, 10)
mesh = cube.export_mesh()
vtkmesh = vtkMesh(mesh)

cube = sm.Cube(10, 10, 10) mesh = cube.export_mesh() vtkmesh = vtkMesh(mesh)

In [4]:

with Display():
    vtkmesh.plot()

with Display(): vtkmesh.plot()

In [5]:

def map_to_sphere(x, y, z):
    import numpy as np
    # cube has side range 0 to 1, map to [-1, 1]
    x1 = 2*x-1 ; y1 = 2*y-1 ; z1 = 2*z-1
    x2 = x1 * np.sqrt(1-y1**2/2-z1**2/2+(y1*z1)**2/3)
    y2 = y1 * np.sqrt(1-x1**2/2-z1**2/2+(x1*z1)**2/3)
    z2 = z1 * np.sqrt(1-x1**2/2-y1**2/2+(x1*y1)**2/3)
    return x2, y2, z2

mesh.morph(map_to_sphere)

def map_to_sphere(x, y, z): import numpy as np # cube has side range 0 to 1, map to [-1, 1] x1 = 2*x-1 ; y1 = 2*y-1 ; z1 = 2*z-1 x2 = x1 * np.sqrt(1-y1**2/2-z1**2/2+(y1*z1)**2/3) y2 = y1 * np.sqrt(1-x1**2/2-z1**2/2+(x1*z1)**2/3) z2 = z1 * np.sqrt(1-x1**2/2-y1**2/2+(x1*y1)**2/3) return x2, y2, z2 mesh.morph(map_to_sphere)

In [6]:

vtkmesh = vtkMesh(mesh)
with Display():
    vtkmesh.plot()

vtkmesh = vtkMesh(mesh) with Display(): vtkmesh.plot()