// Copyright (c) 2022 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.

#pragma once

#include <stdlib.h>
#include <cstdio>

#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>

#if defined(_WIN32)
#ifdef FASTDEPLOY_LIB
#define FASTDEPLOY_DECL __declspec(dllexport)
#else
#define FASTDEPLOY_DECL __declspec(dllimport)
#endif  // FASTDEPLOY_LIB
#else
#define FASTDEPLOY_DECL __attribute__((visibility("default")))
#endif  // _WIN32

namespace fastdeploy {

class FASTDEPLOY_DECL FDLogger {
 public:
  static bool disable_info;
  static bool disable_warning;
  FDLogger() {
    line_ = "";
    prefix_ = "[FastDeploy]";
    level_ = 0;
  }
  // 0: INFO
  // 1: WARNING
  // 2: ERROR
  explicit FDLogger(int level = 0, const std::string& prefix = "[FastDeploy]");

  bool verbose() {
    if (disable_info && level_ == 0) {
      return false;
    }
    if (disable_warning && level_ == 1) {
      return false;
    }
    return true;
  }

  template <typename T>
  FDLogger& operator<<(const T& val) {
    if (!verbose()) {
      return *this;
    }
    std::stringstream ss;
    ss << val;
    line_ += ss.str();
    return *this;
  }
  FDLogger& operator<<(std::ostream& (*os)(std::ostream&));
  ~FDLogger() {
    if (!verbose() && line_ != "") {
      std::cout << line_ << std::endl;
    }
  }

 private:
  std::string line_;
  std::string prefix_;
  int level_ = 0;
};

FASTDEPLOY_DECL bool ReadBinaryFromFile(const std::string& file,
                                        std::string* contents);

#ifndef __REL_FILE__
#define __REL_FILE__ __FILE__
#endif

#define FDERROR                                                \
  FDLogger(2, "[ERROR]") << __REL_FILE__ << "(" << __LINE__ \
                            << ")::" << __FUNCTION__ << "\t"

#define FDWARNING                                                \
  FDLogger(1, "[WARNING]") << __REL_FILE__ << "(" << __LINE__ \
                              << ")::" << __FUNCTION__ << "\t"

#define FDINFO                                                \
  FDLogger(0, "[INFO]") << __REL_FILE__ << "(" << __LINE__ \
                           << ")::" << __FUNCTION__ << "\t"

#define FDASSERT(condition, format, ...)                        \
  if (!(condition)) {                                           \
    int n = std::snprintf(nullptr, 0, format, ##__VA_ARGS__);   \
    std::vector<char> buffer(n + 1);                            \
    std::snprintf(buffer.data(), n + 1, format, ##__VA_ARGS__); \
    FDERROR << buffer.data() << std::endl;                      \
    std::abort();                                               \
  }

///////// Basic Marco ///////////

#define FD_PRIVATE_CASE_TYPE_USING_HINT(NAME, enum_type, type, HINT, ...) \
  case enum_type: {                                                       \
    using HINT = type;                                                    \
    __VA_ARGS__();                                                        \
    break;                                                                \
  }

#define FD_PRIVATE_CASE_TYPE(NAME, enum_type, type, ...) \
  FD_PRIVATE_CASE_TYPE_USING_HINT(NAME, enum_type, type, data_t, __VA_ARGS__)

// Visit different data type to match the corresponding function of FDTensor
#define FD_VISIT_ALL_TYPES(TYPE, NAME, ...)                                    \
  [&] {                                                                        \
    const auto& __dtype__ = TYPE;                                              \
    switch (__dtype__) {                                                       \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::BOOL, bool,         \
                           __VA_ARGS__)                                        \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT32, int32_t,     \
                           __VA_ARGS__)                                        \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT64, int64_t,     \
                           __VA_ARGS__)                                        \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP32, float,        \
                           __VA_ARGS__)                                        \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP64, double,       \
                           __VA_ARGS__)                                        \
      default:                                                                 \
        FDASSERT(                                                              \
            false,                                                             \
            "Invalid enum data type. Expect to accept data type BOOL, INT32, " \
            "INT64, FP32, FP64, but receive type %s.",                         \
            Str(__dtype__).c_str());                                           \
    }                                                                          \
  }()

#define FD_VISIT_INT_FLOAT_TYPES(TYPE, NAME, ...)                             \
  [&] {                                                                       \
    const auto& __dtype__ = TYPE;                                             \
    switch (__dtype__) {                                                      \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT32, int32_t,    \
                           __VA_ARGS__)                                       \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT64, int64_t,    \
                           __VA_ARGS__)                                       \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP32, float,       \
                           __VA_ARGS__)                                       \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP64, double,      \
                           __VA_ARGS__)                                       \
      default:                                                                \
        FDASSERT(false,                                                       \
                 "Invalid enum data type. Expect to accept data type INT32, " \
                 "INT64, FP32, FP64, but receive type %s.",                   \
                 Str(__dtype__).c_str());                                     \
    }                                                                         \
  }()

#define FD_VISIT_FLOAT_TYPES(TYPE, NAME, ...)                                \
  [&] {                                                                      \
    const auto& __dtype__ = TYPE;                                            \
    switch (__dtype__) {                                                     \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP32, float,      \
                           __VA_ARGS__)                                      \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::FP64, double,     \
                           __VA_ARGS__)                                      \
      default:                                                               \
        FDASSERT(false,                                                      \
                 "Invalid enum data type. Expect to accept data type FP32, " \
                 "FP64, but receive type %s.",                               \
                 Str(__dtype__).c_str());                                    \
    }                                                                        \
  }()

#define FD_VISIT_INT_TYPES(TYPE, NAME, ...)                                   \
  [&] {                                                                       \
    const auto& __dtype__ = TYPE;                                             \
    switch (__dtype__) {                                                      \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT32, int32_t,    \
                           __VA_ARGS__)                                       \
      FD_PRIVATE_CASE_TYPE(NAME, ::fastdeploy::FDDataType::INT64, int64_t,    \
                           __VA_ARGS__)                                       \
      default:                                                                \
        FDASSERT(false,                                                       \
                 "Invalid enum data type. Expect to accept data type INT32, " \
                 "INT64, but receive type %s.",                               \
                 Str(__dtype__).c_str());                                     \
    }                                                                         \
  }()

FASTDEPLOY_DECL std::vector<int64_t> GetStride(
    const std::vector<int64_t>& dims);

}  // namespace fastdeploy