atom.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 <llvm/ADT/ilist.h>
 
#include "core/value.h"
#include "core/global.h"
#include "core/symbol_table.h"
#include "core/item.h"
 
class Data;
class Expr;
 
 
 
 
class Atom
  : public llvm::ilist_node_with_parent<Atom, Object>
  , public Global
{
public:
  static constexpr Global::Kind kGlobalKind = Global::Kind::ATOM;
 
  using ItemListType = llvm::ilist<Item>;
  // Iterators over items.
  typedef ItemListType::iterator iterator;
  typedef ItemListType::const_iterator const_iterator;
 
public:
  Atom(
      const std::string_view name,
      Visibility visibility = Visibility::LOCAL,
      std::optional<llvm::Align> align = std::nullopt)
    : Global(Global::Kind::ATOM, name, visibility, 0)
    , parent_(nullptr)
    , align_(align)
  {
  }
 
  ~Atom() override;
 
  void removeFromParent() override;
  void eraseFromParent() override;
 
  Object *getParent() const { return parent_; }
 
  void remove(iterator it) { items_.remove(it); }
  void erase(iterator it) { items_.erase(it); }
  void AddItem(Item *atom, Item *before = nullptr);
 
  // Iterators over items.
  bool empty() const { return items_.empty(); }
  size_t size() const { return items_.size(); }
  iterator begin() { return items_.begin(); }
  iterator end() { return items_.end(); }
  const_iterator begin() const { return items_.begin(); }
  const_iterator end() const { return items_.end(); }
  void clear() { items_.clear(); }
 
  size_t GetByteSize() const;
  void SetAlignment(llvm::Align align) { align_ = align; }
  std::optional<llvm::Align> GetAlignment() const override { return align_; }
 
  Prog *getProg() override;
 
  void dump(llvm::raw_ostream &os = llvm::errs()) const;
 
private:
  friend struct SymbolTableListTraits<Atom>;
  friend struct llvm::ilist_traits<Item>;
  static ItemListType Atom::*getSublistAccess(Item *) {
    return &Atom::items_;
  }
 
  void setParent(Object *parent) { parent_ = parent; }
 
private:
  Object *parent_;
  ItemListType items_;
  std::optional<llvm::Align> align_;
};
 
inline llvm::raw_ostream &operator<<(llvm::raw_ostream &os, const Atom &atom)
{
  atom.dump(os);
  return os;
}