//! Отпечаток голоса: вектор модели WeSpeaker для куска звука 16 кГц.

use std::ffi::CString;
use std::path::Path;

#[derive(Debug, thiserror::Error)]
pub enum Error {
    #[error("модель голосов не найдена: {0}")]
    NoModel(String),
    #[error("модель голосов не загрузилась: {0}")]
    Load(String),
}

/// Снимает отпечатки. Держит модель в памяти; создаётся на одну расшифровку.
pub struct Extractor {
    raw: *const sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractor,
    dim: usize,
}

// Указатель используется из одного потока за раз: отпечатки снимаются
// последовательно в `spawn_blocking`, модель сама раскладывает работу по ядрам.
unsafe impl Send for Extractor {}

impl Extractor {
    pub fn open(model: &Path, threads: usize) -> Result<Self, Error> {
        if !model.is_file() {
            return Err(Error::NoModel(model.display().to_string()));
        }
        let path =
            CString::new(model.to_string_lossy().as_bytes()).map_err(|e| Error::Load(e.to_string()))?;
        let cpu = CString::new("cpu").unwrap();
        let config = sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorConfig {
            model: path.as_ptr(),
            num_threads: threads.clamp(1, 16) as i32,
            debug: 0,
            provider: cpu.as_ptr(),
        };
        let raw = unsafe { sherpa_rs_sys::SherpaOnnxCreateSpeakerEmbeddingExtractor(&config) };
        if raw.is_null() {
            return Err(Error::Load(model.display().to_string()));
        }
        let dim = unsafe { sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorDim(raw) } as usize;
        Ok(Self { raw, dim })
    }

    /// Отпечаток куска (16 кГц, моно). Пусто, если звука слишком мало, чтобы
    /// модель что-то сказала.
    pub fn embed(&mut self, samples: &[i16]) -> Vec<f32> {
        let x: Vec<f32> = samples.iter().map(|&s| s as f32 / 32768.0).collect();
        unsafe {
            let stream = sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorCreateStream(self.raw);
            sherpa_rs_sys::SherpaOnnxOnlineStreamAcceptWaveform(stream, 16_000, x.as_ptr(), x.len() as i32);
            sherpa_rs_sys::SherpaOnnxOnlineStreamInputFinished(stream);
            let out = if sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorIsReady(self.raw, stream) != 0 {
                let p = sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorComputeEmbedding(self.raw, stream);
                let v = std::slice::from_raw_parts(p, self.dim).to_vec();
                sherpa_rs_sys::SherpaOnnxSpeakerEmbeddingExtractorDestroyEmbedding(p);
                v
            } else {
                Vec::new()
            };
            sherpa_rs_sys::SherpaOnnxDestroyOnlineStream(stream);
            out
        }
    }
}

impl Drop for Extractor {
    fn drop(&mut self) {
        unsafe { sherpa_rs_sys::SherpaOnnxDestroySpeakerEmbeddingExtractor(self.raw) };
    }
}