One-KVM/src/video/encoder/mod.rs

//! Video encoder implementations
//!
//! This module provides video encoding capabilities including:
//! - JPEG encoding for raw frames (YUYV, NV12, etc.)
//! - H264 encoding (hardware + software)
//! - H265 encoding (hardware + software)
//! - VP8 encoding (hardware + software)
//! - VP9 encoding (hardware + software)
//! - WebRTC video codec abstraction
//! - Encoder registry for automatic detection

use hwcodec::common::DataFormat;
use hwcodec::ffmpeg_ram::CodecInfo;

pub mod codec;
pub mod h264;
pub mod h265;
pub mod jpeg;
pub mod registry;
pub mod self_check;
pub mod traits;
pub mod vp8;
pub mod vp9;

// Core traits and types
pub use traits::{
    BitratePreset, EncodedFormat, EncodedFrame, Encoder, EncoderConfig, EncoderFactory,
};

// WebRTC codec abstraction
pub use codec::{CodecFrame, VideoCodec, VideoCodecConfig, VideoCodecFactory, VideoCodecType};

// Encoder registry
pub use registry::{AvailableEncoder, EncoderBackend, EncoderRegistry, VideoEncoderType};
pub use self_check::{
    build_hardware_self_check_runtime_error, run_hardware_self_check, VideoEncoderSelfCheckCell,
    VideoEncoderSelfCheckCodec, VideoEncoderSelfCheckResponse, VideoEncoderSelfCheckRow,
};

// H264 encoder
pub use h264::{H264Config, H264Encoder, H264EncoderType, H264InputFormat};

// H265 encoder
pub use h265::{H265Config, H265Encoder, H265EncoderType, H265InputFormat};

// VP8 encoder
pub use vp8::{VP8Config, VP8Encoder, VP8EncoderType, VP8InputFormat};

// VP9 encoder
pub use vp9::{VP9Config, VP9Encoder, VP9EncoderType, VP9InputFormat};

// JPEG encoder
pub use jpeg::JpegEncoder;

pub(crate) fn select_codec_for_format<F>(
    encoders: &[CodecInfo],
    format: DataFormat,
    preferred: F,
) -> Option<&CodecInfo>
where
    F: Fn(&CodecInfo) -> bool,
{
    encoders
        .iter()
        .find(|codec| codec.format == format && preferred(codec))
        .or_else(|| encoders.iter().find(|codec| codec.format == format))
}

pub(crate) fn detect_best_codec_for_format<T, F>(
    encoders: &[CodecInfo],
    format: DataFormat,
    preferred: F,
) -> Option<(T, String)>
where
    T: From<EncoderBackend>,
    F: Fn(&CodecInfo) -> bool,
{
    select_codec_for_format(encoders, format, preferred).map(|codec| {
        (
            T::from(EncoderBackend::from_codec_name(&codec.name)),
            codec.name.clone(),
        )
    })
}