VisitAll.hpp

Go to the documentation of this file.

#pragma once
 
#include "slang/ast/ASTVisitor.h"
#include "slang/ast/statements/MiscStatements.h"
#include "slang/ast/symbols/InstanceSymbols.h"
#include "slang/ast/symbols/ValueSymbol.h"
#include "slang/ast/symbols/VariableSymbols.h"
 
#include <cstdint>
 
namespace slang::netlist {

struct VisitAll : public ast::ASTVisitor<VisitAll, ast::VisitFlags::AllGood> {
  uint64_t count = 0;
 
  void handle(const ast::ValueSymbol &symbol) { count++; }

  void handle(const ast::InstanceSymbol &symbol) {
    if (!symbol.body.flags.has(ast::InstanceFlags::Uninstantiated)) {
      visitDefault(symbol);
    }
  }

  void handle(const ast::VariableDeclStatement &stmt) {
    if (stmt.symbol.lifetime != ast::VariableLifetime::Static) {
      if (auto *init = stmt.symbol.getInitializer()) {
        init->visit(*this);
      }
    }
  }
};
 
} // namespace slang::netlist