From fc7016a8c1de9d6bb2d83cbc034c65e9383274ee Mon Sep 17 00:00:00 2001 From: meh Date: Sat, 16 May 2015 22:50:01 +0200 Subject: [PATCH] codec/encoder/video: add some accessors --- src/codec/encoder/video.rs | 203 ++++++++++++++++++++++++++++++++++++- 1 file changed, 200 insertions(+), 3 deletions(-) diff --git a/src/codec/encoder/video.rs b/src/codec/encoder/video.rs index 861ab97..06cb081 100644 --- a/src/codec/encoder/video.rs +++ b/src/codec/encoder/video.rs @@ -1,11 +1,12 @@ use std::ops::Deref; -use libc::c_int; +use libc::{c_int, c_float}; use ffi::*; -use super::Encoder; -use ::{Packet, Error}; +use super::{Encoder, MotionEstimation, Prediction, Comparison, Decision}; +use ::{Packet, Error, Rational}; use ::frame; +use ::format; pub struct Video(pub Encoder); @@ -20,6 +21,202 @@ impl Video { } } } + + pub fn set_width(&mut self, value: usize) { + unsafe { + (*self.ptr).width = value as c_int; + } + } + + pub fn set_height(&mut self, value: usize) { + unsafe { + (*self.ptr).height = value as c_int; + } + } + + pub fn set_gop(&mut self, value: usize) { + unsafe { + (*self.ptr).gop_size = value as c_int; + } + } + + pub fn set_format(&mut self, value: format::Pixel) { + unsafe { + (*self.ptr).pix_fmt = value.into(); + } + } + + pub fn set_motion_estimation(&mut self, value: MotionEstimation) { + unsafe { + (*self.ptr).me_method = value.into(); + } + } + + pub fn set_max_b_frames(&mut self, value: usize) { + unsafe { + (*self.ptr).max_b_frames = value as c_int; + } + } + + pub fn set_b_quant_factor(&mut self, value: f32) { + unsafe { + (*self.ptr).b_quant_factor = value as c_float; + } + } + + pub fn set_b_quant_offset(&mut self, value: f32) { + unsafe { + (*self.ptr).b_quant_offset = value as c_float; + } + } + + pub fn set_i_quant_factor(&mut self, value: f32) { + unsafe { + (*self.ptr).i_quant_factor = value as c_float; + } + } + + pub fn set_i_quant_offset(&mut self, value: f32) { + unsafe { + (*self.ptr).i_quant_offset = value as c_float; + } + } + + pub fn set_lumi_masking(&mut self, value: f32) { + unsafe { + (*self.ptr).lumi_masking = value as c_float; + } + } + + pub fn set_temporal_cplx_masking(&mut self, value: f32) { + unsafe { + (*self.ptr).temporal_cplx_masking = value as c_float; + } + } + + pub fn set_spatial_cplx_masking(&mut self, value: f32) { + unsafe { + (*self.ptr).spatial_cplx_masking = value as c_float; + } + } + + pub fn set_p_masking(&mut self, value: f32) { + unsafe { + (*self.ptr).p_masking = value as c_float; + } + } + + pub fn set_dark_masking(&mut self, value: f32) { + unsafe { + (*self.ptr).dark_masking = value as c_float; + } + } + + pub fn set_prediction(&mut self, value: Prediction) { + unsafe { + (*self.ptr).prediction_method = value.into(); + } + } + + pub fn set_aspect_ratio(&mut self, value: Rational) { + unsafe { + (*self.ptr).sample_aspect_ratio = value.0; + } + } + + pub fn set_me_comparison(&mut self, value: Comparison) { + unsafe { + (*self.ptr).me_cmp = value.into(); + } + } + + pub fn set_me_sub_comparison(&mut self, value: Comparison) { + unsafe { + (*self.ptr).me_sub_cmp = value.into(); + } + } + + pub fn set_mb_comparison(&mut self, value: Comparison) { + unsafe { + (*self.ptr).mb_cmp = value.into(); + } + } + + pub fn set_ildct_comparison(&mut self, value: Comparison) { + unsafe { + (*self.ptr).ildct_cmp = value.into(); + } + } + + pub fn set_dia_size(&mut self, value: usize) { + unsafe { + (*self.ptr).dia_size = value as c_int; + } + } + + pub fn set_last_predictors(&mut self, value: usize) { + unsafe { + (*self.ptr).last_predictor_count = value as c_int; + } + } + + pub fn set_pre_me(&mut self, value: MotionEstimation) { + unsafe { + (*self.ptr).pre_me = value.into(); + } + } + + pub fn set_me_pre_comparison(&mut self, value: Comparison) { + unsafe { + (*self.ptr).me_pre_cmp = value.into(); + } + } + + pub fn set_pre_dia_size(&mut self, value: usize) { + unsafe { + (*self.ptr).pre_dia_size = value as c_int; + } + } + + pub fn set_me_subpel_quality(&mut self, value: usize) { + unsafe { + (*self.ptr).me_subpel_quality = value as c_int; + } + } + + pub fn set_me_range(&mut self, value: usize) { + unsafe { + (*self.ptr).me_range = value as c_int; + } + } + + pub fn set_intra_quant_bias(&mut self, value: Option) { + unsafe { + if let Some(value) = value { + (*self.ptr).intra_quant_bias = value as c_int; + } + else { + (*self.ptr).intra_quant_bias = FF_DEFAULT_QUANT_BIAS; + } + } + } + + pub fn set_inter_quant_bias(&mut self, value: Option) { + unsafe { + if let Some(value) = value { + (*self.ptr).inter_quant_bias = value as c_int; + } + else { + (*self.ptr).inter_quant_bias = FF_DEFAULT_QUANT_BIAS; + } + } + } + + pub fn set_mb_decision(&mut self, value: Decision) { + unsafe { + (*self.ptr).mb_decision = value.into(); + } + } } impl Deref for Video {