summaryrefslogtreecommitdiff
path: root/testing
diff options
context:
space:
mode:
Diffstat (limited to 'testing')
-rw-r--r--testing/embedder_test.cpp31
-rw-r--r--testing/utils/bitmap_saver.cpp48
-rw-r--r--testing/utils/bitmap_saver.h21
3 files changed, 71 insertions, 29 deletions
diff --git a/testing/embedder_test.cpp b/testing/embedder_test.cpp
index 3f953d22df..215009ceac 100644
--- a/testing/embedder_test.cpp
+++ b/testing/embedder_test.cpp
@@ -6,13 +6,11 @@
#include <limits.h>
-#include <fstream>
#include <list>
#include <map>
#include <memory>
#include <string>
#include <utility>
-#include <vector>
#include "core/fdrm/crypto/fx_crypt.h"
#include "public/cpp/fpdf_scopers.h"
@@ -21,8 +19,8 @@
#include "public/fpdf_text.h"
#include "public/fpdfview.h"
#include "testing/gmock/include/gmock/gmock.h"
-#include "testing/image_diff/image_diff_png.h"
#include "testing/test_support.h"
+#include "testing/utils/bitmap_saver.h"
#include "testing/utils/path_service.h"
#include "third_party/base/logging.h"
#include "third_party/base/ptr_util.h"
@@ -494,32 +492,7 @@ std::string EmbedderTest::HashBitmap(FPDF_BITMAP bitmap) {
// static
void EmbedderTest::WriteBitmapToPng(FPDF_BITMAP bitmap,
const std::string& filename) {
- const int stride = FPDFBitmap_GetStride(bitmap);
- const int width = FPDFBitmap_GetWidth(bitmap);
- const int height = FPDFBitmap_GetHeight(bitmap);
- const auto* buffer =
- static_cast<const unsigned char*>(FPDFBitmap_GetBuffer(bitmap));
-
- std::vector<unsigned char> png_encoding;
- bool encoded;
- if (FPDFBitmap_GetFormat(bitmap) == FPDFBitmap_Gray) {
- encoded = image_diff_png::EncodeGrayPNG(buffer, width, height, stride,
- &png_encoding);
- } else {
- encoded = image_diff_png::EncodeBGRAPNG(buffer, width, height, stride,
- /*discard_transparency=*/false,
- &png_encoding);
- }
-
- ASSERT_TRUE(encoded);
- ASSERT_LT(filename.size(), 256u);
-
- std::ofstream png_file;
- png_file.open(filename, std::ios_base::out | std::ios_base::binary);
- png_file.write(reinterpret_cast<char*>(&png_encoding.front()),
- png_encoding.size());
- ASSERT_TRUE(png_file.good());
- png_file.close();
+ BitmapSaver::WriteBitmapToPng(bitmap, filename);
}
#endif
diff --git a/testing/utils/bitmap_saver.cpp b/testing/utils/bitmap_saver.cpp
new file mode 100644
index 0000000000..38f2cf2738
--- /dev/null
+++ b/testing/utils/bitmap_saver.cpp
@@ -0,0 +1,48 @@
+// Copyright 2018 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "testing/utils/bitmap_saver.h"
+
+#include <fstream>
+#include <vector>
+
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/image_diff/image_diff_png.h"
+
+// static
+void BitmapSaver::WriteBitmapToPng(FPDF_BITMAP bitmap,
+ const std::string& filename) {
+ const int stride = FPDFBitmap_GetStride(bitmap);
+ const int width = FPDFBitmap_GetWidth(bitmap);
+ const int height = FPDFBitmap_GetHeight(bitmap);
+ const auto* buffer =
+ static_cast<const unsigned char*>(FPDFBitmap_GetBuffer(bitmap));
+
+ std::vector<unsigned char> png_encoding;
+ bool encoded;
+ if (FPDFBitmap_GetFormat(bitmap) == FPDFBitmap_Gray) {
+ encoded = image_diff_png::EncodeGrayPNG(buffer, width, height, stride,
+ &png_encoding);
+ } else {
+ encoded = image_diff_png::EncodeBGRAPNG(buffer, width, height, stride,
+ /*discard_transparency=*/false,
+ &png_encoding);
+ }
+
+ ASSERT_TRUE(encoded);
+ ASSERT_LT(filename.size(), 256u);
+
+ std::ofstream png_file;
+ png_file.open(filename, std::ios_base::out | std::ios_base::binary);
+ png_file.write(reinterpret_cast<char*>(&png_encoding.front()),
+ png_encoding.size());
+ ASSERT_TRUE(png_file.good());
+ png_file.close();
+}
+
+// static
+void BitmapSaver::WriteBitmapToPng(CFX_DIBitmap* bitmap,
+ const std::string& filename) {
+ WriteBitmapToPng(reinterpret_cast<FPDF_BITMAP>(bitmap), filename);
+}
diff --git a/testing/utils/bitmap_saver.h b/testing/utils/bitmap_saver.h
new file mode 100644
index 0000000000..9f931fc323
--- /dev/null
+++ b/testing/utils/bitmap_saver.h
@@ -0,0 +1,21 @@
+// Copyright 2018 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef TESTING_UTILS_BITMAP_SAVER_H_
+#define TESTING_UTILS_BITMAP_SAVER_H_
+
+#include <string>
+
+#include "public/fpdfview.h"
+
+class CFX_DIBitmap;
+
+class BitmapSaver {
+ public:
+ static void WriteBitmapToPng(FPDF_BITMAP bitmap, const std::string& filename);
+ static void WriteBitmapToPng(CFX_DIBitmap* bitmap,
+ const std::string& filename);
+};
+
+#endif // TESTING_UTILS_BITMAP_SAVER_H_