// Chunk data type, and derived types

#pragma once

#include <cstdint>
#include <string>
#include <vector>

// TODO: use it everywhere!
using OP_T = std::vector<uint8_t>;

class Chunk
{
public:
 virtual ~Chunk(){}
 virtual std::vector<uint8_t> getCode() = 0;
 virtual size_t size() = 0; ///< returns size in bytes
};

// can be added via multiple inheritance to chunks with addresses
struct AddressFeature
{
 std::string label;

 bool relativeAddressing{true};

 std::vector<uint8_t> machine_code;
 size_t addr_size{0};
 size_t addr_offs{0}; ///< offset inside code

 std::vector<uint8_t> alternative_code;
 size_t alternative_size{0};
 size_t alternative_offs{0}; ///< offset inside code
};

class Label: public Chunk
{
public:
 Label(const std::string& name) : m_name(name) {}
 std::string name(){return m_name;}
 std::vector<uint8_t> getCode() override { return {}; }
 size_t size() override { return 0; }

private:
 std::string m_name;
};

class Data: public Chunk
{
public:
 Data(const std::vector<uint8_t>& data): m_data(data) {}
 virtual ~Data(){}

 std::vector<uint8_t> getCode() override
 {
  return m_data;
 }

 size_t size() override
 {
  return m_data.size();
 }

protected:
 std::vector<uint8_t> m_data;
};

class Op: public Chunk
{
public:
 virtual ~Op(){};
 virtual bool optimize() = 0; ///< returns true if changed
};

class OpSimple: public Op
{
public:
 OpSimple(std::vector<uint8_t> machine_code): machine_code(machine_code) {}

 std::vector<uint8_t> getCode() override
 {
  return machine_code;
 }

 size_t size() override
 {
  return machine_code.size();
 }

 bool optimize() override ///< returns true if changed
 {
  return false;
 }

protected:
 std::vector<uint8_t> machine_code;
};