alpha2-host 0.5.2

//! Decode corruption policy and source-attributed recovery reports.
use alpha2_media::SourceId;
use alpha2_native::{DecodeDiagnostic, DecodeDiagnosticCounts, DecodePolicy};
use serde_json::{Value, json};
use std::collections::BTreeMap;
use std::path::Path;
use std::sync::{Arc, Mutex};

#[cfg(test)]
thread_local! {
    /// Overrides `ALPHA2_DECODE_POLICY` per thread, since parallel tests share the environment.
    pub static TEST_DECODE_POLICY: std::cell::Cell<Option<DecodePolicy>> =
        const { std::cell::Cell::new(None) };
}

#[derive(Debug, Default)]
pub struct ExportDecoding {
    policy: DecodePolicy,
    sources: Mutex<BTreeMap<String, DecodeDiagnosticCounts>>,
    /// Reported after the sources.
    short_of_sessions: Mutex<Option<Value>>,
    /// What a source's decoder leaves out (alpha, extra HEVC layers), reported after the sources.
    dropped_layers: Mutex<BTreeMap<String, alpha2_native::DroppedLayer>>,
}

impl ExportDecoding {
    /// Reads `ALPHA2_DECODE_POLICY` as an export starts.
    ///
    /// # Errors
    /// The variable is set to something other than `strict` or `tolerant`.
    pub fn from_env() -> Result<Self, String> {
        #[cfg(test)]
        if let Some(policy) = TEST_DECODE_POLICY.with(std::cell::Cell::get) {
            return Ok(Self::new(policy));
        }
        DecodePolicy::from_env().map(Self::new)
    }

    pub fn new(policy: DecodePolicy) -> Self {
        Self {
            policy,
            sources: Mutex::default(),
            short_of_sessions: Mutex::default(),
            dropped_layers: Mutex::default(),
        }
    }

    pub fn note_dropped_layer(&self, source: &SourceId, dropped: &alpha2_native::DroppedLayer) {
        if let Ok(mut noted) = self.dropped_layers.lock() {
            noted
                .entry(source.as_str().to_owned())
                .or_insert_with(|| dropped.clone());
        }
    }

    /// Videos past the session count reopen a decoder (and decode from a keyframe) as they come
    /// round.
    pub fn note_decoder_sessions(&self, count: crate::config::DecoderSessionCount, reopens: u64) {
        if count.sessions >= count.peak {
            return;
        }
        let entry = json!({
            "type": "decoderSessionsShort",
            "peak": count.peak,
            "decoderSessions": count.sessions,
            "limit": count.limit.as_str(),
            "reopens": reopens,
            "message": format!(
                "{} videos are read at once but {} decoder sessions fit ({}); {reopens} decoders were reopened",
                count.peak,
                count.sessions,
                count.limit.as_str(),
            ),
        });
        if let Ok(mut noted) = self.short_of_sessions.lock() {
            *noted = Some(entry);
        }
    }

    /// The same policy with its own report, for a pass that decodes again what the export's
    /// decoders report on.
    pub fn unreported(&self) -> Self {
        Self::new(self.policy)
    }

    fn record(&self, source: &str, diagnostic: &DecodeDiagnostic) {
        if let Ok(mut sources) = self.sources.lock() {
            sources
                .entry(source.to_owned())
                .or_default()
                .record(diagnostic);
        }
    }

    /// `[]` when none reported anything.
    pub fn diagnostics_json(&self) -> Value {
        let sources = self
            .sources
            .lock()
            .map(|sources| sources.clone())
            .unwrap_or_default();
        let short_of_sessions = self
            .short_of_sessions
            .lock()
            .ok()
            .and_then(|noted| noted.clone());
        let dropped_layers = self
            .dropped_layers
            .lock()
            .map(|noted| noted.clone())
            .unwrap_or_default();
        Value::Array(
            sources
                .into_iter()
                .map(|(source, counts)| {
                    json!({
                        "source": source,
                        "corruptPackets": counts.corrupt_packets,
                        "videoGaps": counts.video_gaps,
                        "concealedAudioSamples": counts.concealed_audio_samples,
                        "timestampGapSamples": counts.timestamp_gap_samples,
                        "droppedAudioSamples": counts.dropped_audio_samples,
                    })
                })
                .chain(dropped_layers.into_iter().map(|(source, dropped)| {
                    json!({
                        "type": "layerDropped",
                        "code": dropped.code,
                        "source": source,
                        "reason": dropped.reason,
                    })
                }))
                .chain(short_of_sessions)
                .collect(),
        )
    }
}

// Defines `pub fn $name(decoder, source, path, decoding)`, which applies the decode policy and
// counts and emits each diagnostic against `source`.
macro_rules! configure {
    ($name:ident, $ty:ty) => {
        pub fn $name(
            mut decoder: $ty,
            source: &SourceId,
            path: &Path,
            decoding: &Arc<ExportDecoding>,
        ) -> $ty {
            decoder.set_decode_policy(decoding.policy);
            let path = path.display().to_string();
            let source = source.as_str().to_owned();
            let decoding = Arc::clone(decoding);
            decoder.set_decode_reporter(move |diagnostic| {
                decoding.record(&source, &diagnostic);
                let mut report = match diagnostic {
                    DecodeDiagnostic::Corruption(packet) => serde_json::json!({
                        "type": "decodeCorruption", "corruptPackets": 1,
                        "stage": packet.stage, "pts": packet.pts,
                        "timeBase": packet.time_base
                    }),
                    DecodeDiagnostic::VideoGap(gap) => serde_json::json!({
                        "type": "decodeVideoGap", "previousPts": gap.previous_pts,
                        "recoveredPts": gap.recovered_pts, "timeBase": gap.time_base
                    }),
                    DecodeDiagnostic::AudioGap(gap) => serde_json::json!({
                        "type": "decodeAudioConcealment", "reason": gap.reason,
                        "startSample": gap.start_sample, "endSample": gap.end_sample,
                        "samples": gap.end_sample - gap.start_sample, "sampleRate": gap.sample_rate
                    }),
                    DecodeDiagnostic::AudioOverlap(overlap) => serde_json::json!({
                        "type": "decodeAudioOverlap", "startSample": overlap.start_sample,
                        "endSample": overlap.end_sample,
                        "samples": overlap.end_sample - overlap.start_sample,
                        "sampleRate": overlap.sample_rate
                    }),
                };
                report["source"] = serde_json::json!(path);
                report["sourceId"] = serde_json::json!(source);
                alpha2_os::diagnostic!("{report}");
            });
            decoder
        }
    };
}
configure!(audio, alpha2_native::NativeAudioDecoder);
configure!(video, alpha2_native::NativeDecoder);
configure!(export, alpha2_native::NativeExportDecoder);