revng-revng/include/revng-c/PHIASAPAssignmentInfo/PHIASAPAssignmentInfo.h

#ifndef REVNGC_PHIASAPASSIGNMENTINFO_H
#define REVNGC_PHIASAPASSIGNMENTINFO_H
//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

// LLVM includes
#include <llvm/Pass.h>

// revng includes
#include <revng/ADT/SmallMap.h>

struct PHIASAPAssignmentInfo : public llvm::FunctionPass {

private:
  using PHIIncomingMap = SmallMap<llvm::PHINode *, unsigned, 4>;
  using BBPHIMap = SmallMap<llvm::BasicBlock *, PHIIncomingMap, 4>;

public:
  static char ID;

  PHIASAPAssignmentInfo() : llvm::FunctionPass(ID), PHIInfoMap() {}

  void getAnalysisUsage(llvm::AnalysisUsage &AU) const override {
    AU.setPreservesAll();
  }

  bool runOnFunction(llvm::Function &F) override;

  const BBPHIMap &getBBToPHIIncomingMap() const { return PHIInfoMap; }

  BBPHIMap &&extractBBToPHIIncomingMap() { return std::move(PHIInfoMap); }

private:
  BBPHIMap PHIInfoMap;
};

#endif /* ifndef REVNGC_PHIASAPASSIGNMENTINFO_H */