Modify original data by using parameter data.
(
data, window, picks, first_samp, last_samp, base_tmin, base_tmax, mode
)
| 21 | |
| 22 | |
| 23 | def _fix_artifact( |
| 24 | data, window, picks, first_samp, last_samp, base_tmin, base_tmax, mode |
| 25 | ): |
| 26 | """Modify original data by using parameter data.""" |
| 27 | if mode == "linear": |
| 28 | x = np.array([first_samp, last_samp]) |
| 29 | f = interp1d(x, data[:, (first_samp, last_samp)][picks]) |
| 30 | xnew = np.arange(first_samp, last_samp) |
| 31 | interp_data = f(xnew) |
| 32 | data[picks, first_samp:last_samp] = interp_data |
| 33 | if mode == "window": |
| 34 | data[picks, first_samp:last_samp] = ( |
| 35 | data[picks, first_samp:last_samp] * window[np.newaxis, :] |
| 36 | ) |
| 37 | if mode == "constant": |
| 38 | data[picks, first_samp:last_samp] = data[picks, base_tmin:base_tmax].mean( |
| 39 | axis=1 |
| 40 | )[:, None] |
| 41 | |
| 42 | |
| 43 | @fill_doc |
no test coverage detected