data.h

// This file if part of the llir-opt project.
// Licensing information can be found in the LICENSE file.
// (C) 2018 Nandor Licker. All rights reserved.
 
#pragma once
 
#include <cstdint>
#include <string_view>
#include <unordered_map>
 
#include <llvm/ADT/ilist.h>
 
#include "core/object.h"
#include "core/global.h"
#include "core/symbol_table.h"
 
class Prog;
class Data;
class Object;
 
 
 
template <> struct llvm::ilist_traits<Data> {
private:
  using instr_iterator = simple_ilist<Data>::iterator;
 
public:
  void deleteNode(Data *inst);
  void addNodeToList(Data *inst);
  void removeNodeFromList(Data *inst);
  void transferNodesFromList(
      ilist_traits &from,
      instr_iterator first,
      instr_iterator last
  );
 
  Prog *getParent();
};
 
 
class Data final : public llvm::ilist_node_with_parent<Data, Prog> {
private:
  using ObjectListType = llvm::ilist<Object>;
 
  using iterator = ObjectListType::iterator;
  using const_iterator = ObjectListType::const_iterator;
  using reverse_iterator = ObjectListType::reverse_iterator;
  using const_reverse_iterator = ObjectListType::const_reverse_iterator;
 
public:
  // Initialises the data segment.
  Data(const std::string_view name);
 
  ~Data();
 
  bool IsConstant();
  bool IsZeroed() const;
  bool IsWritable() const;
 
  void removeFromParent();
  void eraseFromParent();
 
  Prog *getParent() const { return parent_; }
 
  // Returns the segment name.
  std::string_view GetName() const { return name_; }
  llvm::StringRef getName() const { return name_; }
 
  // Checks if the section is empty.
  bool IsEmpty() const { return objects_.empty(); }
 
  void remove(iterator it) { objects_.remove(it); }
  void erase(iterator it) { objects_.erase(it); }
  void AddObject(Object *object, Object *before = nullptr);
 
  // Iterators over objects.
  bool empty() const { return objects_.empty(); }
  size_t size() const { return objects_.size(); }
  iterator begin() { return objects_.begin(); }
  iterator end() { return objects_.end(); }
  const_iterator begin() const { return objects_.begin(); }
  const_iterator end() const { return objects_.end(); }
  reverse_iterator rbegin() { return objects_.rbegin(); }
  reverse_iterator rend() { return objects_.rend(); }
  const_reverse_iterator rbegin() const { return objects_.rbegin(); }
  const_reverse_iterator rend() const { return objects_.rend(); }
 
private:
  friend struct llvm::ilist_traits<Data>;
  friend struct llvm::ilist_traits<Object>;
  static ObjectListType Data::*getSublistAccess(Object *) {
    return &Data::objects_;
  }
 
  void setParent(Prog *parent) { parent_ = parent; }
 
private:
  Prog *parent_;
  const std::string name_;
  ObjectListType objects_;
};