//===-- ControlFlowContext.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

//

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

//

//  This file defines a ControlFlowContext class that is used by dataflow

//  analyses that run over Control-Flow Graphs (CFGs).

//

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

#ifndef LLVM_CLANG_ANALYSIS_FLOWSENSITIVE_CONTROLFLOWCONTEXT_H

#define LLVM_CLANG_ANALYSIS_FLOWSENSITIVE_CONTROLFLOWCONTEXT_H

#include "clang/AST/ASTContext.h"

#include "clang/AST/Decl.h"

#include "clang/AST/Stmt.h"

#include "clang/Analysis/CFG.h"

#include "llvm/ADT/DenseMap.h"

#include "llvm/Support/Error.h"

#include <memory>

#include <utility>

namespace clang {

namespace dataflow {

/// Holds CFG and other derived context that is needed to perform dataflow

/// analysis.

class ControlFlowContext {

public:

  /// Builds a ControlFlowContext from an AST node.

  static llvm::Expected<ControlFlowContext> build(const Decl *D, Stmt *S,

                                                  ASTContext *C);

  /// Returns the CFG that is stored in this context.

  const CFG &getCFG() const { return *Cfg; }

  /// Returns a mapping from statements to basic blocks that contain them.

  const llvm::DenseMap<const Stmt *, const CFGBlock *> &getStmtToBlock() const {

    return StmtToBlock;

  }

private:

  ControlFlowContext(std::unique_ptr<CFG> Cfg,

                     llvm::DenseMap<const Stmt *, const CFGBlock *> StmtToBlock)

      : Cfg(std::move(Cfg)), StmtToBlock(std::move(StmtToBlock)) {}

  std::unique_ptr<CFG> Cfg;

  llvm::DenseMap<const Stmt *, const CFGBlock *> StmtToBlock;

};

} // namespace dataflow

} // namespace clang

#endif // LLVM_CLANG_ANALYSIS_FLOWSENSITIVE_CONTROLFLOWCONTEXT_H