sound.py

from __future__ import annotations
from typing import NamedTuple

import numpy as np
from numpy import ndarray


class Sound(NamedTuple):
    data: ndarray
    sr: int

    @property
    def duration(self) -> float:
        return self.data.shape[-1] / self.sr

    @staticmethod
    def equals(a: "Sound", b: "Sound") -> bool:
        return a.sr == b.sr and a.data is b.data

    @staticmethod
    def equals_content(a: "Sound", b: "Sound", approx: bool = False) -> bool:
        if Sound.equals(a, b):
            return True
        if a.sr != b.sr:
            return False
        if approx:
            return np.allclose(a.data, b.data)
        else:
            return np.array_equal(a.data, b.data)