mirror
/
corrscope
kopia lustrzana https://github.com/corrscope/corrscope
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

Debug period estimator

debug
nyanpasu64 2022-03-16 12:52:49 -07:00
rodzic 9f512babca
commit ce911d656d
2 zmienionych plików z 41 dodań i 40 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

74
corrscope/triggers.py
Wyświetl plik

@ -625,33 +625,33 @@ class CorrelationTrigger(MainTrigger):
edge_score *= -cfg.edge_strength edge_score *= -cfg.edge_strength
peaks += edge_score peaks += edge_score
self.custom_line( # self.custom_line(
"corr_kernel", # "corr_kernel",
corr_kernel, # corr_kernel,
np.arange(-self.A, self.B) * stride, # np.arange(-self.A, self.B) * stride,
False, # False,
) # )
self.custom_line( # self.custom_line(
"peak_kernel", # "peak_kernel",
self._corr_buffer, # self._corr_buffer,
np.arange(-self.A, self.B) * stride, # np.arange(-self.A, self.B) * stride,
False, # False,
) # )
self.custom_line( # self.custom_line(
"corr", # "corr",
corr, # corr,
np.arange(trigger_begin, trigger_begin + stride * len(corr), stride), # np.arange(trigger_begin, trigger_begin + stride * len(corr), stride),
True, # True,
False, # False,
) # )
self.custom_line( # self.custom_line(
"peaks", # "peaks",
peaks, # peaks,
np.arange(trigger_begin, trigger_begin + stride * len(corr), stride), # np.arange(trigger_begin, trigger_begin + stride * len(corr), stride),
True, # True,
False, # False,
) # )
# Don't pick peaks more than `period * trigger_radius_periods` away from the # Don't pick peaks more than `period * trigger_radius_periods` away from the
# center. # center.
@ -690,17 +690,17 @@ class CorrelationTrigger(MainTrigger):
corr[1:][peaks[1:] < peaks[:-1]] = min_corr corr[1:][peaks[1:] < peaks[:-1]] = min_corr
corr[0] = corr[-1] = min_corr corr[0] = corr[-1] = min_corr
self.custom_line( # self.custom_line(
"corr_peaks", # "corr_peaks",
corr, # corr,
np.arange( # np.arange(
trigger_begin + stride * (begin_offset), # trigger_begin + stride * (begin_offset),
trigger_begin + stride * (begin_offset + len(corr)), # trigger_begin + stride * (begin_offset + len(corr)),
stride, # stride,
), # ),
True, # True,
False, # False,
) # )
# Find optimal offset # Find optimal offset
peak_offset = np.argmax(corr) + begin_offset # type: int peak_offset = np.argmax(corr) + begin_offset # type: int

7
corrscope/utils/trigger_util.py
Wyświetl plik

@ -107,9 +107,10 @@ def get_period(
else: else:
peakX = temp_peakX peakX = temp_peakX
# self.custom_line( self.custom_vline("minX", minX, False)
# "autocorr", corr, np.arange(len(corr)) * self._stride, False, False self.custom_line(
# ) "autocorr", corr, np.arange(len(corr)) * self._stride, False, False
)
return int(peakX) return int(peakX)