blob: 920645ab92128e55a9e171a5eb1f8b6aeab16336 [file] [log] [blame]
// 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.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#include "core/fxge/dib/cstretchengine.h"
#include <math.h>
#include <algorithm>
#include <type_traits>
#include <utility>
#include "core/fxcrt/fx_safe_types.h"
#include "core/fxcrt/fx_system.h"
#include "core/fxcrt/pauseindicator_iface.h"
#include "core/fxge/calculate_pitch.h"
#include "core/fxge/dib/cfx_dibbase.h"
#include "core/fxge/dib/cfx_dibitmap.h"
#include "core/fxge/dib/fx_dib.h"
#include "core/fxge/dib/scanlinecomposer_iface.h"
#include "third_party/base/check.h"
static_assert(
std::is_trivially_destructible<CStretchEngine::PixelWeight>::value,
"PixelWeight storage may be re-used without invoking its destructor");
// static
bool CStretchEngine::UseInterpolateBilinear(
const FXDIB_ResampleOptions& options,
int dest_width,
int dest_height,
int src_width,
int src_height) {
return !options.bInterpolateBilinear && !options.bNoSmoothing &&
abs(dest_width) != 0 &&
abs(dest_height) / 8 <
static_cast<long long>(src_width) * src_height / abs(dest_width);
}
// static
size_t CStretchEngine::PixelWeight::TotalBytesForWeightCount(
size_t weight_count) {
// Always room for one weight even for empty ranges due to declaration
// of m_Weights[1] in the header. Don't shrink below this since
// CalculateWeights() relies on this later.
const size_t extra_weights = weight_count > 0 ? weight_count - 1 : 0;
FX_SAFE_SIZE_T total_bytes = extra_weights;
total_bytes *= sizeof(m_Weights[0]);
total_bytes += sizeof(PixelWeight);
return total_bytes.ValueOrDie();
}
CStretchEngine::WeightTable::WeightTable() = default;
CStretchEngine::WeightTable::~WeightTable() = default;
bool CStretchEngine::WeightTable::CalculateWeights(
int dest_len,
int dest_min,
int dest_max,
int src_len,
int src_min,
int src_max,
const FXDIB_ResampleOptions& options) {
// 512MB should be large enough for this while preventing OOM.
static constexpr size_t kMaxTableBytesAllowed = 512 * 1024 * 1024;
// Help the compiler realize that these can't change during a loop iteration:
const bool bilinear = options.bInterpolateBilinear;
m_DestMin = 0;
m_ItemSizeBytes = 0;
m_WeightTablesSizeBytes = 0;
m_WeightTables.clear();
if (dest_len == 0)
return true;
if (dest_min > dest_max)
return false;
m_DestMin = dest_min;
const double scale = static_cast<double>(src_len) / dest_len;
const double base = dest_len < 0 ? src_len : 0;
const size_t weight_count = static_cast<size_t>(ceil(fabs(scale))) + 1;
m_ItemSizeBytes = PixelWeight::TotalBytesForWeightCount(weight_count);
const size_t dest_range = static_cast<size_t>(dest_max - dest_min);
const size_t kMaxTableItemsAllowed = kMaxTableBytesAllowed / m_ItemSizeBytes;
if (dest_range > kMaxTableItemsAllowed)
return false;
m_WeightTablesSizeBytes = dest_range * m_ItemSizeBytes;
m_WeightTables.resize(m_WeightTablesSizeBytes);
if (options.bNoSmoothing || fabs(scale) < 1.0f) {
for (int dest_pixel = dest_min; dest_pixel < dest_max; ++dest_pixel) {
PixelWeight& pixel_weights = *GetPixelWeight(dest_pixel);
double src_pos = dest_pixel * scale + scale / 2 + base;
if (bilinear) {
int src_start = static_cast<int>(floor(src_pos - 0.5));
int src_end = static_cast<int>(floor(src_pos + 0.5));
src_start = std::max(src_start, src_min);
src_end = std::min(src_end, src_max - 1);
pixel_weights.SetStartEnd(src_start, src_end, weight_count);
if (pixel_weights.m_SrcStart >= pixel_weights.m_SrcEnd) {
// Always room for one weight per size calculation.
pixel_weights.m_Weights[0] = kFixedPointOne;
} else {
pixel_weights.m_Weights[1] =
FixedFromDouble(src_pos - pixel_weights.m_SrcStart - 0.5f);
pixel_weights.m_Weights[0] =
kFixedPointOne - pixel_weights.m_Weights[1];
}
} else {
int pixel_pos = static_cast<int>(floor(src_pos));
int src_start = std::max(pixel_pos, src_min);
int src_end = std::min(pixel_pos, src_max - 1);
pixel_weights.SetStartEnd(src_start, src_end, weight_count);
pixel_weights.m_Weights[0] = kFixedPointOne;
}
}
return true;
}
for (int dest_pixel = dest_min; dest_pixel < dest_max; ++dest_pixel) {
PixelWeight& pixel_weights = *GetPixelWeight(dest_pixel);
double src_start = dest_pixel * scale + base;
double src_end = src_start + scale;
int start_i = floor(std::min(src_start, src_end));
int end_i = floor(std::max(src_start, src_end));
start_i = std::max(start_i, src_min);
end_i = std::min(end_i, src_max - 1);
if (start_i > end_i) {
start_i = std::min(start_i, src_max - 1);
pixel_weights.SetStartEnd(start_i, start_i, weight_count);
continue;
}
pixel_weights.SetStartEnd(start_i, end_i, weight_count);
uint32_t remaining = kFixedPointOne;
double rounding_error = 0.0;
for (int j = start_i; j < end_i; ++j) {
double dest_start = (j - base) / scale;
double dest_end = (j + 1 - base) / scale;
if (dest_start > dest_end)
std::swap(dest_start, dest_end);
double area_start = std::max(dest_start, static_cast<double>(dest_pixel));
double area_end = std::min(dest_end, static_cast<double>(dest_pixel + 1));
double weight = std::max(0.0, area_end - area_start);
uint32_t fixed_weight = FixedFromDouble(weight + rounding_error);
pixel_weights.SetWeightForPosition(j, fixed_weight);
remaining -= fixed_weight;
rounding_error =
weight - static_cast<double>(fixed_weight) / kFixedPointOne;
}
// Note: underflow is defined behaviour for unsigned types and will
// result in an out-of-range value.
if (remaining && remaining <= kFixedPointOne) {
pixel_weights.SetWeightForPosition(end_i, remaining);
} else {
pixel_weights.RemoveLastWeightAndAdjust(remaining);
}
}
return true;
}
const CStretchEngine::PixelWeight* CStretchEngine::WeightTable::GetPixelWeight(
int pixel) const {
DCHECK(pixel >= m_DestMin);
return reinterpret_cast<const PixelWeight*>(
&m_WeightTables[(pixel - m_DestMin) * m_ItemSizeBytes]);
}
CStretchEngine::PixelWeight* CStretchEngine::WeightTable::GetPixelWeight(
int pixel) {
return const_cast<PixelWeight*>(std::as_const(*this).GetPixelWeight(pixel));
}
CStretchEngine::CStretchEngine(ScanlineComposerIface* pDestBitmap,
FXDIB_Format dest_format,
int dest_width,
int dest_height,
const FX_RECT& clip_rect,
const RetainPtr<const CFX_DIBBase>& pSrcBitmap,
const FXDIB_ResampleOptions& options)
: m_DestFormat(dest_format),
m_DestBpp(GetBppFromFormat(dest_format)),
m_SrcBpp(pSrcBitmap->GetBPP()),
m_bHasAlpha(pSrcBitmap->IsAlphaFormat()),
m_pSource(pSrcBitmap),
m_pSrcPalette(pSrcBitmap->GetPaletteSpan()),
m_SrcWidth(pSrcBitmap->GetWidth()),
m_SrcHeight(pSrcBitmap->GetHeight()),
m_pDestBitmap(pDestBitmap),
m_DestWidth(dest_width),
m_DestHeight(dest_height),
m_DestClip(clip_rect) {
if (m_bHasAlpha) {
DCHECK_EQ(m_DestFormat, FXDIB_Format::kArgb);
DCHECK_EQ(m_DestBpp, GetBppFromFormat(FXDIB_Format::kArgb));
DCHECK_EQ(m_pSource->GetFormat(), FXDIB_Format::kArgb);
DCHECK_EQ(m_SrcBpp, GetBppFromFormat(FXDIB_Format::kArgb));
}
absl::optional<uint32_t> maybe_size =
fxge::CalculatePitch32(m_DestBpp, clip_rect.Width());
if (!maybe_size.has_value())
return;
m_DestScanline.resize(maybe_size.value());
if (dest_format == FXDIB_Format::kRgb32)
std::fill(m_DestScanline.begin(), m_DestScanline.end(), 255);
m_InterPitch = fxge::CalculatePitch32OrDie(m_DestBpp, m_DestClip.Width());
m_ExtraMaskPitch = fxge::CalculatePitch32OrDie(8, m_DestClip.Width());
if (options.bNoSmoothing) {
m_ResampleOptions.bNoSmoothing = true;
} else {
if (UseInterpolateBilinear(options, dest_width, dest_height, m_SrcWidth,
m_SrcHeight)) {
m_ResampleOptions.bInterpolateBilinear = true;
} else {
m_ResampleOptions = options;
}
}
double scale_x = static_cast<float>(m_SrcWidth) / m_DestWidth;
double scale_y = static_cast<float>(m_SrcHeight) / m_DestHeight;
double base_x = m_DestWidth > 0 ? 0.0f : m_DestWidth;
double base_y = m_DestHeight > 0 ? 0.0f : m_DestHeight;
double src_left = scale_x * (clip_rect.left + base_x);
double src_right = scale_x * (clip_rect.right + base_x);
double src_top = scale_y * (clip_rect.top + base_y);
double src_bottom = scale_y * (clip_rect.bottom + base_y);
if (src_left > src_right)
std::swap(src_left, src_right);
if (src_top > src_bottom)
std::swap(src_top, src_bottom);
m_SrcClip.left = static_cast<int>(floor(src_left));
m_SrcClip.right = static_cast<int>(ceil(src_right));
m_SrcClip.top = static_cast<int>(floor(src_top));
m_SrcClip.bottom = static_cast<int>(ceil(src_bottom));
FX_RECT src_rect(0, 0, m_SrcWidth, m_SrcHeight);
m_SrcClip.Intersect(src_rect);
switch (m_SrcBpp) {
case 1:
m_TransMethod = m_DestBpp == 8 ? TransformMethod::k1BppTo8Bpp
: TransformMethod::k1BppToManyBpp;
break;
case 8:
m_TransMethod = m_DestBpp == 8 ? TransformMethod::k8BppTo8Bpp
: TransformMethod::k8BppToManyBpp;
break;
default:
m_TransMethod = m_bHasAlpha ? TransformMethod::kManyBpptoManyBppWithAlpha
: TransformMethod::kManyBpptoManyBpp;
break;
}
}
CStretchEngine::~CStretchEngine() = default;
bool CStretchEngine::Continue(PauseIndicatorIface* pPause) {
while (m_State == State::kHorizontal) {
if (ContinueStretchHorz(pPause))
return true;
m_State = State::kVertical;
StretchVert();
}
return false;
}
bool CStretchEngine::StartStretchHorz() {
if (m_DestWidth == 0 || m_InterPitch == 0 || m_DestScanline.empty())
return false;
FX_SAFE_SIZE_T safe_size = m_SrcClip.Height();
safe_size *= m_InterPitch;
const size_t size = safe_size.ValueOrDefault(0);
if (size == 0)
return false;
m_InterBuf = FixedTryAllocZeroedDataVector<uint8_t>(size);
if (m_InterBuf.empty())
return false;
if (!m_WeightTable.CalculateWeights(
m_DestWidth, m_DestClip.left, m_DestClip.right, m_SrcWidth,
m_SrcClip.left, m_SrcClip.right, m_ResampleOptions)) {
return false;
}
m_CurRow = m_SrcClip.top;
m_State = State::kHorizontal;
return true;
}
bool CStretchEngine::ContinueStretchHorz(PauseIndicatorIface* pPause) {
if (!m_DestWidth)
return false;
if (m_pSource->SkipToScanline(m_CurRow, pPause))
return true;
int Bpp = m_DestBpp / 8;
static const int kStrechPauseRows = 10;
int rows_to_go = kStrechPauseRows;
for (; m_CurRow < m_SrcClip.bottom; ++m_CurRow) {
if (rows_to_go == 0) {
if (pPause && pPause->NeedToPauseNow())
return true;
rows_to_go = kStrechPauseRows;
}
const uint8_t* src_scan = m_pSource->GetScanline(m_CurRow).data();
pdfium::span<uint8_t> dest_span = m_InterBuf.writable_span().subspan(
(m_CurRow - m_SrcClip.top) * m_InterPitch, m_InterPitch);
size_t dest_span_index = 0;
// TODO(npm): reduce duplicated code here
switch (m_TransMethod) {
case TransformMethod::k1BppTo8Bpp:
case TransformMethod::k1BppToManyBpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
PixelWeight* pWeights = m_WeightTable.GetPixelWeight(col);
uint32_t dest_a = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
if (src_scan[j / 8] & (1 << (7 - j % 8)))
dest_a += pixel_weight * 255;
}
dest_span[dest_span_index++] = PixelFromFixed(dest_a);
}
break;
}
case TransformMethod::k8BppTo8Bpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
PixelWeight* pWeights = m_WeightTable.GetPixelWeight(col);
uint32_t dest_a = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
dest_a += pixel_weight * src_scan[j];
}
dest_span[dest_span_index++] = PixelFromFixed(dest_a);
}
break;
}
case TransformMethod::k8BppToManyBpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
PixelWeight* pWeights = m_WeightTable.GetPixelWeight(col);
uint32_t dest_r = 0;
uint32_t dest_g = 0;
uint32_t dest_b = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
unsigned long argb = m_pSrcPalette[src_scan[j]];
if (m_DestFormat == FXDIB_Format::kRgb) {
dest_r += pixel_weight * static_cast<uint8_t>(argb >> 16);
dest_g += pixel_weight * static_cast<uint8_t>(argb >> 8);
dest_b += pixel_weight * static_cast<uint8_t>(argb);
} else {
dest_b += pixel_weight * static_cast<uint8_t>(argb >> 24);
dest_g += pixel_weight * static_cast<uint8_t>(argb >> 16);
dest_r += pixel_weight * static_cast<uint8_t>(argb >> 8);
}
}
dest_span[dest_span_index++] = PixelFromFixed(dest_b);
dest_span[dest_span_index++] = PixelFromFixed(dest_g);
dest_span[dest_span_index++] = PixelFromFixed(dest_r);
}
break;
}
case TransformMethod::kManyBpptoManyBpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
PixelWeight* pWeights = m_WeightTable.GetPixelWeight(col);
uint32_t dest_r = 0;
uint32_t dest_g = 0;
uint32_t dest_b = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
const uint8_t* src_pixel = src_scan + j * Bpp;
dest_b += pixel_weight * (*src_pixel++);
dest_g += pixel_weight * (*src_pixel++);
dest_r += pixel_weight * (*src_pixel);
}
dest_span[dest_span_index++] = PixelFromFixed(dest_b);
dest_span[dest_span_index++] = PixelFromFixed(dest_g);
dest_span[dest_span_index++] = PixelFromFixed(dest_r);
dest_span_index += Bpp - 3;
}
break;
}
case TransformMethod::kManyBpptoManyBppWithAlpha: {
DCHECK(m_bHasAlpha);
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
PixelWeight* pWeights = m_WeightTable.GetPixelWeight(col);
uint32_t dest_a = 0;
uint32_t dest_r = 0;
uint32_t dest_g = 0;
uint32_t dest_b = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
const uint8_t* src_pixel = src_scan + j * Bpp;
uint32_t pixel_weight =
pWeights->GetWeightForPosition(j) * src_pixel[3] / 255;
dest_b += pixel_weight * (*src_pixel++);
dest_g += pixel_weight * (*src_pixel++);
dest_r += pixel_weight * (*src_pixel);
dest_a += pixel_weight;
}
dest_span[dest_span_index++] = PixelFromFixed(dest_b);
dest_span[dest_span_index++] = PixelFromFixed(dest_g);
dest_span[dest_span_index++] = PixelFromFixed(dest_r);
dest_span[dest_span_index] = PixelFromFixed(255 * dest_a);
dest_span_index += Bpp - 3;
}
break;
}
}
rows_to_go--;
}
return false;
}
void CStretchEngine::StretchVert() {
if (m_DestHeight == 0)
return;
WeightTable table;
if (!table.CalculateWeights(m_DestHeight, m_DestClip.top, m_DestClip.bottom,
m_SrcHeight, m_SrcClip.top, m_SrcClip.bottom,
m_ResampleOptions)) {
return;
}
const int DestBpp = m_DestBpp / 8;
for (int row = m_DestClip.top; row < m_DestClip.bottom; ++row) {
unsigned char* dest_scan = m_DestScanline.data();
PixelWeight* pWeights = table.GetPixelWeight(row);
switch (m_TransMethod) {
case TransformMethod::k1BppTo8Bpp:
case TransformMethod::k1BppToManyBpp:
case TransformMethod::k8BppTo8Bpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
pdfium::span<const uint8_t> src_span =
m_InterBuf.span().subspan((col - m_DestClip.left) * DestBpp);
uint32_t dest_a = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
dest_a +=
pixel_weight * src_span[(j - m_SrcClip.top) * m_InterPitch];
}
*dest_scan = PixelFromFixed(dest_a);
dest_scan += DestBpp;
}
break;
}
case TransformMethod::k8BppToManyBpp:
case TransformMethod::kManyBpptoManyBpp: {
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
pdfium::span<const uint8_t> src_span =
m_InterBuf.span().subspan((col - m_DestClip.left) * DestBpp);
uint32_t dest_r = 0;
uint32_t dest_g = 0;
uint32_t dest_b = 0;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
pdfium::span<const uint8_t> src_pixel =
src_span.subspan((j - m_SrcClip.top) * m_InterPitch, 3);
dest_b += pixel_weight * src_pixel[0];
dest_g += pixel_weight * src_pixel[1];
dest_r += pixel_weight * src_pixel[2];
}
dest_scan[0] = PixelFromFixed(dest_b);
dest_scan[1] = PixelFromFixed(dest_g);
dest_scan[2] = PixelFromFixed(dest_r);
dest_scan += DestBpp;
}
break;
}
case TransformMethod::kManyBpptoManyBppWithAlpha: {
DCHECK(m_bHasAlpha);
for (int col = m_DestClip.left; col < m_DestClip.right; ++col) {
pdfium::span<const uint8_t> src_span =
m_InterBuf.span().subspan((col - m_DestClip.left) * DestBpp);
uint32_t dest_a = 0;
uint32_t dest_r = 0;
uint32_t dest_g = 0;
uint32_t dest_b = 0;
constexpr size_t kPixelBytes = 4;
for (int j = pWeights->m_SrcStart; j <= pWeights->m_SrcEnd; ++j) {
uint32_t pixel_weight = pWeights->GetWeightForPosition(j);
pdfium::span<const uint8_t> src_pixel = src_span.subspan(
(j - m_SrcClip.top) * m_InterPitch, kPixelBytes);
dest_b += pixel_weight * src_pixel[0];
dest_g += pixel_weight * src_pixel[1];
dest_r += pixel_weight * src_pixel[2];
dest_a += pixel_weight * src_pixel[3];
}
if (dest_a) {
int r = static_cast<uint32_t>(dest_r) * 255 / dest_a;
int g = static_cast<uint32_t>(dest_g) * 255 / dest_a;
int b = static_cast<uint32_t>(dest_b) * 255 / dest_a;
dest_scan[0] = std::clamp(b, 0, 255);
dest_scan[1] = std::clamp(g, 0, 255);
dest_scan[2] = std::clamp(r, 0, 255);
}
dest_scan[3] = PixelFromFixed(dest_a);
dest_scan += DestBpp;
}
break;
}
}
m_pDestBitmap->ComposeScanline(row - m_DestClip.top, m_DestScanline);
}
}