81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
|
"""Локальная WAV-запись обеих сторон учебного голосового вызова."""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
import wave
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from pathlib import Path
|
||
|
|
from uuid import UUID
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from app.config import get_settings
|
||
|
|
|
||
|
|
TARGET_RATE = 16_000
|
||
|
|
|
||
|
|
|
||
|
|
def recording_path(session_id: UUID) -> Path:
|
||
|
|
return Path(get_settings().recordings_dir).resolve() / f"{session_id}.wav"
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class _Segment:
|
||
|
|
offset: int
|
||
|
|
samples: np.ndarray
|
||
|
|
|
||
|
|
|
||
|
|
class CallRecorder:
|
||
|
|
"""Смешивает PCM16 разных частот на монотаймлайн 16 кГц.
|
||
|
|
|
||
|
|
Вход курсанта приходит по 16 кГц, TTS звонящего — по 24 кГц. Метка
|
||
|
|
monotonic сохраняет паузы и взаимное расположение реплик; системные часы
|
||
|
|
и изменение времени на хосте на запись не влияют.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, path: Path, *, clock=time.monotonic) -> None:
|
||
|
|
self.path = path
|
||
|
|
self._clock = clock
|
||
|
|
self._started = clock()
|
||
|
|
self._segments: list[_Segment] = []
|
||
|
|
self._finalized = False
|
||
|
|
|
||
|
|
def add_pcm(self, pcm: bytes, *, sample_rate: int) -> None:
|
||
|
|
if self._finalized or not pcm or sample_rate <= 0 or len(pcm) % 2:
|
||
|
|
return
|
||
|
|
source = np.frombuffer(pcm, dtype="<i2").astype(np.int32)
|
||
|
|
if source.size == 0:
|
||
|
|
return
|
||
|
|
if sample_rate != TARGET_RATE:
|
||
|
|
length = max(1, round(source.size * TARGET_RATE / sample_rate))
|
||
|
|
points = np.linspace(0, source.size - 1, length)
|
||
|
|
source = np.rint(np.interp(points, np.arange(source.size), source)).astype(np.int32)
|
||
|
|
offset = max(0, round((self._clock() - self._started) * TARGET_RATE))
|
||
|
|
self._segments.append(_Segment(offset=offset, samples=source))
|
||
|
|
|
||
|
|
def finalize(self) -> Path | None:
|
||
|
|
if self._finalized:
|
||
|
|
return self.path if self.path.is_file() else None
|
||
|
|
self._finalized = True
|
||
|
|
if not self._segments:
|
||
|
|
return None
|
||
|
|
total = max(item.offset + item.samples.size for item in self._segments)
|
||
|
|
mixed = np.zeros(total, dtype=np.int32)
|
||
|
|
for item in self._segments:
|
||
|
|
mixed[item.offset:item.offset + item.samples.size] += item.samples
|
||
|
|
pcm = np.clip(mixed, -32768, 32767).astype("<i2").tobytes()
|
||
|
|
|
||
|
|
self.path.parent.mkdir(parents=True, exist_ok=True)
|
||
|
|
temporary = self.path.with_suffix(".wav.tmp")
|
||
|
|
with wave.open(str(temporary), "wb") as target:
|
||
|
|
target.setnchannels(1)
|
||
|
|
target.setsampwidth(2)
|
||
|
|
target.setframerate(TARGET_RATE)
|
||
|
|
target.writeframes(pcm)
|
||
|
|
os.replace(temporary, self.path)
|
||
|
|
return self.path
|
||
|
|
|
||
|
|
|
||
|
|
def start_recording(session_id: UUID) -> CallRecorder | None:
|
||
|
|
if not get_settings().record_calls:
|
||
|
|
return None
|
||
|
|
return CallRecorder(recording_path(session_id))
|