public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a basic Prismify to turn a flat VMobject into something with depth

Browse source
This commit is contained in:
Grant Sanderson 2022-03-17 12:00:10 -07:00
parent bd6c731e67
commit f249da95fb
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
manimlib/mobject/three_dimensions.py
View file

@ -9,6 +9,7 @@ from manimlib.mobject.geometry import Square
from manimlib.mobject.geometry import Polygon from manimlib.mobject.geometry import Polygon
from manimlib.utils.bezier import interpolate from manimlib.utils.bezier import interpolate
from manimlib.utils.config_ops import digest_config from manimlib.utils.config_ops import digest_config
from manimlib.utils.iterables import adjacent_pairs
from manimlib.utils.space_ops import get_norm from manimlib.utils.space_ops import get_norm
from manimlib.utils.space_ops import z_to_vector from manimlib.utils.space_ops import z_to_vector
from manimlib.utils.space_ops import compass_directions from manimlib.utils.space_ops import compass_directions
@ -278,3 +279,22 @@ class Prism(Cube):
Cube.init_points(self) Cube.init_points(self)
for dim, value in enumerate(self.dimensions): for dim, value in enumerate(self.dimensions):
self.rescale_to_fit(value, dim, stretch=True) self.rescale_to_fit(value, dim, stretch=True)
class Prismify(VGroup):
CONFIG = {
"apply_depth_test": True
}
def __init__(self, vmobject, depth=1.0, direction=IN, **kwargs):
# At the moment, this assume stright edges
super().__init__(**kwargs)
vect = depth * direction
self.add(vmobject.copy())
points = vmobject.get_points()[::vmobject.n_points_per_curve]
for p1, p2 in adjacent_pairs(points):
wall = VMobject()
wall.match_style(vmobject)
wall.set_points_as_corners([p1, p2, p2 + vect, p1 + vect])
self.add(wall)
self.add(vmobject.copy().shift(vect).reverse_points())