commit a4819140bca93820340f21cadc0a911ee10c2568
author Dichen Zhang <dichenzhang@google.com> Fri Oct 21 04:12:30 2022 +0000
committer Dichen Zhang <dichenzhang@google.com> Tue Oct 25 17:05:23 2022 +0000
tree 7ac88023b73ee4b6f92fe1aeb23130f95d4d09dd
parent 4a66b3cdf1a66827767ebf495cdebdf2f81d974a

libjpegrecoverymap: add JPEG encoder for single channel image

test: jpegencoder_test
bug: b/252835416
Change-Id: I6a13438ae4695569058645ad13c1182b7a626524

libs/jpegrecoverymap/tests/jpegencoder_test.cpp

diff --git a/libs/jpegrecoverymap/tests/jpegencoder_test.cpp b/libs/jpegrecoverymap/tests/jpegencoder_test.cpp
index 2d144f0..4cd2a5e 100644
--- a/libs/jpegrecoverymap/tests/jpegencoder_test.cpp
+++ b/libs/jpegrecoverymap/tests/jpegencoder_test.cpp
@@ -25,6 +25,9 @@
 #define VALID_IMAGE "/sdcard/Documents/minnie-320x240.yu12"
 #define VALID_IMAGE_WIDTH 320
 #define VALID_IMAGE_HEIGHT 240
+#define SINGLE_CHANNEL_IMAGE "/sdcard/Documents/minnie-320x240.y"
+#define SINGLE_CHANNEL_IMAGE_WIDTH VALID_IMAGE_WIDTH
+#define SINGLE_CHANNEL_IMAGE_HEIGHT VALID_IMAGE_HEIGHT
 #define INVALID_SIZE_IMAGE "/sdcard/Documents/minnie-318x240.yu12"
 #define INVALID_SIZE_IMAGE_WIDTH 318
 #define INVALID_SIZE_IMAGE_HEIGHT 240
@@ -43,7 +46,7 @@
     virtual void SetUp();
     virtual void TearDown();
 
-    Image mValidImage, mInvalidSizeImage;
+    Image mValidImage, mInvalidSizeImage, mSingleChannelImage;
 };
 
 JpegEncoderTest::JpegEncoderTest() {}
@@ -89,6 +92,11 @@
     }
     mInvalidSizeImage.width = INVALID_SIZE_IMAGE_WIDTH;
     mInvalidSizeImage.height = INVALID_SIZE_IMAGE_HEIGHT;
+    if (!loadFile(SINGLE_CHANNEL_IMAGE, &mSingleChannelImage)) {
+        FAIL() << "Load file " << SINGLE_CHANNEL_IMAGE << " failed";
+    }
+    mSingleChannelImage.width = SINGLE_CHANNEL_IMAGE_WIDTH;
+    mSingleChannelImage.height = SINGLE_CHANNEL_IMAGE_HEIGHT;
 }
 
 void JpegEncoderTest::TearDown() {}
@@ -106,5 +114,12 @@
                                           mInvalidSizeImage.height, JPEG_QUALITY, NULL, 0));
 }
 
+TEST_F(JpegEncoderTest, singleChannelImage) {
+    JpegEncoder encoder;
+    EXPECT_TRUE(encoder.compressImage(mSingleChannelImage.buffer.get(), mSingleChannelImage.width,
+                                         mSingleChannelImage.height, JPEG_QUALITY, NULL, 0, true));
+    ASSERT_GT(encoder.getCompressedImageSize(), static_cast<uint32_t>(0));
+}
+
 }