// Copyright 2016 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/fpdfapi/page/cpdf_meshstream.h"

#include <utility>

#include "core/fpdfapi/page/cpdf_colorspace.h"
#include "core/fpdfapi/page/cpdf_function.h"
#include "core/fpdfapi/parser/cpdf_array.h"
#include "core/fpdfapi/parser/cpdf_dictionary.h"
#include "core/fpdfapi/parser/cpdf_stream.h"
#include "core/fpdfapi/parser/cpdf_stream_acc.h"
#include "core/fxcrt/cfx_bitstream.h"
#include "core/fxcrt/check.h"
#include "core/fxcrt/compiler_specific.h"
#include "core/fxcrt/span.h"

namespace {

// See PDF Reference 1.7, page 315, table 4.32. (Also table 4.33 and 4.34)
bool ShouldCheckBPC(ShadingType type) {
  switch (type) {
    case kFreeFormGouraudTriangleMeshShading:
    case kLatticeFormGouraudTriangleMeshShading:
    case kCoonsPatchMeshShading:
    case kTensorProductPatchMeshShading:
      return true;
    default:
      return false;
  }
}

// Same references as ShouldCheckBPC() above.
bool IsValidBitsPerComponent(uint32_t x) {
  switch (x) {
    case 1:
    case 2:
    case 4:
    case 8:
    case 12:
    case 16:
      return true;
    default:
      return false;
  }
}

// Same references as ShouldCheckBPC() above.
bool IsValidBitsPerCoordinate(uint32_t x) {
  switch (x) {
    case 1:
    case 2:
    case 4:
    case 8:
    case 12:
    case 16:
    case 24:
    case 32:
      return true;
    default:
      return false;
  }
}

// See PDF Reference 1.7, page 315, table 4.32. (Also table 4.34)
bool ShouldCheckBitsPerFlag(ShadingType type) {
  switch (type) {
    case kFreeFormGouraudTriangleMeshShading:
    case kCoonsPatchMeshShading:
    case kTensorProductPatchMeshShading:
      return true;
    default:
      return false;
  }
}

// Same references as ShouldCheckBitsPerFlag() above.
bool IsValidBitsPerFlag(uint32_t x) {
  switch (x) {
    case 2:
    case 4:
    case 8:
      return true;
    default:
      return false;
  }
}

}  // namespace

CPDF_MeshVertex::CPDF_MeshVertex() = default;

CPDF_MeshVertex::CPDF_MeshVertex(const CPDF_MeshVertex&) = default;

CPDF_MeshVertex::~CPDF_MeshVertex() = default;

CPDF_MeshStream::CPDF_MeshStream(
    ShadingType type,
    const std::vector<std::unique_ptr<CPDF_Function>>& funcs,
    RetainPtr<const CPDF_Stream> pShadingStream,
    RetainPtr<CPDF_ColorSpace> pCS)
    : type_(type),
      funcs_(funcs),
      shading_stream_(std::move(pShadingStream)),
      cs_(std::move(pCS)),
      stream_(pdfium::MakeRetain<CPDF_StreamAcc>(shading_stream_)) {}

CPDF_MeshStream::~CPDF_MeshStream() = default;

bool CPDF_MeshStream::Load() {
  stream_->LoadAllDataFiltered();
  bit_stream_ = std::make_unique<CFX_BitStream>(stream_->GetSpan());

  RetainPtr<const CPDF_Dictionary> dict = shading_stream_->GetDict();
  coord_bits_ = dict->GetIntegerFor("BitsPerCoordinate");
  component_bits_ = dict->GetIntegerFor("BitsPerComponent");
  if (ShouldCheckBPC(type_)) {
    if (!IsValidBitsPerCoordinate(coord_bits_)) {
      return false;
    }
    if (!IsValidBitsPerComponent(component_bits_)) {
      return false;
    }
  }

  flag_bits_ = dict->GetIntegerFor("BitsPerFlag");
  if (ShouldCheckBitsPerFlag(type_) && !IsValidBitsPerFlag(flag_bits_)) {
    return false;
  }

  uint32_t nComponents = cs_->ComponentCount();
  if (nComponents > kMaxComponents) {
    return false;
  }

  components_ = funcs_.empty() ? nComponents : 1;
  RetainPtr<const CPDF_Array> pDecode = dict->GetArrayFor("Decode");
  if (!pDecode || pDecode->size() != 4 + components_ * 2) {
    return false;
  }

  xmin_ = pDecode->GetFloatAt(0);
  xmax_ = pDecode->GetFloatAt(1);
  ymin_ = pDecode->GetFloatAt(2);
  ymax_ = pDecode->GetFloatAt(3);
  for (uint32_t i = 0; i < components_; ++i) {
    color_min_[i] = pDecode->GetFloatAt(i * 2 + 4);
    color_max_[i] = pDecode->GetFloatAt(i * 2 + 5);
  }
  if (ShouldCheckBPC(type_)) {
    coord_max_ = coord_bits_ == 32 ? -1 : (1 << coord_bits_) - 1;
    component_max_ = (1 << component_bits_) - 1;
  }
  return true;
}

void CPDF_MeshStream::SkipBits(uint32_t nbits) {
  bit_stream_->SkipBits(nbits);
}

void CPDF_MeshStream::ByteAlign() {
  bit_stream_->ByteAlign();
}

bool CPDF_MeshStream::IsEOF() const {
  return bit_stream_->IsEOF();
}

bool CPDF_MeshStream::CanReadFlag() const {
  return bit_stream_->BitsRemaining() >= flag_bits_;
}

bool CPDF_MeshStream::CanReadCoords() const {
  return bit_stream_->BitsRemaining() / 2 >= coord_bits_;
}

bool CPDF_MeshStream::CanReadColor() const {
  return bit_stream_->BitsRemaining() / component_bits_ >= components_;
}

uint32_t CPDF_MeshStream::ReadFlag() const {
  DCHECK(ShouldCheckBitsPerFlag(type_));
  return bit_stream_->GetBits(flag_bits_) & 0x03;
}

CFX_PointF CPDF_MeshStream::ReadCoords() const {
  DCHECK(ShouldCheckBPC(type_));

  CFX_PointF pos;
  if (coord_bits_ == 32) {
    pos.x = xmin_ + bit_stream_->GetBits(coord_bits_) * (xmax_ - xmin_) /
                        static_cast<double>(coord_max_);
    pos.y = ymin_ + bit_stream_->GetBits(coord_bits_) * (ymax_ - ymin_) /
                        static_cast<double>(coord_max_);
  } else {
    pos.x = xmin_ +
            bit_stream_->GetBits(coord_bits_) * (xmax_ - xmin_) / coord_max_;
    pos.y = ymin_ +
            bit_stream_->GetBits(coord_bits_) * (ymax_ - ymin_) / coord_max_;
  }
  return pos;
}

FX_RGB_STRUCT<float> CPDF_MeshStream::ReadColor() const {
  DCHECK(ShouldCheckBPC(type_));

  std::array<float, kMaxComponents> color_value;
  for (uint32_t i = 0; i < components_; ++i) {
    color_value[i] = color_min_[i] + bit_stream_->GetBits(component_bits_) *
                                         (color_max_[i] - color_min_[i]) /
                                         component_max_;
  }
  if (funcs_.empty()) {
    return cs_->GetRGBOrZerosOnError(color_value);
  }
  return {color_value[0], 0.0f, 0.0f};
}

bool CPDF_MeshStream::ReadVertex(const CFX_Matrix& pObject2Bitmap,
                                 CPDF_MeshVertex* vertex,
                                 uint32_t* flag) {
  if (!CanReadFlag()) {
    return false;
  }
  *flag = ReadFlag();

  if (!CanReadCoords()) {
    return false;
  }
  vertex->position = pObject2Bitmap.Transform(ReadCoords());

  if (!CanReadColor()) {
    return false;
  }
  vertex->rgb = ReadColor();
  bit_stream_->ByteAlign();
  return true;
}

std::vector<CPDF_MeshVertex> CPDF_MeshStream::ReadVertexRow(
    const CFX_Matrix& pObject2Bitmap,
    int count) {
  std::vector<CPDF_MeshVertex> vertices;
  for (int i = 0; i < count; ++i) {
    if (bit_stream_->IsEOF() || !CanReadCoords()) {
      return std::vector<CPDF_MeshVertex>();
    }

    vertices.emplace_back();
    CPDF_MeshVertex& vertex = vertices.back();
    vertex.position = pObject2Bitmap.Transform(ReadCoords());
    if (!CanReadColor()) {
      return std::vector<CPDF_MeshVertex>();
    }

    vertex.rgb = ReadColor();
    bit_stream_->ByteAlign();
  }
  return vertices;
}