//===- ProfileSummary.h - Profile summary data structure. -------*- 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 defines the profile summary data structure.
 
//
 
//===----------------------------------------------------------------------===//
 
 
 
#ifndef LLVM_IR_PROFILESUMMARY_H
 
#define LLVM_IR_PROFILESUMMARY_H
 
 
 
#include <algorithm>
 
#include <cassert>
 
#include <cstdint>
 
#include <vector>
 
 
 
namespace llvm {
 
 
 
class LLVMContext;
 
class Metadata;
 
class raw_ostream;
 
 
 
// The profile summary is one or more (Cutoff, MinCount, NumCounts) triplets.
 
// The semantics of counts depend on the type of profile. For instrumentation
 
// profile, counts are block counts and for sample profile, counts are
 
// per-line samples. Given a target counts percentile, we compute the minimum
 
// number of counts needed to reach this target and the minimum among these
 
// counts.
 
struct ProfileSummaryEntry {
 
  const uint32_t Cutoff;    ///< The required percentile of counts.
 
  const uint64_t MinCount;  ///< The minimum count for this percentile.
 
  const uint64_t NumCounts; ///< Number of counts >= the minimum count.
 
 
 
  ProfileSummaryEntry(uint32_t TheCutoff, uint64_t TheMinCount,
 
                      uint64_t TheNumCounts)
 
      : Cutoff(TheCutoff), MinCount(TheMinCount), NumCounts(TheNumCounts) {}
 
};
 
 
 
using SummaryEntryVector = std::vector<ProfileSummaryEntry>;
 
 
 
class ProfileSummary {
 
public:
 
  enum Kind { PSK_Instr, PSK_CSInstr, PSK_Sample };
 
 
 
private:
 
  const Kind PSK;
 
  const SummaryEntryVector DetailedSummary;
 
  const uint64_t TotalCount, MaxCount, MaxInternalCount, MaxFunctionCount;
 
  const uint32_t NumCounts, NumFunctions;
 
  /// If 'Partial' is false, it means the profile being used to optimize
 
  /// a target is collected from the same target.
 
  /// If 'Partial' is true, it means the profile is for common/shared
 
  /// code. The common profile is usually merged from profiles collected
 
  /// from running other targets.
 
  bool Partial = false;
 
  /// This approximately represents the ratio of the number of profile counters
 
  /// of the program being built to the number of profile counters in the
 
  /// partial sample profile. When 'Partial' is false, it is undefined. This is
 
  /// currently only available under thin LTO mode.
 
  double PartialProfileRatio = 0.0;
 
  /// Return detailed summary as metadata.
 
  Metadata *getDetailedSummaryMD(LLVMContext &Context);
 
 
 
public:
 
  static const int Scale = 1000000;
 
 
 
  ProfileSummary(Kind K, const SummaryEntryVector &DetailedSummary,
 
                 uint64_t TotalCount, uint64_t MaxCount,
 
                 uint64_t MaxInternalCount, uint64_t MaxFunctionCount,
 
                 uint32_t NumCounts, uint32_t NumFunctions,
 
                 bool Partial = false, double PartialProfileRatio = 0)
 
      : PSK(K), DetailedSummary(std::move(DetailedSummary)),
 
        TotalCount(TotalCount), MaxCount(MaxCount),
 
        MaxInternalCount(MaxInternalCount), MaxFunctionCount(MaxFunctionCount),
 
        NumCounts(NumCounts), NumFunctions(NumFunctions), Partial(Partial),
 
        PartialProfileRatio(PartialProfileRatio) {}
 
 
 
  Kind getKind() const { return PSK; }
 
  /// Return summary information as metadata.
 
  Metadata *getMD(LLVMContext &Context, bool AddPartialField = true,
 
                  bool AddPartialProfileRatioField = true);
 
  /// Construct profile summary from metdata.
 
  static ProfileSummary *getFromMD(Metadata *MD);
 
  const SummaryEntryVector &getDetailedSummary() { return DetailedSummary; }
 
  uint32_t getNumFunctions() const { return NumFunctions; }
 
  uint64_t getMaxFunctionCount() const { return MaxFunctionCount; }
 
  uint32_t getNumCounts() const { return NumCounts; }
 
  uint64_t getTotalCount() const { return TotalCount; }
 
  uint64_t getMaxCount() const { return MaxCount; }
 
  uint64_t getMaxInternalCount() const { return MaxInternalCount; }
 
  void setPartialProfile(bool PP) { Partial = PP; }
 
  bool isPartialProfile() const { return Partial; }
 
  double getPartialProfileRatio() const { return PartialProfileRatio; }
 
  void setPartialProfileRatio(double R) {
 
    assert(isPartialProfile() && "Unexpected when not partial profile");
 
    PartialProfileRatio = R;
 
  }
 
  void printSummary(raw_ostream &OS) const;
 
  void printDetailedSummary(raw_ostream &OS) const;
 
};
 
 
 
} // end namespace llvm
 
 
 
#endif // LLVM_IR_PROFILESUMMARY_H