effy 0.10.0

A small and friendly terminal FFmpeg helper that simplifies common tasks
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use std::sync::Arc;

use tracing::debug;

use crate::{
    model::{InputConstraints, InputType},
    params::{
        Parameter, ParameterData, PresetParameter, SelectOption,
        macros::select_non_default_custom_value,
    },
    visitors::{CommandBuilder, HWAccel, VisitorContext},
};

pub(crate) struct VideoScale;

impl VideoScale {
    pub(crate) const ID: &'static str = "scale";
    pub(crate) const NAME: &'static str = "Video Scale";
    const DEFAULT: &'static str = "0";
    const VARIANTS: [&str; 7] = ["144", "240", "360", "0", "480", "720", "1080"];

    pub fn new_parameter() -> Parameter {
        Parameter::new(
            Self::ID,
            Self::NAME,
            ParameterData::CustomSelect {
                options: SelectOption::from_slice(&Self::VARIANTS),
                selected_index: 3,
                value: Self::DEFAULT.to_owned(),
                constraints: InputConstraints {
                    length: 4,
                    input_type: InputType::PositiveInteger,
                },
                validator: Arc::new(Self::validate),
                formatter: Some(Arc::new(Self::format_value)),
            },
        )
    }

    fn validate(value: &str) -> Result<String, &str> {
        Self::validate_range(value)
            .and_then(Self::validate_divisible_by_2)
            .map(|num| num.to_string())
    }

    fn validate_range(value: &str) -> Result<i32, &str> {
        if let Ok(num) = value.parse::<i32>()
            && (num == 0 || (64..=8192).contains(&num))
        {
            Ok(num)
        } else {
            Err("Invalid value. Expected a number in range 64..8192, or 0 - original")
        }
    }

    fn validate_divisible_by_2<'a>(num: i32) -> Result<i32, &'a str> {
        if num % 2 == 0 {
            Ok(num)
        } else {
            Err("Invalid value. Expected a number divisible by 2")
        }
    }

    fn format_value(value: &str) -> String {
        if value == Self::DEFAULT {
            "original".to_owned()
        } else {
            format!("{}p", value)
        }
    }

    pub fn build_command(cb: &mut CommandBuilder, data: &ParameterData) {
        if let Some(value) = select_non_default_custom_value!(data) {
            debug!(value, "build_command");
            // Use nvenc cuda scale only if there is no other video filter
            #[cfg(target_os = "windows")]
            {
                if (cb.hwaccel == HWAccel::Nvenc) && (cb.video_filters.is_empty()) {
                    cb.video_filters.push(format!("scale_cuda=-2:{}", &value));
                } else {
                    cb.video_filters.push(format!("scale=-2:{}", &value));
                }
            }

            #[cfg(target_os = "linux")]
            {
                if (cb.hwaccel == HWAccel::Nvenc) && (cb.video_filters.is_empty()) {
                    cb.video_filters.push(format!("scale_cuda=-2:{}", value));
                } else if cb.hwaccel == HWAccel::Vaapi {
                    cb.video_filters.push(format!("scale_vaapi=-2:{}", value));
                } else {
                    cb.video_filters.push(format!("scale=-2:{}", value));
                }
            }

            #[cfg(not(any(target_os = "windows", target_os = "linux")))]
            {
                cb.video_filters.push(format!("scale=-2:{}", &value));
            }
        }
    }
}

impl<'a> PresetParameter<'a> for VideoScale {
    fn apply_preset(_ctx: &VisitorContext, data: &mut ParameterData, preset_value: &str) {
        if Self::validate(preset_value).is_ok() {
            Self::set_parameter_value(data, preset_value);
        }
    }

    fn save_preset(_ctx: &VisitorContext, data: &'a ParameterData) -> Option<String> {
        select_non_default_custom_value!(data).cloned()
    }
}