1 //===--- StmtGraphTraits.h - Graph Traits for the class Stmt ----*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines a template specialization of llvm::GraphTraits to
11 // treat ASTs (Stmt*) as graphs
13 //===----------------------------------------------------------------------===//
15 #ifndef LLVM_CLANG_AST_STMTGRAPHTRAITS_H
16 #define LLVM_CLANG_AST_STMTGRAPHTRAITS_H
18 #include "clang/AST/Stmt.h"
19 #include "llvm/ADT/DepthFirstIterator.h"
20 #include "llvm/ADT/GraphTraits.h"
24 //template <typename T> struct GraphTraits;
27 template <> struct GraphTraits<clang::Stmt*> {
28 typedef clang::Stmt NodeType;
29 typedef clang::Stmt * NodeRef;
30 typedef clang::Stmt::child_iterator ChildIteratorType;
31 typedef llvm::df_iterator<clang::Stmt*> nodes_iterator;
33 static NodeType* getEntryNode(clang::Stmt* S) { return S; }
35 static inline ChildIteratorType child_begin(NodeType* N) {
36 if (N) return N->child_begin();
37 else return ChildIteratorType();
40 static inline ChildIteratorType child_end(NodeType* N) {
41 if (N) return N->child_end();
42 else return ChildIteratorType();
45 static nodes_iterator nodes_begin(clang::Stmt* S) {
49 static nodes_iterator nodes_end(clang::Stmt* S) {
55 template <> struct GraphTraits<const clang::Stmt*> {
56 typedef const clang::Stmt NodeType;
57 typedef const clang::Stmt * NodeRef;
58 typedef clang::Stmt::const_child_iterator ChildIteratorType;
59 typedef llvm::df_iterator<const clang::Stmt*> nodes_iterator;
61 static NodeType* getEntryNode(const clang::Stmt* S) { return S; }
63 static inline ChildIteratorType child_begin(NodeType* N) {
64 if (N) return N->child_begin();
65 else return ChildIteratorType();
68 static inline ChildIteratorType child_end(NodeType* N) {
69 if (N) return N->child_end();
70 else return ChildIteratorType();
73 static nodes_iterator nodes_begin(const clang::Stmt* S) {
77 static nodes_iterator nodes_end(const clang::Stmt* S) {
83 } // end namespace llvm