image_reader: Add api-level-34 AImageReader_newWithDataSpace()

Android 14 (U, API level 34) adds `DataSpace` support to `ImageReader` with a new constructor and a getter, and finally switches to the regular `HardwareBufferFormat` instead of the restricted `ImageFormat` enum to select the underlying buffer format. After all, an `AImage` can be turned into a `HardwareBuffer` which supports a wider range of formats.
rust-mobile · May 14, 2024 · 9fadd6c · 9fadd6c
1 parent ce3d82f
commit 9fadd6c
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 7 deletions.
diff --git a/ndk/CHANGELOG.md b/ndk/CHANGELOG.md
@@ -1,5 +1,7 @@
 # Unreleased
 
+- image_reader: Add `ImageReader::new_with_data_space()` constructor and `ImageReader::data_space()` getter from API level 34. (#474)
+
 # 0.9.0 (2024-04-26)
 
 - Move `MediaFormat` from `media::media_codec` to its own `media::media_format` module. (#442)

diff --git a/ndk/Cargo.toml b/ndk/Cargo.toml
@@ -14,7 +14,7 @@ rust-version = "1.66"
 
 [features]
 default = ["rwh_06"]
-all = ["audio", "bitmap", "media", "nativewindow", "sync", "api-level-33", "rwh_04", "rwh_05", "rwh_06"]
+all = ["audio", "bitmap", "media", "nativewindow", "sync", "api-level-34", "rwh_04", "rwh_05", "rwh_06"]
 
 audio = ["ffi/audio", "api-level-26"]
 bitmap = ["ffi/bitmap"]
@@ -33,6 +33,7 @@ api-level-30 = ["api-level-29"]
 api-level-31 = ["api-level-30"]
 api-level-32 = ["api-level-31"]
 api-level-33 = ["api-level-32"]
+api-level-34 = ["api-level-33"]
 
 test = ["ffi/test", "jni", "all"]
 

diff --git a/ndk/src/media/image_reader.rs b/ndk/src/media/image_reader.rs
@@ -4,22 +4,25 @@
 //! [`AImage`]: https://developer.android.com/ndk/reference/group/media#aimage
 #![cfg(feature = "api-level-24")]
 
-use crate::media_error::{construct, construct_never_null, MediaError, Result};
-use crate::native_window::NativeWindow;
-use crate::utils::abort_on_panic;
-use num_enum::{FromPrimitive, IntoPrimitive};
-use std::{ffi::c_void, fmt, mem::MaybeUninit, ptr::NonNull};
-
 #[cfg(feature = "api-level-26")]
 use std::os::fd::{FromRawFd, IntoRawFd, OwnedFd};
+use std::{ffi::c_void, fmt, mem::MaybeUninit, ptr::NonNull};
+
+use num_enum::{FromPrimitive, IntoPrimitive};
 
 #[cfg(feature = "api-level-26")]
 use crate::hardware_buffer::{HardwareBuffer, HardwareBufferUsage};
+use crate::media_error::{construct, construct_never_null, MediaError, Result};
+use crate::native_window::NativeWindow;
+use crate::utils::abort_on_panic;
+#[cfg(feature = "api-level-34")]
+use crate::{data_space::DataSpace, hardware_buffer_format::HardwareBufferFormat};
 
 #[repr(i32)]
 #[derive(Copy, Clone, Debug, PartialEq, Eq, FromPrimitive, IntoPrimitive)]
 #[allow(non_camel_case_types)]
 #[non_exhaustive]
+#[doc(alias = "AIMAGE_FORMATS")]
 pub enum ImageFormat {
     RGBA_8888 = ffi::AIMAGE_FORMATS::AIMAGE_FORMAT_RGBA_8888.0 as i32,
     RGBX_8888 = ffi::AIMAGE_FORMATS::AIMAGE_FORMAT_RGBX_8888.0 as i32,
@@ -44,9 +47,13 @@ pub enum ImageFormat {
     __Unknown(i32),
 }
 
+#[doc(alias = "AImageReader_ImageCallback")]
+#[doc(alias = "AImageReader_ImageListener")]
 pub type ImageListener = Box<dyn FnMut(&ImageReader) + Send>;
 
 #[cfg(feature = "api-level-26")]
+#[doc(alias = "AImageReader_BufferRemovedCallback")]
+#[doc(alias = "AImageReader_BufferRemovedListener")]
 pub type BufferRemovedListener = Box<dyn FnMut(&ImageReader, &HardwareBuffer) + Send>;
 
 /// Result returned by:
@@ -108,6 +115,7 @@ impl AcquireResult<Image> {
 /// A native [`AImageReader *`]
 ///
 /// [`AImageReader *`]: https://developer.android.com/ndk/reference/group/media#aimagereader
+#[doc(alias = "AImageReader")]
 pub struct ImageReader {
     inner: NonNull<ffi::AImageReader>,
     image_cb: Option<Box<ImageListener>>,
@@ -144,6 +152,7 @@ impl ImageReader {
         self.inner.as_ptr()
     }
 
+    #[doc(alias = "AImageReader_new")]
     pub fn new(width: i32, height: i32, format: ImageFormat, max_images: i32) -> Result<Self> {
         let inner = construct_never_null(|res| unsafe {
             ffi::AImageReader_new(width, height, format.into(), max_images, res)
@@ -153,6 +162,7 @@ impl ImageReader {
     }
 
     #[cfg(feature = "api-level-26")]
+    #[doc(alias = "AImageReader_newWithUsage")]
     pub fn new_with_usage(
         width: i32,
         height: i32,
@@ -174,6 +184,33 @@ impl ImageReader {
         Ok(Self::from_ptr(inner))
     }
 
+    #[cfg(feature = "api-level-34")]
+    #[doc(alias = "AImageReader_newWithDataSpace")]
+    pub fn new_with_data_space(
+        width: i32,
+        height: i32,
+        usage: HardwareBufferUsage,
+        max_images: i32,
+        format: HardwareBufferFormat,
+        data_space: DataSpace,
+    ) -> Result<Self> {
+        let inner = construct_never_null(|res| unsafe {
+            ffi::AImageReader_newWithDataSpace(
+                width,
+                height,
+                usage.bits(),
+                max_images,
+                i32::from(format)
+                    .try_into()
+                    .expect("Unexpected sign bits in format"),
+                data_space.into(),
+                res,
+            )
+        })?;
+
+        Ok(Self::from_ptr(inner))
+    }
+
     #[doc(alias = "AImageReader_setImageListener")]
     pub fn set_image_listener(&mut self, listener: ImageListener) -> Result<()> {
         let mut boxed = Box::new(listener);
@@ -444,6 +481,13 @@ impl Image {
         unsafe { ffi::AImage_deleteAsync(self.as_ptr(), release_fence_fd.into_raw_fd()) };
         std::mem::forget(self);
     }
+
+    #[cfg(feature = "api-level-34")]
+    #[doc(alias = "AImage_getDataSpace")]
+    pub fn data_space(&self) -> Result<DataSpace> {
+        construct(|res| unsafe { ffi::AImage_getDataSpace(self.as_ptr(), res) })
+            .map(DataSpace::from)
+    }
 }
 
 impl Drop for Image {