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

Fixed bugs in partial_bezier_points and integer_interpolate

Browse source
This commit is contained in:
Grant Sanderson 2019-02-05 15:25:44 -08:00
parent 272c5f4deb
commit 55081c75ef
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
manimlib/utils/bezier.py
View file

@ -24,12 +24,16 @@ def partial_bezier_points(points, a, b):
This algorithm is pretty nifty, and pretty dense. This algorithm is pretty nifty, and pretty dense.
""" """
if a == 1:
return [points[-1]] * len(points)
a_to_1 = np.array([ a_to_1 = np.array([
bezier(points[i:])(a) bezier(points[i:])(a)
for i in range(len(points)) for i in range(len(points))
]) ])
end_prop = (b - a) / (1. - a)
return np.array([ return np.array([
bezier(a_to_1[:i + 1])((b - a) / (1. - a)) bezier(a_to_1[:i + 1])(end_prop)
for i in range(len(points)) for i in range(len(points))
]) ])
@ -54,6 +58,8 @@ def integer_interpolate(start, end, alpha):
""" """
if alpha >= 1: if alpha >= 1:
return (end - 1, 1.0) return (end - 1, 1.0)
if alpha <= 0:
return (start, 0)
value = int(interpolate(start, end, alpha)) value = int(interpolate(start, end, alpha))
residue = ((end - start) * alpha) % 1 residue = ((end - start) * alpha) % 1
return (value, residue) return (value, residue)