//===- IRBindings.cpp - Additional bindings for ir ------------------------===//

//

//                     The LLVM Compiler Infrastructure

//

// This file is distributed under the University of Illinois Open Source

// License. See LICENSE.TXT for details.

//

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

//

// This file defines additional C bindings for the ir component.

//

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

/*

 * Shamelessly based on IRBindings.cpp from LLVM's GO bindings. The original

 * file could be found at bindings/go/llvm/IRBindings.cpp in LLVM source tree.

 */

#include <llvm-c-ext/CoreExt.h>

#include "llvm/IR/Attributes.h"

#include "llvm/IR/DebugLoc.h"

#include "llvm/IR/Function.h"

#include "llvm/IR/IRBuilder.h"

#include "llvm/IR/LLVMContext.h"

#include "llvm/IR/DebugInfoMetadata.h"

#include "llvm/IR/Module.h"

using namespace llvm;

void LLVMAddFunctionAttr2(LLVMValueRef Fn, uint64_t PA) {

  Function *Func = unwrap<Function>(Fn);

  const AttributeSet PAL = Func->getAttributes();

  AttrBuilder B(PA);

  const AttributeSet PALnew =

    PAL.addAttributes(Func->getContext(), AttributeSet::FunctionIndex,

                      AttributeSet::get(Func->getContext(),

                                        AttributeSet::FunctionIndex, B));

  Func->setAttributes(PALnew);

}

uint64_t LLVMGetFunctionAttr2(LLVMValueRef Fn) {

  Function *Func = unwrap<Function>(Fn);

  const AttributeSet PAL = Func->getAttributes();

  return PAL.Raw(AttributeSet::FunctionIndex);

}

void LLVMRemoveFunctionAttr2(LLVMValueRef Fn, uint64_t PA) {

  Function *Func = unwrap<Function>(Fn);

  const AttributeSet PAL = Func->getAttributes();

  AttrBuilder B(PA);

  const AttributeSet PALnew =

    PAL.removeAttributes(Func->getContext(), AttributeSet::FunctionIndex,

                         AttributeSet::get(Func->getContext(),

                                           AttributeSet::FunctionIndex, B));

  Func->setAttributes(PALnew);

}

LLVMMetadataRef LLVMConstantAsMetadata(LLVMValueRef C) {

  return wrap(ConstantAsMetadata::get(unwrap<Constant>(C)));

}

LLVMMetadataRef LLVMMDString2(LLVMContextRef C, const char *Str, unsigned SLen) {

  return wrap(MDString::get(*unwrap(C), StringRef(Str, SLen)));

}

LLVMMetadataRef LLVMMDNode2(LLVMContextRef C, LLVMMetadataRef *MDs,

                            unsigned Count) {

  return wrap(

      MDNode::get(*unwrap(C), ArrayRef<Metadata *>(unwrap(MDs), Count)));

}

LLVMMetadataRef LLVMTemporaryMDNode(LLVMContextRef C, LLVMMetadataRef *MDs,

                                    unsigned Count) {

  return wrap(MDTuple::getTemporary(*unwrap(C),

                                    ArrayRef<Metadata *>(unwrap(MDs), Count))

                  .release());

}

void LLVMAddNamedMetadataOperand2(LLVMModuleRef M, const char *name,

                                  LLVMMetadataRef Val) {

  NamedMDNode *N = unwrap(M)->getOrInsertNamedMetadata(name);

  if (!N)

    return;

  if (!Val)

    return;

  N->addOperand(unwrap<MDNode>(Val));

}

void LLVMSetMetadata2(LLVMValueRef Inst, unsigned KindID, LLVMMetadataRef MD) {

  MDNode *N = MD ? unwrap<MDNode>(MD) : nullptr;

  unwrap<Instruction>(Inst)->setMetadata(KindID, N);

}

void LLVMMetadataReplaceAllUsesWith(LLVMMetadataRef MD, LLVMMetadataRef New) {

  auto *Node = unwrap<MDNode>(MD);

  Node->replaceAllUsesWith(unwrap<Metadata>(New));

  MDNode::deleteTemporary(Node);

}

void LLVMSetCurrentDebugLocation2(LLVMBuilderRef Bref, unsigned Line,

                                  unsigned Col, LLVMMetadataRef Scope,

                                  LLVMMetadataRef InlinedAt) {

  unwrap(Bref)->SetCurrentDebugLocation(

      DebugLoc::get(Line, Col, Scope ? unwrap<MDNode>(Scope) : nullptr,

                    InlinedAt ? unwrap<MDNode>(InlinedAt) : nullptr));

}

LLVMMetadataRef LLVMGetCurrentDebugLocation2(LLVMBuilderRef Bref) {

	return wrap(unwrap(Bref)->getCurrentDebugLocation());

}

void LLVMModuleAddModuleFlag(LLVMModuleRef M, LLVMModuleFlagBehavior Behavior, const char *Key, LLVMMetadataRef Val) {

	unwrap(M)->addModuleFlag((Module::ModFlagBehavior)Behavior, StringRef(Key), unwrap(Val));

}

LLVMValueRef LLVMMetadataAsValue(LLVMMetadataRef MD) {

	return LLVMMetadataAsValueInContext(LLVMGetGlobalContext(), MD);

}

LLVMValueRef LLVMMetadataAsValueInContext(LLVMContextRef C, LLVMMetadataRef MD) {

	return wrap(MetadataAsValue::get(*unwrap(C), unwrap(MD)));

}

LLVMMetadataRef LLVMValueAsMetadata(LLVMValueRef Value) {

	return wrap(ValueAsMetadata::get(unwrap(Value)));

}

int64_t LLVMValueAsSInt64(LLVMValueRef Value) {

	if (ConstantInt* CI = dyn_cast<ConstantInt>(unwrap(Value))) {

		if (CI->getBitWidth() <= 64) {

			return CI->getSExtValue();

		}

	}

	// Not a constant integer value,

	return 0;

}

uint64_t LLVMValueAsUInt64(LLVMValueRef Value) {

	if (ConstantInt* CI = dyn_cast<ConstantInt>(unwrap(Value))) {

		if (CI->getBitWidth() <= 64) {

			return CI->getZExtValue();

		}

	}

	// Not a constant integer value,

	return 0;

}