Skip to content

Commit

Permalink
add preprocessing names_to_funcitons dict
Browse files Browse the repository at this point in the history
  • Loading branch information
chrishalcrow committed Dec 6, 2024
1 parent 8252f8f commit 8436eb2
Showing 1 changed file with 40 additions and 0 deletions.
40 changes: 40 additions & 0 deletions src/spikeinterface/preprocessing/preprocessinglist.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,46 @@
from .astype import AstypeRecording, astype
from .unsigned_to_signed import UnsignedToSignedRecording, unsigned_to_signed

from .motion import correct_motion

pp_name_to_function = {
# filter stuff
"filter": filter,
"bandpass_filter": bandpass_filter,
"notch_filter": notch_filter,
"highpass_filter": highpass_filter,
"gaussian_filter": gaussian_filter,
# gain offset stuff
"normalize_by_quantile": normalize_by_quantile,
"scale": scale,
"zscore": zscore,
"center": center,
# decorrelation stuff
"whiten": whiten,
# re-reference
"common_reference": common_reference,
"phase_shift": phase_shift,
# misc
"rectify": rectify,
"clip": clip,
"blank_staturation": blank_staturation,
"silence_periods": silence_periods,
"remove_artifacts": remove_artifacts,
"zero_channel_pad": zero_channel_pad,
"deepinterpolate": deepinterpolate,
"resample": resample,
"decimate": decimate,
"highpass_spatial_filter": highpass_spatial_filter,
"interpolate_bad_channels": interpolate_bad_channels,
"depth_order": depth_order,
"average_across_direction": average_across_direction,
"directional_derivative": directional_derivative,
"astype": astype,
"unsigned_to_signed": unsigned_to_signed,
"unsigned_to_signed": unsigned_to_signed,
# motion correction
"correct_motion": correct_motion,
}

pp_function_to_class = {
# filter stuff
Expand Down

0 comments on commit 8436eb2

Please sign in to comment.