opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-01 17:29:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

3d stuff is still rudimentary, but slightly less buggy now

Browse Source
This commit is contained in:
Grant Sanderson
2017-02-23 18:14:08 -08:00
parent c2eb360684
commit ab2b6c2ba4
1 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
topics/three_dimensions.py
View File

@ -1,31 +1,20 @@
from helpers import * from helpers import *
from mobject.vectorized_mobject import VGroup from mobject.vectorized_mobject import VGroup, VMobject
from topics.geometry import Square from topics.geometry import Square
from scene import Scene from scene import Scene
from camera import Camera from camera import Camera
class ThreeDCamera(Camera): class ThreeDCamera(Camera):
CONFIG = { CONFIG = {
"sun_vect" : UP+LEFT, "sun_vect" : 3*UP+LEFT,
"shading_factor" : 0.5, "shading_factor" : 0.5,
} }
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
Camera.__init__(self, *args, **kwargs) Camera.__init__(self, *args, **kwargs)
self.unit_sun_vect = self.sun_vect/np.linalg.norm(self.sun_vect) self.unit_sun_vect = self.sun_vect/np.linalg.norm(self.sun_vect)
def display_multiple_vectorized_mobjects(self, vmobjects):
def cmp_vmobs(*vmobs):
return cmp(*[
vm.get_edge_center(IN)[2]
for vm in vmobs
])
Camera.display_multiple_vectorized_mobjects(
self,
sorted(vmobjects, cmp = cmp_vmobs)
)
def get_stroke_color(self, vmobject): def get_stroke_color(self, vmobject):
return Color(rgb = self.get_shaded_rgb( return Color(rgb = self.get_shaded_rgb(
color_to_rgb(vmobject.get_stroke_color()), color_to_rgb(vmobject.get_stroke_color()),
@ -59,6 +48,19 @@ class ThreeDCamera(Camera):
return OUT return OUT
return normal/length return normal/length
def display_multiple_vectorized_mobjects(self, vmobjects):
def z_cmp(*vmobs):
if all([hasattr(vm, "part_of_three_d_mobject") for vm in vmobs]):
return cmp(*[
vm.get_edge_center(IN)[2]
for vm in vmobs
])
else:
return 0
Camera.display_multiple_vectorized_mobjects(
self, sorted(vmobjects, cmp = z_cmp)
)
class ThreeDScene(Scene): class ThreeDScene(Scene):
CONFIG = { CONFIG = {
@ -67,7 +69,13 @@ class ThreeDScene(Scene):
############## ##############
class Cube(VGroup): class ThreeDMobject(VMobject):
def __init__(self, **kwargs):
VMobject.__init__(self, **kwargs)
for submobject in self.submobject_family():
submobject.part_of_three_d_mobject = True
class Cube(ThreeDMobject):
CONFIG = { CONFIG = {
"fill_opacity" : 0.75, "fill_opacity" : 0.75,
"fill_color" : BLUE, "fill_color" : BLUE,