printer.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 <ostream>
#include <unordered_map>
 
#include <llvm/Support/raw_ostream.h>
 
#include "core/cond.h"
#include "core/calling_conv.h"
#include "core/visibility.h"
#include "core/inst.h"
 
class Atom;
class Block;
class Data;
class Func;
class Prog;
class Object;
 
 
 
class Printer {
public:
  Printer(llvm::raw_ostream &os) : os_(os) {}
 
  virtual void Print(const Prog &prog);
  virtual void Print(const Data &data);
  virtual void Print(const Object &object);
  virtual void Print(const Atom &atom);
  virtual void Print(const Func &func);
  virtual void Print(const Block &block);
  virtual void Print(const Inst &inst);
  virtual void Print(const Expr &expr);
  virtual void Print(ConstRef<Value> val);
  virtual void Print(const std::string_view str);
 
protected:
  virtual void PrintFuncHeader(const Func &func) {}
  virtual void PrintInstHeader(const Inst &inst) {}
 
private:
  void PrintImpl(const Inst &inst);
 
protected:
  llvm::raw_ostream &os_;
 
private:
  std::unordered_map<ConstRef<Inst>, unsigned> insts_;
};