a RG5d'@sddlmZmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZddZd d Zed d d dZddZddZdS))Ssympify)Dummysymbols) Piecewisepiecewise_fold)And)Interval) lru_cachecCsRt|trBt|jdkrB|j\}}|j|kr6||}}|j|jfStd|dS)zreturn the interval corresponding to the condition Conditions in spline's Piecewise give the range over which an expression is valid like (lo <= x) & (x <= hi). This function returns (lo, hi). zunexpected cond type: %sN) isinstancerlenargsltsgts TypeError)condxabr\/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/functions/special/bsplines.py_ivl s     rcCs@tj||fvrt||}ntj||fvr>> from sympy import bspline_basis >>> from sympy.abc import x >>> d = 0 >>> knots = tuple(range(5)) >>> bspline_basis(d, knots, 0, x) Piecewise((1, (x >= 0) & (x <= 1)), (0, True)) For a given ``(d, knots)`` there are ``len(knots)-d-1`` B-splines defined, that are indexed by ``n`` (starting at 0). Here is an example of a cubic B-spline: >>> bspline_basis(3, tuple(range(5)), 0, x) Piecewise((x**3/6, (x >= 0) & (x <= 1)), (-x**3/2 + 2*x**2 - 2*x + 2/3, (x >= 1) & (x <= 2)), (x**3/2 - 4*x**2 + 10*x - 22/3, (x >= 2) & (x <= 3)), (-x**3/6 + 2*x**2 - 8*x + 32/3, (x >= 3) & (x <= 4)), (0, True)) By repeating knot points, you can introduce discontinuities in the B-splines and their derivatives: >>> d = 1 >>> knots = (0, 0, 2, 3, 4) >>> bspline_basis(d, knots, 0, x) Piecewise((1 - x/2, (x >= 0) & (x <= 2)), (0, True)) It is quite time consuming to construct and evaluate B-splines. If you need to evaluate a B-spline many times, it is best to lambdify them first: >>> from sympy import lambdify >>> d = 3 >>> knots = tuple(range(10)) >>> b0 = bspline_basis(d, knots, 0, x) >>> f = lambdify(x, b0) >>> y = f(0.5) Parameters ========== d : integer degree of bspline knots : list of integer values list of knots points of bspline n : integer $n$-th B-spline x : symbol See Also ======== bspline_basis_set References ========== .. [1] https://en.wikipedia.org/wiki/B-spline css|]}t|VqdSNr).0krrr z bspline_basis..z(n + d + 1 must not exceed len(knots) - 1rrzdegree must be non-negative: %r)rtupleintr ValueErrorrrOner containsr bspline_basisr1xreplace) r%knotsnrxvarZn_knotsZ n_intervalsresultdenomBr&Ar$rrrr@Ts6S"      r@cs*td}fddt|DS)a| Return the ``len(knots)-d-1`` B-splines at *x* of degree *d* with *knots*. Explanation =========== This function returns a list of piecewise polynomials that are the ``len(knots)-d-1`` B-splines of degree *d* for the given knots. This function calls ``bspline_basis(d, knots, n, x)`` for different values of *n*. Examples ======== >>> from sympy import bspline_basis_set >>> from sympy.abc import x >>> d = 2 >>> knots = range(5) >>> splines = bspline_basis_set(d, knots, x) >>> splines [Piecewise((x**2/2, (x >= 0) & (x <= 1)), (-x**2 + 3*x - 3/2, (x >= 1) & (x <= 2)), (x**2/2 - 3*x + 9/2, (x >= 2) & (x <= 3)), (0, True)), Piecewise((x**2/2 - x + 1/2, (x >= 1) & (x <= 2)), (-x**2 + 5*x - 11/2, (x >= 2) & (x <= 3)), (x**2/2 - 4*x + 8, (x >= 3) & (x <= 4)), (0, True))] Parameters ========== d : integer degree of bspline knots : list of integers list of knots points of bspline x : symbol See Also ======== bspline_basis r:csg|]}tt|qSr)r@r;r6r-r%rBrrr r9z%bspline_basis_set..)r range)r%rBrZ n_splinesrrJrbspline_basis_sets0rMcsddlm}ddlm}t|}|jr,|js8td|t|t|krPtdt||dkrhtdt dd t ||dd Dstd d d |D}|j r|dd}||| }n8|d}dd t ||| d||d| D}|dg|dt ||dg|d}t ||fdd |D} ||| ||ftdt|td} t | d} ddD} fdd | D} t | | } t| ddd} dd | D} dd D}g}| D]4tfdd t | |Dtj}||fqt|S)a Return spline of degree *d*, passing through the given *X* and *Y* values. Explanation =========== This function returns a piecewise function such that each part is a polynomial of degree not greater than *d*. The value of *d* must be 1 or greater and the values of *X* must be strictly increasing. Examples ======== >>> from sympy import interpolating_spline >>> from sympy.abc import x >>> interpolating_spline(1, x, [1, 2, 4, 7], [3, 6, 5, 7]) Piecewise((3*x, (x >= 1) & (x <= 2)), (7 - x/2, (x >= 2) & (x <= 4)), (2*x/3 + 7/3, (x >= 4) & (x <= 7))) >>> interpolating_spline(3, x, [-2, 0, 1, 3, 4], [4, 2, 1, 1, 3]) Piecewise((7*x**3/117 + 7*x**2/117 - 131*x/117 + 2, (x >= -2) & (x <= 1)), (10*x**3/117 - 2*x**2/117 - 122*x/117 + 77/39, (x >= 1) & (x <= 4))) Parameters ========== d : integer Degree of Bspline strictly greater than equal to one x : symbol X : list of strictly increasing integer values list of X coordinates through which the spline passes Y : list of strictly increasing integer values list of Y coordinates through which the spline passes See Also ======== bspline_basis_set, interpolating_poly r)linsolve)Matrixz1Spline degree must be a positive integer, not %s.z/Number of X and Y coordinates must be the same.r:z6Degree must be less than the number of control points.css|]\}}||kVqdSr4rr6rrrrrr89r9z'interpolating_spline..Nz.The x-coordinates must be strictly increasing.cSsg|] }t|qSrr5rIrrrrK;r9z(interpolating_spline..r cSsg|]\}}||dqS)r rrPrrrrKCsrcs g|]fddDqS)csg|]}|qSr)subsr6r)vrrrrKKr9z3interpolating_spline...r)r6)basisr)rSrrKKr9zc0:{})clscSs(h|] }|jD]\}}|dkr|qqS)Tr)r6rer#rrr Or9z'interpolating_spline..csg|]}t|qSr)r)r6r#rrrrKSr9cSs|dS)NrrrYrrrUr9z&interpolating_spline..)keycSsg|] \}}|qSrr)r6ryrrrrKVr9cSsg|]}dd|jDqS)cSsi|]\}}||qSrr)r6rWr#rrr Xr9z3interpolating_spline...rVrRrrrrKXr9cs"g|]\}}||tjqSr)getrr)r6r#r%)r-rrrK\r9)sympy.solvers.solvesetrNsympy.matrices.denserOr is_Integer is_positiver=r allzipis_oddrrMrformatrsortedsumrrr r)r%rXYrNrOjZinterior_knotsrBrHcoeff intervalsivalcomZ basis_dictssplinepiecer)rTr-rrinterpolating_splinesJ.      $, (  rrN) sympy.corerrsympy.core.symbolrrsympy.functionsrrsympy.logic.boolalgrsympy.sets.setsr functoolsr rr1r@rMrrrrrrs   ; w4