#pragma once

#include <cstdint>
#include <memory>
#include <optional>
#include <string>
#include <vector>
#include "t81/bigint.hpp"
#include "t81/fraction.hpp"
#include "t81/isa/opcodes.hpp"
#include "t81/isa/type_alias.hpp"
#include "t81/tensor.hpp"
#include "t81/types/T81Complex.hpp"
#include "t81/weights.hpp"

namespace t81::tisc {
enum class LiteralKind : std::uint8_t {
  Int = 0,
  Bool,
  FloatHandle,
  FractionHandle,
  SymbolHandle,
  TensorHandle,
  ShapeHandle,
  ComplexHandle,
  BigIntHandle,
};

struct Insn {
  Opcode opcode{Opcode::Nop};
  std::int32_t a{0};
  std::int64_t b{0};
  std::int32_t c{0};
  LiteralKind literal_kind{LiteralKind::Int};
};

struct EnumVariantMetadata {
  std::string name;
  std::optional<std::string> payload;
  int variant_id = -1;
};

struct EnumMetadata {
  int enum_id = -1;
  std::string name;
  std::vector<EnumVariantMetadata> variants;
};

struct FunctionMetadata {
  std::string name;
  bool is_axion_verify{false};
};

struct Program {
  std::vector<Insn> insns;
  std::vector<double> float_pool;
  std::vector<t81::T81BigInt> bigint_pool;
  std::vector<t81::T81Fraction> fraction_pool;
  std::vector<std::string> symbol_pool;
  std::vector<t81::T729DynamicTensor> tensor_pool;
  std::vector<std::vector<int>> shape_pool;
  std::vector<t81::T81Complex<18>> complex_pool;
  std::string axion_policy_text;
  std::string match_metadata_text;
  std::shared_ptr<t81::weights::ModelFile> weights_model;
  std::vector<tisc::TypeAliasMetadata> type_aliases;
  std::vector<EnumMetadata> enum_metadata;
  std::vector<FunctionMetadata> function_metadata;
};
}  // namespace t81::tisc