cerc-io/solidity
15
0
Fork 0
mirror of https://github.com/ethereum/solidity synced 2023-10-03 13:03:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add referencedSourceUnits() helper

Browse Source
This commit is contained in:
Alex Beregszaszi 2017-07-13 00:08:28 +01:00
parent 1b0ef0b785
commit a9bddf71aa
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libsolidity/ast/AST.cpp
View File

@ -84,6 +84,16 @@ SourceUnitAnnotation& SourceUnit::annotation() const
return dynamic_cast<SourceUnitAnnotation&>(*m_annotation); return dynamic_cast<SourceUnitAnnotation&>(*m_annotation);
} }
vector<SourceUnit const*> SourceUnit::referencedSourceUnits() const
{
vector<SourceUnit const*> sourceUnits;
for (ImportDirective const* importDirective: filteredNodes<ImportDirective>(nodes()))
{
sourceUnits.push_back(importDirective->annotation().sourceUnit);
}
return sourceUnits;
}
SourceUnit const& Declaration::sourceUnit() const SourceUnit const& Declaration::sourceUnit() const
{ {
solAssert(!!m_scope, ""); solAssert(!!m_scope, "");

3
libsolidity/ast/AST.h
View File

@ -136,6 +136,9 @@ public:
std::vector<ASTPointer<ASTNode>> nodes() const { return m_nodes; } std::vector<ASTPointer<ASTNode>> nodes() const { return m_nodes; }
/// @returns a vector of referenced SourceUnits.
std::vector<SourceUnit const*> referencedSourceUnits() const;
private: private:
std::vector<ASTPointer<ASTNode>> m_nodes; std::vector<ASTPointer<ASTNode>> m_nodes;
}; };