def draw(self, axes: Axes3D):
R = (self.orientation * (self.orientation_origin.get_inverse())).to_rot_matrix()
verts = [vectorize(np.dot(R, v)) for v in self.vertices]
for plane in self.planes:
collection = Poly3DCollection(
[[self.center + verts[index] for index in plane]])
axes.add_collection3d(collection)
def cylinder_map(hist: Union[Histogram2D, CylinderSurfaceHistogram],
ax: Axes3D,
*,
show_zero: bool = True,
**kwargs):
"""Heat map plotted on the surface of a cylinder."""
data = get_data(hist,
cumulative=False,
flatten=False,
density=kwargs.pop("density", False))
cmap = _get_cmap(kwargs)
norm, cmap_data = _get_cmap_data(data, kwargs)
colors = cmap(cmap_data)
if hasattr(hist, "radius"):
r = kwargs.pop("radius", hist.radius)
else:
r = kwargs.pop("radius", 1)
xs = r * np.outer(np.cos(hist.numpy_bins[0]), np.ones(hist.shape[1] + 1))
ys = r * np.outer(np.sin(hist.numpy_bins[0]), np.ones(hist.shape[1] + 1))
zs = np.outer(np.ones(hist.shape[0] + 1), hist.numpy_bins[1])
for i in range(hist.shape[0]):
for j in range(hist.shape[1]):
if not show_zero and not data[i, j]:
continue
x = xs[i, j], xs[i, j + 1], xs[i + 1, j + 1], xs[i + 1, j]
y = ys[i, j], ys[i, j + 1], ys[i + 1, j + 1], ys[i + 1, j]
z = zs[i, j], zs[i, j + 1], zs[i + 1, j + 1], zs[i + 1, j]
verts = [list(zip(x, y, z))]
col = Poly3DCollection(verts)
col.set_facecolor(colors[i, j])
ax.add_collection3d(col)
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")
if matplotlib.__version__ < "2":
ax.plot_surface([], [], [], color="b")
ax.set_xlim(-r * 1.1, r * 1.1)
ax.set_ylim(-r * 1.1, r * 1.1)
ax.set_zlim(zs.min(), zs.max())
def globe_map(hist: Union[Histogram2D, DirectionalHistogram],
ax: Axes3D,
*,
show_zero: bool = True,
**kwargs):
"""Heat map plotted on the surface of a sphere."""
data = get_data(hist,
cumulative=False,
flatten=False,
density=kwargs.pop("density", False))
cmap = _get_cmap(kwargs)
norm, cmap_data = _get_cmap_data(data, kwargs)
colors = cmap(cmap_data)
lw = kwargs.pop("lw", 1)
r = 1
xs = r * np.outer(np.sin(hist.numpy_bins[0]), np.cos(hist.numpy_bins[1]))
ys = r * np.outer(np.sin(hist.numpy_bins[0]), np.sin(hist.numpy_bins[1]))
zs = r * np.outer(np.cos(hist.numpy_bins[0]), np.ones(hist.shape[1] + 1))
for i in range(hist.shape[0]):
for j in range(hist.shape[1]):
if not show_zero and not data[i, j]:
continue
x = xs[i, j], xs[i, j + 1], xs[i + 1, j + 1], xs[i + 1, j]
y = ys[i, j], ys[i, j + 1], ys[i + 1, j + 1], ys[i + 1, j]
z = zs[i, j], zs[i, j + 1], zs[i + 1, j + 1], zs[i + 1, j]
verts = [list(zip(x, y, z))]
col = Poly3DCollection(verts)
col.set_facecolor(colors[i, j])
col.set_edgecolor("black")
col.set_linewidth(lw)
ax.add_collection3d(col)
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")
if matplotlib.__version__ < "2":
ax.plot_surface([], [], [], color="b")
ax.set_xlim(-1.1, 1.1)
ax.set_ylim(-1.1, 1.1)
ax.set_zlim(-1.1, 1.1)
return ax
def surface_map(hist,
ax: Axes3D,
*,
show_zero: bool = True,
x=(lambda x, y: x),
y=(lambda x, y: y),
z=(lambda x, y: 0),
**kwargs):
"""Coloured-rectangle plot of 2D histogram, placed on an arbitrary surface.
Each bin is mapped to a rectangle in 3D space using the x,y,z functions.
Parameters
----------
hist : Histogram2D
show_zero : Optional[bool]
Whether to show coloured box for bins with 0 frequency (otherwise background).
x : function
Function with 2 parameters used to map bins to spatial x coordinate
y : function
Function with 2 parameters used to map bins to spatial y coordinate
z : function
Function with 2 parameters used to map bins to spatial z coordinate
Returns
-------
matplotlib.axes._subplots.Axes3DSubplot
See Also
--------
map, cylinder_map, globe_map
"""
data = get_data(hist,
cumulative=False,
flatten=False,
density=kwargs.pop("density", False))
cmap = _get_cmap(kwargs)
norm, cmap_data = _get_cmap_data(data, kwargs)
colors = cmap(cmap_data)
xs = np.ndarray((hist.shape[0] + 1, hist.shape[1] + 1), dtype=float)
ys = np.ndarray((hist.shape[0] + 1, hist.shape[1] + 1), dtype=float)
zs = np.ndarray((hist.shape[0] + 1, hist.shape[1] + 1), dtype=float)
edges_x = hist.numpy_bins[0]
edges_y = hist.numpy_bins[1]
for i in range(hist.shape[0] + 1):
for j in range(hist.shape[1] + 1):
xs[i, j] = x(edges_x[i], edges_y[j])
ys[i, j] = y(edges_x[i], edges_y[j])
zs[i, j] = z(edges_x[i], edges_y[j])
for i in range(hist.shape[0]):
for j in range(hist.shape[1]):
if not show_zero and not data[i, j]:
continue
x = xs[i, j], xs[i, j + 1], xs[i + 1, j + 1], xs[i + 1, j]
y = ys[i, j], ys[i, j + 1], ys[i + 1, j + 1], ys[i + 1, j]
z = zs[i, j], zs[i, j + 1], zs[i + 1, j + 1], zs[i + 1, j]
verts = [list(zip(x, y, z))]
col = Poly3DCollection(verts)
col.set_facecolor(colors[i, j])
ax.add_collection3d(col)
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")
if matplotlib.__version__ < "2":
ax.plot_surface([], [], [], color="b") # Dummy plot
ax.set_xlim(xs.min(), xs.max())
ax.set_ylim(ys.min(), ys.max())
ax.set_zlim(zs.min(), zs.max())
# ax.plot_surface(x, y, z, rstride=hist.shape[0], color="b")
return ax
