inline_helper.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/PostOrderIterator.h>
 
#include "core/block.h"
#include "core/cast.h"
#include "core/clone.h"
#include "core/func.h"
#include "core/cfg.h"
 
class TrampolineGraph;
 
 
 
class InlineHelper final : public CloneVisitor {
public:
  InlineHelper(CallSite *call, Func *callee, TrampolineGraph &graph);
 
  void Inline();
 
private:
  Inst *Duplicate(Block *block, Inst *inst);
  Ref<Inst> Duplicate(Block *block, ArgInst *arg);
 
  Block *Map(Block *block) override { return blocks_[block]; }
  Ref<Inst> Map(Ref<Inst> inst) override { return insts_[inst]; }
 
  AnnotSet Annot(const Inst *inst) override;
 
  Inst *Convert(
      Type argType,
      Type valType,
      Ref<Inst> valInst,
      AnnotSet &&annot
  );
 
  void DuplicateBlocks();
  void SplitEntry();
 
private:
  const bool isTailCall_;
  const std::vector<Type> types_;
  CallSite *call_;
  const AnnotSet callAnnot_;
  Block *entry_;
  Func *callee_;
  Func *caller_;
  llvm::DenseMap<unsigned, unsigned> frameIndices_;
  Block *exit_;
  Block *throw_;
  Block *throwSplit_;
  std::vector<Ref<PhiInst>> raisePhis_;
  std::vector<Ref<PhiInst>> landPhis_;
  std::vector<Ref<PhiInst>> phis_;
  unsigned numExits_;
  llvm::SmallVector<Ref<Inst>, 8> args_;
  llvm::DenseMap<Block *, Block *> blocks_;
  std::unordered_map<Ref<Inst>, Ref<Inst>> insts_;
  llvm::ReversePostOrderTraversal<Func *> rpot_;
  TrampolineGraph &graph_;
};