riscvcall.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 <vector>
 
#include <llvm/ADT/ArrayRef.h>
#include <llvm/ADT/iterator_range.h>
#include <llvm/MC/MCRegister.h>
 
#include "core/inst.h"
#include "core/func.h"
#include "emitter/call_lowering.h"
 
class CallInst;
class InvokeInst;
class ReturnInst;
class RaiseInst;
class Func;
 
 
 
class RISCVCall final : public CallLowering {
public:
  RISCVCall(const Func *func)
    : CallLowering(func)
    , numFixedArgs_(func->GetNumParams())
  {
    AnalyseFunc(func);
  }
 
  RISCVCall(const CallSite *inst)
    : CallLowering(inst)
    , numFixedArgs_(inst->GetNumFixedArgs().value_or(inst->arg_size()))
  {
    AnalyseCall(inst);
  }
 
  RISCVCall(const ReturnInst *inst)
    : CallLowering(inst)
  {
    AnalyseReturn(inst);
  }
 
  RISCVCall(const LandingPadInst *inst)
    : CallLowering(inst)
  {
    AnalysePad(inst);
  }
 
  RISCVCall(const RaiseInst *inst)
    : CallLowering(inst)
  {
    AnalyseRaise(inst);
  }
 
  llvm::ArrayRef<llvm::MCPhysReg> GetUnusedGPRs() const;
  llvm::ArrayRef<llvm::MCPhysReg> GetUsedGPRs() const;
  llvm::ArrayRef<llvm::MCPhysReg> GetUnusedFPRs() const;
  llvm::ArrayRef<llvm::MCPhysReg> GetUsedFPRs() const;
 
  unsigned GetFrameSize() const override { return stack_; }
 
private:
  void AssignArgC(unsigned i, FlaggedType type) override;
  void AssignArgOCaml(unsigned i, FlaggedType type) override;
  void AssignArgOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignArgOCamlGc(unsigned i, FlaggedType type) override;
  void AssignArgXen(unsigned i, FlaggedType type) override;
  void AssignArgWin64(unsigned i, FlaggedType type) override;
 
  void AssignRetC(unsigned i, FlaggedType type) override;
  void AssignRetOCaml(unsigned i, FlaggedType type) override;
  void AssignRetOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignRetOCamlGc(unsigned i, FlaggedType type) override;
  void AssignRetXen(unsigned i, FlaggedType type) override;
  void AssignRetWin64(unsigned i, FlaggedType type) override;
 
  void AssignArgReg(ArgLoc &loc, llvm::MVT vt, llvm::Register reg);
  void AssignArgStack(ArgLoc &loc, llvm::MVT type, unsigned size);
  void AssignRetReg(RetLoc &loc, llvm::MVT vt, llvm::Register reg);
 
  llvm::ArrayRef<llvm::MCPhysReg> GetGPRs() const;
  llvm::ArrayRef<llvm::MCPhysReg> GetFPRs() const;
 
private:
  unsigned numFixedArgs_ = 0;
  unsigned argI_ = 0;
  unsigned argF_ = 0;
  unsigned retI_ = 0;
  unsigned retF_ = 0;
  unsigned stack_ = 0;
};