def quiver_demo(**kwargs):
'''
quiver_demo(length = 1.0,
normalize = True,
facecolor = 'b',
edgecolor = None,
arrow_length_ratio = 0.3,
shaftsize = 0.01,
headsize = 0.01)
'''
from ifigure.interactive import quiver, threed, figure, lighting, view, isec, nsec
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2), np.arange(-0.8, 1, 0.2),
np.arange(-0.8, 1, 0.8))
u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
np.sin(np.pi * z))
figure()
nsec(2)
isec(0)
threed('on')
lighting(light=0.5, ambient=0.7)
view('noclip')
quiver(x, y, z, u, v, w, length=0.3, facecolor='b')
isec(1)
threed('on')
lighting(light=0.5, ambient=0.7)
view('noclip')
quiver(x, y, z, u, v, w, cz=True, cdata=y, length=0.3)
def contourf_demo(**kwargs):
import mpl_toolkits.mplot3d.axes3d as axes3d
from ifigure.interactive import contourf, threed, figure
X, Y, Z = axes3d.get_test_data(0.05)
v = figure(gl = True)
threed('on')
contourf(X, Y, Z)
def contourf_demo(**kwargs):
import mpl_toolkits.mplot3d.axes3d as axes3d
from ifigure.interactive import contourf, threed, figure
X, Y, Z = axes3d.get_test_data(0.05)
v = figure(gl=True)
threed('on')
contourf(X, Y, Z)
def quiver_demo(**kwargs):
'''
quiver_demo(length = 1.0,
normalize = True,
facecolor = 'b',
edgecolor = None,
arrow_length_ratio = 0.3,
shaftsize = 0.01,
headsize = 0.01)
'''
from ifigure.interactive import quiver, threed, figure, lighting, view, isec, nsec
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
np.arange(-0.8, 1, 0.2),
np.arange(-0.8, 1, 0.8))
u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
np.sin(np.pi * z))
figure()
nsec(2)
isec(0)
threed('on')
lighting(light = 0.5, ambient = 0.7)
view('noclip')
quiver(x, y, z, u, v, w, length = 0.3, facecolor = 'b')
isec(1)
threed('on')
lighting(light = 0.5, ambient = 0.7)
view('noclip')
quiver(x, y, z, u, v, w, cz = True, cdata = y, length = 0.3)
def plot_tiles3():
x, y, z = read_tile_data(list=True)
figure()
threed('on')
hold('on')
for k in range(100):
m = k + 100
plot(x[m], y[m], z[m], facecolor=[1, 0, 0, 1])
def plot_tiles():
'''
line plot of tile data
'''
x, y, z = read_tile_data()
figure()
threed('on')
plot(x, y, z)
def plot_tiles3():
x, y, z = read_tile_data(list = True)
figure()
threed('on')
hold('on')
for k in range(100):
m = k+100
plot(x[m], y[m], z[m], facecolor = [1, 0, 0, 1])
def plot_tiles():
'''
line plot of tile data
'''
x, y, z = read_tile_data()
figure()
threed('on')
plot(x, y, z)
def contour_demo2(**kwargs):
import mpl_toolkits.mplot3d.axes3d as axes3d
from ifigure.interactive import contourf, threed, figure, hold
X, Y, Z = axes3d.get_test_data(0.05)
v = figure(gl=True)
threed('on')
hold('on')
contourf(X, Y, Z)
contourf(X, Y, Z, zdir='x', offset=-40)
def contour_demo2(**kwargs):
import mpl_toolkits.mplot3d.axes3d as axes3d
from ifigure.interactive import contourf, threed, figure, hold
X, Y, Z = axes3d.get_test_data(0.05)
v = figure(gl = True)
threed('on')
hold('on')
contourf(X, Y, Z)
contourf(X, Y, Z, zdir = 'x', offset = -40)
def surf_demo(**kwargs):
from ifigure.interactive import surf, threed
threed('on')
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf(X, Y, Z, cmap='coolwarm', shade=True, **kwargs)
def surf_demo(**kwargs):
from ifigure.interactive import surf, threed
threed('on')
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf(X,Y,Z, cmap='coolwarm', shade=True, **kwargs)
def solid_stl_demo(**kwargs):
from stl import mesh
from ifigure.interactive import solid, figure, threed
import ifigure, os
from os.path import dirname
path = dirname(dirname(dirname(ifigure.__file__)))
mymesh = mesh.Mesh.from_file(os.path.join(path, 'example', 'D_antenna.stl'))
v = figure()
threed('on')
solid(mymesh.vectors, alpha = 0.5, **kwargs)
def solid_stl_demo(**kwargs):
from stl import mesh
from ifigure.interactive import solid, figure, threed
import ifigure, os
from os.path import dirname
path = dirname(dirname(dirname(ifigure.__file__)))
mymesh = mesh.Mesh.from_file(os.path.join(path, 'example',
'D_antenna.stl'))
v = figure()
threed('on')
solid(mymesh.vectors, alpha=0.5, **kwargs)
def revolve_demo(**kwargs):
m = np.linspace(0, np.pi*2, 30)
r = 1.0 + 0.2*np.cos(m)
z = 0.25 * np.sin(m)
from ifigure.interactive import revolve, figure, threed
v = figure(gl = True)
threed('on')
revolve(r, z, raxis = [0.5, 1], rtheta = [0, 2*np.pi/3])
v.xlim(-1.5, 1.5)
v.ylim(-1.5, 1.5)
v.zlim(-1.5, 1.5)
def revolve_demo(**kwargs):
m = np.linspace(0, np.pi * 2, 30)
r = 1.0 + 0.2 * np.cos(m)
z = 0.25 * np.sin(m)
from ifigure.interactive import revolve, figure, threed
v = figure(gl=True)
threed('on')
revolve(r, z, raxis=[0.5, 1], rtheta=[0, 2 * np.pi / 3])
v.xlim(-1.5, 1.5)
v.ylim(-1.5, 1.5)
v.zlim(-1.5, 1.5)
def surf_demo(**kwargs):
from ifigure.interactive import surf, threed, figure, hold, lighting
figure()
threed('on')
# X = np.arange(-5, 5, 0.25)
# Y = np.arange(-5, 5, 0.25)
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
lighting(light = 0.5, ambient = 0.7)
surf(X, Y, Z, cmap='coolwarm', shade = 'linear', **kwargs)
def surf_demo(**kwargs):
from ifigure.interactive import surf, threed, figure, hold, lighting
figure()
threed('on')
# X = np.arange(-5, 5, 0.25)
# Y = np.arange(-5, 5, 0.25)
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
lighting(light=0.5, ambient=0.7)
surf(X, Y, Z, cmap='coolwarm', shade='linear', **kwargs)
def image_demo():
from ifigure.interactive import image, threed, figure, hold
figure(gl = True)
threed('on')
hold('on')
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
image(Z, cmap='coolwarm', im_center = [0,0,1])
image(Z, cmap='coolwarm', im_center = [0,0,1],
im_axes = ([0,0,1], [0,1,0]))
image(Z, cmap='coolwarm', im_center = [0,0,1],
im_axes = ([1/np.sqrt(2), 0,1/np.sqrt(2)],
[0, 1, 0]))
def image_demo():
from ifigure.interactive import image, threed, figure, hold
figure(gl=True)
threed('on')
hold('on')
X = np.linspace(-5, 5, 40)
Y = np.linspace(-5, 5, 40)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
image(Z, cmap='coolwarm', im_center=[0, 0, 1])
image(Z,
cmap='coolwarm',
im_center=[0, 0, 1],
im_axes=([0, 0, 1], [0, 1, 0]))
image(Z,
cmap='coolwarm',
im_center=[0, 0, 1],
im_axes=([1 / np.sqrt(2), 0, 1 / np.sqrt(2)], [0, 1, 0]))
def trisurf3d_demo(**kwargs):
from matplotlib import cm
import numpy as np
from ifigure.interactive import figure, threed, trisurf, nsec, isec, lighting
# preparing the data, the same as mplot3d demo
n_angles = 36
n_radii = 8
radii = np.linspace(0.125, 1.0, n_radii)
angles = np.linspace(0, 2 * np.pi, n_angles, endpoint=False)
angles = np.repeat(angles[..., np.newaxis], n_radii, axis=1)
x = np.append(0, (radii * np.cos(angles)).flatten())
y = np.append(0, (radii * np.sin(angles)).flatten())
z = np.sin(-x * y)
v = figure()
nsec(3)
isec(0)
threed('on')
trisurf(x, y, z, cmap=cm.jet, linewidth=0.2, cz=True)
isec(1)
threed('on')
trisurf(x, y, z, linewidth=0.2, color='b')
lighting(light=0.5, ambient=0.5)
isec(2)
threed('on')
trisurf(x,
y,
z * 0,
cmap=cm.jet,
linewidth=0.2,
cz=True,
cdata=z,
edgecolor=None)
def trisurf3d_demo(**kwargs):
from matplotlib import cm
import numpy as np
from ifigure.interactive import figure, threed, trisurf , nsec, isec, lighting
# preparing the data, the same as mplot3d demo
n_angles = 36
n_radii = 8
radii = np.linspace(0.125, 1.0, n_radii)
angles = np.linspace(0, 2*np.pi, n_angles, endpoint=False)
angles = np.repeat(angles[...,np.newaxis], n_radii, axis=1)
x = np.append(0, (radii*np.cos(angles)).flatten())
y = np.append(0, (radii*np.sin(angles)).flatten())
z = np.sin(-x*y)
v = figure()
nsec(3)
isec(0)
threed('on')
trisurf(x, y, z, cmap=cm.jet, linewidth=0.2, cz = True)
isec(1)
threed('on')
trisurf(x, y, z, linewidth=0.2, color='b')
lighting(light = 0.5, ambient = 0.5)
isec(2)
threed('on')
trisurf(x, y, z*0, cmap=cm.jet, linewidth=0.2,
cz = True, cdata = z, edgecolor = None)
def surf_demo2(**kwargs):
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from ifigure.interactive import surf, threed, figure, isec, nsec
figure(gl = True)
nsec(3)
isec(0)
threed('on')
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)
x = 10 * np.outer(np.cos(u), np.sin(v))
y = 10 * np.outer(np.sin(u), np.sin(v))
z = 10 * np.outer(np.ones(np.size(u)), np.cos(v))
surf(x, y, z, cz = False, facecolor='b', edgecolor = 'k')
isec(1)
threed('on')
surf(x, y, z, cstride=4, cz = True, edgecolor = 'k')
isec(2)
threed('on')
surf(x, y, z, cz = True, cdata = y, edgecolor = None)
def surf_demo2(**kwargs):
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
from ifigure.interactive import surf, threed, figure, isec, nsec
figure(gl=True)
nsec(3)
isec(0)
threed('on')
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)
x = 10 * np.outer(np.cos(u), np.sin(v))
y = 10 * np.outer(np.sin(u), np.sin(v))
z = 10 * np.outer(np.ones(np.size(u)), np.cos(v))
surf(x, y, z, cz=False, facecolor='b', edgecolor='k')
isec(1)
threed('on')
surf(x, y, z, cstride=4, cz=True, edgecolor='k')
isec(2)
threed('on')
surf(x, y, z, cz=True, cdata=y, edgecolor=None)
def solid_demo(**kwargs):
'''
solid_demo2(cz = True, linewidths = 1.0, edgecolor='red')
solid_demo2(facecolor='b', linewidths = 1.0, edgecolor='red')
'''
from ifigure.interactive import solid, figure, threed, isec, nsec, lighting
import ifigure, os
# preparing the data, the same as mplot3d demo
theta = np.linspace(0, 2.0 * np.pi, endpoint=True, num=50)
r = np.linspace(0.01, 4.0, endpoint=True, num=50)
# This is the Mobius mapping, taking a u, v pair and returning an x, y, z
# triple
THETA, R = np.meshgrid(theta, r)
X = R*np.cos(THETA)
Y = R*np.sin(THETA)
Z = np.exp(- R*R)
import matplotlib.tri as mtri
tri = mtri.Triangulation(X.flatten(), Y.flatten())
# Triangulate parameter space to determine the triangles
if 'cz' in kwargs and kwargs['cz']:
v = np.dstack((tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],))
v = np.vstack((tri.x, tri.y, Z.flatten()))
kwargs['cdata'] = 5*Z.flatten()[tri.triangles]
kwargs['cdata'] = 5*Z.flatten()
kwargs['cz'] = True
# or #
v = np.dstack((tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],
5 * Z.flatten()[tri.triangles],))
v = np.vstack((tri.x, tri.y, Z.flatten(), 5*Z.flatten()))
elif 'twod' in kwargs and kwargs['twod']:
v = np.dstack((tri.x[tri.triangles],
tri.y[tri.triangles]),)
v = np.vstack((tri.x, tri.y, ))
kwargs.pop('twod')
kwargs['zvalue'] = 1.0
else:
v = np.dstack((tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],))
v = np.vstack((tri.x, tri.y, Z.flatten()))
idxset=tri.triangles
v = v.transpose()
print v.shape, idxset.shape
viewer = figure()
nsec(3)
isec(0)
threed('on')
solid(v, idxset, edgecolor = 'k', facecolor = 'b')
lighting(light = 0.5, ambient = 0.5)
isec(1)
threed('on')
solid(v, idxset, cz = True)
lighting(light = 0.5, ambient = 0.5)
isec(2)
threed('on')
solid(v, idxset, cz = True, cdata = v[...,0], shade='linear')
lighting(light = 0.5, ambient = 0.5)
def plot_tiles2():
x, y, z = read_tile_data()
figure()
threed('on')
plot(x, y, z, facecolor = [1, 0, 0, 1])
def plot_tiles2():
x, y, z = read_tile_data()
figure()
threed('on')
plot(x, y, z, facecolor=[1, 0, 0, 1])
def solid_demo(**kwargs):
'''
solid_demo2(cz = True, linewidths = 1.0, edgecolor='red')
solid_demo2(facecolor='b', linewidths = 1.0, edgecolor='red')
'''
from ifigure.interactive import solid, figure, threed, isec, nsec, lighting
import ifigure, os
# preparing the data, the same as mplot3d demo
theta = np.linspace(0, 2.0 * np.pi, endpoint=True, num=50)
r = np.linspace(0.01, 4.0, endpoint=True, num=50)
# This is the Mobius mapping, taking a u, v pair and returning an x, y, z
# triple
THETA, R = np.meshgrid(theta, r)
X = R * np.cos(THETA)
Y = R * np.sin(THETA)
Z = np.exp(-R * R)
import matplotlib.tri as mtri
tri = mtri.Triangulation(X.flatten(), Y.flatten())
# Triangulate parameter space to determine the triangles
if 'cz' in kwargs and kwargs['cz']:
v = np.dstack((
tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],
))
v = np.vstack((tri.x, tri.y, Z.flatten()))
kwargs['cdata'] = 5 * Z.flatten()[tri.triangles]
kwargs['cdata'] = 5 * Z.flatten()
kwargs['cz'] = True
# or #
v = np.dstack((
tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],
5 * Z.flatten()[tri.triangles],
))
v = np.vstack((tri.x, tri.y, Z.flatten(), 5 * Z.flatten()))
elif 'twod' in kwargs and kwargs['twod']:
v = np.dstack((tri.x[tri.triangles], tri.y[tri.triangles]), )
v = np.vstack((
tri.x,
tri.y,
))
kwargs.pop('twod')
kwargs['zvalue'] = 1.0
else:
v = np.dstack((
tri.x[tri.triangles],
tri.y[tri.triangles],
Z.flatten()[tri.triangles],
))
v = np.vstack((tri.x, tri.y, Z.flatten()))
idxset = tri.triangles
v = v.transpose()
print v.shape, idxset.shape
viewer = figure()
nsec(3)
isec(0)
threed('on')
solid(v, idxset, edgecolor='k', facecolor='b')
lighting(light=0.5, ambient=0.5)
isec(1)
threed('on')
solid(v, idxset, cz=True)
lighting(light=0.5, ambient=0.5)
isec(2)
threed('on')
solid(v, idxset, cz=True, cdata=v[..., 0], shade='linear')
lighting(light=0.5, ambient=0.5)
