// Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <algorithm>
#include <vector>
#include "paddle/fluid/inference/capi/c_api_internal.h"
#include "paddle/fluid/inference/capi/paddle_c_api.h"
#include "paddle/fluid/platform/enforce.h"

using paddle::ConvertToACPrecision;
using paddle::ConvertToPaddleDType;
using paddle::ConvertToPDDataType;

extern "C" {

PD_PaddleBuf* PD_NewPaddleBuf() { return new PD_PaddleBuf; }

void PD_DeletePaddleBuf(PD_PaddleBuf* buf) {
  if (buf) {
    delete buf;
    buf = nullptr;
    VLOG(3) << "PD_PaddleBuf delete successfully. ";
  }
}

void PD_PaddleBufResize(PD_PaddleBuf* buf, size_t length) {
  PADDLE_ENFORCE_NOT_NULL(buf,
                          paddle::platform::errors::InvalidArgument(
                              "The pointer of Buffer shouldn't be nullptr"));
  buf->buf.Resize(length);
}

void PD_PaddleBufReset(PD_PaddleBuf* buf, void* data, size_t length) {
  PADDLE_ENFORCE_NOT_NULL(buf,
                          paddle::platform::errors::InvalidArgument(
                              "The pointer of Buffer shouldn't be nullptr"));
  buf->buf.Reset(data, length);
}

bool PD_PaddleBufEmpty(PD_PaddleBuf* buf) {
  PADDLE_ENFORCE_NOT_NULL(buf,
                          paddle::platform::errors::InvalidArgument(
                              "The pointer of Buffer shouldn't be nullptr"));
  return buf->buf.empty();
}

void* PD_PaddleBufData(PD_PaddleBuf* buf) {
  PADDLE_ENFORCE_NOT_NULL(buf,
                          paddle::platform::errors::InvalidArgument(
                              "The pointer of Buffer shouldn't be nullptr"));
  return buf->buf.data();
}

size_t PD_PaddleBufLength(PD_PaddleBuf* buf) {
  PADDLE_ENFORCE_NOT_NULL(buf,
                          paddle::platform::errors::InvalidArgument(
                              "The pointer of Buffer shouldn't be nullptr"));
  return buf->buf.length();
}

}  // extern "C"

namespace paddle {
paddle::PaddleDType ConvertToPaddleDType(PD_DataType dtype) {
  switch (dtype) {
    case PD_FLOAT32:
      return PD_PaddleDType::FLOAT32;
    case PD_INT32:
      return PD_PaddleDType::INT32;
    case PD_INT64:
      return PD_PaddleDType::INT64;
    case PD_UINT8:
      return PD_PaddleDType::UINT8;
    default:
      CHECK(false) << "Unsupport dtype.";
      return PD_PaddleDType::FLOAT32;
  }
}

PD_DataType ConvertToPDDataType(PD_PaddleDType dtype) {
  switch (dtype) {
    case PD_PaddleDType::FLOAT32:
      return PD_DataType::PD_FLOAT32;
    case PD_PaddleDType::INT32:
      return PD_DataType::PD_INT32;
    case PD_PaddleDType::INT64:
      return PD_DataType::PD_INT64;
    case PD_PaddleDType::UINT8:
      return PD_DataType::PD_UINT8;
    default:
      CHECK(false) << "Unsupport dtype.";
      return PD_DataType::PD_UNKDTYPE;
  }
}

PD_ACPrecision ConvertToACPrecision(Precision dtype) {
  switch (dtype) {
    case Precision::kFloat32:
      return PD_ACPrecision::kFloat32;
    case Precision::kInt8:
      return PD_ACPrecision::kInt8;
    case Precision::kHalf:
      return PD_ACPrecision::kHalf;
    default:
      CHECK(false) << "Unsupport precision.";
      return PD_ACPrecision::kFloat32;
  }
}
}  // namespace paddle
