public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Better ThreeDCamera.add_fixed_orientation_mobjects

Browse source
This commit is contained in:
Grant Sanderson 2018-08-23 14:44:22 -07:00
parent 781820ba9f
commit 90b21a4a8a
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
camera/three_d_camera.py
View file

@ -42,7 +42,7 @@ class ThreeDCamera(Camera):
self.gamma_tracker = ValueTracker(self.gamma)
self.light_source = Point(self.light_source_start_point)
self.frame_center = Point(self.frame_center)
self.fixed_orientation_mobjects = set()
self.fixed_orientation_mobjects = dict()
self.fixed_in_frame_mobjects = set()
self.reset_rotation_matrix()
@ -189,15 +189,19 @@ class ThreeDCamera(Camera):
if fixed_in_frame:
return points
if fixed_orientation:
center = center_of_mass(points)
# center = center_of_mass(points)
center_func = self.fixed_orientation_mobjects[mobject]
center = center_func()
new_center = self.project_point(center)
return points + (new_center - center)
else:
return self.project_points(points)
def add_fixed_orientation_mobjects(self, *mobjects):
for mobject in self.extract_mobject_family_members(mobjects):
self.fixed_orientation_mobjects.add(mobject)
def add_fixed_orientation_mobjects(self, *mobjects, center_func=None):
for mobject in mobjects:
func = center_func or mobject.get_center
for submob in mobject.get_family():
self.fixed_orientation_mobjects[submob] = func
def add_fixed_in_frame_mobjects(self, *mobjects):
for mobject in self.extract_mobject_family_members(mobjects):