o eN!@sdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZddZed/d d Zd dZd/ddZddZddZed/ddZed/ddZddZddZed/ddZed/dd Zd!d"Zed/d#d$Zd%d&Zed0d'd(Zd)d*Z d+d,Z!d/d-d.Z"d S)1z:Efficient functions for generating orthogonal polynomials.)Dummy)dup_muldup_mul_ground dup_lshiftdup_subdup_add)ZZQQ) named_poly)publiccCs|dkr|jgS|jg|||d|j|||dg}}td|dD]}||||||||d||d}|||d||j|||||d|}|||d||j|||d||d|||d||d|} |||j|||j|||d||} t|||} tt|d|| |} t|| |} |tt| | || |}}q'|S)z/Low-level implementation of Jacobi polynomials.)onerangerrrr)nabKm2m1iZdenZf0f1f2p0p1p2rFD:\Projects\ConvertPro\env\Lib\site-packages\sympy/polys/orthopolys.py dup_jacobi s006V4  rNFcCst|tdd|||f|S)aGenerates the Jacobi polynomial `P_n^{(a,b)}(x)`. Parameters ========== n : int Degree of the polynomial. a Lower limit of minimal domain for the list of coefficients. b Upper limit of minimal domain for the list of coefficients. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. NzJacobi polynomial)r r)rrrxpolysrrr jacobi_polysr!cCs|dkr|jgS|jg|d||jg}}td|dD]8}tt|d||d||j|||d|}t||d||j|||j|}|t|||}}q|S)z3Low-level implementation of Gegenbauer polynomials.r r rzerorrrr)rrrrrrrrrrrdup_gegenbauer,s2(r$cCst|tdd||f|S)a?Generates the Gegenbauer polynomial `C_n^{(a)}(x)`. Parameters ========== n : int Degree of the polynomial. x : optional a Decides minimal domain for the list of coefficients. polys : bool, optional If True, return a Poly, otherwise (default) return an expression. NzGegenbauer polynomial)r r$)rrrr rrrgegenbauer_poly7sr%cCsd|dkr|jgS|jg|j|jg}}td|dD]}|ttt|d||d|||}}q|S)zDLow-level implementation of Chebyshev polynomials of the first kind.r r rr#rrrrrrrrrrrrdup_chebyshevtGs (r(cCsf|dkr|jgS|jg|d|jg}}td|dD]}|ttt|d||d|||}}q|S)zELow-level implementation of Chebyshev polynomials of the second kind.r r r&r'rrrdup_chebyshevuPs (r)cCt|ttd|f|S)aGenerates the Chebyshev polynomial of the first kind `T_n(x)`. Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. z&Chebyshev polynomial of the first kind)r r(rrrr rrrchebyshevt_polyY r,cCr*)aGenerates the Chebyshev polynomial of the second kind `U_n(x)`. Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. z'Chebyshev polynomial of the second kind)r r)rr+rrrchebyshevu_polyir-r.cCs~|dkr|jgS|jg|d|jg}}td|dD]!}t|d|}t|||d|}|tt||||d|}}q|S)z0Low-level implementation of Hermite polynomials.r r rr#rrrrrrrrrrrrrr dup_hermiteys  r1cCsp|dkr|jgS|jg|j|jg}}td|dD]}t|d|}t|||d|}|t|||}}q|S)z>Low-level implementation of probabilist's Hermite polynomials.r r r/r0rrrdup_hermite_probs r2cCr*)zGenerates the Hermite polynomial `H_n(x)`. Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. zHermite polynomial)r r1rr+rrr hermite_poly r3cCr*)aGenerates the probabilist's Hermite polynomial `He_n(x)`. Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. z probabilist's Hermite polynomial)r r2rr+rrrhermite_prob_polyr-r5cCs|dkr|jgS|jg|j|jg}}td|dD]'}tt|d||d|d||}t|||d||}|t|||}}q|S)z1Low-level implementation of Legendre polynomials.r r r"r0rrr dup_legendres"r6cCr*)zGenerates the Legendre polynomial `P_n(x)`. Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. zLegendre polynomial)r r6r r+rrr legendre_polyr4r7cCs|jg|jg}}td|dD]4}t||j ||||j|||dg|}t|||j|||j|}|t|||}}q|S)z1Low-level implementation of Laguerre polynomials.r r )r#rrrrr)ralpharrrrrrrrr dup_laguerres 2 r9cCst|tdd||f|S)aQGenerates the Laguerre polynomial `L_n^{(\alpha)}(x)`. Parameters ========== n : int Degree of the polynomial. x : optional alpha : optional Decides minimal domain for the list of coefficients. polys : bool, optional If True, return a Poly, otherwise (default) return an expression. NzLaguerre polynomial)r r9)rrr8r rrr laguerre_polysr:cCsx|dkr |j|jgS|jg|j|jg}}td|dD]}|ttt|d||d|d|||}}qt|d|S)z%Low-level implementation of fn(n, x).r r r&r'rrrdup_spherical_bessel_fns  0 r;c Cs\|j|jg|jg}}td|dD]}|ttt|d||dd||||}}q|S)z&Low-level implementation of fn(-n, x).r r r&r'rrrdup_spherical_bessel_fn_minuss0r=cCs@|durtd}|dkrtnt}tt||tdtd|f|S)a Coefficients for the spherical Bessel functions. These are only needed in the jn() function. The coefficients are calculated from: fn(0, z) = 1/z fn(1, z) = 1/z**2 fn(n-1, z) + fn(n+1, z) == (2*n+1)/z * fn(n, z) Parameters ========== n : int Degree of the polynomial. x : optional polys : bool, optional If True, return a Poly, otherwise (default) return an expression. Examples ======== >>> from sympy.polys.orthopolys import spherical_bessel_fn as fn >>> from sympy import Symbol >>> z = Symbol("z") >>> fn(1, z) z**(-2) >>> fn(2, z) -1/z + 3/z**3 >>> fn(3, z) -6/z**2 + 15/z**4 >>> fn(4, z) 1/z - 45/z**3 + 105/z**5 Nrrr )rr=r;r absrr )rrr frrrspherical_bessel_fns% rA)NF)NrF)#__doc__Zsympy.core.symbolrZsympy.polys.densearithrrrrrZsympy.polys.domainsrr Zsympy.polys.polytoolsr Zsympy.utilitiesr rr!r$r%r(r)r,r.r1r2r3r5r6r7r9r:r;r=rArrrrs@