U ZʹjY@sdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZdd lmZmZmZdd lmZdd lmZdd lmZddlmZddgiZddZddZ ddZ!ddZ"GdddeZ#Gddde#Z$d!dd Z%dS)"zFourier Series)oopi)Wild)Expr)Add)Tuple)S)DummySymbol)sympify)sincossinc) SeriesBase) SeqFormula)Interval) is_sequence)fourier_seriesZ matplotlibc Csddlm}|d|d|d}}td|t||}d||||||}||tjd}|td|||||||dtffS)z,Returns the cos sequence in a Fourier seriesr integrate) sympy.integralsrr rsubsrZerorr) funclimitsnrxLZcos_termZformulaa0r!S/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/series/fourier.pyfourier_cos_seqs r#cCsdddlm}|d|d|d}}td|t||}td|||||||dtfS)z,Returns the sin sequence in a Fourier seriesrrrr)rrr rrr)rrrrrrZsin_termr!r!r"fourier_sin_seq s r$cCsdd}d\}}}|dkr0||t t}}}t|trnt|dkrR|\}}}nt|dkrn||}|\}}t|tr|dks|dkrtdt|tj tj g}||ks||krtdt |||fS) a Limits should be of the form (x, start, stop). x should be a symbol. Both start and stop should be bounded. Explanation =========== * If x is not given, x is determined from func. * If limits is None. Limit of the form (x, -pi, pi) is returned. Examples ======== >>> from sympy.series.fourier import _process_limits as pari >>> from sympy.abc import x >>> pari(x**2, (x, -2, 2)) (x, -2, 2) >>> pari(x**2, (-2, 2)) (x, -2, 2) >>> pari(x**2, None) (x, -pi, pi) cSs6|j}t|dkr|S|s&tdStd|dS)Nrkz specify dummy variables for %s. If the function contains more than one free symbol, a dummy variable should be supplied explicitly e.g. FourierSeries(m*n**2, (n, -pi, pi))) free_symbolslenpopr ValueError)rfreer!r!r"_find_x@s z _process_limits.._find_x)NNNNrzInvalid limits given: %sz.Both the start and end value should be bounded) rrrr' isinstancer r)strrNegativeInfinityInfinityr )rrr+rstartstopZ unboundedr!r!r"_process_limits)s       r3c sdd}fdd}ddlm}m}m}||||}|} tddd d d gd td fd d gd | dD]@} | d} | D]*} || s|| |sd|fSqqxd|fS)NcSs ||jkSNr&)exprsrr!r!r"check_fxcszfinite_check..check_fxcsBt|ttfr>|jd}|t||dk r:dSdSdS)NrTF)r-r r argsmatchr)_exprrrZ sincos_args)abr!r" check_sincosfs  z"finite_check..check_sincosr)TR2TR1 sincos_to_sumr;cSs|jSr4Z is_Integerr%r!r!r"szfinite_check..cSs |tjkSr4rrrBr!r!r"rCsrD propertiesr<cs |jkSr4r5rBrr!r"rCtrDrFT)sympy.simplify.fur>r?r@ as_coeff_addrZ as_coeff_mul) frrr7r=r>r?r@r:Z add_coeffsZ mul_coeffstr!)r;r<rr" finite_checkas   rNc@seZdZdZddZeddZeddZedd Zed d Z ed d Z eddZ eddZ eddZ eddZeddZeddZddZd5ddZd6dd Zd!d"Zd#d$Zd%d&Zd'd(Zd7d+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd)S)8 FourierSeriesa9Represents Fourier sine/cosine series. Explanation =========== This class only represents a fourier series. No computation is performed. For how to compute Fourier series, see the :func:`fourier_series` docstring. See Also ======== sympy.series.fourier.fourier_series cGstt|}tj|f|Sr4)mapr r__new__)clsr8r!r!r"rQs zFourierSeries.__new__cCs |jdSNrr8selfr!r!r"functionszFourierSeries.functioncCs|jddSNrrrTrUr!r!r"rszFourierSeries.xcCs|jdd|jddfS)NrrrTrUr!r!r"periodszFourierSeries.periodcCs|jddS)NrrrTrUr!r!r"r szFourierSeries.a0cCs|jddS)NrrrTrUr!r!r"anszFourierSeries.ancCs|jddS)NrrTrUr!r!r"bnszFourierSeries.bncCs tdtSrS)rrrUr!r!r"intervalszFourierSeries.intervalcCs|jjSr4)r\infrUr!r!r"r1szFourierSeries.startcCs|jjSr4)r\suprUr!r!r"r2szFourierSeries.stopcCstSr4)rrUr!r!r"lengthszFourierSeries.lengthcCst|jd|jddS)Nrrr)absrYrUr!r!r"rszFourierSeries.LcCs|j}||r|SdSr4)rhas)rVoldnewrr!r!r" _eval_subss zFourierSeries._eval_subsr,cCsJ|dkrt|Sg}|D](}t||kr,qB|tjk r||qt|S)a Return the first n nonzero terms of the series. If ``n`` is None return an iterator. Parameters ========== n : int or None Amount of non-zero terms in approximation or None. Returns ======= Expr or iterator : Approximation of function expanded into Fourier series. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x, (x, -pi, pi)) >>> s.truncate(4) 2*sin(x) - sin(2*x) + 2*sin(3*x)/3 - sin(4*x)/2 See Also ======== sympy.series.fourier.FourierSeries.sigma_approximation N)iterr'rrappendr)rVrtermsrMr!r!r"truncates    zFourierSeries.truncatecs&fddt|dD}t|S)a Return :math:`\sigma`-approximation of Fourier series with respect to order n. Explanation =========== Sigma approximation adjusts a Fourier summation to eliminate the Gibbs phenomenon which would otherwise occur at discontinuities. A sigma-approximated summation for a Fourier series of a T-periodical function can be written as .. math:: s(\theta) = \frac{1}{2} a_0 + \sum _{k=1}^{m-1} \operatorname{sinc} \Bigl( \frac{k}{m} \Bigr) \cdot \left[ a_k \cos \Bigl( \frac{2\pi k}{T} \theta \Bigr) + b_k \sin \Bigl( \frac{2\pi k}{T} \theta \Bigr) \right], where :math:`a_0, a_k, b_k, k=1,\ldots,{m-1}` are standard Fourier series coefficients and :math:`\operatorname{sinc} \Bigl( \frac{k}{m} \Bigr)` is a Lanczos :math:`\sigma` factor (expressed in terms of normalized :math:`\operatorname{sinc}` function). Parameters ========== n : int Highest order of the terms taken into account in approximation. Returns ======= Expr : Sigma approximation of function expanded into Fourier series. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x, (x, -pi, pi)) >>> s.sigma_approximation(4) 2*sin(x)*sinc(pi/4) - 2*sin(2*x)/pi + 2*sin(3*x)*sinc(3*pi/4)/3 See Also ======== sympy.series.fourier.FourierSeries.truncate Notes ===== The behaviour of :meth:`~sympy.series.fourier.FourierSeries.sigma_approximation` is different from :meth:`~sympy.series.fourier.FourierSeries.truncate` - it takes all nonzero terms of degree smaller than n, rather than first n nonzero ones. References ========== .. [1] https://en.wikipedia.org/wiki/Gibbs_phenomenon .. [2] https://en.wikipedia.org/wiki/Sigma_approximation cs.g|]&\}}|tjk rtt||qSr!)rrrr).0irMrr!r" 3s z5FourierSeries.sigma_approximation..N) enumerater)rVrrgr!rkr"sigma_approximationsBz!FourierSeries.sigma_approximationcCs\t||j}}||jkr*td||f|j|}|j|}|||jd||j|j fS)a Shift the function by a term independent of x. Explanation =========== f(x) -> f(x) + s This is fast, if Fourier series of f(x) is already computed. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x**2, (x, -pi, pi)) >>> s.shift(1).truncate() -4*cos(x) + cos(2*x) + 1 + pi**2/3 '%s' should be independent of %sr) r rr&r)r rWrr8rZr[)rVrLrr sfuncr!r!r"shift7s    zFourierSeries.shiftcCs|t||j}}||jkr*td||f|j|||}|j|||}|j|||}|||j d|j ||fS)a Shift x by a term independent of x. Explanation =========== f(x) -> f(x + s) This is fast, if Fourier series of f(x) is already computed. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x**2, (x, -pi, pi)) >>> s.shiftx(1).truncate() -4*cos(x + 1) + cos(2*x + 2) + pi**2/3 ror r rr&r)rZrr[rWrr8r rVrLrrZr[rpr!r!r"shiftxVs zFourierSeries.shiftxcCstt||j}}||jkr*td||f|j|}|j|}|j|}|jd|}| ||jd|||fS)a Scale the function by a term independent of x. Explanation =========== f(x) -> s * f(x) This is fast, if Fourier series of f(x) is already computed. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x**2, (x, -pi, pi)) >>> s.scale(2).truncate() -8*cos(x) + 2*cos(2*x) + 2*pi**2/3 rorr) r rr&r)rZZ coeff_mulr[r r8r)rVrLrrZr[r rpr!r!r"scalevs    zFourierSeries.scalecCs|t||j}}||jkr*td||f|j|||}|j|||}|j|||}|||j d|j ||fS)a Scale x by a term independent of x. Explanation =========== f(x) -> f(s*x) This is fast, if Fourier series of f(x) is already computed. Examples ======== >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> s = fourier_series(x**2, (x, -pi, pi)) >>> s.scalex(2).truncate() -4*cos(2*x) + cos(4*x) + pi**2/3 rorrrrsr!r!r"scalexs zFourierSeries.scalexNrcCs |D]}|tjk r|SqdSr4rE)rVrZlogxcdirrMr!r!r"_eval_as_leading_terms z#FourierSeries._eval_as_leading_termcCs&|dkr|jS|j||j|SrS)r rZZcoeffr[)rVptr!r!r" _eval_termszFourierSeries._eval_termcCs |dS)N)rurUr!r!r"__neg__szFourierSeries.__neg__cCst|tr|j|jkrtd|j|j}}|j|j||}|j|jkrP|S|j|j}|j |j }|j |j }| ||j d|||fSt ||S)N(Both the series should have same periodsr)r-rOrYr)rrWrr&rZr[r rr8r)rVotherryrWrZr[r r!r!r"__add__s      zFourierSeries.__add__cCs || Sr4)r)rVr~r!r!r"__sub__szFourierSeries.__sub__)r,)r,)Nr)__name__ __module__ __qualname____doc__rQpropertyrWrrYr rZr[r\r1r2r_rrdrhrnrqrtrurvrxrzr|rrr!r!r!r"rOsH            , F ! rOc@sXeZdZdZddZeddZeddZdd Zd d Z d d Z ddZ ddZ dS)FiniteFourierSeriesaRepresents Finite Fourier sine/cosine series. For how to compute Fourier series, see the :func:`fourier_series` docstring. Parameters ========== f : Expr Expression for finding fourier_series limits : ( x, start, stop) x is the independent variable for the expression f (start, stop) is the period of the fourier series exprs: (a0, an, bn) or Expr a0 is the constant term a0 of the fourier series an is a dictionary of coefficients of cos terms an[k] = coefficient of cos(pi*(k/L)*x) bn is a dictionary of coefficients of sin terms bn[k] = coefficient of sin(pi*(k/L)*x) or exprs can be an expression to be converted to fourier form Methods ======= This class is an extension of FourierSeries class. Please refer to sympy.series.fourier.FourierSeries for further information. See Also ======== sympy.series.fourier.FourierSeries sympy.series.fourier.fourier_series cst|}t|}t|}t|tr0t|dks|\}}ddlm|tfdd|D}|jddddd\}}|dt |d|d d} t d d d d d gd} t dfdd gd} i} i} |D]}| | t | t | }| | t| t | }|r@|| | || tj| || <q|rj|| | || tj| || <q||7}qt|| | }t||||S)Nr,rTR10csg|] }|qSr!r!)rirjrr!r"rl sz/FiniteFourierSeries.__new__..F)ZtrigZ power_baseZ power_explogrrr;cSs|jSr4rArBr!r!r"rCrDz-FiniteFourierSeries.__new__..cSs |tjk Sr4rErBr!r!r"rCrDrFr<cs |jkSr4r5rBrHr!r"rCrD)r r-rr'rJrIrrexpandr`rr9r rr getrrrrQ)rRrKrr6ceZrexprr Zexp_lsrr;r<rZr[prMqr!)rrr"rQs0  $$  zFiniteFourierSeries.__new__cCsB|jr dnd}|tt|jt|jd7}td|SrX)r maxsetrZkeysunionr[r)rV_lengthr!r!r"r\&s*zFiniteFourierSeries.intervalcCs |j|jSr4)r2r1rUr!r!r"r_,szFiniteFourierSeries.lengthcCsdt||j}}||jkr*td||f||||}|j|||}|||jd|SNror r rr&r)rhrrWrr8rVrLrr:rpr!r!r"rt0s  zFiniteFourierSeries.shiftxcCsTt||j}}||jkr*td||f||}|j|}|||jd|Sr)r rr&r)rhrWrr8rr!r!r"ru;s    zFiniteFourierSeries.scalecCsdt||j}}||jkr*td||f||||}|j|||}|||jd|Srrrr!r!r"rvFs  zFiniteFourierSeries.scalexcCsb|dkr|jS|j|tjt|t|j|j|j |tjt |t|j|j}|SrS) r rZrrrr rrrr[r )rVry_termr!r!r"rzQs &&zFiniteFourierSeries._eval_termcCst|tr&|t|j|jdddSt|tr|j|jkrDtd|j |j }}|j|j ||}|j |j krv|St||jddSdS)NrF)finiter})r) r-rOrrrWr8rrYr)rrr&)rVr~rrrWr!r!r"rYs    zFiniteFourierSeries.__add__N) rrrrrQrr\r_rtrurvrzrr!r!r!r"rs&$     rNTc Cs:t|}t||}|d}||jkr(|S|rdt|d|dd}t|||\}}|rdt|||Std}|d|dd}|jr ||| } || krt |||\} } t ddt f} t ||| | | fS|| kr t j} t ddt f} t|||} t ||| | | fSt |||\} } t|||} t ||| | | fS)a`Computes the Fourier trigonometric series expansion. Explanation =========== Fourier trigonometric series of $f(x)$ over the interval $(a, b)$ is defined as: .. math:: \frac{a_0}{2} + \sum_{n=1}^{\infty} (a_n \cos(\frac{2n \pi x}{L}) + b_n \sin(\frac{2n \pi x}{L})) where the coefficients are: .. math:: L = b - a .. math:: a_0 = \frac{2}{L} \int_{a}^{b}{f(x) dx} .. math:: a_n = \frac{2}{L} \int_{a}^{b}{f(x) \cos(\frac{2n \pi x}{L}) dx} .. math:: b_n = \frac{2}{L} \int_{a}^{b}{f(x) \sin(\frac{2n \pi x}{L}) dx} The condition whether the function $f(x)$ given should be periodic or not is more than necessary, because it is sufficient to consider the series to be converging to $f(x)$ only in the given interval, not throughout the whole real line. This also brings a lot of ease for the computation because you do not have to make $f(x)$ artificially periodic by wrapping it with piecewise, modulo operations, but you can shape the function to look like the desired periodic function only in the interval $(a, b)$, and the computed series will automatically become the series of the periodic version of $f(x)$. This property is illustrated in the examples section below. Parameters ========== limits : (sym, start, end), optional *sym* denotes the symbol the series is computed with respect to. *start* and *end* denotes the start and the end of the interval where the fourier series converges to the given function. Default range is specified as $-\pi$ and $\pi$. Returns ======= FourierSeries A symbolic object representing the Fourier trigonometric series. Examples ======== Computing the Fourier series of $f(x) = x^2$: >>> from sympy import fourier_series, pi >>> from sympy.abc import x >>> f = x**2 >>> s = fourier_series(f, (x, -pi, pi)) >>> s1 = s.truncate(n=3) >>> s1 -4*cos(x) + cos(2*x) + pi**2/3 Shifting of the Fourier series: >>> s.shift(1).truncate() -4*cos(x) + cos(2*x) + 1 + pi**2/3 >>> s.shiftx(1).truncate() -4*cos(x + 1) + cos(2*x + 2) + pi**2/3 Scaling of the Fourier series: >>> s.scale(2).truncate() -8*cos(x) + 2*cos(2*x) + 2*pi**2/3 >>> s.scalex(2).truncate() -4*cos(2*x) + cos(4*x) + pi**2/3 Computing the Fourier series of $f(x) = x$: This illustrates how truncating to the higher order gives better convergence. .. plot:: :context: reset :format: doctest :include-source: True >>> from sympy import fourier_series, pi, plot >>> from sympy.abc import x >>> f = x >>> s = fourier_series(f, (x, -pi, pi)) >>> s1 = s.truncate(n = 3) >>> s2 = s.truncate(n = 5) >>> s3 = s.truncate(n = 7) >>> p = plot(f, s1, s2, s3, (x, -pi, pi), show=False, legend=True) >>> p[0].line_color = (0, 0, 0) >>> p[0].label = 'x' >>> p[1].line_color = (0.7, 0.7, 0.7) >>> p[1].label = 'n=3' >>> p[2].line_color = (0.5, 0.5, 0.5) >>> p[2].label = 'n=5' >>> p[3].line_color = (0.3, 0.3, 0.3) >>> p[3].label = 'n=7' >>> p.show() This illustrates how the series converges to different sawtooth waves if the different ranges are specified. .. plot:: :context: close-figs :format: doctest :include-source: True >>> s1 = fourier_series(x, (x, -1, 1)).truncate(10) >>> s2 = fourier_series(x, (x, -pi, pi)).truncate(10) >>> s3 = fourier_series(x, (x, 0, 1)).truncate(10) >>> p = plot(x, s1, s2, s3, (x, -5, 5), show=False, legend=True) >>> p[0].line_color = (0, 0, 0) >>> p[0].label = 'x' >>> p[1].line_color = (0.7, 0.7, 0.7) >>> p[1].label = '[-1, 1]' >>> p[2].line_color = (0.5, 0.5, 0.5) >>> p[2].label = '[-pi, pi]' >>> p[3].line_color = (0.3, 0.3, 0.3) >>> p[3].label = '[0, 1]' >>> p.show() Notes ===== Computing Fourier series can be slow due to the integration required in computing an, bn. It is faster to compute Fourier series of a function by using shifting and scaling on an already computed Fourier series rather than computing again. e.g. If the Fourier series of ``x**2`` is known the Fourier series of ``x**2 - 1`` can be found by shifting by ``-1``. See Also ======== sympy.series.fourier.FourierSeries References ========== .. [1] https://mathworld.wolfram.com/FourierSeries.html rrrr)r r3r&r`rNrr is_zerorr#rrrOrrr$) rKrrrr is_finiteZres_frcenterZneg_fr rZr[r!r!r"rjs6%      r)NT)&rZsympy.core.numbersrrZsympy.core.symbolrZsympy.core.exprrZsympy.core.addrZsympy.core.containersrZsympy.core.singletonrr r Zsympy.core.sympifyr Z(sympy.functions.elementary.trigonometricr r rZsympy.series.series_classrZsympy.series.sequencesrZsympy.sets.setsrZsympy.utilities.iterablesrZ__doctest_requires__r#r$r3rNrOrrr!r!r!r"s0             8^