#pragma once

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include <limits>
#include <string>

#include "revng/ADT/SortedVector.h"
#include "revng/Model/VerifyHelper.h"
#include "revng/Support/MetaAddress.h"
#include "revng/Support/MetaAddress/YAMLTraits.h"
#include "revng/Yield/TagType.h"

namespace yield {
using ByteContainer = llvm::SmallVector<uint8_t, 16>;
}

/* TUPLE-TREE-YAML

name: Instruction
type: struct
fields:
  - name: Address
    doc: >
      Indicates the address of the first byte of the instruction.
    type: MetaAddress
  - name: RawBytes
    type: yield::ByteContainer
  - name: Disassembled
    type: string

  - name: Tags
    sequence:
      type: SortedVector
      elementType: Tag
    optional: true

  - name: OpcodeIdentifier
    type: string
    optional: true
  - name: Comment
    doc: >
      Contains any extra information deduced based on the disassembly of this
      instruction that could be relevant for the user.
    type: string
    optional: true
  - name: Error
    doc: >
      Contains any extra extra warning/error style information deduced based on
      the disassembly of this instruction that could be relevant for the user.
    type: string
    optional: true

key:
  - Address

TUPLE-TREE-YAML */

#include "revng/Yield/Generated/Early/Instruction.h"

namespace yield {

class Instruction : public generated::Instruction {
public:
  using generated::Instruction::Instruction;

public:
  bool verify(model::VerifyHelper &VH) const;
  void dump() const debug_function;

public:
  inline bool verify() const debug_function { return verify(false); }
  inline bool verify(bool Assert) const debug_function {
    model::VerifyHelper VH(Assert);
    return verify(VH);
  }

public:
  inline MetaAddress getRelativeAddressBase() const {
    switch (Address().type()) {
    case MetaAddressType::Code_x86:
    case MetaAddressType::Code_x86_64:
      return Address() + RawBytes().size(); // Start of the next instruction.
    case MetaAddressType::Code_arm:
      return Address() + 8; // Two instructions later
    case MetaAddressType::Code_arm_thumb:
      return Address() + 4; // Two instructions later
    case MetaAddressType::Code_aarch64:
    case MetaAddressType::Code_systemz:
    case MetaAddressType::Code_mips:
    case MetaAddressType::Code_mipsel:
      return Address();
    default:
      revng_abort("Unsupported instruction.");
    }
  }
};

} // namespace yield

#include "revng/Yield/Generated/Late/Instruction.h"