mirrors/gst-plugins-rs
mirrors/gst-plugins-rs
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git synced 2025-02-18 05:45:14 +00:00
Code Issues Projects Releases Wiki Activity

rav1enc: Expose various new settings

Browse source
This commit is contained in:
Sebastian Dröge 2022-02-06 21:16:09 +02:00 committed by Sebastian Dröge
parent 12f2b9282a
commit 1451e267cc
1 changed files with 142 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

142
video/rav1e/src/rav1enc/imp.rs
View file

@ -19,15 +19,29 @@ use rav1e::config;
use rav1e::data; use rav1e::data;
use std::sync::Mutex; use std::sync::Mutex;
#[derive(Debug, PartialEq, Eq, Clone, Copy, glib::Enum)]
#[repr(u32)]
#[enum_type(name = "GstRav1eEncTune")]
pub enum Tune {
Psnr,
Psychovisual,
}
const DEFAULT_SPEED_PRESET: u32 = 6; const DEFAULT_SPEED_PRESET: u32 = 6;
const DEFAULT_LOW_LATENCY: bool = false; const DEFAULT_LOW_LATENCY: bool = false;
const DEFAULT_MIN_KEY_FRAME_INTERVAL: u64 = 12; const DEFAULT_MIN_KEY_FRAME_INTERVAL: u64 = 12;
const DEFAULT_MAX_KEY_FRAME_INTERVAL: u64 = 240; const DEFAULT_MAX_KEY_FRAME_INTERVAL: u64 = 240;
const DEFAULT_SWITCH_FRAME_INTERVAL: u64 = 0;
const DEFAULT_BITRATE: i32 = 0; const DEFAULT_BITRATE: i32 = 0;
const DEFAULT_QUANTIZER: usize = 100; const DEFAULT_QUANTIZER: usize = 100;
const DEFAULT_MIN_QUANTIZER: u8 = 0;
const DEFAULT_TILE_COLS: usize = 0; const DEFAULT_TILE_COLS: usize = 0;
const DEFAULT_TILE_ROWS: usize = 0; const DEFAULT_TILE_ROWS: usize = 0;
const DEFAULT_TILES: usize = 0; const DEFAULT_TILES: usize = 0;
const DEFAULT_RDO_LOOKAHEAD_FRAMES: i32 = -1;
const DEFAULT_TUNE: Tune = Tune::Psychovisual;
const DEFAULT_RESERVOIR_FRAME_DELAY: i32 = i32::MIN;
const DEFAULT_ERROR_RESILIENT: bool = false;
const DEFAULT_THREADS: usize = 0; const DEFAULT_THREADS: usize = 0;
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
@ -36,11 +50,17 @@ struct Settings {
low_latency: bool, low_latency: bool,
min_key_frame_interval: u64, min_key_frame_interval: u64,
max_key_frame_interval: u64, max_key_frame_interval: u64,
switch_frame_interval: u64,
bitrate: i32, bitrate: i32,
quantizer: usize, quantizer: usize,
min_quantizer: u8,
tile_cols: usize, tile_cols: usize,
tile_rows: usize, tile_rows: usize,
tiles: usize, tiles: usize,
rdo_lookahead_frames: i32,
tune: Tune,
reservoir_frame_delay: i32,
error_resilient: bool,
threads: usize, threads: usize,
} }
@ -51,11 +71,17 @@ impl Default for Settings {
low_latency: DEFAULT_LOW_LATENCY, low_latency: DEFAULT_LOW_LATENCY,
min_key_frame_interval: DEFAULT_MIN_KEY_FRAME_INTERVAL, min_key_frame_interval: DEFAULT_MIN_KEY_FRAME_INTERVAL,
max_key_frame_interval: DEFAULT_MAX_KEY_FRAME_INTERVAL, max_key_frame_interval: DEFAULT_MAX_KEY_FRAME_INTERVAL,
switch_frame_interval: DEFAULT_SWITCH_FRAME_INTERVAL,
bitrate: DEFAULT_BITRATE, bitrate: DEFAULT_BITRATE,
quantizer: DEFAULT_QUANTIZER, quantizer: DEFAULT_QUANTIZER,
min_quantizer: DEFAULT_MIN_QUANTIZER,
tile_cols: DEFAULT_TILE_COLS, tile_cols: DEFAULT_TILE_COLS,
tile_rows: DEFAULT_TILE_ROWS, tile_rows: DEFAULT_TILE_ROWS,
tiles: DEFAULT_TILES, tiles: DEFAULT_TILES,
rdo_lookahead_frames: DEFAULT_RDO_LOOKAHEAD_FRAMES,
tune: DEFAULT_TUNE,
reservoir_frame_delay: DEFAULT_RESERVOIR_FRAME_DELAY,
error_resilient: DEFAULT_ERROR_RESILIENT,
threads: DEFAULT_THREADS, threads: DEFAULT_THREADS,
} }
} }
@ -251,6 +277,15 @@ impl ObjectImpl for Rav1Enc {
DEFAULT_MAX_KEY_FRAME_INTERVAL, DEFAULT_MAX_KEY_FRAME_INTERVAL,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY, glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
), ),
glib::ParamSpecUInt64::new(
"switch-frame-interval",
"Switch Frame Interval",
"Switch Frame Interval",
0,
std::u64::MAX,
DEFAULT_SWITCH_FRAME_INTERVAL,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecInt::new( glib::ParamSpecInt::new(
"bitrate", "bitrate",
"Bitrate", "Bitrate",
@ -269,6 +304,15 @@ impl ObjectImpl for Rav1Enc {
DEFAULT_QUANTIZER as u32, DEFAULT_QUANTIZER as u32,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY, glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
), ),
glib::ParamSpecUInt::new(
"min-quantizer",
"Min Quantizer",
"Min Quantizer",
0,
std::u8::MAX as u32,
DEFAULT_MIN_QUANTIZER as u32,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecUInt::new( glib::ParamSpecUInt::new(
"tile-cols", "tile-cols",
"Tile Cols", "Tile Cols",
@ -296,6 +340,39 @@ impl ObjectImpl for Rav1Enc {
DEFAULT_TILES as u32, DEFAULT_TILES as u32,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY, glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
), ),
glib::ParamSpecInt::new(
"rdo-lookahead-frames",
"RDO Lookahead Frames",
"RDO Lookahead Frames",
-1,
std::i32::MAX,
DEFAULT_RDO_LOOKAHEAD_FRAMES,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecEnum::new(
"tune",
"Tune",
"Tune",
Tune::static_type(),
DEFAULT_TUNE as i32,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecInt::new(
"reservoir-frame-delay",
"Reservoir Frame Delay",
"Reservoir Frame Delay",
std::i32::MIN,
std::i32::MAX,
DEFAULT_RESERVOIR_FRAME_DELAY,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecBoolean::new(
"error-resilient",
"Error Resilient",
"Error Resilient",
DEFAULT_ERROR_RESILIENT,
glib::ParamFlags::READWRITE | gst::PARAM_FLAG_MUTABLE_READY,
),
glib::ParamSpecUInt::new( glib::ParamSpecUInt::new(
"threads", "threads",
"Threads", "Threads",
@ -335,6 +412,10 @@ impl ObjectImpl for Rav1Enc {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.max_key_frame_interval = value.get().expect("type checked upstream"); settings.max_key_frame_interval = value.get().expect("type checked upstream");
} }
"switch-frame-interval" => {
let mut settings = self.settings.lock().unwrap();
settings.switch_frame_interval = value.get().expect("type checked upstream");
}
"bitrate" => { "bitrate" => {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.bitrate = value.get().expect("type checked upstream"); settings.bitrate = value.get().expect("type checked upstream");
@ -343,6 +424,10 @@ impl ObjectImpl for Rav1Enc {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.quantizer = value.get::<u32>().expect("type checked upstream") as usize; settings.quantizer = value.get::<u32>().expect("type checked upstream") as usize;
} }
"min-quantizer" => {
let mut settings = self.settings.lock().unwrap();
settings.min_quantizer = value.get::<u32>().expect("type checked upstream") as u8;
}
"tile-cols" => { "tile-cols" => {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.tile_cols = value.get::<u32>().expect("type checked upstream") as usize; settings.tile_cols = value.get::<u32>().expect("type checked upstream") as usize;
@ -355,6 +440,22 @@ impl ObjectImpl for Rav1Enc {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.tiles = value.get::<u32>().expect("type checked upstream") as usize; settings.tiles = value.get::<u32>().expect("type checked upstream") as usize;
} }
"rdo-lookahead-frames" => {
let mut settings = self.settings.lock().unwrap();
settings.rdo_lookahead_frames = value.get::<i32>().expect("type checked upstream");
}
"tune" => {
let mut settings = self.settings.lock().unwrap();
settings.tune = value.get::<Tune>().expect("type checked upstream");
}
"reservoir-frame-delay" => {
let mut settings = self.settings.lock().unwrap();
settings.reservoir_frame_delay = value.get::<i32>().expect("type checked upstream");
}
"error-resilient" => {
let mut settings = self.settings.lock().unwrap();
settings.error_resilient = value.get::<bool>().expect("type checked upstream");
}
"threads" => { "threads" => {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.threads = value.get::<u32>().expect("type checked upstream") as usize; settings.threads = value.get::<u32>().expect("type checked upstream") as usize;
@ -381,6 +482,10 @@ impl ObjectImpl for Rav1Enc {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
settings.max_key_frame_interval.to_value() settings.max_key_frame_interval.to_value()
} }
"switch-frame-interval" => {
let settings = self.settings.lock().unwrap();
settings.switch_frame_interval.to_value()
}
"bitrate" => { "bitrate" => {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
settings.bitrate.to_value() settings.bitrate.to_value()
@ -389,6 +494,10 @@ impl ObjectImpl for Rav1Enc {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
(settings.quantizer as u32).to_value() (settings.quantizer as u32).to_value()
} }
"min-quantizer" => {
let settings = self.settings.lock().unwrap();
(settings.min_quantizer as u32).to_value()
}
"tile-cols" => { "tile-cols" => {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
(settings.tile_cols as u32).to_value() (settings.tile_cols as u32).to_value()
@ -401,6 +510,22 @@ impl ObjectImpl for Rav1Enc {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
(settings.tiles as u32).to_value() (settings.tiles as u32).to_value()
} }
"rdo-lookahead-frames" => {
let settings = self.settings.lock().unwrap();
(settings.rdo_lookahead_frames as i32).to_value()
}
"tune" => {
let settings = self.settings.lock().unwrap();
settings.tune.to_value()
}
"reservoir-frame-delay" => {
let settings = self.settings.lock().unwrap();
settings.reservoir_frame_delay.to_value()
}
"error-resilient" => {
let settings = self.settings.lock().unwrap();
settings.error_resilient.to_value()
}
"threads" => { "threads" => {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
(settings.threads as u32).to_value() (settings.threads as u32).to_value()
@ -614,11 +739,28 @@ impl VideoEncoderImpl for Rav1Enc {
low_latency: settings.low_latency, low_latency: settings.low_latency,
min_key_frame_interval: settings.min_key_frame_interval, min_key_frame_interval: settings.min_key_frame_interval,
max_key_frame_interval: settings.max_key_frame_interval, max_key_frame_interval: settings.max_key_frame_interval,
switch_frame_interval: settings.switch_frame_interval,
bitrate: settings.bitrate, bitrate: settings.bitrate,
quantizer: settings.quantizer, quantizer: settings.quantizer,
min_quantizer: settings.min_quantizer,
tile_cols: settings.tile_cols, tile_cols: settings.tile_cols,
tile_rows: settings.tile_rows, tile_rows: settings.tile_rows,
tiles: settings.tiles, tiles: settings.tiles,
rdo_lookahead_frames: if settings.rdo_lookahead_frames < 0 {
config::SpeedSettings::rdo_lookahead_frames(settings.speed_preset as usize)
} else {
settings.rdo_lookahead_frames as usize
},
tune: match settings.tune {
Tune::Psnr => rav1e::prelude::Tune::Psnr,
Tune::Psychovisual => rav1e::prelude::Tune::Psychovisual,
},
reservoir_frame_delay: if settings.reservoir_frame_delay == i32::MIN {
None
} else {
Some(settings.reservoir_frame_delay)
},
error_resilient: settings.error_resilient,
..Default::default() ..Default::default()
}) })
.with_threads(settings.threads); .with_threads(settings.threads);