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 7, 2024 · 6aa2005 · 6aa2005
1 parent 49bbbba
commit 6aa2005
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 7 deletions.
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,27 +4,30 @@
 //! [`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};
+#[cfg(feature = "api-level-26")]
+use std::os::fd::{FromRawFd, IntoRawFd, OwnedFd};
 use std::{
     ffi::c_void,
     fmt::{self, Debug, Formatter},
     mem::MaybeUninit,
     ptr::NonNull,
 };
 
-#[cfg(feature = "api-level-26")]
-use std::os::fd::{FromRawFd, IntoRawFd, OwnedFd};
+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,
@@ -49,9 +52,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:
@@ -113,6 +120,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>>,
@@ -149,6 +157,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)
@@ -158,6 +167,7 @@ impl ImageReader {
     }
 
     #[cfg(feature = "api-level-26")]
+    #[doc(alias = "AImageReader_newWithUsage")]
     pub fn new_with_usage(
         width: i32,
         height: i32,
@@ -179,6 +189,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);
@@ -449,6 +486,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 {