def _create_testing_scene(empty_scene, show=False, off_screen=False):
if empty_scene:
plotter = BackgroundPlotter(show=show, off_screen=off_screen)
else:
plotter = BackgroundPlotter(shape=(2, 2),
border=True,
border_width=10,
border_color='grey',
show=show,
off_screen=off_screen)
plotter.set_background('black', top='blue')
plotter.subplot(0, 0)
cone = pyvista.Cone(resolution=4)
actor = plotter.add_mesh(cone)
plotter.remove_actor(actor)
plotter.add_text('Actor is removed')
plotter.subplot(0, 1)
plotter.add_mesh(pyvista.Box(), color='green', opacity=0.8)
plotter.subplot(1, 0)
cylinder = pyvista.Cylinder(resolution=6)
plotter.add_mesh(cylinder, smooth_shading=True)
plotter.show_bounds()
plotter.subplot(1, 1)
sphere = pyvista.Sphere(phi_resolution=6, theta_resolution=6)
plotter.add_mesh(sphere)
plotter.enable_cell_picking()
return plotter
def draw_3D(
self,
atom_scale: float = 1,
background_color: str = "white",
cone_color: str = "steelblue",
cone_opacity: float = 0.75,
) -> None:
"""Draw a 3D representation of the molecule with the cone.
Args:
atom_scale: Scaling factor for atom size
background_color: Background color for plot
cone_color: Cone color
cone_opacity: Cone opacity
"""
# Set up plotter
p = BackgroundPlotter()
p.set_background(background_color)
# Draw molecule
for atom in self._atoms:
color = hex2color(jmol_colors[atom.element])
radius = atom.radius * atom_scale
sphere = pv.Sphere(center=list(atom.coordinates), radius=radius)
p.add_mesh(sphere, color=color, opacity=1, name=str(atom.index))
# Determine direction and extension of cone
angle = math.degrees(self._cone.angle)
coordinates: Array2DFloat = np.array([atom.coordinates for atom in self._atoms])
radii: Array1DFloat = np.array([atom.radius for atom in self._atoms])
if angle > 180:
normal = -self._cone.normal
else:
normal = self._cone.normal
projected = np.dot(normal, coordinates.T) + np.array(radii)
max_extension = np.max(projected)
if angle > 180:
max_extension += 1
# Make the cone
cone = get_drawing_cone(
center=[0, 0, 0] + (max_extension * normal) / 2,
direction=-normal,
angle=angle,
height=max_extension,
capping=False,
resolution=100,
)
p.add_mesh(cone, opacity=cone_opacity, color=cone_color)
def draw_3D(
self,
atom_scale: float = 1,
background_color: str = "white",
point_color: str = "steelblue",
opacity: float = 0.25,
size: float = 1,
) -> None:
"""Draw a 3D representation.
Draws the molecule with the solvent accessible surface area.
Args:
atom_scale: Scaling factor for atom size
background_color: Background color for plot
point_color: Color of surface points
opacity: Point opacity
size: Point size
"""
# Set up plotter
p = BackgroundPlotter()
p.set_background(background_color)
# Draw molecule
for atom in self._atoms:
color = hex2color(jmol_colors[atom.element])
radius = atom.radius * atom_scale - self._probe_radius
sphere = pv.Sphere(center=list(atom.coordinates), radius=radius)
p.add_mesh(sphere, color=color, opacity=1, name=str(atom.index))
# Draw surface points
surface_points: Array2DFloat = np.vstack(
[atom.accessible_points for atom in self._atoms])
p.add_points(surface_points,
color=point_color,
opacity=opacity,
point_size=size)
def draw_3D(
self,
atom_scale: float = 0.5,
background_color: str = "white",
arrow_color: str = "steelblue",
) -> None:
"""Draw a 3D representation of the molecule with the Sterimol vectors.
Args:
atom_scale: Scaling factor for atom size
background_color: Background color for plot
arrow_color: Arrow color
"""
# Set up plotter
p = BackgroundPlotter()
p.set_background(background_color)
# Draw molecule
for atom in self._atoms:
color = hex2color(jmol_colors[atom.element])
if atom.element == 0:
radius = 0.5 * atom_scale
else:
radius = atom.radius * atom_scale
sphere = pv.Sphere(center=list(atom.coordinates), radius=radius)
if atom.index in self._excluded_atoms:
opacity = 0.25
else:
opacity = 1
p.add_mesh(sphere,
color=color,
opacity=opacity,
name=str(atom.index))
# Draw sphere for Buried Sterimol
if hasattr(self, "_sphere_radius"):
sphere = pv.Sphere(center=self._dummy_atom.coordinates,
radius=self._sphere_radius)
p.add_mesh(sphere, opacity=0.25)
if hasattr(self, "_points"):
p.add_points(self._points, color="gray")
# Get arrow starting points
start_L = self._dummy_atom.coordinates
start_B = self._attached_atom.coordinates
# Add L arrow with label
length = np.linalg.norm(self.L)
direction = self.L / length
stop_L = start_L + length * direction
L_arrow = get_drawing_arrow(start=start_L,
direction=direction,
length=length)
p.add_mesh(L_arrow, color=arrow_color)
# Add B_1 arrow
length = np.linalg.norm(self.B_1)
direction = self.B_1 / length
stop_B_1 = start_B + length * direction
B_1_arrow = get_drawing_arrow(start=start_B,
direction=direction,
length=length)
p.add_mesh(B_1_arrow, color=arrow_color)
# Add B_5 arrow
length = np.linalg.norm(self.B_5)
direction = self.B_5 / length
stop_B_5 = start_B + length * direction
B_5_arrow = get_drawing_arrow(start=start_B,
direction=direction,
length=length)
p.add_mesh(B_5_arrow, color=arrow_color)
# Add labels
points = np.vstack([stop_L, stop_B_1, stop_B_5])
labels = ["L", "B1", "B5"]
p.add_point_labels(
points,
labels,
text_color="black",
font_size=30,
bold=False,
show_points=False,
point_size=1,
)
self._plotter = p