//===--- AttrImpl.cpp - Classes for representing attributes -----*- 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 contains out-of-line methods for Attr classes.

//

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

#include "clang/AST/ASTContext.h"

#include "clang/AST/Attr.h"

#include "clang/AST/Expr.h"

#include "clang/AST/Type.h"

using namespace clang;

void LoopHintAttr::printPrettyPragma(raw_ostream &OS,

                                     const PrintingPolicy &Policy) const {

  unsigned SpellingIndex = getAttributeSpellingListIndex();

  // For "#pragma unroll" and "#pragma nounroll" the string "unroll" or

  // "nounroll" is already emitted as the pragma name.

  if (SpellingIndex == Pragma_nounroll ||

      SpellingIndex == Pragma_nounroll_and_jam)

    return;

  else if (SpellingIndex == Pragma_unroll ||

           SpellingIndex == Pragma_unroll_and_jam) {

    OS << ' ' << getValueString(Policy);

    return;

  }

  assert(SpellingIndex == Pragma_clang_loop && "Unexpected spelling");

  OS << ' ' << getOptionName(option) << getValueString(Policy);

}

// Return a string containing the loop hint argument including the

// enclosing parentheses.

std::string LoopHintAttr::getValueString(const PrintingPolicy &Policy) const {

  std::string ValueName;

  llvm::raw_string_ostream OS(ValueName);

  OS << "(";

  if (state == Numeric)

    value->printPretty(OS, nullptr, Policy);

  else if (state == Enable)

    OS << "enable";

  else if (state == Full)

    OS << "full";

  else if (state == AssumeSafety)

    OS << "assume_safety";

  else

    OS << "disable";

  OS << ")";

  return OS.str();

}

// Return a string suitable for identifying this attribute in diagnostics.

std::string

LoopHintAttr::getDiagnosticName(const PrintingPolicy &Policy) const {

  unsigned SpellingIndex = getAttributeSpellingListIndex();

  if (SpellingIndex == Pragma_nounroll)

    return "#pragma nounroll";

  else if (SpellingIndex == Pragma_unroll)

    return "#pragma unroll" +

           (option == UnrollCount ? getValueString(Policy) : 
""4
);

  else if (SpellingIndex == Pragma_nounroll_and_jam)

    return "#pragma nounroll_and_jam";

  else if (SpellingIndex == Pragma_unroll_and_jam)

    return "#pragma unroll_and_jam" +

           (option == UnrollAndJamCount ? getValueString(Policy) : 
""0
);

  assert(SpellingIndex == Pragma_clang_loop && "Unexpected spelling");

  return getOptionName(option) + getValueString(Policy);

}

void OMPDeclareSimdDeclAttr::printPrettyPragma(

    raw_ostream &OS, const PrintingPolicy &Policy) const {

  if (getBranchState() != BS_Undefined)

    OS << ' ' << ConvertBranchStateTyToStr(getBranchState());

  if (auto *E = getSimdlen()) {

    OS << " simdlen(";

    E->printPretty(OS, nullptr, Policy);

    OS << ")";

  }

  if (uniforms_size() > 0) {

    OS << " uniform";

    StringRef Sep = "(";

    for (auto *E : uniforms()) {

      OS << Sep;

      E->printPretty(OS, nullptr, Policy);

      Sep = ", ";

    }

    OS << ")";

  }

  alignments_iterator NI = alignments_begin();

  for (auto *E : aligneds()) {

    OS << " aligned(";

    E->printPretty(OS, nullptr, Policy);

    if (*NI) {

      OS << ": ";

      (*NI)->printPretty(OS, nullptr, Policy);

    }

    OS << ")";

    ++NI;

  }

  steps_iterator I = steps_begin();

  modifiers_iterator MI = modifiers_begin();

  for (auto *E : linears()) {

    OS << " linear(";

    if (*MI != OMPC_LINEAR_unknown)

      OS << getOpenMPSimpleClauseTypeName(llvm::omp::Clause::OMPC_linear, *MI)

         << "(";

    E->printPretty(OS, nullptr, Policy);

    if (*MI != OMPC_LINEAR_unknown)

      OS << ")";

    if (*I) {

      OS << ": ";

      (*I)->printPretty(OS, nullptr, Policy);

    }

    OS << ")";

    ++I;

    ++MI;

  }

}

void OMPDeclareTargetDeclAttr::printPrettyPragma(

    raw_ostream &OS, const PrintingPolicy &Policy) const {

  // Use fake syntax because it is for testing and debugging purpose only.

  if (getDevType() != DT_Any)

    OS << " device_type(" << ConvertDevTypeTyToStr(getDevType()) << ")";

  if (getMapType() != MT_To)

    OS << ' ' << ConvertMapTypeTyToStr(getMapType());

}

llvm::Optional<OMPDeclareTargetDeclAttr::MapTypeTy>

OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(const ValueDecl *VD) {

  if (!VD->hasAttrs())

    return llvm::None;

  unsigned Level = 0;

  const OMPDeclareTargetDeclAttr *FoundAttr = nullptr;

  for (const auto *Attr : VD->specific_attrs<OMPDeclareTargetDeclAttr>()) {

    if (Level < Attr->getLevel()) {

      Level = Attr->getLevel();

      FoundAttr = Attr;

    }

  }

  if (FoundAttr)

    return FoundAttr->getMapType();

  return llvm::None;

}

llvm::Optional<OMPDeclareTargetDeclAttr::DevTypeTy>

OMPDeclareTargetDeclAttr::getDeviceType(const ValueDecl *VD) {

  if (!VD->hasAttrs())

    return llvm::None;

  unsigned Level = 0;

  const OMPDeclareTargetDeclAttr *FoundAttr = nullptr;

  for (const auto *Attr : VD->specific_attrs<OMPDeclareTargetDeclAttr>()) {

    if (Level < Attr->getLevel()) {

      Level = Attr->getLevel();

      FoundAttr = Attr;

    }

  }

  if (FoundAttr)

    return FoundAttr->getDevType();

  return llvm::None;

}

llvm::Optional<SourceLocation>

OMPDeclareTargetDeclAttr::getLocation(const ValueDecl *VD) {

  if (!VD->hasAttrs())

    return llvm::None;

  unsigned Level = 0;

  const OMPDeclareTargetDeclAttr *FoundAttr = nullptr;

  for (const auto *Attr : VD->specific_attrs<OMPDeclareTargetDeclAttr>()) {

    if (Level < Attr->getLevel()) {

      Level = Attr->getLevel();

      FoundAttr = Attr;

    }

  }

  if (FoundAttr)

    return FoundAttr->getRange().getBegin();

  return llvm::None;

}

namespace clang {

llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, const OMPTraitInfo &TI);

llvm::raw_ostream &operator<<(llvm::raw_ostream &OS, const OMPTraitInfo *TI);

}

void OMPDeclareVariantAttr::printPrettyPragma(

    raw_ostream &OS, const PrintingPolicy &Policy) const {

  if (const Expr *E = getVariantFuncRef()) {

    OS << "(";

    E->printPretty(OS, nullptr, Policy);

    OS << ")";

  }

  OS << " match(" << traitInfos << ")";

}

#include "clang/AST/AttrImpl.inc"