def get_re_params(self) -> REParams:
"""
Gets the ridge extraction params which are passed to the algorithm.
"""
return create(
signals=self.signals,
params_type=REParams,
intervals=self.view.get_interval_tuples(),
# fmin=fmin,
# fmax=fmax,
f0=self.view.get_f0(),
# Only one of these two will be used, depending on the selected transform.
window=self.view.get_wt_wft_type(),
wavelet=self.view.get_wt_wft_type(),
cut_edges=self.view.get_cut_edges(),
preprocess=self.view.get_preprocess(),
transform=self.view.get_transform_type(),
)
def get_params(self, all_signals=True) -> PCParams:
if all_signals:
self.signals_calc = self.signals
else:
self.signals_calc = self.signals.only(self.selected_signal_name)
return create(
params_type=PCParams,
signals=self.signals_calc,
fmin=self.view.get_fmin(),
fmax=self.view.get_fmax(),
f0=self.view.get_f0(),
wavelet=self.view.get_wt_wft_type(),
cut_edges=self.view.get_cut_edges(),
preprocess=self.view.get_preprocess(),
transform="wt",
surr_count=self.view.get_surr_count(),
surr_method=self.view.get_surr_method(),
surr_enabled=self.view.get_surr_enabled(),
)
def get_params(self, all_signals: bool = True) -> DHParams:
if all_signals:
self.signals_calc = self.signals
else:
self.signals_calc = self.signals.only(self.selected_signal_name)
fmin = self.view.get_fmin()
fmax = self.view.get_fmax()
scale_min = 1 / fmax
scale_max = 1 / fmin
return create(
signals=self.signals_calc,
params_type=DHParams,
scale_min=scale_min,
scale_max=scale_max,
sigma=self.view.get_sigma(),
time_res=self.view.get_time_res(),
surr_count=self.view.get_surr_count(),
crop=self.view.get_cut_edges(),
)
def get_params(self, all_signals=True) -> TFParams:
"""
Creates the parameters to use when performing the calculations.
"""
if all_signals:
self.signals_calc = self.signals
else:
self.signals_calc = self.signals.only(self.selected_signal_name)
return create(
params_type=TFParams,
signals=self.signals_calc,
fmin=self.view.get_fmin(),
fmax=self.view.get_fmax(),
f0=self.view.get_f0(),
# Only one of these two will be used, depending on the selected transform.
window=self.view.get_wt_wft_type(),
wavelet=self.view.get_wt_wft_type(),
cut_edges=self.view.get_cut_edges(),
preprocess=self.view.get_preprocess(),
transform=self.view.get_transform_type(),
)