1 files changed, 446 insertions, 0 deletions
diff --git a/gtest/cjpeg-gtest-wrapper.cpp b/gtest/cjpeg-gtest-wrapper.cpp
new file mode 100644
index 0000000..c4ba600
--- /dev/null
+++ b/gtest/cjpeg-gtest-wrapper.cpp
@@ -0,0 +1,446 @@
+/*
+ * Copyright 2020 The Chromium Authors. All Rights Reserved.
+ *
+ * This software is provided 'as-is', without any express or implied
+ * warranty.  In no event will the authors be held liable for any damages
+ * arising from the use of this software.
+ *
+ * Permission is granted to anyone to use this software for any purpose,
+ * including commercial applications, and to alter it and redistribute it
+ * freely, subject to the following restrictions:
+ *
+ * 1. The origin of this software must not be misrepresented; you must not
+ *    claim that you wrote the original software. If you use this software
+ *    in a product, an acknowledgment in the product documentation would be
+ *    appreciated but is not required.
+ * 2. Altered source versions must be plainly marked as such, and must not be
+ *    misrepresented as being the original software.
+ * 3. This notice may not be removed or altered from any source distribution.
+ */
+
+#include "base/files/file.h"
+#include "base/files/file_util.h"
+#include "base/hash/md5.h"
+#include "base/path_service.h"
+
+#include <gtest/gtest.h>
+#include <string>
+
+extern "C" int cjpeg(int argc, char *argv[]);
+
+static std::string GetTargetDirectory() {
+#if defined(ANDROID)
+  return "/sdcard";
+#else
+  base::FilePath path;
+  base::PathService::Get(base::DIR_CURRENT, &path);
+  return path.MaybeAsASCII();
+#endif
+}
+
+static void GetTestFilePath(base::FilePath* path, std::string filename) {
+  ASSERT_TRUE(base::PathService::Get(base::DIR_SOURCE_ROOT, path));
+  *path = path->AppendASCII("third_party");
+  *path = path->AppendASCII("libjpeg_turbo");
+  *path = path->AppendASCII("testimages");
+  *path = path->AppendASCII(filename);
+  ASSERT_TRUE(base::PathExists(*path));
+}
+
+static bool CompareFileAndMD5(const base::FilePath& path,
+                              const std::string expected_md5) {
+  // Read the output file and compute the MD5.
+  std::string output;
+  if (!base::ReadFileToString(path, &output)) {
+    return false;
+  }
+  const std::string md5 = base::MD5String(output);
+  return expected_md5 == md5;
+}
+
+TEST(CJPEGTest, RGBISlow) {
+
+  base::FilePath icc_path;
+  GetTestFilePath(&icc_path, "test1.icc");
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_rgb_islow.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-rgb";
+  std::string arg2 = "-dct";
+  std::string arg3 = "int";
+  std::string arg4 = "-icc";
+  std::string arg5 = icc_path.MaybeAsASCII();
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0],
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "1d44a406f61da743b5fd31c0a9abdca3";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, IFastOpt422) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_422_ifast_opt.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "2x1";
+  std::string arg3 = "-dct";
+  std::string arg4 = "fast";
+  std::string arg5 = "-opt";
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0],
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "2540287b79d913f91665e660303ab2c8";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, IFastProg420Q100) {
+
+  base::FilePath scans_path;
+  GetTestFilePath(&scans_path, "test.scan");
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_420_q100_ifast_prog.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "2x2";
+  std::string arg3 = "-quality";
+  std::string arg4 = "100";
+  std::string arg5 = "-dct";
+  std::string arg6 = "fast";
+  std::string arg7 = "-scans";
+  std::string arg8 = scans_path.MaybeAsASCII();
+  std::string arg9 = "-outfile";
+  std::string arg10 = output_path.MaybeAsASCII();
+  std::string arg11 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0], &arg9[0], &arg10[0],
+                           &arg11[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(12, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "e59bb462016a8d9a748c330a3474bb55";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, GrayISlow) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_gray_islow.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-gray";
+  std::string arg2 = "-dct";
+  std::string arg3 = "int";
+  std::string arg4 = "-outfile";
+  std::string arg5 = output_path.MaybeAsASCII();
+  std::string arg6 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0],
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(7, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "72b51f894b8f4a10b3ee3066770aa38d";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, IFastOpt420S) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_420s_ifast_opt.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "2x2";
+  std::string arg3 = "-smooth";
+  std::string arg4 = "1";
+  std::string arg5 = "-dct";
+  std::string arg6 = "int";
+  std::string arg7 = "-opt";
+  std::string arg8 = "-outfile";
+  std::string arg9 = output_path.MaybeAsASCII();
+  std::string arg10 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0], &arg9[0], &arg10[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(11, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "388708217ac46273ca33086b22827ed8";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, FloatProg3x2) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(GetTargetDirectory());
+  output_path = output_path.AppendASCII("testout_3x2_float_prog.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "3x2";
+  std::string arg3 = "-dct";
+  std::string arg4 = "float";
+  std::string arg5 = "-prog";
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+#if defined(__arm__) || defined(__aarch64__)
+  const std::string EXPECTED_MD5 = "9bca803d2042bd1eb03819e2bf92b3e5";
+#else
+  const std::string EXPECTED_MD5 = "343e3f8caf8af5986ebaf0bdc13b5c71";
+#endif
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, IFastProg3x2) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_3x2_ifast_prog.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "3x2";
+  std::string arg3 = "-dct";
+  std::string arg4 = "fast";
+  std::string arg5 = "-prog";
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "1ee5d2c1a77f2da495f993c8c7cceca5";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+#ifdef C_ARITH_CODING_SUPPORTED
+TEST(CJPEGTest, ISlowAri420) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_420_islow_ari.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-dct";
+  std::string arg2 = "int";
+  std::string arg3 = "-arithmetic";
+  std::string arg4 = "-outfile";
+  std::string arg5 = output_path.MaybeAsASCII();
+  std::string arg6 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(7, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "e986fb0a637a8d833d96e8a6d6d84ea1";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, ISlowProgAri444) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_444_islow_progari.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-sample";
+  std::string arg2 = "1x1";
+  std::string arg3 = "-dct";
+  std::string arg4 = "int";
+  std::string arg5 = "-prog";
+  std::string arg6 = "-arithmetic";
+  std::string arg7 = "-outfile";
+  std::string arg8 = output_path.MaybeAsASCII();
+  std::string arg9 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0], &arg9[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(10, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "0a8f1c8f66e113c3cf635df0a475a617";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, ISlowAri444) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_444_islow_ari.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-dct";
+  std::string arg2 = "int";
+  std::string arg3 = "-arithmetic";
+  std::string arg4 = "-sample";
+  std::string arg5 = "1x1";
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "dd1b827fc504feb0259894e7132585b4";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+#endif
+
+TEST(CJPEGTest, ISlowProg420) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_420_islow_prog.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-dct";
+  std::string arg2 = "int";
+  std::string arg3 = "-prog";
+  std::string arg4 = "-outfile";
+  std::string arg5 = output_path.MaybeAsASCII();
+  std::string arg6 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(7, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "1c4afddc05c0a43489ee54438a482d92";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, ISlow444) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_444_islow.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-dct";
+  std::string arg2 = "int";
+  std::string arg3 = "-sample";
+  std::string arg4 = "1x1";
+  std::string arg5 = "-outfile";
+  std::string arg6 = output_path.MaybeAsASCII();
+  std::string arg7 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(8, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "62a8665a2e08e90c6fffa3a94b894ce2";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}
+
+TEST(CJPEGTest, ISlowProg444) {
+
+  base::FilePath input_image_path;
+  GetTestFilePath(&input_image_path, "testorig.ppm");
+  base::FilePath output_path(FILE_PATH_LITERAL(GetTargetDirectory()));
+  output_path = output_path.AppendASCII("testout_444_islow_prog.jpg");
+
+  std::string prog_name = "cjpeg";
+  std::string arg1 = "-dct";
+  std::string arg2 = "int";
+  std::string arg3 = "-prog";
+  std::string arg4 = "-sample";
+  std::string arg5 = "1x1";
+  std::string arg6 = "-outfile";
+  std::string arg7 = output_path.MaybeAsASCII();
+  std::string arg8 = input_image_path.MaybeAsASCII();
+
+  char *command_line[] = { &prog_name[0],
+                           &arg1[0], &arg2[0], &arg3[0], &arg4[0], &arg5[0],
+                           &arg6[0], &arg7[0], &arg8[0]
+                         };
+  // Generate test image file.
+  EXPECT_EQ(cjpeg(9, command_line), 0);
+
+  // Compare expected MD5 sum against that of test image.
+  const std::string EXPECTED_MD5 = "666ef192fff72570e332db7610e1a7d1";
+  EXPECT_TRUE(CompareFileAndMD5(output_path, EXPECTED_MD5));
+}