import pandas as pd

from opendbm.api_lib.model import AudioModel
from opendbm.dbm_lib.dbm_features.raw_features.audio.jitter import run_jitter


class Jitter(AudioModel):
    def __init__(self):
        super().__init__()
        self._params = ["aco_jitter"]

    @AudioModel.prep_func
    def _fit_transform(self, path, **kwargs):
        return run_jitter(path, ".", self.r_config, save=False, ff_df=kwargs["ff_df"])