opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-31 14:03:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Extended Mobject.align_to to handle aligning centers of mobjects

Browse Source
This commit is contained in:
Grant Sanderson
2018-01-23 13:41:14 -08:00
parent 1294b9b2fe
commit 64b5399bc5
1 changed files with 20 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
mobject/mobject.py
View File

@ -316,14 +316,29 @@ class Mobject(object):
self.shift(target_point - point_to_align + buff*direction) self.shift(target_point - point_to_align + buff*direction)
return self return self
def align_to(self, mobject_or_point, direction = UP): def align_to(self, mobject_or_point, direction = ORIGIN, alignment_vect = UP):
"""
Examples:
mob1.align_to(mob2, UP) moves mob1 vertically so that its
top edge lines ups with mob2's top edge.
mob1.align_to(mob2, alignment_vector = RIGHT) moves mob1
horizontally so that it's center is directly above/below
the center of mob2
"""
if isinstance(mobject_or_point, Mobject): if isinstance(mobject_or_point, Mobject):
mob = mobject_or_point mob = mobject_or_point
point = mob.get_edge_center(direction) target_point = mob.get_critical_point(direction)
else: else:
point = mobject_or_point target_point = mobject_or_point
diff = point - self.get_edge_center(direction) direction_norm = np.linalg.norm(direction)
self.shift(direction*np.dot(diff, direction)) if direction_norm > 0:
alignment_vect = np.array(direction)/direction_norm
reference_point = self.get_critical_point(direction)
else:
reference_point = self.get_center()
diff = target_point - reference_point
self.shift(alignment_vect*np.dot(diff, alignment_vect))
return self return self
def shift_onto_screen(self, **kwargs): def shift_onto_screen(self, **kwargs):