From 10df6080647e5904a6c03b497586a554b1985a99 Mon Sep 17 00:00:00 2001 From: Leonard Kugis Date: Fri, 21 Aug 2020 22:30:37 +0200 Subject: Curves Added Linear and Bezier Curves. --- AutopyExtended/Curve/CurveBezier.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 AutopyExtended/Curve/CurveBezier.py (limited to 'AutopyExtended/Curve/CurveBezier.py') diff --git a/AutopyExtended/Curve/CurveBezier.py b/AutopyExtended/Curve/CurveBezier.py new file mode 100644 index 0000000..64d9539 --- /dev/null +++ b/AutopyExtended/Curve/CurveBezier.py @@ -0,0 +1,25 @@ +import numpy as np + +from 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.dot(x, poly) + self.__y = np.dot(y, poly) + + def point(self, t, t_target=None): + if(t_target == None): + t_target = self.__tn + if(t_target != self.__tn): + raise ValueError("t_target must be equal to tn.") + return (self.__x[t], self.__y[t]) + + @staticmethod + def bernstein(i, n, t): + return comb(n, i) * (t ** (n - 1)) * (1 - t) ** i -- cgit v1.2.1