// Different kinds of abstract storage locations: Constants, Global and Local Storage, Temporaries, ...
#pragma once

#include "data.h"
#include "scope.h"

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

namespace FlowGraph {

 // Parameter to Data class, defining where the data is stored
 // Explicitely not including size
 // But including label/data/pointer
 // -> includes identity of certain data point
 class Storage
 {
 public:
  virtual ~Storage() {} // force class to be polymorphic, for smart pointers
 };

 class Constant: public Storage
 {
 public:
  Constant(std::vector<uint8_t>& value): m_value(value) {} // little endian data
  const std::vector<uint8_t>& value() const { return m_value; }
 private:
  std::vector<uint8_t> m_value;
 };

 class GlobalStorage : public Storage
 {
 public:
  GlobalStorage(const std::string& name): m_name(name) {}
  const std::string& name() const { return m_name; }
 private:
  std::string m_name;
 };

 // named values
 class LocalStorage : public Storage
 {
 public:
  LocalStorage(LocalScope& scope, const std::string& name): m_name(name), m_scope(scope) {}
  const std::string& name() const { return m_name; }
  index_t indexOfStorage() const;
 private:
  std::string m_name;
  LocalScope& m_scope;
 };

 // intermediate results, anonymous values
 // use generated name
 class TemporaryStorage : public Storage
 {
 public:
  TemporaryStorage(LocalScope& scope);
  std::string name() const;
  index_t indexOfStorage() const;
 private:
  LocalScope& m_scope;
 };

 // dereferenced pointer
 class PointeeStorage : public Storage
 {
 public:
  PointeeStorage(const Data& pointer, const Data& offset): m_pointer(pointer), m_offset(offset) {
   if (pointer.type() != DataType::Pointer)
    throw std::runtime_error("Pointer argument must be a DataType::Pointer");
   if (offset.type() != DataType::Size)
    throw std::runtime_error("Offset argument must be a DataType::Size");
  }
  Data pointer() { return m_pointer; }
  Data offset() { return m_offset; }
 private:
  Data m_pointer;
  Data m_offset;
 };

}