U -e@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZmZmZdd lmZdd lmZmZdd lmZdd lmZdd lmZm Z m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mGddde=Z?Gddde=Z@Gddde=ZAGd d!d!e=ZBGd"d#d#e=ZCd$d%ZDGd&d'd'e=ZEd(d)ZFd*d+ZGGd,d-d-eEZHGd.d/d/eEZIGd0d1d1eEZJGd2d3d3eJZKGd4d5d5eJZLdGd8d9ZMGd:d;d;e ZNGdd?d?eNZPGd@dAdAeNZQGdBdCdCeNZRGdDdEdEe ZSdFS)Hwraps)S)Add)cacheit)Expr)FunctionArgumentIndexError_mexpand)fuzzy_or fuzzy_not)RationalpiI)Pow)DummyWild)sympify) factorial)sincoscsccot)ceiling)explog)cbrtsqrtroot)Absreim polar_lift unpolarify)gammadigamma uppergamma)hyper)spherical_bessel_fn)mpworkprecc@s^eZdZdZeddZeddZeddZdd d Z d d Z d dZ ddZ ddZ dS) BesselBasea Abstract base class for Bessel-type functions. This class is meant to reduce code duplication. All Bessel-type functions can 1) be differentiated, with the derivatives expressed in terms of similar functions, and 2) be rewritten in terms of other Bessel-type functions. Here, Bessel-type functions are assumed to have one complex parameter. To use this base class, define class attributes ``_a`` and ``_b`` such that ``2*F_n' = -_a*F_{n+1} + b*F_{n-1}``. cCs |jdS)z( The order of the Bessel-type function. rargsselfr0_/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sympy/functions/special/bessel.pyorder4szBesselBase.ordercCs |jdS)z+ The argument of the Bessel-type function. r,r.r0r0r1argument9szBesselBase.argumentcCsdSNr0clsnuzr0r0r1eval>szBesselBase.evalcCsN|dkrt|||jd||jd|j|jd||jd|jSNr;r3)r _b __class__r2r4_ar/argindexr0r0r1fdiffBs  zBesselBase.fdiffcCs*|j}|jdkr&||j|SdSNF)r4is_extended_negativer>r2 conjugater/r9r0r0r1_eval_conjugateHs zBesselBase._eval_conjugatecCsx|j|j}}||rdS|||s,dS|||}|jrdt|ttt t t t fsZ|j sdt|jStt|j |jgSrC)r2r4has_eval_is_meromorphicsubs is_integer isinstancebesseljbesselihn1hn2jnynis_zeror is_infiniter )r/xar8r9Zz0r0r0r1rIMs    zBesselBase._eval_is_meromorphiccKs|j|j|j}}}|jr|djrn|j |j||d|d|j|d||d||S|djrd|j|d||d|||j|j||d|S|SNr3r;) r2r4r>Zis_real is_positiver?r=_eval_expand_func is_negative)r/hintsr8r9fr0r0r1rYZs & &zBesselBase._eval_expand_funccKsddlm}||S)Nr) besselsimp)Zsympy.simplify.simplifyr])r/kwargsr]r0r0r1_eval_simplifyes zBesselBase._eval_simplifyN)r;)__name__ __module__ __qualname____doc__propertyr2r4 classmethodr:rBrGrIrYr_r0r0r0r1r+$s      r+csheZdZdZejZejZeddZ ddZ ddZ dd Z dfd d Z ddZdfdd ZZS)rMa4 Bessel function of the first kind. Explanation =========== The Bessel $J$ function of order $\nu$ is defined to be the function satisfying Bessel's differential equation .. math :: z^2 \frac{\mathrm{d}^2 w}{\mathrm{d}z^2} + z \frac{\mathrm{d}w}{\mathrm{d}z} + (z^2 - \nu^2) w = 0, with Laurent expansion .. math :: J_\nu(z) = z^\nu \left(\frac{1}{\Gamma(\nu + 1) 2^\nu} + O(z^2) \right), if $\nu$ is not a negative integer. If $\nu=-n \in \mathbb{Z}_{<0}$ *is* a negative integer, then the definition is .. math :: J_{-n}(z) = (-1)^n J_n(z). Examples ======== Create a Bessel function object: >>> from sympy import besselj, jn >>> from sympy.abc import z, n >>> b = besselj(n, z) Differentiate it: >>> b.diff(z) besselj(n - 1, z)/2 - besselj(n + 1, z)/2 Rewrite in terms of spherical Bessel functions: >>> b.rewrite(jn) sqrt(2)*sqrt(z)*jn(n - 1/2, z)/sqrt(pi) Access the parameter and argument: >>> b.order n >>> b.argument z See Also ======== bessely, besseli, besselk References ========== .. [1] Abramowitz, Milton; Stegun, Irene A., eds. (1965), "Chapter 9", Handbook of Mathematical Functions with Formulas, Graphs, and Mathematical Tables .. [2] Luke, Y. L. (1969), The Special Functions and Their Approximations, Volume 1 .. [3] https://en.wikipedia.org/wiki/Bessel_function .. [4] https://functions.wolfram.com/Bessel-TypeFunctions/BesselJ/ cCsZ|jrX|jrtjS|jr"|jdks,t|jr2tjSt|jrL|jdk rLtjS|j rXtj S|tj tj fkrntjS| r||| | t|| S|jr| rtj| t| |S|t}|rt|t||S|jrt|}||kr:t||Sn8|\}}|dkr:td|t|tt||St|}||krVt||SdS)NFTrr;)rSrOnerKr rXZerorZComplexInfinity is_imaginaryNaNInfinityNegativeInfinitycould_extract_minus_signrM NegativeOneextract_multiplicativelyrrNr#extract_branch_factorrrr7r8r9ZnewznZnnur0r0r1r:s:      " z besselj.evalcKs(ttt|dt|tt |SNr;)rrrrNr"r/r8r9r^r0r0r1_eval_rewrite_as_besselisz besselj._eval_rewrite_as_besselicKs<|jdkr8tt|t| |tt|t||SdSrC)rKrrbesselyrrtr0r0r1_eval_rewrite_as_besselys z besselj._eval_rewrite_as_besselycKs"td|tt|tj|jSrs)rrrQrHalfr4rtr0r0r1_eval_rewrite_as_jnszbesselj._eval_rewrite_as_jnNrc s|j\}}z||}Wntk r0|YSX||\}}|jrb||d|t|dS|jr|dkrtdn|}|||} | jstdt|t d|ddtt |S|St t | |||S)Nr;r3r) r-as_leading_termNotImplementedErroras_coeff_exponentrXr$rZrrrsuperrM_eval_as_leading_term r/rUlogxcdirr8r9argcesignr>r0r1rs   0zbesselj._eval_as_leading_termcCs|j\}}|jr|jrdSdSNTr-rKis_extended_realr/r8r9r0r0r1_eval_is_extended_reals  zbesselj._eval_is_extended_realc s&ddlm}|j\}}z||\}} Wnttfk rD|YSX| jrt|| } ||||} |d|||| } | t j kr| St | d|  } | |t |d}|g}td| ddD]4}|| |||9}t ||  }||qt|| Stt|||||SNrOrderr;r3)sympy.series.orderrr-leadterm ValueErrorr|rXr _eval_nseriesremoveOrrgr r$rangeappendrr~rMr/rUrrrrrr8r9_rnewnorttermskrr0r1rs*       zbesselj._eval_nseries)Nr)r)r`rarbrcrrfr?r=rer:rurwryrrr __classcell__r0r0rr1rMjsD #rMcsheZdZdZejZejZeddZ ddZ ddZ dd Z dfd d Z ddZdfdd ZZS)rva` Bessel function of the second kind. Explanation =========== The Bessel $Y$ function of order $\nu$ is defined as .. math :: Y_\nu(z) = \lim_{\mu \to \nu} \frac{J_\mu(z) \cos(\pi \mu) - J_{-\mu}(z)}{\sin(\pi \mu)}, where $J_\mu(z)$ is the Bessel function of the first kind. It is a solution to Bessel's equation, and linearly independent from $J_\nu$. Examples ======== >>> from sympy import bessely, yn >>> from sympy.abc import z, n >>> b = bessely(n, z) >>> b.diff(z) bessely(n - 1, z)/2 - bessely(n + 1, z)/2 >>> b.rewrite(yn) sqrt(2)*sqrt(z)*yn(n - 1/2, z)/sqrt(pi) See Also ======== besselj, besseli, besselk References ========== .. [1] https://functions.wolfram.com/Bessel-TypeFunctions/BesselY/ cCs|jr6|jrtjSt|jdkr&tjSt|jr6tjS|tjtjfkrLtjS|ttjkrxt tt |ddtjS|ttjkrt t t |ddtjS|j r| rtj | t| |SdS)NFr3r;)rSrrlr rhrjrkrgrrrrKrmrnrvr6r0r0r1r:Fs   z bessely.evalcKs<|jdkr8tt|tt|t||t| |SdSrC)rKrrrrMrtr0r0r1_eval_rewrite_as_besseljZs z bessely._eval_rewrite_as_besseljcKs|j|j}|r|tSdSr5)rr-rewriterNr/r8r9r^Zajr0r0r1ru^s z bessely._eval_rewrite_as_besselicKs"td|tt|tj|jSrs)rrrRrrxr4rtr0r0r1_eval_rewrite_as_yncszbessely._eval_rewrite_as_ynNrc s~|j\}}z||}Wntk r0|YSX||\}}|jrdtt|dt||} |jr|d|  t|dtnt j } |d| tt|t |dt j } t | | | gj||d}|S|jrj|dkrdn|}|||} | jsftdtt|d|td dtt|d|tdd|td|ttS|Stt||||S)Nr;r3rrrz)r-r{r|r}rXrrrMrrrgr% EulerGammarrZrrrr~rvr) r/rUrrr8r9rrrterm_oneterm_two term_threerrr0r1rfs&  ,, bzbessely._eval_as_leading_termcCs|j\}}|jr|jrdSdSrr-rKrXrr0r0r1rs  zbessely._eval_is_extended_realc s8ddlm}|j\}}z||\}} Wnttfk rD|YSX| jr"|jr"t|| } t ||} dt t |d|  ||||} gg} }||||}|d |||| }|tjkr|St|d| }|tjkrp|| t|dt }| |td|D]R}|||}|tjkrF|||9}n |||9}t|| }| |q||t t|}|t|dtj}||td| ddD]V}|| |||9}t|| }|t||dt|d}||q| t| t|Stt| ||||Sr)rrr-rrr|rXrKrrMrrrrrrgr rrrr%rrr~rvr/rUrrrrrr8r9rrrZbnrVbrrrrrrdenomprr0r1rsH     $         zbessely._eval_nseries)Nr)r)r`rarbrcrrfr?r=rer:rrurrrrrr0r0rr1rvs( rvcsjeZdZdZej ZejZeddZ ddZ ddZ dd Z d d Z dfdd Zdfdd ZZS)rNa Modified Bessel function of the first kind. Explanation =========== The Bessel $I$ function is a solution to the modified Bessel equation .. math :: z^2 \frac{\mathrm{d}^2 w}{\mathrm{d}z^2} + z \frac{\mathrm{d}w}{\mathrm{d}z} + (z^2 + \nu^2)^2 w = 0. It can be defined as .. math :: I_\nu(z) = i^{-\nu} J_\nu(iz), where $J_\nu(z)$ is the Bessel function of the first kind. Examples ======== >>> from sympy import besseli >>> from sympy.abc import z, n >>> besseli(n, z).diff(z) besseli(n - 1, z)/2 + besseli(n + 1, z)/2 See Also ======== besselj, bessely, besselk References ========== .. [1] https://functions.wolfram.com/Bessel-TypeFunctions/BesselI/ cCs|jrX|jrtjS|jr"|jdks,t|jr2tjSt|jrL|jdk rLtjS|j rXtj St |tj tj fkrrtjS|tj krtj S|tj krd|tj S|r||| | t|| S|jr|rt| |S|t}|rt| t|| S|jr*t|}||krbt||Sn8|\}}|dkrbtd|t|tt||St|}||kr~t||SdS)NFTrr;)rSrrfrKr rXrgrZrhrirjr!rkrlrmrNrorrMr#rprrrqr0r0r1r:sB         " z besseli.evalcKs(tt t|dt|tt|Srs)rrrrMr"rtr0r0r1r sz besseli._eval_rewrite_as_besseljcKs|j|j}|r|tSdSr5rr-rrvrr0r0r1rw s z besseli._eval_rewrite_as_besselycKs|j|jtSr5)rr-rrQrtr0r0r1ryszbesseli._eval_rewrite_as_jncCs|j\}}|jr|jrdSdSrrrr0r0r1rs  zbesseli._eval_is_extended_realNrc s|j\}}z||}Wntk r0|YSX||\}}|jrb||d|t|dS|jr|dkrtdn|}|||} | jst|tdt |S|St t | |||SNr;r3r) r-r{r|r}rXr$rZrrrr~rNrrrr0r1rs   zbesseli._eval_as_leading_termc s$ddlm}|j\}}z||\}} Wnttfk rD|YSX| jrt|| } ||||} |d|||| } | t j kr| St | d|  } | |t |d}|g}td| ddD]2}|| |||9}t ||  }||qt|| Stt|||||Sr)rrr-rrr|rXrrrrrgr r$rrrr~rNrrr0r1r.s*       zbesseli._eval_nseries)Nr)r)r`rarbrcrrfr?r=rer:rrwryrrrrr0r0rr1rNs' 'rNcsreZdZdZejZej ZeddZ ddZ ddZ dd Z d d Z d d Zdfdd Zdfdd ZZS)besselka Modified Bessel function of the second kind. Explanation =========== The Bessel $K$ function of order $\nu$ is defined as .. math :: K_\nu(z) = \lim_{\mu \to \nu} \frac{\pi}{2} \frac{I_{-\mu}(z) -I_\mu(z)}{\sin(\pi \mu)}, where $I_\mu(z)$ is the modified Bessel function of the first kind. It is a solution of the modified Bessel equation, and linearly independent from $Y_\nu$. Examples ======== >>> from sympy import besselk >>> from sympy.abc import z, n >>> besselk(n, z).diff(z) -besselk(n - 1, z)/2 - besselk(n + 1, z)/2 See Also ======== besselj, besseli, bessely References ========== .. [1] https://functions.wolfram.com/Bessel-TypeFunctions/BesselK/ cCsv|jr6|jrtjSt|jdkr&tjSt|jr6tjS|tjttjttjfkrXtjS|j rr| rrt | |SdSrC) rSrrkr rhrjrrlrgrKrmrr6r0r0r1r:ws z besselk.evalcKs8|jdkr4ttt|t| |t||dSdS)NFr;)rKrrrNrtr0r0r1rus z besselk._eval_rewrite_as_besselicKs|j|j}|r|tSdSr5)rur-rrM)r/r8r9r^Zair0r0r1rs z besselk._eval_rewrite_as_besseljcKs|j|j}|r|tSdSr5rrr0r0r1rws z besselk._eval_rewrite_as_besselycKs|j|j}|r|tSdSr5)rwr-rrR)r/r8r9r^Zayr0r0r1rs zbesselk._eval_rewrite_as_yncCs|j\}}|jr|jrdSdSrrrr0r0r1rs  zbesselk._eval_is_extended_realNrc s|j\}}z||}Wntk r0|YSX||\}}|jrd|dt|dt||} |jr|d| t|ddntj } d||d|dt|t |dtj } t | | | gj||d}|S|j rttt| td|Stt||||S)Nrr3r;r)r-r{r|r}rXrrNrrrgr%rrrZrrrr~rr) r/rUrrr8r9rrrrrrrr0r1rs  "*2zbesselk._eval_as_leading_termc sBddlm}|j\}}z||\}} Wnttfk rD|YSX| jr,|jr,t|| } t ||} d|dt |d|  ||||} gg} }||||}|d |||| }|t jkr|St|d| }|t jkrt|| t|dd}| |td|D]R}|||}|t jkrJ|||9}n |||9}t|| }| |q ||d|dt|}|t|dt j}||td| ddD]T}|||||9}t|| }|t||dt|d}||q| t| t|Stt| ||||S)Nrrrr3r;)rrr-rrr|rXrKrrNrrrrrgr rrrr%rrr~rrrr0r1rsH     (         zbesselk._eval_nseries)Nr)r)r`rarbrcrrfr?r=rer:rurrwrrrrrr0r0rr1rNs% rc@s$eZdZdZejZejZddZdS)hankel1a Hankel function of the first kind. Explanation =========== This function is defined as .. math :: H_\nu^{(1)} = J_\nu(z) + iY_\nu(z), where $J_\nu(z)$ is the Bessel function of the first kind, and $Y_\nu(z)$ is the Bessel function of the second kind. It is a solution to Bessel's equation. Examples ======== >>> from sympy import hankel1 >>> from sympy.abc import z, n >>> hankel1(n, z).diff(z) hankel1(n - 1, z)/2 - hankel1(n + 1, z)/2 See Also ======== hankel2, besselj, bessely References ========== .. [1] https://functions.wolfram.com/Bessel-TypeFunctions/HankelH1/ cCs(|j}|jdkr$t|j|SdSrC)r4rDhankel2r2rErFr0r0r1rGs zhankel1._eval_conjugateN r`rarbrcrrfr?r=rGr0r0r0r1rs$rc@s$eZdZdZejZejZddZdS)ra Hankel function of the second kind. Explanation =========== This function is defined as .. math :: H_\nu^{(2)} = J_\nu(z) - iY_\nu(z), where $J_\nu(z)$ is the Bessel function of the first kind, and $Y_\nu(z)$ is the Bessel function of the second kind. It is a solution to Bessel's equation, and linearly independent from $H_\nu^{(1)}$. Examples ======== >>> from sympy import hankel2 >>> from sympy.abc import z, n >>> hankel2(n, z).diff(z) hankel2(n - 1, z)/2 - hankel2(n + 1, z)/2 See Also ======== hankel1, besselj, bessely References ========== .. [1] https://functions.wolfram.com/Bessel-TypeFunctions/HankelH2/ cCs(|j}|jdkr$t|j|SdSrC)r4rDrr2rErFr0r0r1rG=s zhankel2._eval_conjugateNrr0r0r0r1rs%rcstfdd}|S)Ncs|jr|||SdSr5)rKrfnr0r1gDszassume_integer_order..gr)rrr0rr1assume_integer_orderCsrc@s*eZdZdZddZddZd ddZd S) SphericalBesselBasea- Base class for spherical Bessel functions. These are thin wrappers around ordinary Bessel functions, since spherical Bessel functions differ from the ordinary ones just by a slight change in order. To use this class, define the ``_eval_evalf()`` and ``_expand()`` methods. cKs tddS)z@ Expand self into a polynomial. Nu is guaranteed to be Integer. Z expansionNr|r/r[r0r0r1_expandWszSphericalBesselBase._expandcKs|jjr|jf|S|Sr5)r2 is_Integerrrr0r0r1rY[s z%SphericalBesselBase._eval_expand_funcr;cCs:|dkrt||||jd|j||jd|jSr<)r r>r2r4r@r0r0r1rB`s  zSphericalBesselBase.fdiffN)r;)r`rarbrcrrYrBr0r0r0r1rKs rcCs8t||t|tj|dt| d|t|SNr3)r(rrrnrrrr9r0r0r1_jngs$rcCs8tj|dt| d|t|t||t|Sr)rrnr(rrrr0r0r1_ynls$rc@sDeZdZdZeddZddZddZdd Zd d Z d d Z dS)rQa Spherical Bessel function of the first kind. Explanation =========== This function is a solution to the spherical Bessel equation .. math :: z^2 \frac{\mathrm{d}^2 w}{\mathrm{d}z^2} + 2z \frac{\mathrm{d}w}{\mathrm{d}z} + (z^2 - \nu(\nu + 1)) w = 0. It can be defined as .. math :: j_\nu(z) = \sqrt{\frac{\pi}{2z}} J_{\nu + \frac{1}{2}}(z), where $J_\nu(z)$ is the Bessel function of the first kind. The spherical Bessel functions of integral order are calculated using the formula: .. math:: j_n(z) = f_n(z) \sin{z} + (-1)^{n+1} f_{-n-1}(z) \cos{z}, where the coefficients $f_n(z)$ are available as :func:`sympy.polys.orthopolys.spherical_bessel_fn`. Examples ======== >>> from sympy import Symbol, jn, sin, cos, expand_func, besselj, bessely >>> z = Symbol("z") >>> nu = Symbol("nu", integer=True) >>> print(expand_func(jn(0, z))) sin(z)/z >>> expand_func(jn(1, z)) == sin(z)/z**2 - cos(z)/z True >>> expand_func(jn(3, z)) (-6/z**2 + 15/z**4)*sin(z) + (1/z - 15/z**3)*cos(z) >>> jn(nu, z).rewrite(besselj) sqrt(2)*sqrt(pi)*sqrt(1/z)*besselj(nu + 1/2, z)/2 >>> jn(nu, z).rewrite(bessely) (-1)**nu*sqrt(2)*sqrt(pi)*sqrt(1/z)*bessely(-nu - 1/2, z)/2 >>> jn(2, 5.2+0.3j).evalf(20) 0.099419756723640344491 - 0.054525080242173562897*I See Also ======== besselj, bessely, besselk, yn References ========== .. [1] https://dlmf.nist.gov/10.47 cCsD|jr*|jrtjS|jr*|jr$tjStjS|tjtjfkr@tjSdSr5) rSrrfrKrXrgrhrlrkr6r0r0r1r:szjn.evalcKs ttd|t|tj|Srs)rrrMrrxrtr0r0r1rszjn._eval_rewrite_as_besseljcKs,tj|ttd|t| tj|Srs)rrnrrrvrxrtr0r0r1rwszjn._eval_rewrite_as_besselycKstj|t| d|Sr)rrnrRrtr0r0r1rszjn._eval_rewrite_as_yncKst|j|jSr5)rr2r4rr0r0r1rsz jn._expandcCs|jjr|t|SdSr5r2rrrM _eval_evalfr/precr0r0r1rszjn._eval_evalfN) r`rarbrcrer:rrwrrrr0r0r0r1rQrs9 rQc@s@eZdZdZeddZeddZddZdd Zd d Z d S) rRa Spherical Bessel function of the second kind. Explanation =========== This function is another solution to the spherical Bessel equation, and linearly independent from $j_n$. It can be defined as .. math :: y_\nu(z) = \sqrt{\frac{\pi}{2z}} Y_{\nu + \frac{1}{2}}(z), where $Y_\nu(z)$ is the Bessel function of the second kind. For integral orders $n$, $y_n$ is calculated using the formula: .. math:: y_n(z) = (-1)^{n+1} j_{-n-1}(z) Examples ======== >>> from sympy import Symbol, yn, sin, cos, expand_func, besselj, bessely >>> z = Symbol("z") >>> nu = Symbol("nu", integer=True) >>> print(expand_func(yn(0, z))) -cos(z)/z >>> expand_func(yn(1, z)) == -cos(z)/z**2-sin(z)/z True >>> yn(nu, z).rewrite(besselj) (-1)**(nu + 1)*sqrt(2)*sqrt(pi)*sqrt(1/z)*besselj(-nu - 1/2, z)/2 >>> yn(nu, z).rewrite(bessely) sqrt(2)*sqrt(pi)*sqrt(1/z)*bessely(nu + 1/2, z)/2 >>> yn(2, 5.2+0.3j).evalf(20) 0.18525034196069722536 + 0.014895573969924817587*I See Also ======== besselj, bessely, besselk, jn References ========== .. [1] https://dlmf.nist.gov/10.47 cKs0tj|dttd|t| tj|SrW)rrnrrrMrxrtr0r0r1rszyn._eval_rewrite_as_besseljcKs ttd|t|tj|Srs)rrrvrrxrtr0r0r1rwszyn._eval_rewrite_as_besselycKstj|dt| d|Sr)rrnrQrtr0r0r1ryszyn._eval_rewrite_as_jncKst|j|jSr5)rr2r4rr0r0r1rsz yn._expandcCs|jjr|t|SdSr5)r2rrrvrrr0r0r1rszyn._eval_evalfN) r`rarbrcrrrwryrrr0r0r0r1rRs.  rRc@sLeZdZeddZeddZddZddZd d Zd d Z d dZ dS)SphericalHankelBasecKsN|j}ttd|t|tj||ttj|dt| tj|Sr<)_hankel_kind_signrrrMrrxrrnr/r8r9r^hksr0r0r1rs&z,SphericalHankelBase._eval_rewrite_as_besseljcKsJ|j}ttd|tj|t| tj||tt|tj|Srs)rrrrrnrvrxrrr0r0r1rws(z,SphericalHankelBase._eval_rewrite_as_besselycKs(|j}t||t|tt||Sr5)rrQrrRrrr0r0r1r sz'SphericalHankelBase._eval_rewrite_as_yncKs(|j}t|||tt||tSr5)rrQrrRrrr0r0r1ry$sz'SphericalHankelBase._eval_rewrite_as_jncKsF|jjr|jf|S|j}|j}|j}t|||tt||SdSr5)r2rrr4rrQrrR)r/r[r8r9rr0r0r1rY(s  z%SphericalHankelBase._eval_expand_funccKs2|j}|j}|j}t|||tt||Sr5)r2r4rrrrexpand)r/r[rrr9rr0r0r1r1s zSphericalHankelBase._expandcCs|jjr|t|SdSr5rrr0r0r1r@szSphericalHankelBase._eval_evalfN) r`rarbrrrwrryrYrrr0r0r0r1r s   rc@s"eZdZdZejZeddZdS)rOa Spherical Hankel function of the first kind. Explanation =========== This function is defined as .. math:: h_\nu^(1)(z) = j_\nu(z) + i y_\nu(z), where $j_\nu(z)$ and $y_\nu(z)$ are the spherical Bessel function of the first and second kinds. For integral orders $n$, $h_n^(1)$ is calculated using the formula: .. math:: h_n^(1)(z) = j_{n}(z) + i (-1)^{n+1} j_{-n-1}(z) Examples ======== >>> from sympy import Symbol, hn1, hankel1, expand_func, yn, jn >>> z = Symbol("z") >>> nu = Symbol("nu", integer=True) >>> print(expand_func(hn1(nu, z))) jn(nu, z) + I*yn(nu, z) >>> print(expand_func(hn1(0, z))) sin(z)/z - I*cos(z)/z >>> print(expand_func(hn1(1, z))) -I*sin(z)/z - cos(z)/z + sin(z)/z**2 - I*cos(z)/z**2 >>> hn1(nu, z).rewrite(jn) (-1)**(nu + 1)*I*jn(-nu - 1, z) + jn(nu, z) >>> hn1(nu, z).rewrite(yn) (-1)**nu*yn(-nu - 1, z) + I*yn(nu, z) >>> hn1(nu, z).rewrite(hankel1) sqrt(2)*sqrt(pi)*sqrt(1/z)*hankel1(nu, z)/2 See Also ======== hn2, jn, yn, hankel1, hankel2 References ========== .. [1] https://dlmf.nist.gov/10.47 cKsttd|t||Srs)rrrrtr0r0r1_eval_rewrite_as_hankel1xszhn1._eval_rewrite_as_hankel1N) r`rarbrcrrfrrrr0r0r0r1rOEs0rOc@s$eZdZdZej ZeddZdS)rPa Spherical Hankel function of the second kind. Explanation =========== This function is defined as .. math:: h_\nu^(2)(z) = j_\nu(z) - i y_\nu(z), where $j_\nu(z)$ and $y_\nu(z)$ are the spherical Bessel function of the first and second kinds. For integral orders $n$, $h_n^(2)$ is calculated using the formula: .. math:: h_n^(2)(z) = j_{n} - i (-1)^{n+1} j_{-n-1}(z) Examples ======== >>> from sympy import Symbol, hn2, hankel2, expand_func, jn, yn >>> z = Symbol("z") >>> nu = Symbol("nu", integer=True) >>> print(expand_func(hn2(nu, z))) jn(nu, z) - I*yn(nu, z) >>> print(expand_func(hn2(0, z))) sin(z)/z + I*cos(z)/z >>> print(expand_func(hn2(1, z))) I*sin(z)/z - cos(z)/z + sin(z)/z**2 + I*cos(z)/z**2 >>> hn2(nu, z).rewrite(hankel2) sqrt(2)*sqrt(pi)*sqrt(1/z)*hankel2(nu, z)/2 >>> hn2(nu, z).rewrite(jn) -(-1)**(nu + 1)*I*jn(-nu - 1, z) + jn(nu, z) >>> hn2(nu, z).rewrite(yn) (-1)**nu*yn(-nu - 1, z) - I*yn(nu, z) See Also ======== hn1, jn, yn, hankel1, hankel2 References ========== .. [1] https://dlmf.nist.gov/10.47 cKsttd|t||Srs)rrrrtr0r0r1_eval_rewrite_as_hankel2szhn2._eval_rewrite_as_hankel2N) r`rarbrcrrfrrrr0r0r0r1rP}s0rPsympyc sddlm}dkrTddlmddlm}||fddtd|dDSd krdd lmzdd l m fd d }Wqt k rddl m fdd }YqXnt dfdd}|}|||}|g} t|dD]} ||||}| |q| S)a Zeros of the spherical Bessel function of the first kind. Explanation =========== This returns an array of zeros of $jn$ up to the $k$-th zero. * method = "sympy": uses `mpmath.besseljzero `_ * method = "scipy": uses the `SciPy's sph_jn `_ and `newton `_ to find all roots, which is faster than computing the zeros using a general numerical solver, but it requires SciPy and only works with low precision floating point numbers. (The function used with method="sympy" is a recent addition to mpmath; before that a general solver was used.) Examples ======== >>> from sympy import jn_zeros >>> jn_zeros(2, 4, dps=5) [5.7635, 9.095, 12.323, 15.515] See Also ======== jn, yn, besselj, besselk, bessely Parameters ========== n : integer order of Bessel function k : integer number of zeros to return r)rr) besseljzero) dps_to_preccs0g|](}ttdt|qS)g?)r _from_mpmathr _to_mpmathint).0l)rrrrr0r1 s zjn_zeros..r3scipy)newton) spherical_jncs |Sr5r0rU)rrrr0r1zjn_zeros..)sph_jncs|ddS)Nrrr0r)rrrr0r1rrUnknown method.cs dkr||}ntd|S)Nrrr)r\rUr)methodrr0r1solvers zjn_zeros..solver)mathrmpmathrZmpmath.libmp.libmpfrrZscipy.optimizerZ scipy.specialr ImportErrorrr|r) rrrrZdpsZmath_pirr\rrrootsir0)rrrrrrrrr1jn_zeross2-         rc@s4eZdZdZddZddZd ddZd d d Zd S)AiryBasezg Abstract base class for Airy functions. This class is meant to reduce code duplication. cCs||jdSNr)funcr-rEr.r0r0r1rGszAiryBase._eval_conjugatecCs |jdjSr)r-rr.r0r0r1rszAiryBase._eval_is_extended_realTcKsL|jd}|}|j}||||d}t||||d}||fS)Nrr;)r-rErr)r/deepr[r9Zzcr\uvr0r0r1 as_real_imags  zAiryBase.as_real_imagcKs$|jfd|i|\}}||tS)Nr)rr)r/rr[Zre_partZim_partr0r0r1_eval_expand_complexszAiryBase._eval_expand_complexN)T)T)r`rarbrcrGrrrr0r0r0r1r s  rc@s^eZdZdZdZdZeddZdddZe e dd Z d d Z d d Z ddZddZdS)airyaia The Airy function $\operatorname{Ai}$ of the first kind. Explanation =========== The Airy function $\operatorname{Ai}(z)$ is defined to be the function satisfying Airy's differential equation .. math:: \frac{\mathrm{d}^2 w(z)}{\mathrm{d}z^2} - z w(z) = 0. Equivalently, for real $z$ .. math:: \operatorname{Ai}(z) := \frac{1}{\pi} \int_0^\infty \cos\left(\frac{t^3}{3} + z t\right) \mathrm{d}t. Examples ======== Create an Airy function object: >>> from sympy import airyai >>> from sympy.abc import z >>> airyai(z) airyai(z) Several special values are known: >>> airyai(0) 3**(1/3)/(3*gamma(2/3)) >>> from sympy import oo >>> airyai(oo) 0 >>> airyai(-oo) 0 The Airy function obeys the mirror symmetry: >>> from sympy import conjugate >>> conjugate(airyai(z)) airyai(conjugate(z)) Differentiation with respect to $z$ is supported: >>> from sympy import diff >>> diff(airyai(z), z) airyaiprime(z) >>> diff(airyai(z), z, 2) z*airyai(z) Series expansion is also supported: >>> from sympy import series >>> series(airyai(z), z, 0, 3) 3**(5/6)*gamma(1/3)/(6*pi) - 3**(1/6)*z*gamma(2/3)/(2*pi) + O(z**3) We can numerically evaluate the Airy function to arbitrary precision on the whole complex plane: >>> airyai(-2).evalf(50) 0.22740742820168557599192443603787379946077222541710 Rewrite $\operatorname{Ai}(z)$ in terms of hypergeometric functions: >>> from sympy import hyper >>> airyai(z).rewrite(hyper) -3**(2/3)*z*hyper((), (4/3,), z**3/9)/(3*gamma(1/3)) + 3**(1/3)*hyper((), (2/3,), z**3/9)/(3*gamma(2/3)) See Also ======== airybi: Airy function of the second kind. airyaiprime: Derivative of the Airy function of the first kind. airybiprime: Derivative of the Airy function of the second kind. References ========== .. [1] https://en.wikipedia.org/wiki/Airy_function .. [2] https://dlmf.nist.gov/9 .. [3] https://encyclopediaofmath.org/wiki/Airy_functions .. [4] https://mathworld.wolfram.com/AiryFunctions.html r3TcCs|jr^|tjkrtjS|tjkr&tjS|tjkr6tjS|jr^tjdtddt tddS|jrtjdtddt tddSdS)Nrr;) is_NumberrrjrkrgrlrSrfr r$r7rr0r0r1r:s   "z airyai.evalcCs$|dkrt|jdSt||dSNr3r) airyaiprimer-r r@r0r0r1rBsz airyai.fdiffcGs:|dkrtjSt|}t|dkr|d}td|| td||dtt|tddtddt|t |dtddtt|tddtddt|dt |dtdd|Stj dtddtt |tj tdttddt|tj t|td||SdS)Nrr3rrr;rz) rrgrlenrrrr rr$rfrrrUZprevious_termsrr0r0r1 taylor_terms" L@Hzairyai.taylor_termcKs`tdd}tdd}t| tdd}t|jr\|t| t| ||t|||SdSNr3rr;r rr rZrrMr/r9r^otttrVr0r0r1rs    zairyai._eval_rewrite_as_besseljcKstdd}tdd}t|tdd}t|jrX|t|t| ||t|||S|t||t| |||t|| t|||SdSrr rr rXrrNrr0r0r1rus    *zairyai._eval_rewrite_as_besselicKs~tjdtddttdd}|tddttdd}|tgtddg|dd|tgtddg|ddS)Nrr;r3 rz)rrfr r$rr'r/r9r^Zpf1Zpf2r0r0r1_eval_rewrite_as_hypers"zairyai._eval_rewrite_as_hyperc Ks |jd}|j}t|dkr|}td|gd}td|gd}td|gd}td|gd}||||||} | dk r| |}d|jr| |}| |}| |}|||||||||} ||||||} tj| tj t | | tj t dt | SdS Nrr3r)excludedmrrr) r- free_symbolsrpoprmatchrKrrxrfrrairybi r/r[rZsymbsr9rrrrrMpfZnewargr0r0r1rYs$   $zairyai._eval_expand_funcN)r3r`rarbrcnargs unbranchedrer:rB staticmethodrrrrurrYr0r0r0r1r$sX   rc@s^eZdZdZdZdZeddZdddZe e dd Z d d Z d d Z ddZddZdS)r a The Airy function $\operatorname{Bi}$ of the second kind. Explanation =========== The Airy function $\operatorname{Bi}(z)$ is defined to be the function satisfying Airy's differential equation .. math:: \frac{\mathrm{d}^2 w(z)}{\mathrm{d}z^2} - z w(z) = 0. Equivalently, for real $z$ .. math:: \operatorname{Bi}(z) := \frac{1}{\pi} \int_0^\infty \exp\left(-\frac{t^3}{3} + z t\right) + \sin\left(\frac{t^3}{3} + z t\right) \mathrm{d}t. Examples ======== Create an Airy function object: >>> from sympy import airybi >>> from sympy.abc import z >>> airybi(z) airybi(z) Several special values are known: >>> airybi(0) 3**(5/6)/(3*gamma(2/3)) >>> from sympy import oo >>> airybi(oo) oo >>> airybi(-oo) 0 The Airy function obeys the mirror symmetry: >>> from sympy import conjugate >>> conjugate(airybi(z)) airybi(conjugate(z)) Differentiation with respect to $z$ is supported: >>> from sympy import diff >>> diff(airybi(z), z) airybiprime(z) >>> diff(airybi(z), z, 2) z*airybi(z) Series expansion is also supported: >>> from sympy import series >>> series(airybi(z), z, 0, 3) 3**(1/3)*gamma(1/3)/(2*pi) + 3**(2/3)*z*gamma(2/3)/(2*pi) + O(z**3) We can numerically evaluate the Airy function to arbitrary precision on the whole complex plane: >>> airybi(-2).evalf(50) -0.41230258795639848808323405461146104203453483447240 Rewrite $\operatorname{Bi}(z)$ in terms of hypergeometric functions: >>> from sympy import hyper >>> airybi(z).rewrite(hyper) 3**(1/6)*z*hyper((), (4/3,), z**3/9)/gamma(1/3) + 3**(5/6)*hyper((), (2/3,), z**3/9)/(3*gamma(2/3)) See Also ======== airyai: Airy function of the first kind. airyaiprime: Derivative of the Airy function of the first kind. airybiprime: Derivative of the Airy function of the second kind. References ========== .. [1] https://en.wikipedia.org/wiki/Airy_function .. [2] https://dlmf.nist.gov/9 .. [3] https://encyclopediaofmath.org/wiki/Airy_functions .. [4] https://mathworld.wolfram.com/AiryFunctions.html r3TcCs|jr^|tjkrtjS|tjkr&tjS|tjkr6tjS|jr^tjdtddt tddS|jrtjdtddt tddSdS)Nrr3r;) rrrjrkrlrgrSrfr r$rr0r0r1r:.s   "z airybi.evalcCs$|dkrt|jdSt||dSr) airybiprimer-r r@r0r0r1rB=sz airybi.fdiffcGs|dkrtjSt|}t|dkr|d}td|tttddt|tj t |tj td|tj tt tddt|tj t |dtd|Stj t ddtt|tj tdtttddt|tj t |td||SdS)Nrr3rrr;r)rrgrrrrrr rrfrrrxrr$rr0r0r1rCs @<Hzairybi.taylor_termcKs`tdd}tdd}t| tdd}t|jr\t| dt| ||t|||SdSrrrr0r0r1rRs    zairybi._eval_rewrite_as_besseljcKstdd}tdd}t|tdd}t|jr\t|tdt| ||t|||St||}t|| }t||t| ||||t|||SdSrrr/r9r^rrrVrrr0r0r1ruYs   .  zairybi._eval_rewrite_as_besselicKsztjtddttdd}|tddttdd}|tgtddg|dd|tgtddg|ddS)Nrrr;r3rrz)rrfrr$r r'rr0r0r1rdszairybi._eval_rewrite_as_hyperc Ks |jd}|j}t|dkr|}td|gd}td|gd}td|gd}td|gd}||||||} | dk r| |}d|jr| |}| |}| |}|||||||||} ||||||} tjt dtj | t | tj | t | SdSr) r-rrrrr rKrrxrrfrr r r0r0r1rYis$   $zairybi._eval_expand_funcN)r3rr0r0r0r1r sZ    r c@sVeZdZdZdZdZeddZdddZdd Z d d Z d d Z ddZ ddZ dS)ra% The derivative $\operatorname{Ai}^\prime$ of the Airy function of the first kind. Explanation =========== The Airy function $\operatorname{Ai}^\prime(z)$ is defined to be the function .. math:: \operatorname{Ai}^\prime(z) := \frac{\mathrm{d} \operatorname{Ai}(z)}{\mathrm{d} z}. Examples ======== Create an Airy function object: >>> from sympy import airyaiprime >>> from sympy.abc import z >>> airyaiprime(z) airyaiprime(z) Several special values are known: >>> airyaiprime(0) -3**(2/3)/(3*gamma(1/3)) >>> from sympy import oo >>> airyaiprime(oo) 0 The Airy function obeys the mirror symmetry: >>> from sympy import conjugate >>> conjugate(airyaiprime(z)) airyaiprime(conjugate(z)) Differentiation with respect to $z$ is supported: >>> from sympy import diff >>> diff(airyaiprime(z), z) z*airyai(z) >>> diff(airyaiprime(z), z, 2) z*airyaiprime(z) + airyai(z) Series expansion is also supported: >>> from sympy import series >>> series(airyaiprime(z), z, 0, 3) -3**(2/3)/(3*gamma(1/3)) + 3**(1/3)*z**2/(6*gamma(2/3)) + O(z**3) We can numerically evaluate the Airy function to arbitrary precision on the whole complex plane: >>> airyaiprime(-2).evalf(50) 0.61825902074169104140626429133247528291577794512415 Rewrite $\operatorname{Ai}^\prime(z)$ in terms of hypergeometric functions: >>> from sympy import hyper >>> airyaiprime(z).rewrite(hyper) 3**(1/3)*z**2*hyper((), (5/3,), z**3/9)/(6*gamma(2/3)) - 3**(2/3)*hyper((), (1/3,), z**3/9)/(3*gamma(1/3)) See Also ======== airyai: Airy function of the first kind. airybi: Airy function of the second kind. airybiprime: Derivative of the Airy function of the second kind. References ========== .. [1] https://en.wikipedia.org/wiki/Airy_function .. [2] https://dlmf.nist.gov/9 .. [3] https://encyclopediaofmath.org/wiki/Airy_functions .. [4] https://mathworld.wolfram.com/AiryFunctions.html r3TcCsR|jr&|tjkrtjS|tjkr&tjS|jrNtjdtddttddSdS)Nrr3) rrrjrkrgrSrnr r$rr0r0r1r:s  zairyaiprime.evalcCs.|dkr |jdt|jdSt||dSr)r-rr r@r0r0r1rBszairyaiprime.fdiffc Cs>|jd|}t|tj|dd}W5QRXt||SNrr3)Z derivative)r-rr*r)rrrr/rr9resr0r0r1rs zairyaiprime._eval_evalfcKsPtdd}t| tdd}t|jrL|dt| ||t|||SdSNr;r)r rr rZrMr/r9r^rrVr0r0r1rs  z$airyaiprime._eval_rewrite_as_besseljcKstdd}tdd}|t|tdd}t|jrP|dt||t| |St|tdd}t||}t|| }||d|t||||t| ||SdSr)r rr rXrNrr0r0r1rus     z$airyaiprime._eval_rewrite_as_besselicKs|dddtddttdd}dtddttdd}|tgtddg|dd|tgtddg|ddS)Nr;rr3r)r r$rr'rr0r0r1rs(z"airyaiprime._eval_rewrite_as_hyperc Ks |jd}|j}t|dkr|}td|gd}td|gd}td|gd}td|gd}||||||} | dk r| |}d|jr| |}| |}| |}|||||||||} ||||||} tj| tj t | | tj t dt | SdSr) r-rrrrr rKrrxrfrrrr r0r0r1rYs$   $zairyaiprime._eval_expand_funcN)r3r`rarbrcrrrer:rBrrrurrYr0r0r0r1rsQ  rc@sVeZdZdZdZdZeddZdddZdd Z d d Z d d Z ddZ ddZ dS)ra6 The derivative $\operatorname{Bi}^\prime$ of the Airy function of the first kind. Explanation =========== The Airy function $\operatorname{Bi}^\prime(z)$ is defined to be the function .. math:: \operatorname{Bi}^\prime(z) := \frac{\mathrm{d} \operatorname{Bi}(z)}{\mathrm{d} z}. Examples ======== Create an Airy function object: >>> from sympy import airybiprime >>> from sympy.abc import z >>> airybiprime(z) airybiprime(z) Several special values are known: >>> airybiprime(0) 3**(1/6)/gamma(1/3) >>> from sympy import oo >>> airybiprime(oo) oo >>> airybiprime(-oo) 0 The Airy function obeys the mirror symmetry: >>> from sympy import conjugate >>> conjugate(airybiprime(z)) airybiprime(conjugate(z)) Differentiation with respect to $z$ is supported: >>> from sympy import diff >>> diff(airybiprime(z), z) z*airybi(z) >>> diff(airybiprime(z), z, 2) z*airybiprime(z) + airybi(z) Series expansion is also supported: >>> from sympy import series >>> series(airybiprime(z), z, 0, 3) 3**(1/6)/gamma(1/3) + 3**(5/6)*z**2/(6*gamma(2/3)) + O(z**3) We can numerically evaluate the Airy function to arbitrary precision on the whole complex plane: >>> airybiprime(-2).evalf(50) 0.27879516692116952268509756941098324140300059345163 Rewrite $\operatorname{Bi}^\prime(z)$ in terms of hypergeometric functions: >>> from sympy import hyper >>> airybiprime(z).rewrite(hyper) 3**(5/6)*z**2*hyper((), (5/3,), z**3/9)/(6*gamma(2/3)) + 3**(1/6)*hyper((), (1/3,), z**3/9)/gamma(1/3) See Also ======== airyai: Airy function of the first kind. airybi: Airy function of the second kind. airyaiprime: Derivative of the Airy function of the first kind. References ========== .. [1] https://en.wikipedia.org/wiki/Airy_function .. [2] https://dlmf.nist.gov/9 .. [3] https://encyclopediaofmath.org/wiki/Airy_functions .. [4] https://mathworld.wolfram.com/AiryFunctions.html r3TcCs~|jrX|tjkrtjS|tjkr&tjS|tjkr6tjS|jrXdtddttddS|jrzdtddttddSdS)Nrr3r) rrrjrkrlrgrSr r$rr0r0r1r:us   zairybiprime.evalcCs.|dkr |jdt|jdSt||dSr)r-r r r@r0r0r1rBszairybiprime.fdiffc Cs>|jd|}t|tj|dd}W5QRXt||Sr)r-rr*r)r rrrr0r0r1rs zairybiprime._eval_evalfcKsRtdd}|t| tdd}t|jrN| tdt| |t||SdSrrrr0r0r1rs  z$airybiprime._eval_rewrite_as_besseljcKstdd}tdd}|t|tdd}t|jrT|tdt| |t||St|tdd}t||}t|| }t||t| |||d|t|||SdSrrrr0r0r1rus   "  z$airybiprime._eval_rewrite_as_besselicKs||ddtddttdd}tddttdd}|tgtddg|dd|tgtddg|ddS)Nr;rrr3rr)rr$r r'rr0r0r1rs$z"airybiprime._eval_rewrite_as_hyperc Ks |jd}|j}t|dkr|}td|gd}td|gd}td|gd}td|gd}||||||} | dk r| |}d|jr| |}| |}| |}|||||||||} ||||||} tjt d| tj t | | tj t | SdSr) r-rrrrr rKrrxrrfrrr r0r0r1rYs$   $zairybiprime._eval_expand_funcN)r3rr0r0r0r1rsS   rc@sFeZdZdZeddZdddZddZd d Zd d Z d dZ dS)marcumqa The Marcum Q-function. Explanation =========== The Marcum Q-function is defined by the meromorphic continuation of .. math:: Q_m(a, b) = a^{- m + 1} \int_{b}^{\infty} x^{m} e^{- \frac{a^{2}}{2} - \frac{x^{2}}{2}} I_{m - 1}\left(a x\right)\, dx Examples ======== >>> from sympy import marcumq >>> from sympy.abc import m, a, b >>> marcumq(m, a, b) marcumq(m, a, b) Special values: >>> marcumq(m, 0, b) uppergamma(m, b**2/2)/gamma(m) >>> marcumq(0, 0, 0) 0 >>> marcumq(0, a, 0) 1 - exp(-a**2/2) >>> marcumq(1, a, a) 1/2 + exp(-a**2)*besseli(0, a**2)/2 >>> marcumq(2, a, a) 1/2 + exp(-a**2)*besseli(0, a**2)/2 + exp(-a**2)*besseli(1, a**2) Differentiation with respect to $a$ and $b$ is supported: >>> from sympy import diff >>> diff(marcumq(m, a, b), a) a*(-marcumq(m, a, b) + marcumq(m + 1, a, b)) >>> diff(marcumq(m, a, b), b) -a**(1 - m)*b**m*exp(-a**2/2 - b**2/2)*besseli(m - 1, a*b) References ========== .. [1] https://en.wikipedia.org/wiki/Marcum_Q-function .. [2] https://mathworld.wolfram.com/MarcumQ-Function.html cCsZ|tjkr@|tjkr$|tjkr$tjSt||dtjt|S|tjkrn|tjkrnddt|dtjS||kr|tjkrdt|d td|dtjS|dkrtjtjt|d td|dt|d td|dS|jr,|jr|jrtjSt||dtjt|S|jrV|jrVddt|dtjSdSr) rrgr&rxr$rrfrNrS)r7rrVrr0r0r1r:s"  &Dz marcumq.evalr;cCs|j\}}}|dkr6|t||| td|||S|dkr|| ||dt|d|d dt|d||St||dS)Nr;r3r)r-rrrNr )r/rArrVrr0r0r1rB s  "Bz marcumq.fdiffcKsjddlm}|dtd}|d||||t|d|d dt|d||||tjgS)Nr)IntegralrUr3r;)Zsympy.integrals.integralsrgetrrrNrrk)r/rrVrr^rrUr0r0r1_eval_rewrite_as_Integrals   @z!marcumq._eval_rewrite_as_IntegralcKsbddlm}|dtd}t|d|d d||||t||||d|tjgS)Nr)Sumrr;r3)Zsympy.concrete.summationsr rrrrNrrk)r/rrVrr^r rr0r0r1_eval_rewrite_as_Sums zmarcumq._eval_rewrite_as_Sumc s|kr|dkr4dtd tdddS|jr|dkrtfddtd|D}tjtd tdddtd |SdS)Nr3r;rcsg|]}t|dqS)r;)rN)rrrVr0r1r$sz4marcumq._eval_rewrite_as_besseli..)rrNrsumrrrx)r/rrVrr^rr0r"r1rus $z marcumq._eval_rewrite_as_besselicCstdd|jDrdSdS)Ncss|] }|jVqdSr5)rS)rrr0r0r1 (sz(marcumq._eval_is_zero..T)allr-r.r0r0r1 _eval_is_zero'szmarcumq._eval_is_zeroN)r;) r`rarbrcrer:rBrr!rur&r0r0r0r1rs0  rN)rr)T functoolsrZ sympy.corerZsympy.core.addrZsympy.core.cacherZsympy.core.exprrZsympy.core.functionrr r Zsympy.core.logicr r Zsympy.core.numbersr rrZsympy.core.powerrZsympy.core.symbolrrZsympy.core.sympifyrZ(sympy.functions.combinatorial.factorialsrZ(sympy.functions.elementary.trigonometricrrrrZ#sympy.functions.elementary.integersrZ&sympy.functions.elementary.exponentialrrZ(sympy.functions.elementary.miscellaneousrrrZ$sympy.functions.elementary.complexesrr r!r"r#Z'sympy.functions.special.gamma_functionsr$r%r&Zsympy.functions.special.hyperr'Zsympy.polys.orthopolysr(rr)r*r+rMrvrNrrrrrrrrQrRrrOrPrrrr rrrr0r0r0r1sf            F1./XB988 T-2%