U f{j@sddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZe dd d Zd d Ze dd dZdddZdddZdS)N)compute_whitener) pick_info) apply_forward)loggerverbosecheck_random_state_check_preload_validate_typeTc Cs|t||||d} |dkr| S|tjkrZt| |||} | j| jt|7_t|| _| ddrx| |d | S)aWGenerate noisy evoked data. .. note:: No projections from ``info`` will be present in the output ``evoked``. You can use e.g. :func:`evoked.add_proj ` or :func:`evoked.set_eeg_reference ` to add them afterward as necessary. Parameters ---------- fwd : instance of Forward A forward solution. stc : SourceEstimate object The source time courses. %(info_not_none)s Used to generate the evoked. cov : Covariance object | None The noise covariance. If None, no noise is added. nave : int Number of averaged epochs (defaults to 30). .. versionadded:: 0.15.0 iir_filter : None | array IIR filter coefficients (denominator) e.g. [1, -1, 0.2]. %(random_state)s %(use_cps)s .. versionadded:: 0.15 %(verbose)s Returns ------- evoked : Evoked object The simulated evoked data. See Also -------- simulate_raw simulate_stc simulate_sparse_stc Notes ----- To make the equivalence between snr and nave, when the snr is given instead of nave:: nave = (1 / 10 ** ((actual_snr - snr)) / 20) ** 2 where actual_snr is the snr to the generated noise before scaling. .. versionadded:: 0.10.0 )use_cpsNZprojs) rnpinf_simulate_noise_evokeddatamathsqrtZint64navegetZadd_projZ apply_proj) ZfwdZstcinfocovr iir_filter random_stater revokednoiserT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/simulation/evoked.pysimulate_evokeds6   rcCs(|}d|jdd<t||||ddS)NrF)allow_subselection)copyr _add_noise)rrrrrrrrrTs  rcCst||||S)aCreate noise as a multivariate Gaussian. The spatial covariance of the noise is given from the cov matrix. Parameters ---------- inst : instance of Evoked, Epochs, or Raw Instance to which to add noise. cov : instance of Covariance The noise covariance. iir_filter : None | array-like IIR filter coefficients (denominator). %(random_state)s %(verbose)s Returns ------- inst : instance of Evoked, Epochs, or Raw The instance, modified to have additional noise. Notes ----- Only channels in both ``inst.info['ch_names']`` and ``cov['names']`` will have noise added to them. This function operates inplace on ``inst``. .. versionadded:: 0.18.0 )r )instrrrrrrr add_noise[s!r"c Cs\ddlm}ddlm}ddlm}ddlm}t||dt||||fddt |d |j } | j d ksnt | j dkr| t j} |j} | td } } |r"tt| d t|d @} t t | d | d} tdt| t| dt|d ft|j| } | t | d} | D]0}|| t| ||||jd| dd7<q&|S)z.Add noise, possibly with channel subselection.r) Covariance)BaseRaw) BaseEpochs)Evokedrr!zRaw, Epochs, or Evokedz Adding noise)rNZch_namesnamesrz3Adding noise to %d/%d channels (%d channels in cov)ZchsZnchan)picks)rr#ior$Zepochsr%rr&r r _datandimAssertionErrorr ZnewaxisrZ_check_consistencyslicelistsetwhereZin1drlenrZarange_generate_noiseshape)r!rrrrr#r$r%r&rrr*Z gen_picksZuse_chsepochrrrr sD            r c Csddlm}t|}t||dd|dd\} } } t| || jd|f} |dk r|dkrttt | t |df}|dg|| d|d \} } nd} | | fS) z;Create spatially colored and temporally IIR-filtered noise.r)lfilterTF)ZpcaZreturn_colorerr*rr)N)Zaxiszi) Z scipy.signalr7rrr dotZstandard_normalr5Zzerosr3) rrrrZ n_samplesr9r*r7rng_Zcolorerrzfrrrr4s   r4)Nr NNTN)NNN)T)NN)rnumpyr rrZio.pickrforwardrutilsrrrr r rrr"r r4rrrrs(    B # !