U `»{j> ã@s8dZddlZddlmZddlZddd„Zddd„ZdS) zBaseline estimation algorithms.éNc CsÖ|dkr d}|dkrd}|dkr$d}|d}t |¡}t t|ƒ ¡d|¡}t d||j¡}| ¡}t  ||¡} t   | ¡} t |ƒD]L} t  | |¡} t  | |¡t  |¡|kr´qÒ| }t  | |¡}t ||¡}q„|S)a Computes the baseline of a given data. Iteratively performs a polynomial fitting in the data to detect its baseline. At every iteration, the fitting weights on the regions with peaks are reduced to identify the baseline only. Parameters ---------- y : ndarray Data to detect the baseline. deg : int (default: 3) Degree of the polynomial that will estimate the data baseline. A low degree may fail to detect all the baseline present, while a high degree may make the data too oscillatory, especially at the edges. max_it : int (default: 100) Maximum number of iterations to perform. tol : float (default: 1e-3) Tolerance to use when comparing the difference between the current fit coefficients and the ones from the last iteration. The iteration procedure will stop when the difference between them is lower than *tol*. Returns ------- ndarray Array with the baseline amplitude for every original point in *y* Néédgü©ñÒMbP?égð?g)ÚnpZonesÚmathÚpowÚabsÚmaxZlinspaceÚsizeÚcopyÚvanderÚLAZpinvÚrangeÚdotZnormÚminimum) ÚyÚdegÚmax_itÚtolÚorderZcoeffsZcondÚxÚbaser Z vander_pinvÚ_Z coeffs_new©rúQ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/peakutils/baseline.pyÚbaselines*      rcCs| ¡t| ¡||||ƒS)ad Computes the upper envelope of a given data. It is implemented in terms of the `baseline` function. Parameters ---------- y : ndarray Data to detect the baseline. deg : int Degree of the polynomial that will estimate the envelope. max_it : int Maximum number of iterations to perform. tol : float Tolerance to use when comparing the difference between the current fit coefficients and the ones from the last iteration. Returns ------- ndarray Array with the envelope amplitude for every original point in *y* )r r)rrrrrrrÚenvelopeAsr)NNN)NNN) Ú__doc__ÚnumpyrZ scipy.linalgZlinalgr rrrrrrrÚs   9