Dimitry Andric
149 lines
4.9 KiB
C++
149 lines
4.9 KiB
C++
//===- AttrIterator.h - Classes for attribute iteration ---------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines the Attr vector and specific_attr_iterator interfaces.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_AST_ATTRITERATOR_H
|
|
#define LLVM_CLANG_AST_ATTRITERATOR_H
|
|
|
|
#include "clang/Basic/LLVM.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/Support/Casting.h"
|
|
#include <cassert>
|
|
#include <cstddef>
|
|
#include <iterator>
|
|
|
|
namespace clang {
|
|
|
|
class ASTContext;
|
|
class Attr;
|
|
|
|
} // namespace clang
|
|
|
|
// Defined in ASTContext.h
|
|
void *operator new(size_t Bytes, const clang::ASTContext &C,
|
|
size_t Alignment = 8);
|
|
|
|
// FIXME: Being forced to not have a default argument here due to redeclaration
|
|
// rules on default arguments sucks
|
|
void *operator new[](size_t Bytes, const clang::ASTContext &C,
|
|
size_t Alignment);
|
|
|
|
// It is good practice to pair new/delete operators. Also, MSVC gives many
|
|
// warnings if a matching delete overload is not declared, even though the
|
|
// throw() spec guarantees it will not be implicitly called.
|
|
void operator delete(void *Ptr, const clang::ASTContext &C, size_t);
|
|
void operator delete[](void *Ptr, const clang::ASTContext &C, size_t);
|
|
|
|
namespace clang {
|
|
|
|
/// AttrVec - A vector of Attr, which is how they are stored on the AST.
|
|
using AttrVec = SmallVector<Attr *, 4>;
|
|
|
|
/// specific_attr_iterator - Iterates over a subrange of an AttrVec, only
|
|
/// providing attributes that are of a specific type.
|
|
template <typename SpecificAttr, typename Container = AttrVec>
|
|
class specific_attr_iterator {
|
|
using Iterator = typename Container::const_iterator;
|
|
|
|
/// Current - The current, underlying iterator.
|
|
/// In order to ensure we don't dereference an invalid iterator unless
|
|
/// specifically requested, we don't necessarily advance this all the
|
|
/// way. Instead, we advance it when an operation is requested; if the
|
|
/// operation is acting on what should be a past-the-end iterator,
|
|
/// then we offer no guarantees, but this way we do not dereference a
|
|
/// past-the-end iterator when we move to a past-the-end position.
|
|
mutable Iterator Current;
|
|
|
|
void AdvanceToNext() const {
|
|
while (!isa<SpecificAttr>(*Current))
|
|
++Current;
|
|
}
|
|
|
|
void AdvanceToNext(Iterator I) const {
|
|
while (Current != I && !isa<SpecificAttr>(*Current))
|
|
++Current;
|
|
}
|
|
|
|
public:
|
|
using value_type = SpecificAttr *;
|
|
using reference = SpecificAttr *;
|
|
using pointer = SpecificAttr *;
|
|
using iterator_category = std::forward_iterator_tag;
|
|
using difference_type = std::ptrdiff_t;
|
|
|
|
specific_attr_iterator() = default;
|
|
explicit specific_attr_iterator(Iterator i) : Current(i) {}
|
|
|
|
reference operator*() const {
|
|
AdvanceToNext();
|
|
return cast<SpecificAttr>(*Current);
|
|
}
|
|
pointer operator->() const {
|
|
AdvanceToNext();
|
|
return cast<SpecificAttr>(*Current);
|
|
}
|
|
|
|
specific_attr_iterator& operator++() {
|
|
++Current;
|
|
return *this;
|
|
}
|
|
specific_attr_iterator operator++(int) {
|
|
specific_attr_iterator Tmp(*this);
|
|
++(*this);
|
|
return Tmp;
|
|
}
|
|
|
|
friend bool operator==(specific_attr_iterator Left,
|
|
specific_attr_iterator Right) {
|
|
assert((Left.Current == nullptr) == (Right.Current == nullptr));
|
|
if (Left.Current < Right.Current)
|
|
Left.AdvanceToNext(Right.Current);
|
|
else
|
|
Right.AdvanceToNext(Left.Current);
|
|
return Left.Current == Right.Current;
|
|
}
|
|
friend bool operator!=(specific_attr_iterator Left,
|
|
specific_attr_iterator Right) {
|
|
return !(Left == Right);
|
|
}
|
|
};
|
|
|
|
template <typename SpecificAttr, typename Container>
|
|
inline specific_attr_iterator<SpecificAttr, Container>
|
|
specific_attr_begin(const Container& container) {
|
|
return specific_attr_iterator<SpecificAttr, Container>(container.begin());
|
|
}
|
|
template <typename SpecificAttr, typename Container>
|
|
inline specific_attr_iterator<SpecificAttr, Container>
|
|
specific_attr_end(const Container& container) {
|
|
return specific_attr_iterator<SpecificAttr, Container>(container.end());
|
|
}
|
|
|
|
template <typename SpecificAttr, typename Container>
|
|
inline bool hasSpecificAttr(const Container& container) {
|
|
return specific_attr_begin<SpecificAttr>(container) !=
|
|
specific_attr_end<SpecificAttr>(container);
|
|
}
|
|
template <typename SpecificAttr, typename Container>
|
|
inline SpecificAttr *getSpecificAttr(const Container& container) {
|
|
specific_attr_iterator<SpecificAttr, Container> i =
|
|
specific_attr_begin<SpecificAttr>(container);
|
|
if (i != specific_attr_end<SpecificAttr>(container))
|
|
return *i;
|
|
else
|
|
return nullptr;
|
|
}
|
|
|
|
} // namespace clang
|
|
|
|
#endif // LLVM_CLANG_AST_ATTRITERATOR_H
|