// Copyright 2014 PDFium Authors. All rights reserved.
// 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/fxcodec/codec/ccodec_gifmodule.h"

#include "core/fxcodec/codec/codec_int.h"
#include "core/fxcodec/fx_codec.h"
#include "core/fxcodec/lgif/cgifcontext.h"
#include "core/fxcodec/lgif/fx_gif.h"
#include "core/fxge/fx_dib.h"
#include "third_party/base/ptr_util.h"

CCodec_GifModule::CCodec_GifModule() {}

CCodec_GifModule::~CCodec_GifModule() {}

std::unique_ptr<CCodec_GifModule::Context> CCodec_GifModule::Start(
    Delegate* pDelegate) {
  return pdfium::MakeUnique<CGifContext>(this, pDelegate);
}

GifDecodeStatus CCodec_GifModule::ReadHeader(Context* pContext,
                                             int* width,
                                             int* height,
                                             int* pal_num,
                                             void** pal_pp,
                                             int* bg_index,
                                             CFX_DIBAttribute* pAttribute) {
  auto* context = static_cast<CGifContext*>(pContext);
  GifDecodeStatus ret = context->ReadHeader();
  if (ret != GifDecodeStatus::Success)
    return ret;

  *width = context->width;
  *height = context->height;
  *pal_num = (2 << context->global_pal_exp);
  *pal_pp = context->m_GlobalPalette.empty() ? nullptr
                                             : context->m_GlobalPalette.data();
  *bg_index = context->bc_index;
  return GifDecodeStatus::Success;
}

GifDecodeStatus CCodec_GifModule::LoadFrameInfo(Context* pContext,
                                                int* frame_num) {
  auto* context = static_cast<CGifContext*>(pContext);
  GifDecodeStatus ret = context->GetFrame();
  if (ret != GifDecodeStatus::Success)
    return ret;

  *frame_num = context->GetFrameNum();
  return GifDecodeStatus::Success;
}

GifDecodeStatus CCodec_GifModule::LoadFrame(Context* pContext,
                                            int frame_num,
                                            CFX_DIBAttribute* pAttribute) {
  auto* context = static_cast<CGifContext*>(pContext);
  GifDecodeStatus ret = context->LoadFrame(frame_num);
  if (ret != GifDecodeStatus::Success || !pAttribute)
    return ret;

  pAttribute->m_nGifLeft = context->m_Images[frame_num]->m_ImageInfo.left;
  pAttribute->m_nGifTop = context->m_Images[frame_num]->m_ImageInfo.top;
  pAttribute->m_fAspectRatio = context->pixel_aspect;
  const uint8_t* buf =
      reinterpret_cast<const uint8_t*>(context->cmt_data.GetBuffer(0));
  uint32_t len = context->cmt_data.GetLength();
  if (len > 21) {
    uint8_t size = *buf++;
    if (size != 0)
      pAttribute->m_strAuthor = ByteString(buf, size);
    else
      pAttribute->m_strAuthor.clear();
  }
  return GifDecodeStatus::Success;
}

uint32_t CCodec_GifModule::GetAvailInput(Context* pContext,
                                         uint8_t** avail_buf_ptr) {
  auto* context = static_cast<CGifContext*>(pContext);
  return context->GetAvailInput(avail_buf_ptr);
}

void CCodec_GifModule::Input(Context* pContext,
                             const uint8_t* src_buf,
                             uint32_t src_size) {
  auto* context = static_cast<CGifContext*>(pContext);
  context->SetInputBuffer((uint8_t*)src_buf, src_size);
}