diff options
author | Leonard Kugis <leonard@kug.is> | 2020-08-22 16:02:17 +0200 |
---|---|---|
committer | Leonard Kugis <leonard@kug.is> | 2020-08-22 16:02:17 +0200 |
commit | f43f909bf20dafb2c10520815282046cffe495f2 (patch) | |
tree | 5be52b3a048c0b2755a670de4269460353bc9760 /AutopyExtended/Curve/Bezier.py | |
parent | 3b47fad88c8b714c83c1c90cf6726d85688b300f (diff) |
Refactoring, Bezier
Refactored module layout.
Added missing dependencies to setup.
Added time scaling to curves.
Diffstat (limited to 'AutopyExtended/Curve/Bezier.py')
-rw-r--r-- | AutopyExtended/Curve/Bezier.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/AutopyExtended/Curve/Bezier.py b/AutopyExtended/Curve/Bezier.py new file mode 100644 index 0000000..ab79ec0 --- /dev/null +++ b/AutopyExtended/Curve/Bezier.py @@ -0,0 +1,24 @@ +import numpy as np + +from AutopyExtended.Curve.Curve import Curve +from scipy.special import comb + +class CurveBezier(Curve): + def __init__(self, cp, tn): + self.__tn = tn + x = np.array([p[0] for p in cp]) + y = np.array([p[1] for p in cp]) + t = np.linspace(0, 1, tn) + poly = np.array([CurveBezier.bernstein(i, len(cp) - 1, t) for i in range(0, len(cp))]) + self.__x = np.flip(np.dot(x, poly)) + self.__y = np.flip(np.dot(y, poly)) + + def point(self, t, t_target): + t_scaled = min([t * self.__tn / t_target, self.__tn - 1]) + return np.add((self.__x[int(np.floor(t_scaled))], self.__y[int(np.floor(t_scaled))]), + np.subtract((self.__x[int(np.ceil(t_scaled))], self.__y[int(np.ceil(t_scaled))]), + (self.__x[int(np.floor(t_scaled))], self.__y[int(np.floor(t_scaled))])) * (t_scaled - np.floor(t_scaled))) + + @staticmethod + def bernstein(i, n, t): + return comb(n, i) * (t ** (n - i)) * (1 - t) ** i |