| // Copyright 2017 The PDFium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "core/fxge/dib/cstretchengine.h" |
| |
| #include <utility> |
| |
| #include "core/fpdfapi/page/cpdf_dib.h" |
| #include "core/fpdfapi/parser/cpdf_dictionary.h" |
| #include "core/fpdfapi/parser/cpdf_number.h" |
| #include "core/fpdfapi/parser/cpdf_stream.h" |
| #include "core/fxge/dib/fx_dib.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| // Discovered experimentally |
| constexpr uint32_t kTooBigSrcLen = 20; |
| constexpr uint32_t kTooBigDestLen = 32 * 1024 * 1024 + 1; |
| |
| uint32_t PixelWeightSum(const CStretchEngine::PixelWeight* weights) { |
| uint32_t sum = 0; |
| for (int i = weights->m_SrcStart; i <= weights->m_SrcEnd; ++i) { |
| sum += weights->GetWeightForPosition(i); |
| } |
| return sum; |
| } |
| |
| void ExecuteOneStretchTest(int32_t dest_width, |
| int32_t src_width, |
| const FXDIB_ResampleOptions& options) { |
| constexpr uint32_t kExpectedSum = CStretchEngine::kFixedPointOne; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(dest_width, 0, dest_width, src_width, 0, |
| src_width, options)); |
| for (int32_t i = 0; i < dest_width; ++i) { |
| EXPECT_EQ(kExpectedSum, PixelWeightSum(table.GetPixelWeight(i))) |
| << "for { " << src_width << ", " << dest_width << " } at " << i; |
| } |
| } |
| |
| void ExecuteOneReversedStretchTest(int32_t dest_width, |
| int32_t src_width, |
| const FXDIB_ResampleOptions& options) { |
| constexpr uint32_t kExpectedSum = CStretchEngine::kFixedPointOne; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(-dest_width, 0, dest_width, src_width, 0, |
| src_width, options)); |
| for (int32_t i = 0; i < dest_width; ++i) { |
| EXPECT_EQ(kExpectedSum, PixelWeightSum(table.GetPixelWeight(i))) |
| << "for { " << src_width << ", " << dest_width << " } at " << i |
| << " (reversed)"; |
| } |
| } |
| |
| void ExecuteStretchTests(const FXDIB_ResampleOptions& options) { |
| // Can't test everything, few random values chosen. |
| constexpr int32_t kDestWidths[] = {1, 2, 337, 512, 808, 2550}; |
| constexpr int32_t kSrcWidths[] = {1, 2, 187, 256, 809, 1110}; |
| for (int32_t src_width : kSrcWidths) { |
| for (int32_t dest_width : kDestWidths) { |
| ExecuteOneStretchTest(dest_width, src_width, options); |
| ExecuteOneReversedStretchTest(dest_width, src_width, options); |
| } |
| } |
| } |
| |
| } // namespace |
| |
| TEST(CStretchEngine, OverflowInCtor) { |
| FX_RECT clip_rect; |
| RetainPtr<CPDF_Dictionary> dict_obj = pdfium::MakeRetain<CPDF_Dictionary>(); |
| dict_obj->SetNewFor<CPDF_Number>("Width", 71000); |
| dict_obj->SetNewFor<CPDF_Number>("Height", 12500); |
| RetainPtr<CPDF_Stream> stream = |
| pdfium::MakeRetain<CPDF_Stream>(std::move(dict_obj)); |
| auto dib_source = pdfium::MakeRetain<CPDF_DIB>(nullptr, stream); |
| dib_source->Load(); |
| CStretchEngine engine(nullptr, FXDIB_Format::k8bppRgb, 500, 500, clip_rect, |
| dib_source, FXDIB_ResampleOptions()); |
| EXPECT_TRUE(engine.GetResampleOptionsForTest().bInterpolateBilinear); |
| EXPECT_FALSE(engine.GetResampleOptionsForTest().bHalftone); |
| EXPECT_FALSE(engine.GetResampleOptionsForTest().bNoSmoothing); |
| EXPECT_FALSE(engine.GetResampleOptionsForTest().bLossy); |
| } |
| |
| TEST(CStretchEngine, WeightRounding) { |
| FXDIB_ResampleOptions options; |
| ExecuteStretchTests(options); |
| } |
| |
| TEST(CStretchEngine, WeightRoundingNoSmoothing) { |
| FXDIB_ResampleOptions options; |
| options.bNoSmoothing = true; |
| ExecuteStretchTests(options); |
| } |
| |
| TEST(CStretchEngine, WeightRoundingBilinear) { |
| FXDIB_ResampleOptions options; |
| options.bInterpolateBilinear = true; |
| ExecuteStretchTests(options); |
| } |
| |
| TEST(CStretchEngine, WeightRoundingNoSmoothingBilinear) { |
| FXDIB_ResampleOptions options; |
| options.bNoSmoothing = true; |
| options.bInterpolateBilinear = true; |
| ExecuteStretchTests(options); |
| } |
| |
| TEST(CStretchEngine, ZeroLengthSrc) { |
| FXDIB_ResampleOptions options; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(100, 0, 100, 0, 0, 0, options)); |
| } |
| |
| TEST(CStretchEngine, ZeroLengthSrcNoSmoothing) { |
| FXDIB_ResampleOptions options; |
| options.bNoSmoothing = true; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(100, 0, 100, 0, 0, 0, options)); |
| } |
| |
| TEST(CStretchEngine, ZeroLengthSrcBilinear) { |
| FXDIB_ResampleOptions options; |
| options.bInterpolateBilinear = true; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(100, 0, 100, 0, 0, 0, options)); |
| } |
| |
| TEST(CStretchEngine, ZeroLengthSrcNoSmoothingBilinear) { |
| FXDIB_ResampleOptions options; |
| options.bNoSmoothing = true; |
| options.bInterpolateBilinear = true; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(100, 0, 100, 0, 0, 0, options)); |
| } |
| |
| TEST(CStretchEngine, ZeroLengthDest) { |
| FXDIB_ResampleOptions options; |
| CStretchEngine::WeightTable table; |
| ASSERT_TRUE(table.CalculateWeights(0, 0, 0, 100, 0, 100, options)); |
| } |
| |
| TEST(CStretchEngine, TooManyWeights) { |
| FXDIB_ResampleOptions options; |
| CStretchEngine::WeightTable table; |
| ASSERT_FALSE(table.CalculateWeights(kTooBigDestLen, 0, kTooBigDestLen, |
| kTooBigSrcLen, 0, kTooBigSrcLen, |
| options)); |
| } |