reference_graph.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 <memory>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <vector>
#include <llvm/Support/raw_ostream.h>
 
class Block;
class CallGraph;
class Func;
class Global;
class Prog;
class MovInst;
class Object;
 
 
 
using OffsetSet = std::set<std::pair<int64_t, int64_t>>;
 
class ReferenceGraph {
public:
  struct Node {
    bool HasIndirectCalls = false;
    bool HasRaise = false;
    bool HasBarrier = false;
    std::unordered_set<Object *> ReadRanges;
    std::unordered_map<Object *, OffsetSet> ReadOffsets;
    std::unordered_set<Object *> WrittenRanges;
    std::unordered_map<Object *, OffsetSet> WrittenOffsets;
    std::unordered_set<Global *> Escapes;
    std::unordered_set<Func *> Called;
    std::unordered_set<Block *> Blocks;
 
    void Merge(const Node &that);
    void AddRead(Object *object);
    void AddWrite(Object *object);
 
    void dump(llvm::raw_ostream &os) const;
  };
 
  ReferenceGraph(Prog &prog, CallGraph &graph);
 
  const Node &operator[](Func &func);
 
protected:
  virtual bool Skip(Func &func) { return false; }
 
private:
  void ExtractReferences(Func &func, Node &node);
  void Build();
  void Classify(Object *o, const MovInst &inst, Node &node);
  void Classify(Object *o, const MovInst &inst, Node &node, int64_t offset);
 
private:
  CallGraph &graph_;
  std::unordered_map<Func *, Node *> funcToNode_;
  std::vector<std::unique_ptr<Node>> nodes_;
  bool built_;
};
 
inline llvm::raw_ostream &operator<<(
    llvm::raw_ostream &os,
    const ReferenceGraph::Node &node)
{
  node.dump(os);
  return os;
}