diff options
Diffstat (limited to 'testing')
| -rw-r--r-- | testing/embedder_test.cpp | 31 | ||||
| -rw-r--r-- | testing/utils/bitmap_saver.cpp | 48 | ||||
| -rw-r--r-- | testing/utils/bitmap_saver.h | 21 |
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_ |