U f{jB<@sddlmZddlZddlmZmZmZddl Z ddl m Z ddl m Z mZmZmZddlmZddlmZmZmZddlmZe jdd Zeed d d Zd ddgZeedZe jde gdddZ ddZ!ddZ"ddZ#ddZ$ddZ%e j&dd Z'd!d"Z(dS)#N)assert_array_almost_equalassert_array_equal assert_equal)testing) read_labelread_forward_solutionpick_types_forwardconvert_forward_solution)Label) simulate_stcsimulate_sparse_stcSourceSimulator) check_versionF)downloadMEGsamplez)sample_audvis_trunc-meg-eeg-oct-6-fwd.fifzAud-lhzAud-rhzVis-rhZsubjectsmodule)scopeparamscCs:tt}t|ddd}t|ddd}ddtD}||fS)NT)Z force_fixedZuse_cpsF)ZmegZeegc Ss&g|]}tttdddd|qS)rrlabelsz%s.label)ropjoin data_path).0labelr_/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/simulation/tests/test_source.py "sz#_get_fwd_labels..)r fname_fwdr r label_names)fwdrrrr_get_fwd_labelssr!cCsZtddd}||j}t|j||j}t|j||}|dkrV|t|jd7}|S)Nrlhrh)dictheminp intersect1dvertices searchsortedlen)rstcZhemi_idx_mappinghemi_idxidxrrr_get_idx_label_stc's  r0c CsF|\}}g}t|D]@\}}t|j|jd|tt|j|j|j d}| |qd}d}d} tt||f} t |d|| || } t | j d|D]>}t|| } t| j| dkst| j| jd |kstqd d } t |d|| || | } t|D]@\}}t|| } d |d tt| |f}t| j| |q|d d}| d d}t |d|||| | } ttt |d||d d|| | ttt |d|dtj|gddd|| | t|ddddkdd}t|g|ddd||d d d fdd}t |d|g| d d || } t t| jd d S)z#Test generation of source estimate.r*posvaluesr'comment rMbP?srcr?r"cSs|dSNr1rxrrrfunNsztest_simulate_stc..fun@N)ZaxisZinuseZrrr$)r*r3r') enumerater r*r3r(onesr,r4r'r5appendr rsubjectr0alldataAssertionErrorshaperpytestraises ValueError RuntimeErrorZ concatenatewhere lh_vertno)r!r rmylabelsir new_labeln_timestmintstepstc_datar-r/r=resZ label_subsetZ data_subsetZlabel_single_vertrrrtest_simulate_stc4sf       ""rVc Cs|\}}d}d}d}tj|tjd||}tjtt|dt|||ddtd g}|D]"}| } | j d | | qXd D]} | d krdnd } t|dt|||| | td } t | jd| jjdt|kst| jjd|kstt|dt|||| | td } t| j| jt| j| jqtjtt|dt|||ddtd |ddd=tjtt|dt|||dtdd|ddd<tjtt|dt|||ddd}tjt|d(t|dt|d||| | td W5QRXd S)*Test generation of sparse source estimate.r6rr7Zdtyper8centerr)rlocationrC subjects_dirr9)randomrYr\N)r random_staterZr[r"ZfooZsubject_his_id)rrZr[)rrZzNumber of labelsmatch)r(arangefloat64rHrIrJr r,r[copyr4fillrBrrCrErGrFrrM rh_vertno)r!r rrQrRrStimesrNrZ this_labelrZr]stc_1stc_2Zerr_strrrrtest_simulate_sparse_stcms    rhc Cs|\}}|dd}g}t|D]@\}}t|j|jd|tt|j|j|j d}| |q d}d} d} tt||f} t |d|| | | } |D]>}t || } t | j| d kst| j| jd|kstqd d }t |d|| | | |} t|D]\}}|jd krd}nd}t| j||j} t| j|| } |dkr^| t| jd7} d |d tt| |f}t| j| |qdS)z0Test generation of source estimate, single hemi.r"Nr1r2r6rr7r8r9cSs|dSr:rr;rrrr=sz*test_generate_stc_single_hemi..funr$r>)r@r r*r3r(rAr,r4r'r5rBr r0rDrErFrGr)r+r)r!r rlabels_single_hemirNrOrrPrQrRrSrTr-r/r=r.rUrrrtest_generate_stc_single_hemis@     rjc Cs|\}}|dd}d}d}d}tj|tjd||}t|dt|||dd}|jjdt|ksjt|jjd|ks~tt|dt|||dd} t|j | j t|j | j t d d rt|dt|||tj dddS) rWr"Nr6rr7rXr8)rr]numpyz1.17)r(r`rar r,rErGrFrrMrdrr\Z default_rng) r!r rrirQrRrSrerfrgrrr$test_simulate_sparse_stc_single_hemis.   rlc Cs|\}}g}t|D]@\}}t|j|jd|tt|j|j|j d}| |q| |d}d}d} tt||f} t j t ddt|d|| || d d W5QRXt|d|| || d d } t| jd | jjd |kstd| jkstdS)z7Test generation of source estimate, overlapping labels.r1r2r6rr7zmust be non-overlappingr^r8F)Z allow_overlapTrr"N)r@r r*r3r(rAr,r4r'r5rBrHrIrKr rrCrErGrF) r!r rrNrOrrPrQrRrSrTr-rrr test_simulate_stc_labels_overlaps2   rmc% Csn|\}}|d}ddd}d}gggg}td|d<tdd|d<td|d <d d d g}g}g} t|D]F\} } t| || d } || | t|||| d | jqngggg} tdddg| d<tdddgtdddgg| d<tdddgg| d <gggg}tdddgdddgg|d<tdddgdddgg|d<tdddgd ddgg|d <ttd|dd }ttd|dd }t t | d| d}t t | d| d}t t| d| d}t t |dg}tddddddg}t ||df|d|ddf<tdd ddddg}t ||df||||ddf<tddddddg}t ||df|||dddf<tddd dddg}t |t | d df}t ||g}t||}tdD] } ||| | | || q|}|}|jdks6t|jdksFt|jdksVtt ||jjdksptt|jd|kstt|jd|kstt|j|t|j|t|j|d}|D](\}}|jjddkst|d7}q|dkstt||dd}tdD] } ||| | | || qtjtdd&||d| d|dt W5QRX|}t|jddddf|j|jddd} | jdkst| j!d|ksttjt"dd|d dW5QRXt|}tjt"dd|W5QRXtjt"dd|d| |W5QRXtjt"dd ||d| dd |W5QRXtjt"d dt|||d }W5QRXt||}d!}!tdD]<} || dddf|!7<||| | | || q|}"|jd"kst|j#dks$t|"jj|jjd|jjd|!fksNtt|||!d#}tdD] } ||| | | || qd|}"|jdkst|j#|!kst|j|!dkst|"jj|jjkstt||d$|d%}#|#|dtdddgdddggd|#_$d&d'|#D}$t |$d ks6t|$djjddksPt|$djjddksjtdS)(zTest Source Simulator.r8rr"r#gUUUUUU?iiir1r$r%)r*r'Zvertnor9g?Ng@g?)durationzarray of integersr^)Z start_sampleZ stop_sample)rrzstart_sample must bezNo simulation parameterszlabel must be a Labelz+Number of waveforms and events should matchzduration must be None or28) first_sampr6)rSrqcSsg|] \}}|qSrr)rr-_rrrrsz)test_source_simulator..)%r(r`r@r rBr)r*arrayranger,Z setdiff1dZzerosZtileZvstackr Zadd_dataZget_stcZget_stim_channelrqrFrQZ last_samprErGrDrZlh_dataZrh_datarHrI TypeErrorZastypefloatrerJrvZ _chk_duration)%r!r rwr8Z hemi_to_indrSZlabel_verticesZhemisrNZ src_verticesrOZvertrPZwfseventsZverts_lhZverts_rhZdiff_01Zdiff_10Zinter_10Zoutput_data_lhtmpZ data_rh_wfZoutput_data_rhZ output_datassr-Z stim_channelcounterZstimZhalf_ssZhalf_stcZpart_stcoffsetZ offset_stcZsource_simulatorZstcsrrrtest_source_simulators       $$        *$    &r))os.pathpathrrkr(Z numpy.testingrrrrHZ mne.datasetsrZmnerrrr Z mne.labelr Zmne.simulationr r r Z mne.utilsrrrrrr[ZfixtureZ _pytest_paramr!r0rVrhrjrlZrequires_testing_datarmrrrrrs0          991 !