StatsManager.h

Go to the documentation of this file.

#ifndef STATS_MANAGER_H
#define STATS_MANAGER_H

#include <string>

#include "clang/AST/ASTConsumer.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/AST/Stmt.h"

class StatsManager {
public:
  static void ComputeStats(std::string &FileName);
  static void Print();

  static int GetNumOfWords() { return NumOfWords; }
  static int GetNumOfStatements() { return NumOfStatements; }
  static int GetNumOfFunctions() { return NumOfFunctions; }

  static void IncreaseNumOfFunctions();
  static void IncreaseNumOfStatements();

  static bool isCountableStatement(clang::Stmt *S);

private:
  StatsManager() {}
  StatsManager(std::string &FileName);
  ~StatsManager() {}

  static int NumOfWords;
  static int NumOfStatements;
  static int NumOfFunctions;
};

class StatsVisitor;

class StatsComputer : public clang::ASTConsumer {
  friend class StatsVisitor;

public:
  StatsComputer() : Visitor(NULL) {}
  ~StatsComputer() { delete Visitor; }

private:
  void Initialize(clang::ASTContext &Ctx);
  bool HandleTopLevelDecl(clang::DeclGroupRef D);

  StatsManager *Manager;
  StatsVisitor *Visitor;
};

class StatsVisitor : public clang::RecursiveASTVisitor<StatsVisitor> {
public:
  StatsVisitor() {}

  bool VisitFunctionDecl(clang::FunctionDecl *FD);
  bool VisitStmt(clang::Stmt *S);
};

#endif // STATS_MANAGER_H