// 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/check.h"
#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"

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);
  UNSAFE_TODO({
    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));
  }

  std::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 = FixedSizeDataVector<uint8_t>::TryZeroed(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.subspan(
        (m_CurRow - m_SrcClip.top) * m_InterPitch, m_InterPitch);
    size_t dest_span_index = 0;
    // TODO(npm): reduce duplicated code here
    UNSAFE_TODO({
      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;
  UNSAFE_TODO({
    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.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.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.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);
    }
  });
}