o h&@s$ddlZdgZgdZdddZdS)N demo_signal)BlocksBumps HeaviSineDopplerRampHiSineLoSineLinChirpTwoChirp QuadChirpMishMash WernerSorrows HypChirps LinChirpsChirpsGabor sineoneoverxz Piece-RegularzPiece-PolynomialRiemannrc0 Cs |dkrtS|dur)|dks|ddkrtdtd|dd|d|}ddg}|}||vr@|dur@td||durO||vrOtd ||d krxgd }gd }d}t||D]\}}||dt||d 7}qb|S|dkrgd }gd}gd} d}t||| D]\}}} ||dt||| d7}q|S|dkrdtdtj |t|dtd|}|S|dkrt |d|td tj d|d}|S|dkr||dk}|S|dkrttj |d|}|S|dkrttj |d|}|S|dkr-ttj ||d|}|S|d krNttj |||ttj d!|||}|S|d"krettj d!|||d }|S|d#krttj d!|||d }|ttj |d|7}|ttj ||d$|7}|S|d%krttj ||d |d }|ttj |d|}|ttj |||}gd } gd} gd} t| | | D]\}}} ||dt||| d7}q|S|d&krd'|tj d(}d)|tj d(}td*|d+d|}t |}t |}t|d,|d-|k|d.k}t|d,|d-|k|d/k}t t d0|}|d}t|}td|dt|}dttj d ||dtj d |d|<|d|ddd1||||<t |}|d||t t|d2d|t t|d2d<|||}|S|d3krd4|tj d(}d5|tj d(}td|d|}t |d|d|}|t||d t||||d }|S|d6krtd|d|d2tj }t|d |d(}d7|d(}td|d|tj }t||d!}|ddd1}t| |d|d8}t|d  d|d(}t|d |d |}t|d9|d9|}td|d|}||td:tj ||d(}||td;tj ||d(} |||| }t|}td|d9dt|d9}dttj d ||d9dtj d |d|d9<|d|d9ddd1|d<|d9|<||}|S|dkrkd=}t| |dd)|}td|d|}t|d  d8}td |dd |d|}t|d|d|}d!||td>|d?tj |}d!||td>|dtj |}||}|S|dkrd(}tj| d|dtd@}dA||dk<||d}tdB|}|d=dC}|S|dDkrt |}t t|dE}!t t|d<}"t t|d)}#t t|d!}$t t|d }%t t|d8}&dFtd|}td|!d|!}td| }td|"d|"}td|td}'td|$d|$}dG}(dHt|d|d d |(d })|)d|"|d|"<d|)|"|#||"|#<|)|#|$||#|$<||$|%||$|%<|||%|%|!<|||%d |!d|%|!dd1<t|%d |!d!|&|%d |!|& dI||%d |!|&|%d |!d!|&<|%d |!d!|&}*|'||*|*|"<|d)|#}+||+ddd1|d)|#|<t||},|,|}|S|dJkrt |}t t|d)}#t t|d2}-t t|d8}&td|#d|#}d8|d!|d d}dKd |d!|d4}d2|d!dL}d?|d d9|d?} d8|d}'t|-d8})||d|#<||d |#d|#dd1<||d |#d!|#<| |d!|#d|#<|'|#dd1|d|#d)|#<|d)|#}+||+ddd1|d)|#|<t|-d2||&|&|-<t|&dM|||-||&|-<t||},||,}|S|dNkrt t t |}.td|.d}/|/|/9}/t|}dOtd|.d||/d<ttj|}|StdP|dQt)Ru8Simple 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. Examples -------- >>> import pywt >>> camera = pywt.data.camera() >>> doppler = pywt.data.demo_signal('doppler', 1024) >>> available_signals = pywt.data.demo_signal('list') >>> print(available_signals) ['Blocks', 'Bumps', 'HeaviSine', 'Doppler', 'Ramp', 'HiSine', 'LoSine', 'LinChirp', 'TwoChirp', 'QuadChirp', 'MishMash', 'WernerSorrows', 'HypChirps', 'LinChirps', 'Chirps', 'Gabor', 'sineoneoverx', 'Piece-Regular', 'Piece-Polynomial', 'Riemann'] listNrzn must be an integer >= 1gaborrz-Parameter n must be set to None when name is z*Parameter n must be provided when name is blocks) 皙?gp= ף?333333?gq= ףp?g?g?g)\(??gRQ?g(\?gQ?) @333333@gr"r!bumps) rr rrr @r"r#g@gffffff@r&) {Gzt?r'g~jtx?{Gz?r(gQ?r(r(r'gMb?r'r heavisineg333333?g ףp= ?dopplerg?g?rampgGz?hisineg_vO?losinegioT?linchirpg?twochirpr quadchirpmishmashg? wernersorrows hypchirpsir gjt?gMbP?g?rg(\?g?r linchirpsdchirps2i^iy?)dtyper(g?iz piece-regular irizpiece-polynomial(-riemanng?zunknown name: z. name must be one of: )lower_implemented_signals ValueErrornparangezipsignabssinpisqrtzerosintroundonesceilcosexpslicefloatfixrsum zeros_likerealfftifft)0nament n_hard_codedt0shsft0hwswposhgtwthpalphabetaf1f2menvelopetmpenvbaA1ixgi1i2jf3f4n_12n_7n_5n_3n_2n_20f5fmaf6kdiffbiasn_10sqnidxrU/var/www/html/scripts/venv/lib/python3.10/site-packages/pywt/data/_wavelab_signals.pyrs 1 &$4.     4    &   "" 4"  2 o 0i  $$ ",*T ((J    C (""( $    "  )rN)numpyrK__all__rIrrrrrs