mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
48 lines
1.3 KiB
C++
48 lines
1.3 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <map>
|
|
#include <set>
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
namespace llvm {
|
|
class ScalarEvolution;
|
|
class SCEV;
|
|
} // end namespace llvm
|
|
|
|
namespace dla {
|
|
struct LayoutTypeSystemNode;
|
|
} // end namespace dla
|
|
|
|
/// Class useful to explore an llvm::SCEV expression to find its base
|
|
/// addresses.
|
|
class SCEVBaseAddressExplorer {
|
|
public:
|
|
using SCEVTypeMap = std::map<const llvm::SCEV *, dla::LayoutTypeSystemNode *>;
|
|
|
|
private:
|
|
llvm::SmallVector<const llvm::SCEV *, 4> Worklist;
|
|
|
|
public:
|
|
SCEVBaseAddressExplorer() = default;
|
|
~SCEVBaseAddressExplorer() = default;
|
|
|
|
/// Returns a set containing the SCEVs of \Root 's base addresses.
|
|
//
|
|
// The function works exploring the AST of the SCEV, going from the \Root
|
|
// towards its operands.
|
|
// If \M is not empty, all the SCEVs with an entry in \M are considered as
|
|
// addresses, and the exploration of the operands does not traverse them, even
|
|
// if the SCEV potentially has the expressive power to do it.
|
|
std::set<const llvm::SCEV *> findBases(llvm::ScalarEvolution *SE,
|
|
const llvm::SCEV *Root,
|
|
const SCEVTypeMap &M);
|
|
|
|
private:
|
|
size_t checkAddressOrTraverse(llvm::ScalarEvolution *SE, const llvm::SCEV *S);
|
|
};
|