gst-plugins-rs/audio/qoa/src/typefind.rs

use crate::qoa::{FrameHeader, QOA_MIN_FILESIZE};
use gst::glib;
use gst::{TypeFind, TypeFindProbability};
use qoaudio::QOA_HEADER_SIZE;

pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
    TypeFind::register(
        Some(plugin),
        "qoa_typefind",
        gst::Rank::None,
        Some("qoa"),
        Some(&gst::Caps::builder("audio/x-qoa").build()),
        |typefind| {
            if let Some(data) = typefind.peek(0, QOA_MIN_FILESIZE as u32) {
                let magic = u32::from_be_bytes(data[0..4].try_into().unwrap());
                if magic == qoaudio::QOA_MAGIC {
                    typefind.suggest(
                        TypeFindProbability::Maximum,
                        &gst::Caps::builder("audio/x-qoa").build(),
                    );
                } else {
                    let raw_header =
                        u64::from_be_bytes(data[0..QOA_HEADER_SIZE].try_into().unwrap());
                    if let Ok(_header) = FrameHeader::parse(raw_header) {
                        typefind.suggest(
                            TypeFindProbability::Likely,
                            &gst::Caps::builder("audio/x-qoa").build(),
                        );
                    }
                }
            }
        },
    )
}