//===- BranchProbAnalysis.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
//
//===----------------------------------------------------------------------===//
//====  Copyright (c) 2015 Advanced Micro Devices, Inc.  All rights reserved.
//
//               Developed by: Advanced Micro Devices, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// with the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// Redistributions of source code must retain the above copyright notice, this
// list of conditions and the following disclaimers.
//
// Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimers in the documentation
// and/or other materials provided with the distribution.
//
// Neither the names of Advanced Micro Devices, Inc., nor the names of its
// contributors may be used to endorse or promote products derived from this
// Software without specific prior written permission.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS WITH
// THE SOFTWARE.
//
//===----------------------------------------------------------------------===//
//
// Compute probability of executing true or false side of a branch.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_ANALYSIS_BRANCH_PROB_ANALYSIS_H
#define LLVM_ANALYSIS_BRANCH_PROB_ANALYSIS_H
#include "llvm/IR/PassManager.h"
#include "llvm/Pass.h"
#include <map>
namespace llvm {
class Module;
class AnalysisUsages;
class BranchProbInfo {
  std::map<BasicBlock *, int> BlockFreq;

public:
  bool invalidate(Function &F, const PreservedAnalyses &PA,
                  FunctionAnalysisManager::Invalidator &);
  int getValue(BasicBlock *);
  void setValue(BasicBlock *, int);
};

class BranchProbAnalysis : public AnalysisInfoMixin<BranchProbAnalysis> {
  friend AnalysisInfoMixin<BranchProbAnalysis>;
  static AnalysisKey Key;

public:
  using Result = BranchProbInfo;
  Result run(Function &F, FunctionAnalysisManager &AM);
};

class BranchProbInfoWrapperPass : public FunctionPass {
  BranchProbInfo BPI;

public:
  static char ID;
};

} // namespace llvm
#endif
