//===--- XRayInstr.h --------------------------------------------*- 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

//

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

//

/// \file

/// Defines the clang::XRayInstrKind enum.

//

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

#ifndef LLVM_CLANG_BASIC_XRAYINSTR_H

#define LLVM_CLANG_BASIC_XRAYINSTR_H

#include "clang/Basic/LLVM.h"

#include "llvm/ADT/StringRef.h"

#include "llvm/Support/MathExtras.h"

#include <cassert>

#include <cstdint>

namespace clang {

using XRayInstrMask = uint32_t;

namespace XRayInstrKind {

// TODO: Auto-generate these as we add more instrumentation kinds.

enum XRayInstrOrdinal : XRayInstrMask {

  XRIO_FunctionEntry,

  XRIO_FunctionExit,

  XRIO_Custom,

  XRIO_Typed,

  XRIO_Count

};

constexpr XRayInstrMask None = 0;

constexpr XRayInstrMask FunctionEntry = 1U << XRIO_FunctionEntry;

constexpr XRayInstrMask FunctionExit = 1U << XRIO_FunctionExit;

constexpr XRayInstrMask Custom = 1U << XRIO_Custom;

constexpr XRayInstrMask Typed = 1U << XRIO_Typed;

constexpr XRayInstrMask All = FunctionEntry | FunctionExit | Custom | Typed;

} // namespace XRayInstrKind

struct XRayInstrSet {

  bool has(XRayInstrMask K) const {

    assert(llvm::isPowerOf2_32(K));

    return Mask & K;

  }

  bool hasOneOf(XRayInstrMask K) const { return Mask & K; }

  void set(XRayInstrMask K, bool Value) {

    Mask = Value ? (Mask | K) : 
(Mask & ~K)0
;

  }

  void clear(XRayInstrMask K = XRayInstrKind::All) { Mask &= ~K; }

  bool empty() const { return Mask == 0; }

  bool full() const { return Mask == XRayInstrKind::All; }

  XRayInstrMask Mask = 0;

};

/// Parses a command line argument into a mask.

XRayInstrMask parseXRayInstrValue(StringRef Value);

/// Serializes a set into a list of command line arguments.

void serializeXRayInstrValue(XRayInstrSet Set,

                             SmallVectorImpl<StringRef> &Values);

} // namespace clang

#endif // LLVM_CLANG_BASIC_XRAYINSTR_H