U f{jv% @sddlmZddlZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZmZd d Zd d Zd dZddZe dddZedddZdS))deepcopyN)_pick_data_channels pick_info)verbosewarnfill_doc_validate_type) parallel_func) AverageTFR _get_datacCs|jd}dd}|dks*||sD||krDdttt|}n||kr\td||f||krtd|jd|||}t|jdd|f|j }tj ||fdd}nd }|||fS) Aux function.cSs|dko||d@ S)Nrr )nrr\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/time_frequency/_stockwell.py_is_power_of_twosz)_check_input_st.._is_power_of_twoNrz6n_fft cannot be smaller than signal size. Got %s < %s.zJThe input signal is shorter ({}) than "n_fft" ({}). Applying zero padding.Zaxisr) shapeintnpceillog2 ValueErrorrformatZzerosdtype concatenate)Zx_inn_fftZn_timesrzero_padZ pad_arrayrrr_check_input_sts$ r cCsddlm}m}||d||}tj|dd|dddddf}|}t||d} tjt| t|ftjd} t | D]v\} } | dkrt t|} n>| t d tj |t d d|d | d |d } | | } || | | <q| S) z;Precompute stockwell Gaussian windows (in the freq domain).r)fftfftfreq?Nr rrgg@g) scipy.fftr!r"rZr_Zarangeemptylen complex128 enumerateZonessqrtpiexpsum)n_sampstart_fstop_fsfreqwidthr!r"ZtwkZf_rangewindowsi_ffwindowrrr_precompute_st_windows)s($ r8c Csddlm}m}|jd}tj|jddt||ftjd}||}tj||gdd}t |D]:\} } || } ||d| | |f| |d| ddf<qb|S)z=Compute ST based on Ali Moukadem MATLAB code (used in tests).rr!ifftrNr$r.) r%r!r:rrr&r'r(rr)) xr/r4r!r:r.STZFxZXFr5r7r6rrr_st=s &,r=cCsDddlm}m}|jd}||} | |t| |} tt|| f} |rVt| nd} ||} tj | | gdd} t |D]\}}||}|| dd|||f|}|dkr|ddd| |f}n|dddd|f}t |}d||dk<|r ||}t tj |dd| |<||9}tj |dd| |<q|| | fS)rrr9rNrr#) r%r!r:rboolrr&r'Z empty_likerr)absZmean)r;r/Z compute_itcrdecimWr!r:r.n_outpsditcXZXXr5r7r6r<ZTFRZTFR_absrrr _st_power_itcKs,    rFr#Fc sddlm} ttjdjdkr4tdjjdd\} } jdtjd} t |\} | | d |}|dkr||dkd}|dkr| }t || t || }||}t jd||||}t| || f}r t| || fnd}tt|\}}|fd d t| D}tt|D](\}\}}|||<|dk rf|||<qf|||fS) aLCompute power and intertrial coherence using Stockwell (S) transform. Same computation as `~mne.time_frequency.tfr_stockwell`, but operates on :class:`NumPy arrays ` instead of `~mne.Epochs` objects. See :footcite:`Stockwell2007,MoukademEtAl2014,WheatEtAl2010,JonesEtAl2006` for more information. Parameters ---------- data : ndarray, shape (n_epochs, n_channels, n_times) The signal to transform. sfreq : float The sampling frequency. fmin : None, float The minimum frequency to include. If None defaults to the minimum fft frequency greater than zero. fmax : None, float The maximum frequency to include. If None defaults to the maximum fft. n_fft : int | None The length of the windows used for FFT. If None, it defaults to the next power of 2 larger than the signal length. width : float The width of the Gaussian window. If < 1, increased temporal resolution, if > 1, increased frequency resolution. Defaults to 1. (classical S-Transform). decim : int The decimation factor on the time axis. To reduce memory usage. return_itc : bool Return intertrial coherence (ITC) as well as averaged power. %(n_jobs)s Returns ------- st_power : ndarray The multitaper power of the Stockwell transformed data. The last two dimensions are frequency and time. itc : ndarray The intertrial coherence. Only returned if return_itc is True. freqs : ndarray The frequencies. See Also -------- mne.time_frequency.tfr_stockwell mne.time_frequency.tfr_multitaper mne.time_frequency.tfr_array_multitaper mne.time_frequency.tfr_morlet mne.time_frequency.tfr_array_morlet References ---------- .. footbibliography:: r)r"dataz@data must be 3D with shape (n_epochs, n_channels, n_times), got Nrrr#c3s2|]*}dd|ddfVqdS)Nr).0crArGr@Zmy_st return_itcr/rrr s z&tfr_array_stockwell..)r%r"r rZndarrayndimrrr>r maxr?Zargminr8r&r rFranger)iter)rGr1fminfmaxrr2r@rLn_jobsr"Zn_epochsZ n_channelsrBZn_fft_freqsr0Zn_freqrCrDparallelZtfrsrJZthis_psdZthis_itcrrKrtfr_array_stockwellfs<9       rWc  Cst||} t|j} t|j| } | dd| ddf} t| | d|||||||d \} } }|jdd|}t| }t| | |||dd}|r|tt | | |||ddf}|S)aCompute Time-Frequency Representation (TFR) using Stockwell Transform. Same computation as `~mne.time_frequency.tfr_array_stockwell`, but operates on `~mne.Epochs` objects instead of :class:`NumPy arrays `. See :footcite:`Stockwell2007,MoukademEtAl2014,WheatEtAl2010,JonesEtAl2006` for more information. Parameters ---------- inst : Epochs | Evoked The epochs or evoked object. fmin : None, float The minimum frequency to include. If None defaults to the minimum fft frequency greater than zero. fmax : None, float The maximum frequency to include. If None defaults to the maximum fft. n_fft : int | None The length of the windows used for FFT. If None, it defaults to the next power of 2 larger than the signal length. width : float The width of the Gaussian window. If < 1, increased temporal resolution, if > 1, increased frequency resolution. Defaults to 1. (classical S-Transform). decim : int The decimation factor on the time axis. To reduce memory usage. return_itc : bool Return intertrial coherence (ITC) as well as averaged power. n_jobs : int The number of jobs to run in parallel (over channels). %(verbose)s Returns ------- power : AverageTFR The averaged power. itc : AverageTFR The intertrial coherence. Only returned if return_itc is True. See Also -------- mne.time_frequency.tfr_array_stockwell mne.time_frequency.tfr_multitaper mne.time_frequency.tfr_array_multitaper mne.time_frequency.tfr_morlet mne.time_frequency.tfr_array_morlet Notes ----- .. versionadded:: 0.9.0 References ---------- .. footbibliography:: Nr1)r1rRrSrr2r@rLrTzstockwell-power)methodz stockwell-itc) r rinforrWtimescopyr'r r)instrRrSrr2r@rLrTrrGZpicksrYpowerrDrUrZZnaveoutrrr tfr_stockwells.<     r_)NNNr#r FN)NNNr#r FNN)r[rnumpyrZio.pickrrutilsrrrr rVr Ztfrr r r r8r=rFrWr_rrrrs0   ]