ppcisel.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 <unordered_map>
 
#include <llvm/ADT/DenseMap.h>
#include <llvm/Analysis/OptimizationRemarkEmitter.h>
#include <llvm/CodeGen/MachineBasicBlock.h>
#include <llvm/IR/Module.h>
#include <llvm/IR/Type.h>
#include <llvm/Pass.h>
#include <llvm/Target/PowerPC/PPCInstrInfo.h>
#include <llvm/Target/PowerPC/PPCISelDAGToDAG.h>
#include <llvm/Target/PowerPC/PPCMachineFunctionInfo.h>
#include <llvm/Target/PowerPC/PPCRegisterInfo.h>
#include <llvm/Target/PowerPC/PPCSubtarget.h>
#include <llvm/Target/PowerPC/PPCTargetMachine.h>
 
#include "core/insts.h"
#include "emitter/isel.h"
#include "emitter/ppc/ppccall.h"
 
class Data;
class Func;
class Inst;
class Prog;
 
 
 
class PPCMatcher : public llvm::PPCDAGMatcher {
public:
  PPCMatcher(
      llvm::PPCTargetMachine &tm,
      llvm::CodeGenOpt::Level ol,
      llvm::MachineFunction &mf
  );
  ~PPCMatcher();
 
  llvm::SelectionDAG &GetDAG() const { return *CurDAG; }
  const llvm::TargetLowering *getTargetLowering() const override
  {
    return CurDAG->getMachineFunction().getSubtarget().getTargetLowering();
  }
 
private:
  llvm::PPCTargetMachine &tm_;
};
 
 
 
class PPCISel final : public ISel {
public:
  static char ID;
 
  PPCISel(
      const Target &target,
      llvm::PPCTargetMachine &tm,
      llvm::TargetLibraryInfo &libInfo,
      const Prog &prog,
      llvm::CodeGenOpt::Level ol,
      bool shared
  );
 
private:
  void Lower(llvm::MachineFunction &mf) override
  {
    m_.reset(new PPCMatcher(tm_, ol_, mf));
  }
 
  SDValue GetRegArch(Register reg) override;
 
  void LowerArch(const Inst *inst) override;
 
  void LowerSyscall(const SyscallInst *inst) override;
  void LowerClone(const CloneInst *inst) override;
  void LowerReturn(const ReturnInst *inst) override;
  void LowerRaise(const RaiseInst *inst) override;
  void LowerSpawn(const SpawnInst *inst) override;
  void LowerLandingPad(const LandingPadInst *inst) override;
  void LowerSet(const SetInst *inst) override;
  void LowerLoadLink(const PPC_LoadLinkInst *inst);
  void LowerStoreCond(const PPC_StoreCondInst *inst);
  void LowerFence(const PPC_FenceInst *inst);
  void LowerIFence(const PPC_IFenceInst *inst);
 
  void LowerArguments(bool hasVAStart) override;
  void LowerVASetup(const PPCCall &ci);
  llvm::SDValue StoreTOC(llvm::SDValue chain);
 
private:
  llvm::SelectionDAG &GetDAG() const override { return m_->GetDAG(); }
  void PreprocessISelDAG() override { m_->PreprocessISelDAG(); }
  void PostprocessISelDAG() override { m_->PostprocessISelDAG(); }
  void Select(SDNode *node) override { m_->Select(node); }
 
  llvm::Register GetStackRegister() const override { return llvm::PPC::X1; }
 
private:
  ConstRef<Value> GetCallee(ConstRef<Inst> inst);
  std::pair<unsigned, llvm::SDValue> LowerCallee(
      ConstRef<Value> inst,
      llvm::SDValue &chain,
      llvm::SDValue &glue
  );
  void LowerCallSite(llvm::SDValue chain, const CallSite *call) override;
 
private:
  llvm::PPCTargetMachine &tm_;
  std::unique_ptr<PPCMatcher> m_;
  llvm::Function *trampoline_;
  bool shared_;
};