aarch64isel.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/AArch64/AArch64InstrInfo.h>
#include <llvm/Target/AArch64/AArch64ISelDAGToDAG.h>
#include <llvm/Target/AArch64/AArch64MachineFunctionInfo.h>
#include <llvm/Target/AArch64/AArch64RegisterInfo.h>
#include <llvm/Target/AArch64/AArch64Subtarget.h>
#include <llvm/Target/AArch64/AArch64TargetMachine.h>
 
#include "core/insts.h"
#include "emitter/isel.h"
#include "emitter/aarch64/aarch64call.h"
 
class Data;
class Func;
class Inst;
class Prog;
 
 
 
class AArch64Matcher : public llvm::AArch64DAGMatcher {
public:
  AArch64Matcher(
      llvm::AArch64TargetMachine &tm,
      llvm::CodeGenOpt::Level ol,
      llvm::MachineFunction &mf
  );
  ~AArch64Matcher();
 
  llvm::SelectionDAG &GetDAG() const { return *CurDAG; }
  const llvm::TargetLowering *getTargetLowering() const override
  {
    return CurDAG->getMachineFunction().getSubtarget().getTargetLowering();
  }
 
private:
  llvm::AArch64TargetMachine &tm_;
};
 
 
 
class AArch64ISel final : public ISel {
public:
  static char ID;
 
  AArch64ISel(
      const Target &target,
      llvm::AArch64TargetMachine &tm,
      llvm::TargetLibraryInfo &libInfo,
      const Prog &prog,
      llvm::CodeGenOpt::Level ol,
      bool shared
  );
 
private:
  void Lower(llvm::MachineFunction &mf) override
  {
    m_.reset(new AArch64Matcher(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 AArch64_LoadLinkInst *inst);
  void LowerStoreCond(const AArch64_StoreCondInst *inst);
  void LowerAsm(const Inst *inst, const char *code);
  void LowerOut(const AArch64_OutInst *inst);
 
  void LowerArguments(bool hasVAStart) override;
  void LowerVASetup(const AArch64Call &ci);
 
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::AArch64::SP; }
 
private:
  void LowerSetSP(SDValue value);
  void SaveVarArgRegisters(const AArch64Call &ci, bool isWin64);
  llvm::SDValue LowerCallee(ConstRef<Inst> inst);
  void LowerCallSite(llvm::SDValue chain, const CallSite *call) override;
 
private:
  llvm::AArch64TargetMachine &tm_;
  std::unique_ptr<AArch64Matcher> m_;
  llvm::Function *trampoline_;
  bool shared_;
};