Add JPEG_R capture to extension sample camera app

android · Jul 4, 2024 · f863493 · f863493
1 parent ab6f960
commit f863493
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 8 deletions.
diff --git a/...ions/app/src/main/java/com/example/android/camera2/extensions/fragments/CameraFragment.kt b/...ions/app/src/main/java/com/example/android/camera2/extensions/fragments/CameraFragment.kt
@@ -651,6 +651,9 @@ class CameraFragment : Fragment(), TextureView.SurfaceTextureListener {
     previewSurface = createPreviewSurface(previewSize)
     stillImageReader = createStillImageReader()
     postviewImageReader = createPostviewImageReader()
+    if (postviewImageReader == null) {
+      isPostviewAvailable = false
+    }
 
     val outputConfig = ArrayList<OutputConfiguration>()
     outputConfig.add(OutputConfiguration(stillImageReader.surface))
@@ -722,14 +725,26 @@ class CameraFragment : Fragment(), TextureView.SurfaceTextureListener {
    * Creates the still image reader and sets up OnImageAvailableListener
    */
   private fun createStillImageReader(): ImageReader {
+    var stillFormat: Int
+    var stillCaptureSize: Size
+
     val yuvColorEncodingSystemSizes = extensionCharacteristics.getExtensionSupportedSizes(
       currentExtension, ImageFormat.YUV_420_888
     )
     val jpegSizes = extensionCharacteristics.getExtensionSupportedSizes(
       currentExtension, ImageFormat.JPEG
     )
-    val stillFormat = if (jpegSizes.isEmpty()) ImageFormat.YUV_420_888 else ImageFormat.JPEG
-    val stillCaptureSize = if (jpegSizes.isEmpty()) yuvColorEncodingSystemSizes[0] else jpegSizes[0]
+    stillFormat = if (jpegSizes.isEmpty()) ImageFormat.YUV_420_888 else ImageFormat.JPEG
+    stillCaptureSize = if (jpegSizes.isEmpty()) yuvColorEncodingSystemSizes[0] else jpegSizes[0]
+
+    val jpegRSizes = extensionCharacteristics.getExtensionSupportedSizes(
+      currentExtension, ImageFormat.JPEG_R
+    )
+    if (args.hdr) {
+      stillFormat = if (jpegRSizes.isEmpty()) stillFormat else ImageFormat.JPEG_R
+      stillCaptureSize = if (jpegRSizes.isEmpty()) stillCaptureSize else jpegRSizes[0]
+    }
+
     val stillImageReader = ImageReader.newInstance(
       stillCaptureSize.width,
       stillCaptureSize.height, stillFormat, 1
@@ -742,7 +757,8 @@ class CameraFragment : Fragment(), TextureView.SurfaceTextureListener {
             hideCaptureProgressUI()
             val file = File(
               requireActivity().getExternalFilesDir(null),
-              if (image.format == ImageFormat.JPEG) "frame.jpg" else "frame.yuv"
+              if (image.format == ImageFormat.JPEG
+                || image.format == ImageFormat.JPEG_R) "frame.jpg" else "frame.yuv"
             )
             output = FileOutputStream(file)
             output.write(getDataFromImage(image))
@@ -790,9 +806,11 @@ class CameraFragment : Fragment(), TextureView.SurfaceTextureListener {
     if (!jpegSupportedSizes.isEmpty()) {
       postviewSize = jpegSupportedSizes[0]
       postviewFormat = ImageFormat.JPEG
-    } else {
+    } else if (!yuvSupportedSizes.isEmpty()){
       postviewSize = yuvSupportedSizes[0]
       postviewFormat = ImageFormat.YUV_420_888
+    } else {
+      return null
     }
     val postviewImageReader =
       ImageReader.newInstance(postviewSize.width, postviewSize.height, postviewFormat, 1)
@@ -1235,7 +1253,7 @@ class CameraFragment : Fragment(), TextureView.SurfaceTextureListener {
       val planes = image.planes
       var buffer: ByteBuffer
       var offset = 0
-      if (format == ImageFormat.JPEG) {
+      if (format == ImageFormat.JPEG || format == ImageFormat.JPEG_R) {
         buffer = planes[0].buffer
         data = ByteArray(buffer.limit())
         buffer.rewind()

diff --git a/...ns/app/src/main/java/com/example/android/camera2/extensions/fragments/SelectorFragment.kt b/...ns/app/src/main/java/com/example/android/camera2/extensions/fragments/SelectorFragment.kt
@@ -18,8 +18,10 @@ package com.example.android.camera2.extensions.fragments
 
 import android.annotation.SuppressLint
 import android.content.Context
+import android.graphics.ImageFormat;
 import android.hardware.camera2.CameraCharacteristics
 import android.hardware.camera2.CameraManager
+import android.os.Build
 import android.os.Bundle
 import android.view.LayoutInflater
 import android.view.View
@@ -61,7 +63,7 @@ class SelectorFragment : Fragment() {
                 view.setOnClickListener {
                     Navigation.findNavController(requireActivity(), R.id.fragment_container)
                             .navigate(SelectorFragmentDirections.actionSelectorToCamera(
-                                    item.cameraId))
+                                    item.cameraId, item.hdr))
                 }
             }
         }
@@ -71,7 +73,8 @@ class SelectorFragment : Fragment() {
 
         private data class CameraInfo(
                 val name: String,
-                val cameraId: String)
+                val cameraId: String,
+                val hdr: Boolean = false)
 
         /** Converts a lens orientation enum into a human-readable string */
         private fun lensOrientationString(value: Int) = when (value) {
@@ -99,9 +102,18 @@ class SelectorFragment : Fragment() {
                                           .REQUEST_AVAILABLE_CAPABILITIES_BACKWARD_COMPATIBLE) &&
                 extensionCharacteristics.supportedExtensions.isNotEmpty()) {
                 availableCameras.add(CameraInfo("$orientation ($id)", id))
+                val currentExtension = extensionCharacteristics.supportedExtensions[0]
+
+                if (Build.VERSION.SDK_INT >= 35) {
+                  var jpegRSizes = extensionCharacteristics.getExtensionSupportedSizes(
+                      currentExtension, ImageFormat.JPEG_R
+                  )
+                  if (jpegRSizes.isNotEmpty()) {
+                    availableCameras.add(CameraInfo("$orientation ($id) JPEG_R", id, true))
+                  }
+                }
               }
             }
-
             return availableCameras
         }
     }

diff --git a/Camera2Extensions/app/src/main/res/navigation/nav_graph.xml b/Camera2Extensions/app/src/main/res/navigation/nav_graph.xml
@@ -43,6 +43,11 @@
             app:argType="string"
             app:nullable="false"/>
 
+        <argument
+            android:name="hdr"
+            app:argType="boolean"
+            app:nullable="false"/>
+
     </fragment>
 
 </navigation>