blob: d11ac9dc267639d07d325138aafb06eafc2e79ac [file] [log] [blame] [edit]
// 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));
}