mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
ef8825bfe6
LLVM has removed llvm::Optional, move over to std::optional. Also clang-format to fix up all the renamed #includes.
93 lines
2.9 KiB
C++
93 lines
2.9 KiB
C++
//===--- PostOrder.h --------------------------------------------*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_SIL_POSTORDER_H
|
|
#define SWIFT_SIL_POSTORDER_H
|
|
|
|
#include "swift/Basic/Range.h"
|
|
#include "swift/SIL/CFG.h"
|
|
#include "swift/SIL/SILBasicBlock.h"
|
|
#include "swift/SIL/SILFunction.h"
|
|
#include "swift/SILOptimizer/Analysis/Analysis.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/PostOrderIterator.h"
|
|
#include "llvm/ADT/iterator_range.h"
|
|
#include <vector>
|
|
|
|
namespace swift {
|
|
|
|
class PostOrderFunctionInfo {
|
|
std::vector<SILBasicBlock *> PostOrder;
|
|
llvm::DenseMap<SILBasicBlock *, unsigned> BBToPOMap;
|
|
|
|
public:
|
|
PostOrderFunctionInfo(SILFunction *F) {
|
|
for (auto *BB : make_range(po_begin(F), po_end(F))) {
|
|
BBToPOMap[BB] = PostOrder.size();
|
|
PostOrder.push_back(BB);
|
|
}
|
|
}
|
|
|
|
using iterator = decltype(PostOrder)::iterator;
|
|
using const_iterator = decltype(PostOrder)::const_iterator;
|
|
using reverse_iterator = decltype(PostOrder)::reverse_iterator;
|
|
using const_reverse_iterator = decltype(PostOrder)::const_reverse_iterator;
|
|
|
|
using range = iterator_range<iterator>;
|
|
using const_range = iterator_range<const_iterator>;
|
|
using reverse_range = iterator_range<reverse_iterator>;
|
|
using const_reverse_range = iterator_range<const_reverse_iterator>;
|
|
|
|
range getPostOrder() {
|
|
return make_range(PostOrder.begin(), PostOrder.end());
|
|
}
|
|
const_range getPostOrder() const {
|
|
return make_range(PostOrder.begin(), PostOrder.end());
|
|
}
|
|
reverse_range getReversePostOrder() {
|
|
return make_range(PostOrder.rbegin(), PostOrder.rend());
|
|
}
|
|
const_reverse_range getReversePostOrder() const {
|
|
return make_range(PostOrder.rbegin(), PostOrder.rend());
|
|
}
|
|
|
|
const_reverse_range getReversePostOrder(SILBasicBlock *StartBlock) const {
|
|
unsigned RPONumber = getRPONumber(StartBlock).value();
|
|
return getReversePostOrder(RPONumber);
|
|
}
|
|
|
|
const_reverse_range getReversePostOrder(unsigned RPONumber) const {
|
|
return make_range(std::next(PostOrder.rbegin(), RPONumber),
|
|
PostOrder.rend());
|
|
}
|
|
|
|
unsigned size() const { return PostOrder.size(); }
|
|
|
|
std::optional<unsigned> getPONumber(SILBasicBlock *BB) const {
|
|
auto Iter = BBToPOMap.find(BB);
|
|
if (Iter != BBToPOMap.end())
|
|
return Iter->second;
|
|
return std::nullopt;
|
|
}
|
|
|
|
std::optional<unsigned> getRPONumber(SILBasicBlock *BB) const {
|
|
auto Iter = BBToPOMap.find(BB);
|
|
if (Iter != BBToPOMap.end())
|
|
return PostOrder.size() - Iter->second - 1;
|
|
return std::nullopt;
|
|
}
|
|
};
|
|
|
|
} // end swift namespace
|
|
|
|
#endif
|