//===-- RISCVTargetMachine.h - Define TargetMachine for RISC-V --*- C++ -*-===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

//

// This file declares the RISC-V specific subclass of TargetMachine.

//

//===----------------------------------------------------------------------===//

#ifndef LLVM_LIB_TARGET_RISCV_RISCVTARGETMACHINE_H

#define LLVM_LIB_TARGET_RISCV_RISCVTARGETMACHINE_H

#include "MCTargetDesc/RISCVMCTargetDesc.h"

#include "RISCVSubtarget.h"

#include "llvm/CodeGen/SelectionDAGTargetInfo.h"

#include "llvm/IR/DataLayout.h"

#include "llvm/Target/TargetMachine.h"

#include <optional>

namespace llvm {

class RISCVTargetMachine : public LLVMTargetMachine {

  std::unique_ptr<TargetLoweringObjectFile> TLOF;

  mutable StringMap<std::unique_ptr<RISCVSubtarget>> SubtargetMap;

public:

  RISCVTargetMachine(const Target &T, const Triple &TT, StringRef CPU,

                     StringRef FS, const TargetOptions &Options,

                     std::optional<Reloc::Model> RM,

                     std::optional<CodeModel::Model> CM, CodeGenOptLevel OL,

                     bool JIT);

  const RISCVSubtarget *getSubtargetImpl(const Function &F) const override;

  // DO NOT IMPLEMENT: There is no such thing as a valid default subtarget,

  // subtargets are per-function entities based on the target-specific

  // attributes of each function.

  const RISCVSubtarget *getSubtargetImpl() const = delete;

  TargetPassConfig *createPassConfig(PassManagerBase &PM) override;

  TargetLoweringObjectFile *getObjFileLowering() const override {

    return TLOF.get();

  }

  MachineFunctionInfo *

  createMachineFunctionInfo(BumpPtrAllocator &Allocator, const Function &F,

                            const TargetSubtargetInfo *STI) const override;

  TargetTransformInfo getTargetTransformInfo(const Function &F) const override;

  bool isNoopAddrSpaceCast(unsigned SrcAS, unsigned DstAS) const override;

  yaml::MachineFunctionInfo *createDefaultFuncInfoYAML() const override;

  yaml::MachineFunctionInfo *

  convertFuncInfoToYAML(const MachineFunction &MF) const override;

  bool parseMachineFunctionInfo(const yaml::MachineFunctionInfo &,

                                PerFunctionMIParsingState &PFS,

                                SMDiagnostic &Error,

                                SMRange &SourceRange) const override;

};

} // namespace llvm

#endif