forked from OSchip/llvm-project
84 lines
2.3 KiB
C++
84 lines
2.3 KiB
C++
//===--- AstGuard.h - Parser Ownership Tracking Utilities -------*- 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 RAII objects for managing ExprTy* and StmtTy*.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_PARSE_ASTGUARD_H
|
|
#define LLVM_CLANG_PARSE_ASTGUARD_H
|
|
|
|
#include "clang/Parse/Action.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
|
|
namespace clang
|
|
{
|
|
/// RAII SmallVector wrapper that holds Action::ExprTy* and similar,
|
|
/// automatically freeing them on destruction unless it's been disowned.
|
|
/// Instantiated for statements and expressions (Action::DeleteStmt and
|
|
/// Action::DeleteExpr).
|
|
template <ASTDestroyer Destroyer, unsigned N>
|
|
class ASTVector : public llvm::SmallVector<void*, N> {
|
|
private:
|
|
#if !defined(DISABLE_SMART_POINTERS)
|
|
Action &Actions;
|
|
bool Owns;
|
|
|
|
void destroy() {
|
|
if (Owns) {
|
|
while (!this->empty()) {
|
|
(Actions.*Destroyer)(this->back());
|
|
this->pop_back();
|
|
}
|
|
}
|
|
}
|
|
|
|
ASTVector(const ASTVector&); // DO NOT IMPLEMENT
|
|
// Reference member prevents copy assignment.
|
|
#endif
|
|
|
|
public:
|
|
#if !defined(DISABLE_SMART_POINTERS)
|
|
ASTVector(Action &actions) : Actions(actions), Owns(true) {}
|
|
|
|
~ASTVector() { destroy(); }
|
|
#else
|
|
ASTVector(Action &) {}
|
|
#endif
|
|
|
|
void **take() {
|
|
#if !defined(DISABLE_SMART_POINTERS)
|
|
Owns = false;
|
|
#endif
|
|
return &(*this)[0];
|
|
}
|
|
|
|
#if !defined(DISABLE_SMART_POINTERS)
|
|
Action &getActions() const { return Actions; }
|
|
#endif
|
|
};
|
|
|
|
/// A SmallVector of statements, with stack size 32 (as that is the only one
|
|
/// used.)
|
|
typedef ASTVector<&Action::DeleteStmt, 32> StmtVector;
|
|
/// A SmallVector of expressions, with stack size 12 (the maximum used.)
|
|
typedef ASTVector<&Action::DeleteExpr, 12> ExprVector;
|
|
|
|
template <ASTDestroyer Destroyer, unsigned N> inline
|
|
ASTMultiPtr<Destroyer> move_arg(ASTVector<Destroyer, N> &vec) {
|
|
#if !defined(DISABLE_SMART_POINTERS)
|
|
return ASTMultiPtr<Destroyer>(vec.getActions(), vec.take(), vec.size());
|
|
#else
|
|
return ASTMultiPtr<Destroyer>(vec.take(), vec.size());
|
|
#endif
|
|
}
|
|
}
|
|
|
|
#endif
|