o e&@s0ddlmZddlZdgZgdZdddZdS))divisionN demo_signal)ZBlocksBumpsZ HeaviSineZDopplerZRampZHiSineZLoSineZLinChirpZTwoChirpZ QuadChirpZMishMashZ WernerSorrowsZ HypChirpsZ LinChirpsZChirpsZGabor sineoneoverxz Piece-RegularzPiece-PolynomialZRiemannrc0 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|dkrdt dtj |t|dtd|}|S|dkrt |d|t d tj d|d}|S|dkr||dk}|S|dkrt tj |d|}|S|dkrt tj |d|}|S|dkr-t tj ||d|}|S|d krNt tj |||t tj d!|||}|S|d"kret tj d!|||d }|S|d#krt tj d!|||d }|t tj |d|7}|t tj ||d$|7}|S|d%krt tj ||d |d }|t tj |d|}|t tj |||}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 td0|}|d}t|}td|dt|}dt tj 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}dt tj 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}t dB|}|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 tt |}.td|.d}/|/|/9}/t|}dOtd|.d||/d<ttj|}|StdP|t)QuLSimple 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. listNrzn must be an integer >= 1Zgaborrz/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@grrZbumps) r rrr r@rrg@gffffff@r) {Gzt?rg~jtx?{Gz?rgQ?rrrgMb?rr Z heavisineg333333?g ףp= ?Zdopplerg?g?ZrampgGz?Zhisineg_vO?ZlosinegioT?Zlinchirpg?ZtwochirprZ quadchirpZmishmashg?Z wernersorrowsZ hypchirpsirgjt?gMbP?g?r g(\?g?r Z linchirpsdZchirps2i^iy?)Zdtyperg?iz piece-regular ir izpiece-polynomial(-Zriemanng?z*unknown name: {}. name must be one of: {})lower_implemented_signals ValueErrornpZarangeformatzipsignabssinpisqrtZzerosintroundZonesceilcosexpslicefloatZfixrsumZ zeros_likerealZfftZifft)0namentZ n_hard_codedZt0shsft0hwswposZhgtZwthpalphabetaf1f2menvelopetmpenvbaA1Zixgi1i2jZf3Zf4Zn_12Zn_7Zn_5Zn_3Zn_2Zn_20Zf5fmaZf6kdiffZbiasZn_10ZsqnidxrZJD:\Projects\ConvertPro\env\Lib\site-packages\pywt/data/_wavelab_signals.pyrs & '$ 4.     4    &   "" 4"  2 p 0j  $$ ",*U ((K    D (""( %    "  )rN) __future__rnumpyr+__all__r)rrZrZrZr[s