o
    7ήc                  	   @   s  d Z ddlmZmZmZ ddlZddlZzddlm	Z	 e	 Z
dZW n ey0   ed dZY nw er7edd	Zg d
ZdZedd eD g ZejdZe
  zi ZeD ]ZeeZejeeddgkddre
deeej  d eej!  n$edkre
deeej" d eej  d eej!  ne
de edkre#e$ddg d Z%de&dde&dde&ddfZ'n
dZ%de&ddfZ'dZ(e
)e(Z*e*d se+d e,e
-d!Z.d"/ed!gZ0e.ee0< e%D ]HZ1e2e1Z3e
d#e3 dZ4e'D ]6Z5e4d7 Z4e
d$e5 d%Z(e
)e(Z*e*d s!e+d e,e
-d&Z6d"/ee4eee1d&gZ7e6ee7< qqqXW e
8  ne
8  w ej9d(i e dS ))zi This script was used to generate dwt_matlabR2012a_result.npz by storing
the outputs from Matlab R2012a.     )divisionprint_functionabsolute_importN)MatlabFzTo run Matlab compatibility tests you need to have MathWorks MATLAB, MathWorks Wavelet Toolbox and the pymatbridge Python package installed.Tz/Can't generate matlab data files without MATLABreduced))zerozpd)constantsp0)	symmetricsym)periodicppd)smoothsp1)periodizationper)gausmexhmorlcgaushanfbspcmorc                 C   s   g | ]}t |qS  )pywtwavelist).0namer   r   O/tmp/pip-target-vg8gfxp4/lib/python/pywt/tests/data/generate_matlab_data_cwt.py
<listcomp>!   s    r    i  r   r   )axiswavelet-r   fulld   e   )r%      i    iP              )r(   i  zpsi = wavefun(wavelet,10)successzXMatlab failed to execute the provided code. Check that the wavelet toolbox is installed.psi_datascalesz"coefs = cwt(data, scales, wavelet)coefscwt_matlabR2015b_result.npz)r3   ):__doc__
__future__r   r   r   numpynpr   pymatbridger   mlab_matlab_missingImportErrorprintEnvironmentErrorsize_setmodesfamiliessumwaveletsrandomRandomStaterstatestartall_matlab_resultsr"   ContinuousWaveletwanyarrayset_variablestrbandwidth_frequencycenter_frequency
fbsp_orderlistrange
data_sizesarangeScales	mlab_coderun_coderesRuntimeErrorasarrayget_variabler.   joinpsi_keyNrandnr0   scale_countr1   r2   	coefs_keystopsavezr   r   r   r   <module>   s    
&4&



-