start work on nihed-cros-libva

[nihav-player.git] / nihed-cros-libva / src / buffer / mpeg2.rs

diff --git a/nihed-cros-libva/src/buffer/mpeg2.rs b/nihed-cros-libva/src/buffer/mpeg2.rs
new file mode 100644 (file)
index 0000000..22df314
--- /dev/null
+++ b/nihed-cros-libva/src/buffer/mpeg2.rs
@@ -0,0 +1,172 @@
+// Copyright 2023 The ChromiumOS Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+//! Wrappers around MPEG2 `VABuffer` types.
+
+use crate::bindings;
+
+/// Wrapper over the `picture_coding_extension` bindgen field in `VAPictureParameterBufferMPEG2`.
+pub struct MPEG2PictureCodingExtension(bindings::_VAPictureParameterBufferMPEG2__bindgen_ty_1);
+
+impl MPEG2PictureCodingExtension {
+    /// Creates the bindgen field.
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        intra_dc_precision: u32,
+        picture_structure: u32,
+        top_field_first: u32,
+        frame_pred_frame_dct: u32,
+        concealment_motion_vectors: u32,
+        q_scale_type: u32,
+        intra_vlc_format: u32,
+        alternate_scan: u32,
+        repeat_first_field: u32,
+        progressive_frame: u32,
+        is_first_field: u32,
+    ) -> Self {
+        let _bitfield_1 =
+            bindings::_VAPictureParameterBufferMPEG2__bindgen_ty_1__bindgen_ty_1::new_bitfield_1(
+                intra_dc_precision,
+                picture_structure,
+                top_field_first,
+                frame_pred_frame_dct,
+                concealment_motion_vectors,
+                q_scale_type,
+                intra_vlc_format,
+                alternate_scan,
+                repeat_first_field,
+                progressive_frame,
+                is_first_field,
+            );
+
+        Self(bindings::_VAPictureParameterBufferMPEG2__bindgen_ty_1 {
+            bits: bindings::_VAPictureParameterBufferMPEG2__bindgen_ty_1__bindgen_ty_1 {
+                _bitfield_align_1: Default::default(),
+                _bitfield_1,
+                __bindgen_padding_0: Default::default(),
+            },
+        })
+    }
+
+    /// Returns the inner FFI type. Useful for testing purposes.
+    pub fn inner(&mut self) -> &bindings::_VAPictureParameterBufferMPEG2__bindgen_ty_1 {
+        &self.0
+    }
+}
+
+/// Wrapper over the `PictureParameterBufferMPEG2` FFI type.
+pub struct PictureParameterBufferMPEG2(Box<bindings::VAPictureParameterBufferMPEG2>);
+
+impl PictureParameterBufferMPEG2 {
+    /// Creates the wrapper.
+    pub fn new(
+        horizontal_size: u16,
+        vertical_size: u16,
+        forward_reference_picture: bindings::VASurfaceID,
+        backward_reference_picture: bindings::VASurfaceID,
+        picture_coding_type: i32,
+        f_code: i32,
+        picture_coding_extension: &MPEG2PictureCodingExtension,
+    ) -> Self {
+        let picture_coding_extension = picture_coding_extension.0;
+
+        Self(Box::new(bindings::VAPictureParameterBufferMPEG2 {
+            horizontal_size,
+            vertical_size,
+            forward_reference_picture,
+            backward_reference_picture,
+            picture_coding_type,
+            f_code,
+            picture_coding_extension,
+            va_reserved: Default::default(),
+        }))
+    }
+
+    pub(crate) fn inner_mut(&mut self) -> &mut bindings::VAPictureParameterBufferMPEG2 {
+        self.0.as_mut()
+    }
+
+    /// Returns the inner FFI type. Useful for testing purposes.
+    pub fn inner(&mut self) -> &bindings::VAPictureParameterBufferMPEG2 {
+        self.0.as_ref()
+    }
+}
+
+/// Wrapper over the `VASliceParameterBufferMPEG2` FFI type.
+pub struct SliceParameterBufferMPEG2(Box<bindings::VASliceParameterBufferMPEG2>);
+
+impl SliceParameterBufferMPEG2 {
+    /// Creates the wrapper.
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        slice_data_size: u32,
+        slice_data_offset: u32,
+        slice_data_flag: u32,
+        macroblock_offset: u32,
+        slice_horizontal_position: u32,
+        slice_vertical_position: u32,
+        quantiser_scale_code: i32,
+        intra_slice_flag: i32,
+    ) -> Self {
+        Self(Box::new(bindings::VASliceParameterBufferMPEG2 {
+            slice_data_size,
+            slice_data_offset,
+            slice_data_flag,
+            macroblock_offset,
+            slice_horizontal_position,
+            slice_vertical_position,
+            quantiser_scale_code,
+            intra_slice_flag,
+            va_reserved: Default::default(),
+        }))
+    }
+
+    pub(crate) fn inner_mut(&mut self) -> &mut bindings::VASliceParameterBufferMPEG2 {
+        self.0.as_mut()
+    }
+
+    /// Returns the inner FFI type. Useful for testing purposes.
+    pub fn inner(&self) -> &bindings::VASliceParameterBufferMPEG2 {
+        self.0.as_ref()
+    }
+}
+
+/// Wrapper over the `VAIQMatrixBufferMPEG2` FFI type.
+pub struct IQMatrixBufferMPEG2(Box<bindings::VAIQMatrixBufferMPEG2>);
+
+impl IQMatrixBufferMPEG2 {
+    /// Creates the wrapper.
+    #[allow(clippy::too_many_arguments)]
+    pub fn new(
+        load_intra_quantiser_matrix: i32,
+        load_non_intra_quantiser_matrix: i32,
+        load_chroma_intra_quantiser_matrix: i32,
+        load_chroma_non_intra_quantiser_matrix: i32,
+        intra_quantiser_matrix: [u8; 64usize],
+        non_intra_quantiser_matrix: [u8; 64usize],
+        chroma_intra_quantiser_matrix: [u8; 64usize],
+        chroma_non_intra_quantiser_matrix: [u8; 64usize],
+    ) -> Self {
+        Self(Box::new(bindings::VAIQMatrixBufferMPEG2 {
+            load_intra_quantiser_matrix,
+            load_non_intra_quantiser_matrix,
+            load_chroma_intra_quantiser_matrix,
+            load_chroma_non_intra_quantiser_matrix,
+            intra_quantiser_matrix,
+            non_intra_quantiser_matrix,
+            chroma_intra_quantiser_matrix,
+            chroma_non_intra_quantiser_matrix,
+            va_reserved: Default::default(),
+        }))
+    }
+
+    pub(crate) fn inner_mut(&mut self) -> &mut bindings::VAIQMatrixBufferMPEG2 {
+        self.0.as_mut()
+    }
+
+    /// Returns the inner FFI type. Useful for testing purposes.
+    pub fn inner(&self) -> &bindings::VAIQMatrixBufferMPEG2 {
+        self.0.as_ref()
+    }
+}