AV1Master/src/workunit.rs

use serde::{Serialize, Deserialize};
use uuid::Uuid;

#[derive(Default, Debug, Serialize, Deserialize, Clone)]
pub struct WUnit {
    pub id: Uuid,
    pub description: WDesc,
    pub status: EStatus
}
impl WUnit {
    pub fn new(id: Uuid, description: WDesc) -> Self {
        WUnit {
            id: id,
            description: description,
            status: EStatus::Queued
        }
    }
}

#[derive(Default, Debug, Serialize, Deserialize, Clone)]
pub struct WDesc {
    pub file_url: String,
    pub priority: u16,
    pub length: u32,
    pub options: EOptions,
}
impl WDesc {
    pub fn new(file_url: &str, priority: Option<u16>, length: u32, options: Option<EOptions>) -> Self {
        WDesc {
            file_url: file_url.to_string(),
            priority: priority.unwrap_or(0),
            length: length,
            options: options.unwrap_or(EOptions::default()),
        }
    }
}


#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct EOptions {
    pub mode: EMode,
    pub resolution: Option<Resolution>,
    pub color_depth: EColorDepth,
    pub enable_fwd_keyframe: bool,
    pub kf_min_dist: Option<u16>,
    pub kf_max_dist: Option<u16>,
    pub two_pass: bool,
    pub speed: u8
}
impl Default for EOptions {
    fn default() -> Self {
        EOptions{
            mode: EMode::default(),
            resolution: Option::default(),
            color_depth: EColorDepth::default(),
            enable_fwd_keyframe: true,
            kf_min_dist: Option::default(),
            kf_max_dist: Option::default(),
            two_pass: false,
            speed: 3
        }
    }
}

#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub enum EMode {
    // Quality (CRF), Constrained Quality, Variable Bitrate, Constant Bitrate
    Q(u8),
    CQ(u8),
    VBR(u32),
    CBR(u32)
}
impl Default for EMode {
    fn default() -> Self {
        EMode::Q(30)
    }
}

#[derive(Default, Debug, Serialize, Deserialize, Clone)]
pub struct Resolution {
    pub width: u16,
    pub height: u16
}

#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub enum EColorDepth {
    Eight = 8,
    Ten = 10,
    Twelve = 12
}
impl Default for EColorDepth {
    fn default() -> Self {
        EColorDepth::Ten
    }
}

#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub enum EStatus {
    Queued,
    Reserved(Client),
    Completed
}
impl Default for EStatus {
    fn default() -> Self {
        EStatus::Queued
    }
}

type Client = String;