Skip to content

Commit

Permalink
Switch to RecursiveASTVisitor.
Browse files Browse the repository at this point in the history
  • Loading branch information
vgvassilev committed Nov 20, 2023
1 parent c9f030a commit 764fa88
Show file tree
Hide file tree
Showing 2 changed files with 78 additions and 120 deletions.
49 changes: 14 additions & 35 deletions include/clad/Differentiator/TBRAnalyzer.h
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#ifndef CLAD_DIFFERENTIATOR_TBRANALYZER_H
#define CLAD_DIFFERENTIATOR_TBRANALYZER_H

#include "clang/AST/StmtVisitor.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/Analysis/CFG.h"

#include "clad/Differentiator/CladUtils.h"
Expand All @@ -14,9 +14,7 @@ using namespace clang;

namespace clad {

class TBRAnalyzer : public clang::ConstStmtVisitor<TBRAnalyzer> {
private:

class TBRAnalyzer : public clang::RecursiveASTVisitor<TBRAnalyzer> {
/// ProfileID is the key type for ArrMap used to represent array indices
/// and object fields.
using ProfileID = clad_compat::FoldingSetNodeID;
Expand Down Expand Up @@ -295,37 +293,18 @@ class TBRAnalyzer : public clang::ConstStmtVisitor<TBRAnalyzer> {
/// Visitors
void Analyze(const clang::FunctionDecl* FD);

void VisitCFGBlock(const clang::CFGBlock* block);

void Visit(const clang::Stmt* stmt) {
clang::ConstStmtVisitor<TBRAnalyzer, void>::Visit(stmt);
}

void VisitArraySubscriptExpr(const clang::ArraySubscriptExpr* ASE);
void VisitBinaryOperator(const clang::BinaryOperator* BinOp);
void VisitCallExpr(const clang::CallExpr* CE);
void VisitCompoundStmt(const clang::CompoundStmt* CS);
void VisitConditionalOperator(const clang::ConditionalOperator* CO);
void VisitCXXConstructExpr(const clang::CXXConstructExpr* CE);
void VisitCXXDefaultArgExpr(const clang::CXXDefaultArgExpr* DE);
void VisitCXXStaticCastExpr(const clang::CXXStaticCastExpr* SCE);
void VisitDeclRefExpr(const clang::DeclRefExpr* DRE);
void VisitDeclStmt(const clang::DeclStmt* DS);
void VisitExprWithCleanups(const clang::ExprWithCleanups* EWC);
void VisitImplicitCastExpr(const clang::ImplicitCastExpr* ICE);
void VisitInitListExpr(const clang::InitListExpr* ILE);
void VisitMemberExpr(const clang::MemberExpr* ME);
void VisitParenExpr(const clang::ParenExpr* PE);
void VisitReturnStmt(const clang::ReturnStmt* RS);
void VisitUnaryOperator(const clang::UnaryOperator* UnOp);

/// FIXME: Make sure these are not necessary
/// Unused Visitors:
// void VisitCXXBoolLiteralExpr(const clang::CXXBoolLiteralExpr* BL);
// void VisitCXXThisExpr(const clang::CXXThisExpr* TE);
// void VisitFloatingLiteral(const clang::FloatingLiteral* FL);
// void VisitIntegerLiteral(const clang::IntegerLiteral* IL);
// void VisitStmt(const clang::Stmt* S);
void VisitCFGBlock(const clang::CFGBlock& block);

bool VisitArraySubscriptExpr(clang::ArraySubscriptExpr* ASE);
bool VisitBinaryOperator(clang::BinaryOperator* BinOp);
bool VisitCallExpr(clang::CallExpr* CE);
bool VisitConditionalOperator(clang::ConditionalOperator* CO);
bool VisitCXXConstructExpr(clang::CXXConstructExpr* CE);
bool VisitDeclRefExpr(clang::DeclRefExpr* DRE);
bool VisitDeclStmt(clang::DeclStmt* DS);
bool VisitInitListExpr(clang::InitListExpr* ILE);
bool VisitMemberExpr(clang::MemberExpr* ME);
bool VisitUnaryOperator(clang::UnaryOperator* UnOp);
};

} // end namespace clad
Expand Down
Loading

0 comments on commit 764fa88

Please sign in to comment.