mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Andrea Gussoni
2e49e21925
Removed the `Switch` BBNode, which was used to create the intermediates nodes for making an original `switch` node a nested tree of `if` checks. Also removed the `IfEqual` AST node, which was used to represent the intermediate check nodes in the AST, for later reconstructing the original `Switch` node in the AST, when possible.
99 lines
2.8 KiB
C++
99 lines
2.8 KiB
C++
/// \file BasicBlockNode.cpp
|
|
/// \brief
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
// Standard includes
|
|
#include <cstdlib>
|
|
#include <map>
|
|
#include <set>
|
|
|
|
// Local libraries includes
|
|
#include "revng-c/RestructureCFGPass/BasicBlockNode.h"
|
|
#include "revng-c/RestructureCFGPass/RegionCFGTree.h"
|
|
#include "revng-c/RestructureCFGPass/Utils.h"
|
|
|
|
using namespace llvm;
|
|
|
|
BasicBlockNode::BasicBlockNode(RegionCFG *Parent,
|
|
llvm::BasicBlock *BB,
|
|
RegionCFG *Collapsed,
|
|
const std::string &Name,
|
|
Type T,
|
|
unsigned Value) :
|
|
ID(Parent->getNewID()),
|
|
Parent(Parent),
|
|
BB(BB),
|
|
CollapsedRegion(Collapsed),
|
|
NodeType(T),
|
|
Name(Name),
|
|
StateVariableValue(Value) {
|
|
}
|
|
|
|
void BasicBlockNode::removeNode() {
|
|
Parent->removeNode(this);
|
|
}
|
|
|
|
// TODO: Check why this implementation is really necessary.
|
|
void BasicBlockNode::printAsOperand(raw_ostream &O, bool PrintType) {
|
|
O << Name;
|
|
}
|
|
|
|
void BasicBlockNode::removeSuccessor(BasicBlockNode *Successor) {
|
|
// TODO: maybe add removeTrue and removeFalse?
|
|
// revng_assert(not isCheck());
|
|
for (auto It = Successors.begin(); It != Successors.end(); It++) {
|
|
if (*It == Successor) {
|
|
Successors.erase(It);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void BasicBlockNode::removePredecessor(BasicBlockNode *Predecessor) {
|
|
for (auto It = Predecessors.begin(); It != Predecessors.end(); It++) {
|
|
if (*It == Predecessor) {
|
|
Predecessors.erase(It);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
using BBNodeMap = std::map<BasicBlockNode *, BasicBlockNode *>;
|
|
|
|
static void handleNeighbors(const BBNodeMap &SubstitutionMap,
|
|
BasicBlockNode::links_container &Neighbors) {
|
|
Neighbors.erase(std::remove_if(Neighbors.begin(),
|
|
Neighbors.end(),
|
|
[&SubstitutionMap](BasicBlockNode *N) {
|
|
return SubstitutionMap.count(N) == 0;
|
|
}),
|
|
Neighbors.end());
|
|
for (BasicBlockNode *&Neighbor : Neighbors) {
|
|
revng_assert(SubstitutionMap.count(Neighbor) != 0);
|
|
Neighbor = SubstitutionMap.at(Neighbor);
|
|
}
|
|
}
|
|
|
|
void BasicBlockNode::updatePointers(const BBNodeMap &SubstitutionMap) {
|
|
handleNeighbors(SubstitutionMap, Predecessors);
|
|
if (not isCheck()) {
|
|
handleNeighbors(SubstitutionMap, Successors);
|
|
} else { // don't screw up the if/then/else of check nodes
|
|
|
|
auto It = SubstitutionMap.find(getTrue());
|
|
if (It != SubstitutionMap.end())
|
|
setTrue(It->second);
|
|
else
|
|
setTrue(nullptr);
|
|
|
|
It = SubstitutionMap.find(getFalse());
|
|
if (It != SubstitutionMap.end())
|
|
setFalse(It->second);
|
|
else
|
|
setFalse(nullptr);
|
|
}
|
|
}
|