cppvisitormatcher.h

Go to the documentation of this file.

//
// Created by saleh on 6/25/23.
//
 
#pragma once
 
#include <algorithm>
#include <vector>
 
#include "cppast.h"
#include "cppvisitorbase.h"
#include "cppvisitorprinter.h"
 
class CppVisitorMatcher : public CppVisitorBase
{
  using MatchedFuncType = std::function<bool(CppObj* p)>;
  const std::vector<CppObjType> typesToMatch;
  MatchedFuncType matchedFunc;
 
public:
  explicit CppVisitorMatcher(const std::vector<CppObjType>& typesToMatch)
    : typesToMatch(typesToMatch)
  {
      matchedFunc = [&](CppObj* p) -> bool {
          return matched(p);
      };
  }
 
  explicit CppVisitorMatcher(const std::vector<CppObjType>& typesToMatch, MatchedFuncType &&matchedFunc)
    : typesToMatch(typesToMatch)
  {
      this->matchedFunc = matchedFunc;
  }
 
  virtual bool matched(CppObj* p)
  {
    std::cout << CppVisitorPrinter::astToString(p) << std::endl;
    return true;
  }
 
  ~CppVisitorMatcher() override = default;
 
private:
  inline bool check(CppObj* p)
  {
    if (std::find(typesToMatch.begin(), typesToMatch.end(), p->objType_) != typesToMatch.end())
    {
      return matchedFunc(p); // continue traversing the tree only if the user wants to.
    }
    return true; // continue traversing the tree to find a possible match.
  }
 
protected:
  bool visit(CppVarType* p) override
  {
    return check(p);
  }
  bool visit(CppCompound* p) override
  {
    return check(p);
  }
  bool visit(CppUnRecogPrePro* p) override
  {
    return check(p);
  }
  bool visit(CppUndef* p) override
  {
    return check(p);
  }
  bool visit(CppUsingDecl* p) override
  {
    return check(p);
  }
  bool visit(CppUsingNamespaceDecl* p) override
  {
    return check(p);
  }
  bool visit(CppVar* p) override
  {
    return check(p);
  }
  bool visit(CppVarList* p) override
  {
    return check(p);
  }
  bool visit(CppCommonBlock<CppObjType::kIfBlock>* p) override
  {
    return check(p);
  }
  bool visit(CppCommonBlock<CppObjType::kWhileBlock>* p) override
  {
    return check(p);
  }
  bool visit(CppCommonBlock<CppObjType::kDoWhileBlock>* p) override
  {
    return check(p);
  }
  bool visit(CppAsmBlock* p) override
  {
    return check(p);
  }
  bool visit(CppBlob* p) override
  {
    return check(p);
  }
  bool visit(CppDefine* p) override
  {
    return check(p);
  }
  bool visit(CppDocComment* p) override
  {
    return check(p);
  }
  bool visit(CppEnum* p) override
  {
    return check(p);
  }
  bool visit(CppExpr* p) override
  {
    return check(p);
  }
  bool visit(CppForBlock* p) override
  {
    return check(p);
  }
  bool visit(CppLambda* p) override
  {
    return check(p);
  }
  bool visit(CppDestructor* p) override
  {
    return check(p);
  }
  bool visit(CppConstructor* p) override
  {
    return check(p);
  }
  bool visit(CppFunction* p) override
  {
    return check(p);
  }
  bool visit(CppFunctionPointer* p) override
  {
    return check(p);
  }
  bool visit(CppTypeConverter* p) override
  {
    return check(p);
  }
  bool visit(CppFwdClsDecl* p) override
  {
    return check(p);
  }
  bool visit(CppHashError* p) override
  {
    return check(p);
  }
  bool visit(CppHashIf* p) override
  {
    return check(p);
  }
  bool visit(CppHashWarning* p) override
  {
    return check(p);
  }
  bool visit(CppImport* p) override
  {
    return check(p);
  }
  bool visit(CppInclude* p) override
  {
    return check(p);
  }
  bool visit(CppLabel* p) override
  {
    return check(p);
  }
  bool visit(CppMacroCall* p) override
  {
    return check(p);
  }
  bool visit(CppNamespaceAlias* p) override
  {
    return check(p);
  }
  bool visit(CppPragma* p) override
  {
    return check(p);
  }
  bool visit(CppRangeForBlock* p) override
  {
    return check(p);
  }
  bool visit(CppSwitchBlock* p) override
  {
    return check(p);
  }
  bool visit(CppTryBlock* p) override
  {
    return check(p);
  }
  bool visit(CppTypedefList* p) override
  {
    return check(p);
  }
  bool visit(CppTypedefName* p) override
  {
    return check(p);
  }
};