XRayInstr.h

Go to the documentation of this file.

//===--- 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);
  }
 
  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