mirror of
https://github.com/3b1b/manim.git
synced 2025-08-01 06:22:54 +08:00
156 lines
4.1 KiB
Python
156 lines
4.1 KiB
Python
|
|
from helpers import *
|
|
|
|
from mobject.vectorized_mobject import VGroup, VMobject
|
|
from topics.geometry import Square
|
|
from scene import Scene
|
|
from camera import Camera
|
|
|
|
class CameraWithPerspective(Camera):
|
|
CONFIG = {
|
|
"camera_distance" : 20,
|
|
}
|
|
def points_to_pixel_coords(self, points):
|
|
distance_ratios = np.divide(
|
|
self.camera_distance,
|
|
self.camera_distance - points[:,2]
|
|
)
|
|
scale_factors = interpolate(0, 1, distance_ratios)
|
|
adjusted_points = np.array(points)
|
|
for i in 0, 1:
|
|
adjusted_points[:,i] *= scale_factors
|
|
|
|
return Camera.points_to_pixel_coords(self, adjusted_points)
|
|
|
|
class ThreeDCamera(CameraWithPerspective):
|
|
CONFIG = {
|
|
"sun_vect" : 5*UP+LEFT,
|
|
"shading_factor" : 0.5,
|
|
}
|
|
def __init__(self, *args, **kwargs):
|
|
Camera.__init__(self, *args, **kwargs)
|
|
self.unit_sun_vect = self.sun_vect/np.linalg.norm(self.sun_vect)
|
|
|
|
def get_color(self, method):
|
|
color = method()
|
|
vmobject = method.im_self
|
|
if is_3d(vmobject):
|
|
return Color(rgb = self.get_shaded_rgb(
|
|
color_to_rgb(color),
|
|
normal_vect = self.get_unit_normal_vect(vmobject)
|
|
))
|
|
else:
|
|
return color
|
|
|
|
def get_stroke_color(self, vmobject):
|
|
return self.get_color(vmobject.get_stroke_color)
|
|
|
|
def get_fill_color(self, vmobject):
|
|
return self.get_color(vmobject.get_fill_color)
|
|
|
|
def get_shaded_rgb(self, rgb, normal_vect):
|
|
brightness = np.dot(normal_vect, self.unit_sun_vect)**2
|
|
if brightness > 0:
|
|
alpha = self.shading_factor*brightness
|
|
return interpolate(rgb, np.ones(3), alpha)
|
|
else:
|
|
alpha = -self.shading_factor*brightness
|
|
return interpolate(rgb, np.zeros(3), alpha)
|
|
|
|
def get_unit_normal_vect(self, vmobject):
|
|
anchors = vmobject.get_anchors()
|
|
if len(anchors) < 3:
|
|
return OUT
|
|
normal = np.cross(anchors[1]-anchors[0], anchors[2]-anchors[1])
|
|
if normal[2] < 0:
|
|
normal = -normal
|
|
length = np.linalg.norm(normal)
|
|
if length == 0:
|
|
return OUT
|
|
return normal/length
|
|
|
|
def display_multiple_vectorized_mobjects(self, vmobjects):
|
|
def z_cmp(*vmobs):
|
|
#Compare to three dimensional mobjects based on their
|
|
#z value, otherwise don't compare.
|
|
three_d_status = map(is_3d, vmobs)
|
|
has_points = [vm.get_num_points() > 0 for vm in vmobs]
|
|
if all(three_d_status) and all(has_points):
|
|
cmp_vect = self.get_unit_normal_vect(vmobs[1])
|
|
return cmp(*[
|
|
np.dot(vm.get_center(), cmp_vect)
|
|
for vm in vmobs
|
|
])
|
|
else:
|
|
return 0
|
|
Camera.display_multiple_vectorized_mobjects(
|
|
self, sorted(vmobjects, cmp = z_cmp)
|
|
)
|
|
|
|
class ThreeDScene(Scene):
|
|
CONFIG = {
|
|
"camera_class" : ThreeDCamera,
|
|
}
|
|
|
|
##############
|
|
|
|
def is_3d(mobject):
|
|
return hasattr(mobject, "part_of_3d_mobject")
|
|
|
|
class ThreeDMobject(VMobject):
|
|
def __init__(self, *args, **kwargs):
|
|
VMobject.__init__(self, *args, **kwargs)
|
|
for submobject in self.submobject_family():
|
|
submobject.part_of_3d_mobject = True
|
|
|
|
class Cube(ThreeDMobject):
|
|
CONFIG = {
|
|
"fill_opacity" : 0.75,
|
|
"fill_color" : BLUE,
|
|
"stroke_width" : 0,
|
|
"propogate_style_to_family" : True,
|
|
"side_length" : 2,
|
|
}
|
|
def generate_points(self):
|
|
for vect in IN, OUT, LEFT, RIGHT, UP, DOWN:
|
|
face = Square(side_length = self.side_length)
|
|
face.shift(self.side_length*OUT/2.0)
|
|
face.apply_function(lambda p : np.dot(p, z_to_vector(vect).T))
|
|
|
|
self.add(face)
|
|
|
|
class Prism(Cube):
|
|
CONFIG = {
|
|
"dimensions" : [3, 2, 1]
|
|
}
|
|
def generate_points(self):
|
|
Cube.generate_points(self)
|
|
for dim, value in enumerate(self.dimensions):
|
|
self.rescale_to_fit(value, dim, stretch = True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|