U f{jG@szddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZd d Zd d ZedddZdS)N)Evoked) BaseEpochs)BaseRaw) find_events) _picks_to_idx)_check_preload _check_optionfill_docc CsRddlm}dtj|dddtt||d|dddfj}|S)z@Return window which has length as much as parameter start - end.r)hannNr)Z scipy.signalr npZr_ZonesabsT)startendr windowrU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/stim.py _get_windows   rc Csddlm}|dkrdt||g}|||dd||ff|}t||} || } | ||||f<|dkr||||f|tjddf||||f<dS)z-Modify original data by using parameter data.r)interp1dlinearNr)Zscipy.interpolaterrarrayarangeZnewaxis) datarpicks first_samp last_sampmoderxfZxnewZ interp_datarrr _fix_artifacts   r#{Gz?rcCstd|ddgtt|jd|}tt|jd|} |dkr\| |dkr\tdd} |dkrrt|| } t|j|dd d }t|d t |t rD|dkrt ||d }t |d krtd|dkrt t |} n|dddf|k} || d f} |j} | D]<}t||j|}t||j| }t| | ||||qnt |tr|jdk rdtdtt|jd|j}||}| |}|j} | D]}t|| ||||qnJt |tr||j}| |j}|j} t| | ||||ntdt||S)a Eliminate stimulation's artifacts from instance. .. note:: This function operates in-place, consider passing ``inst.copy()`` if this is not desired. Parameters ---------- inst : instance of Raw or Epochs or Evoked The data. events : array, shape (n_events, 3) The list of events. Required only when inst is Raw. event_id : int The id of the events generating the stimulation artifacts. If None, read all events. Required only when inst is Raw. tmin : float Start time of the interpolation window in seconds. tmax : float End time of the interpolation window in seconds. mode : 'linear' | 'window' Way to fill the artifacted time interval. 'linear' does linear interpolation 'window' applies a (1 - hanning) window. stim_channel : str | None Stim channel to use. %(picks_all_data)s Returns ------- inst : instance of Raw or Evoked or Epochs Instance with modified data. r rrZsfreqr zGTime range is too short. Use a larger interval or set mode to "linear".Nrr)excludefix_stim_artifact) stim_channelrzNo events are foundrz>Reject is already applied. Use reject=None in the constructor.z'Not a Raw or Epochs or Evoked (got %s).)r intrceilinfo ValueErrorrrr isinstancerrlenr_datarr#rZreject RuntimeErrortminrfirstr TypeErrortype)insteventsZevent_idr1Ztmaxr r(rZs_startZs_endrZ events_selZ event_startrZ event_idxrrZe_startepochrrrr'&sP#           r')NNr$r%rNN)numpyrZevokedrZepochsrioreventrZio.pickrutilsrr r rr#r'rrrrs