Skip to content
Snippets Groups Projects
Commit 1eefe44b authored by Blaß, Michael's avatar Blaß, Michael :speech_balloon:
Browse files

Fixes for TimbreTrackAlt.

parent d069820e
No related branches found
No related tags found
No related merge requests found
...@@ -16,7 +16,7 @@ from apollon.tools import time_stamp ...@@ -16,7 +16,7 @@ from apollon.tools import time_stamp
import comsar import comsar
from comsar._tracks.utilities import (TrackMeta, TrackResult, from comsar._tracks.utilities import (TrackMeta, TrackResult,
TimbreTrackParams) TimbreTrackParams, TimbreTrackAltParams)
STFT_DEFAULT = container.StftParams(fps=44100, window='hamming', n_fft=None, STFT_DEFAULT = container.StftParams(fps=44100, window='hamming', n_fft=None,
...@@ -125,9 +125,8 @@ class TimbreTrackAlt: ...@@ -125,9 +125,8 @@ class TimbreTrackAlt:
""" """
Args: Args:
""" """
self.params = TimbreTrackParams(stft_params or STFT_DEFAULT, self.params = TimbreTrackAltParams(stft_params or STFT_DEFAULT,
corr_dim_params or CORR_DIM_DEFAULT, corr_dim_params or CORR_DIM_DEFAULT)
corr_gram_params or CORR_GRAM_DEFAULT)
self.cutter = Segmentation(self.params.stft.n_perseg, self.cutter = Segmentation(self.params.stft.n_perseg,
self.params.stft.n_overlap, self.params.stft.n_overlap,
......
...@@ -42,6 +42,13 @@ class TimbreTrackParams(TrackParams): ...@@ -42,6 +42,13 @@ class TimbreTrackParams(TrackParams):
corr_gram: signal.container.CorrGramParams corr_gram: signal.container.CorrGramParams
@dataclass
class TimbreTrackAltParams(TrackParams):
"""Parameter set for TimbreTrack"""
stft: signal.container.StftParams
corr_dim: signal.container.CorrDimParams
class TrackResult: class TrackResult:
"""Provide track results.""" """Provide track results."""
def __init__(self, meta: TrackMeta, params: TrackParams, def __init__(self, meta: TrackMeta, params: TrackParams,
......
from . _tracks.timbre import TimbreTrack from . _tracks.timbre import TimbreTrack
from . _tracks.timbre import TimbreTrackAlt
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment