/*========================== begin_copyright_notice ============================

Copyright (C) 2020-2021 Intel Corporation

SPDX-License-Identifier: MIT

============================= end_copyright_notice ===========================*/

#ifndef IGCLLVM_IR_INTRINSICINST_H
#define IGCLLVM_IR_INTRINSICINST_H

#include <llvm/Config/llvm-config.h>
#include <llvm/IR/Constants.h>
#include <llvm/IR/IntrinsicInst.h>
#include "llvm/IR/DebugInfoMetadata.h"
#include "llvm/Support/Casting.h"

#include "Probe/Assertion.h"

namespace IGCLLVM {
inline bool isKillLocation(const llvm::DbgVariableIntrinsic *DbgInst) {
  IGC_ASSERT(DbgInst);
#if LLVM_VERSION_MAJOR <= 12
  return llvm::dyn_cast<llvm::UndefValue>(DbgInst->getVariableLocation());
#elif LLVM_VERSION_MAJOR <= 15
  return DbgInst->isUndef();
#else // LLVM_VERSION_MAJOR >= 16
  return DbgInst->isKillLocation();
#endif
}

inline llvm::Value *getVariableLocation(const llvm::DbgVariableIntrinsic *DbgInst) {
  IGC_ASSERT(DbgInst);
#if LLVM_VERSION_MAJOR <= 12
  return DbgInst->getVariableLocation();
#else
  IGC_ASSERT_MESSAGE((DbgInst->getNumVariableLocationOps() == 1) || isKillLocation(DbgInst),
                     "unsupported number of location ops");
  return DbgInst->getVariableLocationOp(0);
#endif
}

inline void setKillLocation(llvm::DbgVariableIntrinsic *DbgInst) {
  IGC_ASSERT(DbgInst);

#if LLVM_VERSION_MAJOR <= 12
  auto *OP = DbgInst->getVariableLocation();
  IGC_ASSERT_MESSAGE(OP != nullptr, "Empty dbg var not supported");

  auto *Undef = llvm::UndefValue::get(OP->getType());
  DbgInst->setOperand(0, llvm::MetadataAsValue::get(DbgInst->getContext(), llvm::ValueAsMetadata::get(Undef)));
#elif LLVM_VERSION_MAJOR <= 15
  DbgInst->setUndef();
#else // LLVM_VERSION_MAJOR >= 16
  DbgInst->setKillLocation();
#endif
}

inline void setExpression(llvm::DbgVariableIntrinsic *DbgInst, llvm::DIExpression *NewExpr) {
  IGC_ASSERT(DbgInst);
#if LLVM_VERSION_MAJOR <= 12
  DbgInst->setArgOperand(2, llvm::MetadataAsValue::get(DbgInst->getContext(), NewExpr));
#else
  DbgInst->setExpression(NewExpr);
#endif
}
} // namespace IGCLLVM

#endif
