nodes.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 <set>
 
#include "core/adt/bitset.h"
#include "core/analysis/reference_graph.h"
#include "core/dag.h"
#include "core/ref.h"
#include "core/type.h"
 
class Object;
class Inst;
class MemoryStoreInst;
 
 
 
using ObjectOffsetMap = std::unordered_map<ID<Object>, OffsetSet>;
 
struct ObjectClosure {
  BitSet<Func> Funcs;
  BitSet<Object> Objects;
};
 
struct FuncClosure {
  std::unique_ptr<DAGFunc> DAG;
  BitSet<Func> Funcs;
  BitSet<Object> Escaped;
  BitSet<Object> Stored;
  BitSet<Object> Loaded;
  bool Raises;
  bool Indirect;
};
 
struct NodeState {
  BitSet<Func> Funcs;
  BitSet<Object> Escaped;
  BitSet<Object> Stored;
  std::unordered_map
    < ID<Object>
    , std::map<uint64_t, std::pair<Type, Ref<Inst>>>
    > Stores;
 
  void Merge(const NodeState &that);
 
  void Overwrite(ID<Object> changed)
  {
    BitSet<Object> imprecise;
    imprecise.Insert(changed);
    return Overwrite(imprecise);
  }
 
  void Overwrite(const BitSet<Object> &changed);
 
  void dump(llvm::raw_ostream &os);
};
 
 
struct ReverseNodeState {
  DAGBlock &Node;
  llvm::DenseSet<ReverseNodeState *> Succs;
 
  std::unordered_map
    < ID<Object>
    , std::map<uint64_t, std::pair<MemoryStoreInst *, uint64_t>>
    > Stores;
  BitSet<Object> Stored;
 
  BitSet<Object> Loaded;
 
  ReverseNodeState(DAGBlock &node);
 
  void Merge(const ReverseNodeState &that);
 
  void Store(ID<Object> id);
  void Store(
      ID<Object> id,
      uint64_t start,
      uint64_t end,
      MemoryStoreInst *store = nullptr
  );
  void Store(const BitSet<Object> &changed);
 
  void Load(ID<Object> id);
  void Load(ID<Object> id, uint64_t start, uint64_t end);
  void Load(const BitSet<Object> &loaded);
 
  void dump(llvm::raw_ostream &os);
};