public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed issue of buff with curved lines/arrows

Browse source
This commit is contained in:
Grant Sanderson 2017-04-26 16:40:54 -07:00
parent fd20472973
commit 62c43cd306
1 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
topics/geometry.py
View file

@ -83,7 +83,20 @@ class Line(VMobject):
for alpha in np.linspace(0, 1, self.n_arc_anchors) for alpha in np.linspace(0, 1, self.n_arc_anchors)
]) ])
self.considered_smooth = True self.considered_smooth = True
self.account_for_buff()
def account_for_buff(self):
anchors = self.get_anchors()
length = sum([
np.linalg.norm(a2-a1)
for a1, a2 in zip(anchors, anchors[1:])
])
if length < 2*self.buff or self.buff == 0:
return
buff_proportion = self.buff / length
self.pointwise_become_partial(
self, buff_proportion, 1 - buff_proportion
)
def set_start_and_end(self, start, end): def set_start_and_end(self, start, end):
start_to_end = self.pointify(end) - self.pointify(start) start_to_end = self.pointify(end) - self.pointify(start)
@ -96,12 +109,6 @@ class Line(VMobject):
else np.array(arg) else np.array(arg)
for arg, unit in zip([start, end], [1, -1]) for arg, unit in zip([start, end], [1, -1])
] ]
start_to_end = self.end - self.start
length = np.linalg.norm(start_to_end)
if length > 2*self.buff:
start_to_end = start_to_end / np.linalg.norm(start_to_end)
self.start = self.start + self.buff*start_to_end
self.end = self.end - self.buff*start_to_end
def pointify(self, mob_or_point): def pointify(self, mob_or_point):
if isinstance(mob_or_point, Mobject): if isinstance(mob_or_point, Mobject):