///
/// Tensor.hpp
/// This file was generated by nitrogen. DO NOT MODIFY THIS FILE.
/// https://github.com/mrousavy/nitro
/// Copyright © Marc Rousavy @ Margelo
///

#pragma once

#if __has_include(<NitroModules/JSIConverter.hpp>)
#include <NitroModules/JSIConverter.hpp>
#else
#error NitroModules cannot be found! Are you sure you installed NitroModules properly?
#endif
#if __has_include(<NitroModules/NitroDefines.hpp>)
#include <NitroModules/NitroDefines.hpp>
#else
#error NitroModules cannot be found! Are you sure you installed NitroModules properly?
#endif
#if __has_include(<NitroModules/JSIHelpers.hpp>)
#include <NitroModules/JSIHelpers.hpp>
#else
#error NitroModules cannot be found! Are you sure you installed NitroModules properly?
#endif
#if __has_include(<NitroModules/PropNameIDCache.hpp>)
#include <NitroModules/PropNameIDCache.hpp>
#else
#error NitroModules cannot be found! Are you sure you installed NitroModules properly?
#endif

// Forward declaration of `TensorDataType` to properly resolve imports.
namespace margelo::nitro::tflite { enum class TensorDataType; }

#include <string>
#include "TensorDataType.hpp"
#include <vector>

namespace margelo::nitro::tflite {

  /**
   * A struct which can be represented as a JavaScript object (Tensor).
   */
  struct Tensor final {
  public:
    std::string name     SWIFT_PRIVATE;
    TensorDataType dataType     SWIFT_PRIVATE;
    std::vector<double> shape     SWIFT_PRIVATE;

  public:
    Tensor() = default;
    explicit Tensor(std::string name, TensorDataType dataType, std::vector<double> shape): name(name), dataType(dataType), shape(shape) {}

  public:
    friend bool operator==(const Tensor& lhs, const Tensor& rhs) = default;
  };

} // namespace margelo::nitro::tflite

namespace margelo::nitro {

  // C++ Tensor <> JS Tensor (object)
  template <>
  struct JSIConverter<margelo::nitro::tflite::Tensor> final {
    static inline margelo::nitro::tflite::Tensor fromJSI(jsi::Runtime& runtime, const jsi::Value& arg) {
      jsi::Object obj = arg.asObject(runtime);
      return margelo::nitro::tflite::Tensor(
        JSIConverter<std::string>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "name"))),
        JSIConverter<margelo::nitro::tflite::TensorDataType>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "dataType"))),
        JSIConverter<std::vector<double>>::fromJSI(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "shape")))
      );
    }
    static inline jsi::Value toJSI(jsi::Runtime& runtime, const margelo::nitro::tflite::Tensor& arg) {
      jsi::Object obj(runtime);
      obj.setProperty(runtime, PropNameIDCache::get(runtime, "name"), JSIConverter<std::string>::toJSI(runtime, arg.name));
      obj.setProperty(runtime, PropNameIDCache::get(runtime, "dataType"), JSIConverter<margelo::nitro::tflite::TensorDataType>::toJSI(runtime, arg.dataType));
      obj.setProperty(runtime, PropNameIDCache::get(runtime, "shape"), JSIConverter<std::vector<double>>::toJSI(runtime, arg.shape));
      return obj;
    }
    static inline bool canConvert(jsi::Runtime& runtime, const jsi::Value& value) {
      if (!value.isObject()) {
        return false;
      }
      jsi::Object obj = value.getObject(runtime);
      if (!nitro::isPlainObject(runtime, obj)) {
        return false;
      }
      if (!JSIConverter<std::string>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "name")))) return false;
      if (!JSIConverter<margelo::nitro::tflite::TensorDataType>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "dataType")))) return false;
      if (!JSIConverter<std::vector<double>>::canConvert(runtime, obj.getProperty(runtime, PropNameIDCache::get(runtime, "shape")))) return false;
      return true;
    }
  };

} // namespace margelo::nitro