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

Arrow can take one arg

Browse source
This commit is contained in:
Grant Sanderson 2015-12-31 09:24:02 -08:00
parent 4807669e00
commit 0d0e56fd83
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
topics/geometry.py
View file

@ -38,7 +38,7 @@ class Cross(Mobject1D):
class Line(Mobject1D): class Line(Mobject1D):
DEFAULT_CONFIG = { DEFAULT_CONFIG = {
"buffer" : 0 "buff" : 0
} }
def __init__(self, start, end, **kwargs): def __init__(self, start, end, **kwargs):
digest_config(self, kwargs) digest_config(self, kwargs)
@ -64,8 +64,8 @@ class Line(Mobject1D):
] ]
start_to_end = self.end - self.start start_to_end = self.end - self.start
start_to_end /= np.linalg.norm(start_to_end) start_to_end /= np.linalg.norm(start_to_end)
self.start = self.start + self.buffer*start_to_end self.start = self.start + self.buff*start_to_end
self.end = self.end - self.buffer*start_to_end self.end = self.end - self.buff*start_to_end
def generate_points(self): def generate_points(self):
self.add_line(self.start, self.end) self.add_line(self.start, self.end)
@ -84,9 +84,16 @@ class Arrow(Line):
DEFAULT_CONFIG = { DEFAULT_CONFIG = {
"color" : YELLOW_C, "color" : YELLOW_C,
"tip_length" : 0.25, "tip_length" : 0.25,
"buffer" : 0.3, "buff" : 0.3,
} }
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
if len(args) == 1:
target = args[0]
if isinstance(target, Mobject):
point = target.get_center()
else:
point = target
args = (point+UP+LEFT, target)
Line.__init__(self, *args, **kwargs) Line.__init__(self, *args, **kwargs)
self.add_tip() self.add_tip()