U f{jS@sddlZddlmZmZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZe dd d Ze d dddd ddd fddZdddZGdddeZdS)N)SourceEstimateVolSourceEstimate) _ensure_src) rng_uniform)check_random_statewarn _check_optionfill_doc _ensure_int_ensure_events)Label)_compute_nearestrandomspherecCst}t}td|ddgt|} |jdkr8|} d} n|} d} t|| d|j} |dkr|| t| dt| dd} n|j || ||d} | | ||fS) aSelect source positions using a label. Parameters ---------- src : list of dict The source space. label : Label The label. %(random_state)s location : str The label location to choose. Can be 'random' (default) or 'center' to use :func:`mne.Label.center_of_mass` (restricting to vertices both in the label and in the source space). Note that for 'center' mode the label values are used as weights. .. versionadded:: 0.13 subject : str | None The subject the label is defined for. Only used with ``location='center'``. .. versionadded:: 0.13 %(subjects_dir)s .. versionadded:: 0.13 surf : str The surface to use for Euclidean distance center of mass finding. The default here is "sphere", which finds the center of mass on the spherical surface to help avoid potential issues with cortical folding. .. versionadded:: 0.13 Returns ------- lh_vertno : list Selected source coefficients on the left hemisphere. rh_vertno : list Selected source coefficients on the right hemisphere. locationrcenterlhrvertno)Zrestrict_vertices subjects_dirsurf) listr rheminp intersect1dverticesrlenZcenter_of_massappend)srclabel random_statersubjectrr lh_vertno rh_vertnorngrZhemi_idxsrc_selidxr(T/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/simulation/source.pyselect_source_in_labels(*  r*cCsdtdtj|S)NgHz>)rsinpi)tr(r(r)Xr/c  st|t|dd}|j} |dkr(| }n | dk rH|| krHtd|| ft|t|f} t|D]} ||| | ddf<qb|dkrtd|t|d t } t | }fddt ||D}| }n|t|krtd t||fn|t|krt d t|t||t|kr,|d|n|}ggg}td| jdfg}td| jdfg}t|D]\}}t|||||| \}}|d|7<|d|7<t|dkr|| |tjn,t|dkr|| |tjntd qld d|D}d d||fD}tdD]B}t||}|||||<t|dkr(|||||<q(t|}|dt |ddd}}|j| jkstt|dkrtnt}||||||d}|S)aoGenerate sparse (n_dipoles) sources time courses from data_fun. This function randomly selects ``n_dipoles`` vertices in the whole cortex or one single vertex (randomly in or in the center of) each label if ``labels is not None``. It uses ``data_fun`` to generate waveforms for each vertex. Parameters ---------- src : instance of SourceSpaces The source space. n_dipoles : int Number of dipoles to simulate. times : array Time array. data_fun : callable Function to generate the waveforms. The default is a 100 nAm, 10 Hz sinusoid as ``1e-7 * np.sin(20 * pi * t)``. The function should take as input the array of time samples in seconds and return an array of the same length containing the time courses. labels : None | list of Label The labels. The default is None, otherwise its size must be n_dipoles. %(random_state)s location : str The label location to choose. Can be 'random' (default) or 'center' to use :func:`mne.Label.center_of_mass`. Note that for 'center' mode the label values are used as weights. .. versionadded:: 0.13 subject : str | None The subject the label is defined for. Only used with ``location='center'``. .. versionadded:: 0.13 %(subjects_dir)s .. versionadded:: 0.13 surf : str The surface to use for Euclidean distance center of mass finding. The default here is "sphere", which finds the center of mass on the spherical surface to help avoid potential issues with cortical folding. .. versionadded:: 0.13 Returns ------- stc : SourceEstimate The generated source time courses. See Also -------- simulate_raw simulate_evoked simulate_stc Notes ----- .. versionadded:: 0.10.0 F)verboseNzHsubject argument (%s) did not match the source space subject_his_id (%s)rrc s:g|]2\}}|dtt|dd|qS)rZnuseN)rsortZ permutationarange).0nsr%r(r) sz'simulate_sparse_stc..zANumber of labels (%d) smaller than n_dipoles (%d) is not allowed.z]The number of labels is different from the number of dipoles. %s dipole(s) will be generated.zNo vertno found.cSsg|]}t|qSr(rarrayr4vr(r(r)r8scSsg|]}t|qSr()r concatenate)r4dr(r(r)r8srrtmintstepr")rr_subject ValueErrorrzerosrrangeZlinspaceastypeintdiffziprminemptyshape enumerater*rZnewaxisargsortr=AssertionErrorrr)rZ n_dipolestimesZdata_funlabelsr!rr"rrZ subject_srcdataZi_dipoffsetsZ n_dipoles_ssvsZdatasrZlh_dataZrh_datair r#r$iiorderr@rAclsstcr(r7r)simulate_sparse_stcVszA           rZFcst|t|krtdggg}ggg}ddd} t|D]\} } | | j} t|| d| j} t| dkr|| dd}|| d|}|| d| j}t||}|| dt |} d k r t | j| }t fd d | j |D}t ||| }nt|| t| df}|rg}t| D]f\}}t||| kd}t|dkr6t|dkstt|| |d||7<||q6tj||dd }ttt | |} || | || t|q:d d |D}|sBt|dD]:\}}t|tt |}|dkrtd||fqt}tdD]Z} t|| dkrPt|| || <t|| }||| ||| ||| <qPtt|||||jd}|S)aSimulate sources time courses from waveforms and labels. This function generates a source estimate with extended sources by filling the labels with the waveforms given in stc_data. Parameters ---------- src : instance of SourceSpaces The source space. labels : list of Label The labels. stc_data : array, shape (n_labels, n_times) The waveforms. tmin : float The beginning of the timeseries. tstep : float The time step (1 / sampling frequency). value_fun : callable | None Function to apply to the label values to obtain the waveform scaling for each vertex in the label. If None (default), uniform scaling is used. allow_overlap : bool Allow overlapping labels or not. Default value is False. .. versionadded:: 0.18 Returns ------- stc : SourceEstimate The generated source time courses. See Also -------- simulate_raw simulate_evoked simulate_sparse_stc z-labels and stc_data must have the same lengthrr)rZrhrZinuseboolrrNcsg|] }|qSr(r(r; value_funr(r)r8sz simulate_stc..)ZaxiscSsg|]}t|qSr(r9r;r(r(r)r8(s)leftrightzILabels had %s overlaps in the %s hemisphere, they must be non-overlappingrr?)rrCrMrrrrrFruniqueZ searchsortedr:valuesouterZtilewhererOrdeleterextendZ atleast_2drI RuntimeErrorrEZvstackrNrr=rB)rrQstc_datar@rAr^ allow_overlaprZstc_data_extendedZ hemi_to_indrUr Zhemi_indr&r'Zxhsr\Z closest_srcZidx_selZ values_selrR duplicatesZsrc_indZ vertex_indindr<rr>rYr(r]r) simulate_stcsn'        rlc@sbeZdZdZdddZeddZed d Zed d Zd dZ dddZ dddZ ddZ dS)SourceSimulatoraClass to generate simulated Source Estimates. Parameters ---------- src : instance of SourceSpaces Source space. tstep : float Time step between successive samples in data. Default is 0.001 sec. duration : float | None Time interval during which the simulation takes place in seconds. If None, it is computed using existing events and waveform lengths. first_samp : int First sample from which the simulation takes place, as an integer. Comparable to the :term:`first_samp` property of `~mne.io.Raw` objects. Default is 0. Attributes ---------- duration : float The duration of the simulation in seconds. n_times : int The number of time samples of the simulation. MbP?NrcCsb|dk r||krtdt|d|_||_||_g|_g|_tjdt d|_ ||_ g|_ d|_ dS)Nz"duration must be None or >= tstep. first_samp)rdtypei)rCr ro_src_tstep_labels _waveformsrrKrG_events _duration _last_samples _chk_duration)selfrrAdurationror(r(r)__init__Xs zSourceSimulator.__init__cCs|jdk r|jS|j|jS)z2Duration of the simulation in same units as tstep.N)rxn_timesrtr{r(r(r)r|es zSourceSimulator.durationcCsH|jdk rt|j|jS|j}t|jdkr:t|j}||jdS)z)Number of time samples in the simulation.Nrr)rxrGrtrorryrmax)r{Zlsr(r(r)r~ls   zSourceSimulator.n_timescCs|j|jdS)Nr)ror~rr(r(r) last_sampvszSourceSimulator.last_sampcst|tstdt|t|ts:t|dkr:t|}t|tsXt|dkrX|g}t|dkrp|t|}t|t|krtdt|t|ft| tj }j |gt|j |tj|g_jjtj ksttfddtj D_dS)aAdd data to the simulation. Data should be added in the form of a triplet of Label (Where) - Waveform(s) (What) - Event(s) (When) Parameters ---------- label : instance of Label The label (as created for example by mne.read_label). If the label does not match any sources in the SourceEstimate, a ValueError is raised. waveform : array, shape (n_times,) or (n_events, n_times) | list The waveform(s) describing the activity on the label vertices. If list, it must have the same length as events. events : array of int, shape (n_events, 3) Events associated to the waveform(s) to specify when the activity should occur. zlabel must be a Label,not %srrz\Number of waveforms and events should match or there should be a single waveform (%d != %d).cs*g|]"\}}j|dft|dqS)rr)rwr)r4rUwrr(r)r8sz,SourceSimulator.add_data..N) isinstancer rCtyperrndimrr rFint64rurfrvr=rwrrrOr:rMry)r{r waveformeventsr(rr)add_datazs,    zSourceSimulator.add_datacCs|dkr|j}|dkr&||jd}n||kr6td||d}tj|tjd}tt|jdddf|k|jdddf|kd}t |dkr|j|df|}|j|df||<|S)aGet the stim channel from the provided data. Returns the stim channel data according to the simulation parameters which should be added through the add_data method. If both start_sample and stop_sample are not specified, the entire duration is used. Parameters ---------- start_sample : int First sample in chunk. Default is the value of the ``first_samp`` attribute. stop_sample : int | None The final sample of the returned stc. If None, then all samples from start_sample onward are returned. Returns ------- stim_data : ndarray of int, shape (n_samples,) The stimulation channel data. Nrz-Argument start_sample must be >= stop_sample.rqrr) ror~rCrrDrrd logical_andrwr)r{ start_sample stop_sample n_samplesZ stim_dataZstim_indZ relative_indr(r(r)get_stim_channels$  z SourceSimulator.get_stim_channelc svtjdkrtd|dkr$j}|dkr<|jd}n||krLtd||d}ttj|f}ttj |kj dddf|kd}fdd|D}t t |j |D]\}\}} | d} j ||} t t| | dt||d} t t||dt| | d} |||| || 7<q|j8}tjj||jjdd }|S) aSimulate a SourceEstimate from the provided data. Returns a SourceEstimate object constructed according to the simulation parameters which should be added through function add_data. If both start_sample and stop_sample are not specified, the entire duration is used. Parameters ---------- start_sample : int | None First sample in chunk. If ``None`` the value of the ``first_samp`` attribute is used. Defaults to ``None``. stop_sample : int | None The final sample of the returned STC. If ``None``, then all samples past ``start_sample`` are returned. Returns ------- stc : SourceEstimate object The generated source time courses. rz_No simulation parameters were found. Please use function add_data to add simulation parameters.Nrz$start_sample must be >= stop_sample.csg|]}j|qSr()rv)r4rUrr(r)r8sz+SourceSimulator.get_stc..T)ri)rrurCror~rrDrdrryrwrMrIZin1dr3rlrsrt)r{rrrrhrkZsubset_waveformsrUreventZwf_startZwf_stopZ waveform_indZstc_indrYr(rr)get_stcsF   zSourceSimulator.get_stcccsR|j}t|j|d|jD]2}t||jd|}||||||fVqdS)zIterate over 1 second STCs.rN)rrErorzrJrr)r{Z last_samplerrr(r(r)__iter__s   zSourceSimulator.__iter__)rnNr)rN)NN) __name__ __module__ __qualname____doc__r}propertyr|r~rrrrrr(r(r(r)rm?s    . + Crm)NrNNr)NF)numpyrZsource_estimaterrZ source_spacerfixesrutilsrrr r r r r r Zsurfacerr*rZrlobjectrmr(r(r(r) s2      ? } k