15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- StmtVisitor.h - Visitor for Stmt subclasses ------------*- C++ -*-===// 25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// The LLVM Compiler Infrastructure 45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source 60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details. 75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 107999991060b136db49eb9d6b34595383de56278cPeter Collingbourne// This file defines the StmtVisitor and ConstStmtVisitor interfaces. 115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#ifndef LLVM_CLANG_AST_STMTVISITOR_H 155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#define LLVM_CLANG_AST_STMTVISITOR_H 165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 17c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner#include "clang/AST/ExprCXX.h" 18f494b579b22f9950f5af021f0bf9879a91bb8b41Steve Naroff#include "clang/AST/ExprObjC.h" 1916f0049415ec596504891259e2a83e19871c0d52Chris Lattner#include "clang/AST/StmtCXX.h" 2016f0049415ec596504891259e2a83e19871c0d52Chris Lattner#include "clang/AST/StmtObjC.h" 214fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#include "clang/AST/StmtOpenMP.h" 22c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner 235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencernamespace clang { 241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 257999991060b136db49eb9d6b34595383de56278cPeter Collingbournetemplate <typename T> struct make_ptr { typedef T *type; }; 267999991060b136db49eb9d6b34595383de56278cPeter Collingbournetemplate <typename T> struct make_const_ptr { typedef const T *type; }; 271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 287999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// StmtVisitorBase - This class implements a simple visitor for Stmt 297999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// subclasses. Since Expr derives from Stmt, this also includes support for 307999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// visiting Exprs. 317999991060b136db49eb9d6b34595383de56278cPeter Collingbournetemplate<template <typename> class Ptr, typename ImplClass, typename RetTy=void> 327999991060b136db49eb9d6b34595383de56278cPeter Collingbourneclass StmtVisitorBase { 335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic: 347999991060b136db49eb9d6b34595383de56278cPeter Collingbourne 357999991060b136db49eb9d6b34595383de56278cPeter Collingbourne#define PTR(CLASS) typename Ptr<CLASS>::type 367999991060b136db49eb9d6b34595383de56278cPeter Collingbourne#define DISPATCH(NAME, CLASS) \ 377999991060b136db49eb9d6b34595383de56278cPeter Collingbourne return static_cast<ImplClass*>(this)->Visit ## NAME(static_cast<PTR(CLASS)>(S)) 387999991060b136db49eb9d6b34595383de56278cPeter Collingbourne 397999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy Visit(PTR(Stmt) S) { 401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 4103d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner // If we have a binary expr, dispatch to the subcode of the binop. A smart 4203d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner // optimizer (e.g. LLVM) will fold this comparison into the switch stmt 4303d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner // below. 447999991060b136db49eb9d6b34595383de56278cPeter Collingbourne if (PTR(BinaryOperator) BinOp = dyn_cast<BinaryOperator>(S)) { 4503d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner switch (BinOp->getOpcode()) { 462de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_PtrMemD: DISPATCH(BinPtrMemD, BinaryOperator); 472de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_PtrMemI: DISPATCH(BinPtrMemI, BinaryOperator); 482de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Mul: DISPATCH(BinMul, BinaryOperator); 492de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Div: DISPATCH(BinDiv, BinaryOperator); 502de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Rem: DISPATCH(BinRem, BinaryOperator); 512de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Add: DISPATCH(BinAdd, BinaryOperator); 522de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Sub: DISPATCH(BinSub, BinaryOperator); 532de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Shl: DISPATCH(BinShl, BinaryOperator); 542de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Shr: DISPATCH(BinShr, BinaryOperator); 552d5049fcbb4cb9b67086de32b806fffd9f856e30Chris Lattner 562de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_LT: DISPATCH(BinLT, BinaryOperator); 572de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_GT: DISPATCH(BinGT, BinaryOperator); 582de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_LE: DISPATCH(BinLE, BinaryOperator); 592de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_GE: DISPATCH(BinGE, BinaryOperator); 602de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_EQ: DISPATCH(BinEQ, BinaryOperator); 612de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_NE: DISPATCH(BinNE, BinaryOperator); 621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 632de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_And: DISPATCH(BinAnd, BinaryOperator); 642de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Xor: DISPATCH(BinXor, BinaryOperator); 652de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Or : DISPATCH(BinOr, BinaryOperator); 662de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_LAnd: DISPATCH(BinLAnd, BinaryOperator); 672de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_LOr : DISPATCH(BinLOr, BinaryOperator); 682de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Assign: DISPATCH(BinAssign, BinaryOperator); 692de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_MulAssign: DISPATCH(BinMulAssign, CompoundAssignOperator); 702de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_DivAssign: DISPATCH(BinDivAssign, CompoundAssignOperator); 712de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_RemAssign: DISPATCH(BinRemAssign, CompoundAssignOperator); 722de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_AddAssign: DISPATCH(BinAddAssign, CompoundAssignOperator); 732de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_SubAssign: DISPATCH(BinSubAssign, CompoundAssignOperator); 742de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_ShlAssign: DISPATCH(BinShlAssign, CompoundAssignOperator); 752de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_ShrAssign: DISPATCH(BinShrAssign, CompoundAssignOperator); 762de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_AndAssign: DISPATCH(BinAndAssign, CompoundAssignOperator); 772de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_OrAssign: DISPATCH(BinOrAssign, CompoundAssignOperator); 782de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_XorAssign: DISPATCH(BinXorAssign, CompoundAssignOperator); 792de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case BO_Comma: DISPATCH(BinComma, BinaryOperator); 8003d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner } 817999991060b136db49eb9d6b34595383de56278cPeter Collingbourne } else if (PTR(UnaryOperator) UnOp = dyn_cast<UnaryOperator>(S)) { 8213d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner switch (UnOp->getOpcode()) { 832de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_PostInc: DISPATCH(UnaryPostInc, UnaryOperator); 842de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_PostDec: DISPATCH(UnaryPostDec, UnaryOperator); 852de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_PreInc: DISPATCH(UnaryPreInc, UnaryOperator); 862de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_PreDec: DISPATCH(UnaryPreDec, UnaryOperator); 872de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_AddrOf: DISPATCH(UnaryAddrOf, UnaryOperator); 882de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Deref: DISPATCH(UnaryDeref, UnaryOperator); 892de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Plus: DISPATCH(UnaryPlus, UnaryOperator); 902de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Minus: DISPATCH(UnaryMinus, UnaryOperator); 912de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Not: DISPATCH(UnaryNot, UnaryOperator); 922de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_LNot: DISPATCH(UnaryLNot, UnaryOperator); 932de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Real: DISPATCH(UnaryReal, UnaryOperator); 942de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Imag: DISPATCH(UnaryImag, UnaryOperator); 952de56d1d0c3a504ad1529de2677628bdfbb95cd4John McCall case UO_Extension: DISPATCH(UnaryExtension, UnaryOperator); 96224605064a4ef87d1c3d35ad1cb363f8b534012bSebastian Redl } 9703d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner } 981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 99c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner // Top switch stmt: dispatch to VisitFooStmt for each FooStmt. 100c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner switch (S->getStmtClass()) { 101b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie default: llvm_unreachable("Unknown stmt kind!"); 1027381d5cfbd599fa2b9e215011ad7cbd449de231aSean Hunt#define ABSTRACT_STMT(STMT) 103f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define STMT(CLASS, PARENT) \ 10403d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner case Stmt::CLASS ## Class: DISPATCH(CLASS, CLASS); 1054bfe1968410ea8ffe3b4f629addd7c4bcf484765Sean Hunt#include "clang/AST/StmtNodes.inc" 106c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner } 107c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner } 1081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 109c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner // If the implementation chooses not to implement a certain visit method, fall 110c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner // back on VisitExpr or whatever else is the superclass. 111f2cad8633e46cce12fc3d77c0bd451ffd7264bafDouglas Gregor#define STMT(CLASS, PARENT) \ 1127999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy Visit ## CLASS(PTR(CLASS) S) { DISPATCH(PARENT, PARENT); } 1134bfe1968410ea8ffe3b4f629addd7c4bcf484765Sean Hunt#include "clang/AST/StmtNodes.inc" 114c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner 11503d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner // If the implementation doesn't implement binary operator methods, fall back 11603d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner // on VisitBinaryOperator. 11713d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner#define BINOP_FALLBACK(NAME) \ 1187999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy VisitBin ## NAME(PTR(BinaryOperator) S) { \ 11913d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner DISPATCH(BinaryOperator, BinaryOperator); \ 12003d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner } 121224605064a4ef87d1c3d35ad1cb363f8b534012bSebastian Redl BINOP_FALLBACK(PtrMemD) BINOP_FALLBACK(PtrMemI) 12213d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(Mul) BINOP_FALLBACK(Div) BINOP_FALLBACK(Rem) 12313d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(Add) BINOP_FALLBACK(Sub) BINOP_FALLBACK(Shl) 12413d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(Shr) 1251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12613d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(LT) BINOP_FALLBACK(GT) BINOP_FALLBACK(LE) 12713d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(GE) BINOP_FALLBACK(EQ) BINOP_FALLBACK(NE) 12813d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(And) BINOP_FALLBACK(Xor) BINOP_FALLBACK(Or) 12913d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(LAnd) BINOP_FALLBACK(LOr) 13013d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner 13113d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(Assign) 13213d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner BINOP_FALLBACK(Comma) 13313d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner#undef BINOP_FALLBACK 13461919a125a9422002588a7b59e9d1df70089ee17Chris Lattner 13561919a125a9422002588a7b59e9d1df70089ee17Chris Lattner // If the implementation doesn't implement compound assignment operator 13661919a125a9422002588a7b59e9d1df70089ee17Chris Lattner // methods, fall back on VisitCompoundAssignOperator. 13761919a125a9422002588a7b59e9d1df70089ee17Chris Lattner#define CAO_FALLBACK(NAME) \ 1387999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy VisitBin ## NAME(PTR(CompoundAssignOperator) S) { \ 13961919a125a9422002588a7b59e9d1df70089ee17Chris Lattner DISPATCH(CompoundAssignOperator, CompoundAssignOperator); \ 14061919a125a9422002588a7b59e9d1df70089ee17Chris Lattner } 14161919a125a9422002588a7b59e9d1df70089ee17Chris Lattner CAO_FALLBACK(MulAssign) CAO_FALLBACK(DivAssign) CAO_FALLBACK(RemAssign) 14261919a125a9422002588a7b59e9d1df70089ee17Chris Lattner CAO_FALLBACK(AddAssign) CAO_FALLBACK(SubAssign) CAO_FALLBACK(ShlAssign) 14361919a125a9422002588a7b59e9d1df70089ee17Chris Lattner CAO_FALLBACK(ShrAssign) CAO_FALLBACK(AndAssign) CAO_FALLBACK(OrAssign) 14461919a125a9422002588a7b59e9d1df70089ee17Chris Lattner CAO_FALLBACK(XorAssign) 14561919a125a9422002588a7b59e9d1df70089ee17Chris Lattner#undef CAO_FALLBACK 1461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14713d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner // If the implementation doesn't implement unary operator methods, fall back 14813d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner // on VisitUnaryOperator. 14913d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner#define UNARYOP_FALLBACK(NAME) \ 1507999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy VisitUnary ## NAME(PTR(UnaryOperator) S) { \ 15113d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner DISPATCH(UnaryOperator, UnaryOperator); \ 15203d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner } 15313d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(PostInc) UNARYOP_FALLBACK(PostDec) 15413d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(PreInc) UNARYOP_FALLBACK(PreDec) 15513d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(AddrOf) UNARYOP_FALLBACK(Deref) 1561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 15713d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(Plus) UNARYOP_FALLBACK(Minus) 15813d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(Not) UNARYOP_FALLBACK(LNot) 15913d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner UNARYOP_FALLBACK(Real) UNARYOP_FALLBACK(Imag) 160de7e66256b1bdfcf6526994825a8c8fced52a31cEli Friedman UNARYOP_FALLBACK(Extension) 16113d130c3d225cdb5b1e6d9e86cea12eea718c707Chris Lattner#undef UNARYOP_FALLBACK 1621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 163c5598cbc4c3f2fb515af7f142f78ff630bdb5c01Chris Lattner // Base case, ignore it. :) 1647999991060b136db49eb9d6b34595383de56278cPeter Collingbourne RetTy VisitStmt(PTR(Stmt) Node) { return RetTy(); } 16503d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner 1667999991060b136db49eb9d6b34595383de56278cPeter Collingbourne#undef PTR 16703d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner#undef DISPATCH 1687999991060b136db49eb9d6b34595383de56278cPeter Collingbourne}; 1697999991060b136db49eb9d6b34595383de56278cPeter Collingbourne 1707999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// StmtVisitor - This class implements a simple visitor for Stmt subclasses. 1717999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// Since Expr derives from Stmt, this also includes support for visiting Exprs. 1727999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// 1737999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// This class does not preserve constness of Stmt pointers (see also 1747999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// ConstStmtVisitor). 1757999991060b136db49eb9d6b34595383de56278cPeter Collingbournetemplate<typename ImplClass, typename RetTy=void> 1767999991060b136db49eb9d6b34595383de56278cPeter Collingbourneclass StmtVisitor 1777999991060b136db49eb9d6b34595383de56278cPeter Collingbourne : public StmtVisitorBase<make_ptr, ImplClass, RetTy> {}; 1787999991060b136db49eb9d6b34595383de56278cPeter Collingbourne 1797999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// ConstStmtVisitor - This class implements a simple visitor for Stmt 1807999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// subclasses. Since Expr derives from Stmt, this also includes support for 1817999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// visiting Exprs. 1827999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// 1837999991060b136db49eb9d6b34595383de56278cPeter Collingbourne/// This class preserves constness of Stmt pointers (see also StmtVisitor). 1847999991060b136db49eb9d6b34595383de56278cPeter Collingbournetemplate<typename ImplClass, typename RetTy=void> 1857999991060b136db49eb9d6b34595383de56278cPeter Collingbourneclass ConstStmtVisitor 1867999991060b136db49eb9d6b34595383de56278cPeter Collingbourne : public StmtVisitorBase<make_const_ptr, ImplClass, RetTy> {}; 18703d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner 1884fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev/// \brief This class implements a simple visitor for OMPClause 1894fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev/// subclasses. 1904fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevtemplate<class ImplClass, template <typename> class Ptr, typename RetTy> 1914fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevclass OMPClauseVisitorBase { 1924fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevpublic: 1934fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#define PTR(CLASS) typename Ptr<CLASS>::type 1944fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#define DISPATCH(CLASS) \ 1954fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev return static_cast<ImplClass*>(this)->Visit##CLASS(static_cast<PTR(CLASS)>(S)) 1964fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev 1974fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#define OPENMP_CLAUSE(Name, Class) \ 1984fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev RetTy Visit ## Class (PTR(Class) S) { DISPATCH(Class); } 1994fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#include "clang/Basic/OpenMPKinds.def" 2004fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev 2014fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev RetTy Visit(PTR(OMPClause) S) { 2024fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev // Top switch clause: visit each OMPClause. 2034fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev switch (S->getClauseKind()) { 2044fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev default: llvm_unreachable("Unknown clause kind!"); 2054fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#define OPENMP_CLAUSE(Name, Class) \ 2064fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev case OMPC_ ## Name : return Visit ## Class(static_cast<PTR(Class)>(S)); 2074fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#include "clang/Basic/OpenMPKinds.def" 2084fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev } 2094fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev } 2104fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev // Base case, ignore it. :) 2114fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev RetTy VisitOMPClause(PTR(OMPClause) Node) { return RetTy(); } 2124fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#undef PTR 2134fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev#undef DISPATCH 2144fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev}; 2154fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev 2164fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevtemplate<class ImplClass, typename RetTy = void> 2174fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevclass OMPClauseVisitor : 2184fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev public OMPClauseVisitorBase <ImplClass, make_ptr, RetTy> {}; 2194fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevtemplate<class ImplClass, typename RetTy = void> 2204fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataevclass ConstOMPClauseVisitor : 2214fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev public OMPClauseVisitorBase <ImplClass, make_const_ptr, RetTy> {}; 2224fa7eab771ab8212e1058bd1a91061ff120c8fbbAlexey Bataev 22303d6fb99224c36935c9af9f4785cb33453c99b2bChris Lattner} // end namespace clang 2245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 2255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif 226