U f{jc@sFdZddlZddlmZmZmZdddZedd d Zdd d Z dS)z+Utility functions to baseline-correct data.N)loggerverbose _check_optionmeanc Cs2|dk r*td|ddddddgd |}nd }|S) zLog the rescaling method.Nmodelogratioratiozscorerpercent zlogratioz'Applying baseline correction (mode: %s)zNo baseline correction applied)r)baselinermsgrK/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/baseline.py _log_rescale s rTc s|r |}|dk r(t||}t||dks>|jddkrB|S|\}} |dkrXdn8t||kdtdkrtd|ft d| dkrt|ndd} nZ|dkrRdd} nF|dkrfdd} n2|dkrfdd} n|dkrfdd} |dkr| || n2|D],} | |d | ddf| d | ddfq|S)a!Rescale (baseline correct) data. Parameters ---------- data : array It can be of any shape. The only constraint is that the last dimension should be time. times : 1D array Time instants is seconds. %(baseline_rescale)s mode : 'mean' | 'ratio' | 'logratio' | 'percent' | 'zscore' | 'zlogratio' Perform baseline correction by - subtracting the mean of baseline values ('mean') - dividing by the mean of baseline values ('ratio') - dividing by the mean of baseline values and taking the log ('logratio') - subtracting the mean of baseline values followed by dividing by the mean of baseline values ('percent') - subtracting the mean of baseline values and dividing by the standard deviation of baseline values ('zscore') - dividing by the mean of baseline values, taking the log, and dividing by the standard deviation of log baseline values ('zlogratio') copy : bool Whether to return a new instance or modify in place. picks : list of int | None Data to process along the axis=-2 (None, default, processes all). %(verbose)s Returns ------- data_scaled: array Array of same shape as data after rescaling. FNrz9bmin is too large (%s), it exceeds the largest time valuezBbmax is too small (%s), it is smaller than the smallest time valuerz3Bad rescaling slice (%s:%s) from time values %s, %s.TZaxisZkeepdimsrcSs ||8}dSNrdmrrrfun`szrescale..funr cSs ||}dSrrrrrrrcsrcSs||}tj||ddS)Nout)nplog10rrrrrfsr cSs||8}||}dSrrrrrrrjsr cs,||8}|tj|dfddd}dS)N.rTr)rstdrZimaxZiminrrrnsr cs:||}tj||d|tj|dfddd}dS)Nr.rTr)rrrrrrrrrs) copyrrinfoshaperwherelen ValueErrorintr) datatimesr rrZpicksrrZbminbmaxrrpirrrrescales\'                   *r*raisec CsF|dkr dSt|tr"t|dkr2td|d|d|d}}dt|}|dkrh|dkrhtd |\}}|dkr||}t|}|dkr|}t|}||krtd ||f|||ks|||kr>d |d |d |d |d } |dkrt| nD|dkrt| n.|dkr>|||kr,|}|||kr>|}||fS)aCheck if the baseline is valid, and adjust it if requested. ``None`` values inside the baseline parameter will be replaced with ``times[0]`` and ``times[-1]``. Parameters ---------- baseline : tuple | None Beginning and end of the baseline period, in seconds. If ``None``, assume no baseline and return immediately. times : array The time points. sfreq : float The sampling rate. on_baseline_outside_data : 'raise' | 'info' | 'adjust' What do do if the baseline period exceeds the data. If ``'raise'``, raise an exception (default). If ``'info'``, log an info message. If ``'adjust'``, adjust the baseline such that it's within the data range again. Returns ------- (baseline_tmin, baseline_tmax) | None The baseline with ``None`` values replaced with times, and with adjusted times if ``on_baseline_outside_data='adjust'``; or ``None`` if the ``baseline`` parameter is ``None``. Nz `baseline=z=` is an invalid argument, must be a tuple of length 2 or Nonerrg?)NrzOBaseline interval is only one sample. Use `baseline=(0, 0)` if this is desired.z5Baseline min (%s) must be less than baseline max (%s)zBaseline interval [z, z!] sec is outside of epochs data [z$] sec. Epochs were probably cropped.r+r adjust) isinstancetupler#r$floatrr ) r r'ZsfreqZon_baseline_outside_dataZtminZtmaxZtstepZ baseline_tminZ baseline_tmaxrrrr_check_baselinesB     r1)r)rTNN)r+) __doc__numpyrutilsrrrrr*r1rrrrs  g