U ?h%ã@sTddlmZddlZdgZdddddd d d d d ddddddddddgZddd„ZdS)é)ÚdivisionNÚ demo_signalZBlocksÚBumpsZ HeaviSineZDopplerZRampZHiSineZLoSineZLinChirpZTwoChirpZ QuadChirpZMishMashZ WernerSorrowsZ HypChirpsZ LinChirpsZChirpsZGaborÚ sineoneoverxz Piece-RegularzPiece-PolynomialZRiemannc0 Csp| ¡dkrtS|dk rR|dks,|ddkr4tdƒ‚t d|dd|d|¡}ddg}| ¡}||kr‚|dk r‚td |¡ƒ‚n|dkr ||kr td  |¡ƒ‚|d krd d d ddddddddg }dddddddddddg }d}t||ƒD]&\}}||dt ||¡d7}qì nT|d kr´d d d ddddddddg }dddddd!ddd"d#d!g }d$d$d%d&d&d'd&d&d$d(d$g } d}t||| ƒD].\}}} ||dt ||| ¡d7}q€ n¸|d)krödt  dtj |¡t |d*¡t d+|¡} nv|d,kr2t  |d|¡t  dtj d-|d.¡} n:|d/krL||d0k} n |d1krrt  tj |d2|¡} nú|d3kr˜t  tj |d4|¡} nÔ|d5krÂt  tj ||d6|¡} nª|d7krt  tj |||¡t  tj d|||¡} nh|d8kr2t  tj d|||d¡} n:|d9krœt  tj d|||d¡}|t  tj |d2|¡7}|t  tj ||d:|¡7} nÐ|d;krŒt  tj ||d|d¡}|t  tj |d2|¡}|t  tj |||¡}d d d ddddddddg } dddddd!ddd"d#d!g } d$d$d%d&d&d'd&d&d$d(d$g } t| | | ƒD].\}}} ||dt ||| ¡d7}qX nà|d}d|tj d>}t d?|d@d¡|}t  |¡}t  |¡}t  |dA|¡d |k|dBk}t  |dA|¡d |k|dCk}t t d|¡ƒ}|d}t |¡}t d|d¡t |¡}dt  tj d||dtj ¡d|d|…<|d|…dddD…||||…<t  |¡}|d|…|t t |dE¡ƒd|t t |dE¡ƒd…<|||}nX|dFkr¤dG|tj d>}dH|tj d>}t d|d¡|}t  |d|d|¡}|t ||d¡t ||||d¡}nÈ|dIkr”t d|d¡|dEtj }t |d|d>¡}dJ|d>}t d|d¡|tj }t ||d¡}|dddD…}t | |d¡|dK}t |d d|d>¡}t|d|d|ƒ}t|dL|dL|ƒ}t d|d¡|}||t dMtj ||d>¡}||t dNtj ||d>¡} |||| }t |¡}t d|dLd¡t |dL¡}dt  tj d||dLdtj ¡d|d|dL…<|d|dL…dddD…|dO|dL|…<||}nØ|dk rtdP}t | |d¡d|}t d|d¡|}t |d dK¡}td|dd|d|ƒ}t|d|d|ƒ}d||t dQ|dRtj |¡}d||t dQ|dtj |¡}||}nø|dk rÒd>}tj| d|dtdS}d&||dk<||d}t  dT|¡}|dPdU…}nš|dVk r2t  |¡}t t |dW¡ƒ}!t t |dO¡ƒ}"t t |d¡ƒ}#t t |d¡ƒ}$t t |d¡ƒ}%t t |dK¡ƒ}&dXtd |ƒ}t d|!d¡|!}t d|¡ }t d|"d¡|"}t d|¡t d¡}'t d|$d¡|$}d }(dYt |d6|d6 d|(d¡})|)d|"…|d|"…<d6|)|"|#…||"|#…<|)|#|$…||#|$…<||$|%…||$|%…<|||%|%|!…<|||%d|!d|%|!ddD…<t |%d|!d|&|%d|!|&¡ dZ||%d|!|&|%d|!d|&…<|%d|!d|&}*|'||*|*|"…<|d|#}+||+dddD…|d|#|…<t |¡|},|,|}n:|d[k rìt  |¡}t t |d¡ƒ}#t t |dE¡ƒ}-t t |dK¡ƒ}&t d|#d¡|#}dK|d|dd}d\d|d|dG}dE|dd]}dR|ddL|dR} dK|d}'t |-¡dK})||d|#…<||d|#d|#ddD…<||d|#d|#…<| |d|#d|#…<|'|#ddD…|d|#d|#…<|d|#}+||+dddD…|d|#|…<t |-¡dE||&|&|-…<t |&¡d^|||-||&|-…<t |¡|},||,}n€|d_kr\t t t  |¡¡ƒ}.t d|.d¡}/|/|/9}/t |¡}d`t d|.d¡||/d<t tj |¡¡}ntda |t¡ƒ‚|S)buLSimple 1D wavelet test functions. This function can generate a number of common 1D test signals used in papers by David Donoho and colleagues (e.g. [1]_) as well as the wavelet book by Stéphane Mallat [2]_. Parameters ---------- name : {'Blocks', 'Bumps', 'HeaviSine', 'Doppler', ...} The type of test signal to generate (`name` is case-insensitive). If `name` is set to `'list'`, a list of the available test functions is returned. n : int or None The length of the test signal. This should be provided for all test signals except `'Gabor'` and `'sineoneoverx'` which have a fixed length. Returns ------- f : np.ndarray Array of length ``n`` corresponding to the specified test signal type. References ---------- .. [1] D.L. Donoho and I.M. Johnstone. Ideal spatial adaptation by wavelet shrinkage. Biometrika, vol. 81, pp. 425–455, 1994. .. [2] S. Mallat. A Wavelet Tour of Signal Processing: The Sparse Way. Academic Press. 2009. Notes ----- This function is a partial reimplementation of the `MakeSignal` function from the [Wavelab](https://statweb.stanford.edu/~wavelab/) toolbox. These test signals are provided with permission of Dr. Donoho to encourage reproducible research. ÚlistNérzn must be an integer >= 1Zgaborrz/Parameter n must be set to None when name is {}z,Parameter n must be provided when name is {}Úblocksgš™™™™™¹?g¤p= ×£À?g333333Ã?gq= ×£pÍ?gÐ?gš™™™™™Ù?g)\Âõ(Ü?gÍÌÌÌÌÌä?gR¸…ëQè?gö(\Âõè?gìQ¸…ëé?ééûÿÿÿééüÿÿÿégÍÌÌÌÌÌÀgÍÌÌÌÌÌ@g333333@gÍÌÌÌÌÌÀéZbumpsgÍÌÌÌÌÌ@gÍÌÌÌÌÌ@gffffff@g{®Gázt?gú~j¼t“x?g{®Gáz„?g¸…ëQ¸ž?gü©ñÒMb€?Z heavisineg333333Ó?g ×£p= ç?ZdopplergÍÌÌÌÌÌð?gš™™™™™©?Zrampg®Gáz®×?ZhisinegÙ_vOæ?Zlosinegioð…ÉTÕ?Zlinchirpgà?ZtwochirpZ quadchirpZmishmashgÀ?Z wernersorrowsZ hypchirpséigj¼t“ð?gü©ñÒMbP?gš™™™™™é?gÃõ(\Âå?gè?éÿÿÿÿé Z linchirpsédéúZchirpséééé2i^éiyð?é)Zdtypegø?iz piece-regularé iñÿÿÿiºÿÿÿézpiece-polynomialé(é-é–Zriemanngð?z*unknown name: {}. name must be one of: {})ÚlowerÚ_implemented_signalsÚ ValueErrorÚnpZarangeÚformatÚzipÚsignÚabsÚsinÚpiÚsqrtZzerosÚintÚroundZonesÚceilÚcosÚexpÚsliceÚfloatZfixrÚsumZ zeros_likeÚrealZfftZifft)0ÚnameÚnÚtZ n_hard_codedZt0sÚhsÚfÚt0ÚhÚwsÚwÚposZhgtZwthÚpÚalphaÚbetaÚf1Úf2ÚmÚenvelopeÚtmpÚenvÚbÚaÚA1ZixÚgÚi1Úi2ÚjZf3Zf4Zn_12Zn_7Zn_5Zn_3Zn_2Zn_20Zf5ÚfmaZf6ÚkÚdiffZbiasZn_10ZsqnÚidx©rQúL/var/www/html/venv/lib/python3.8/site-packages/pywt/data/_wavelab_signals.pyrs~& ÿÿ $ * 8 2      8 $  $  *   "" 4"  ÿ2 4  $$ "*ÿ*  ((      (""ÿÿ(       "   ÿÿ)rN)Ú __future__rÚnumpyr"Ú__all__r rrQrQrQrRÚs0 ì