Changeset e5c74d1


Ignore:
Timestamp:
Jan 22, 2018, 2:27:24 PM (4 years ago)
Author:
Rob Schluntz <rschlunt@…>
Branches:
aaron-thesis, arm-eh, cleanup-dtors, deferred_resn, demangler, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, resolv-new, with_gc
Children:
b04a408
Parents:
13deae88
Message:

Convert DeclStats? to PassVisitor?

File:
1 edited

Legend:

Unmodified
Added
Removed
  • src/CodeTools/DeclStats.cc

    r13deae88 re5c74d1  
    2323#include <utility>                 // for pair, make_pair
    2424
     25#include "Common/PassVisitor.h"
    2526#include "Common/SemanticError.h"  // for SemanticError
    2627#include "Common/VectorMap.h"      // for VectorMap
     
    3536namespace CodeTools {
    3637
    37         class DeclStats : public Visitor {
     38        struct DeclStats : public WithShortCircuiting {
    3839                template<typename T>
    3940                static void sum(T& a, const T& b) { a += b; }
     
    200201
    201202        public:
    202                 using Visitor::visit;
    203 
    204                 virtual void visit( FunctionDecl *decl ) {
     203                void previsit( FunctionDecl *decl ) {
    205204                        // skip if already seen declaration for this function
    206                         const std::string& mangleName = decl->get_mangleName().empty() ? decl->get_name() : decl->get_mangleName();
    207                         if ( ! seen_names.insert( mangleName ).second ) {
    208                                 maybeAccept( decl->get_statements(), *this );
    209                                 return;
    210                         }
    211 
    212                         Stats& stats = for_linkage[ decl->get_linkage() ];
    213 
    214                         ++stats.n_decls;
    215                         FunctionType* fnTy = decl->get_functionType();
    216                         const Type::ForallList& forall = fnTy->get_forall();
    217                         ++stats.n_type_params.at( forall.size() );
    218                         unsigned n_assns = 0;
    219                         for ( TypeDecl* fdecl : forall ) {
    220                                 n_assns += fdecl->get_assertions().size();
    221                                 for ( DeclarationWithType* assn : fdecl->get_assertions() ) {
    222                                         FunctionType *assnTy = 0;
    223                                         if ( ObjectDecl *assnObj = dynamic_cast<ObjectDecl*>(assn) ) {
    224                                                 if ( PointerType *ptrTy = dynamic_cast<PointerType*>(assnObj->get_type()) ) {
    225                                                         assnTy = dynamic_cast<FunctionType*>(ptrTy->get_base());
    226                                                 } else assnTy = dynamic_cast<FunctionType*>(assnObj->get_type());
    227                                         } else if ( FunctionDecl *assnDecl = dynamic_cast<FunctionDecl*>(assn) ) {
    228                                                 assnTy = assnDecl->get_functionType();
     205                        const std::string& mangleName = decl->get_mangleName().empty() ? decl->name : decl->get_mangleName();
     206                        if ( seen_names.insert( mangleName ).second ) {
     207                                Stats& stats = for_linkage[ decl->linkage ];
     208
     209                                ++stats.n_decls;
     210                                FunctionType* fnTy = decl->type;
     211                                const Type::ForallList& forall = fnTy->forall;
     212                                ++stats.n_type_params.at( forall.size() );
     213                                unsigned n_assns = 0;
     214                                for ( TypeDecl* fdecl : forall ) {
     215                                        n_assns += fdecl->assertions.size();
     216                                        for ( DeclarationWithType* assn : fdecl->assertions ) {
     217                                                FunctionType *assnTy = nullptr;
     218                                                if ( ObjectDecl *assnObj = dynamic_cast<ObjectDecl*>(assn) ) {
     219                                                        if ( PointerType *ptrTy = dynamic_cast<PointerType*>(assnObj->get_type()) ) {
     220                                                                assnTy = dynamic_cast<FunctionType*>(ptrTy->get_base());
     221                                                        } else assnTy = dynamic_cast<FunctionType*>(assnObj->get_type());
     222                                                } else if ( FunctionDecl *assnDecl = dynamic_cast<FunctionDecl*>(assn) ) {
     223                                                        assnTy = assnDecl->type;
     224                                                }
     225                                                if ( assnTy ) analyzeFunc( assnTy, stats, stats.assn_params, stats.assn_returns );
    229226                                        }
    230                                         if ( assnTy ) analyzeFunc( assnTy, stats, stats.assn_params, stats.assn_returns );
    231                                 }
    232                         }
    233                         ++stats.n_assns[ n_assns ];
    234 
    235                         ++stats.by_name[ decl->get_name() ];
    236 
    237                         analyzeFunc( fnTy, stats, stats.params, stats.returns );
    238 
    239                         // analyze expressions in decl statements
    240                         maybeAccept( decl->get_statements(), *this );
    241                 }
    242 
    243                 virtual void visit( UntypedExpr *expr ) {
     227                                }
     228                                ++stats.n_assns[ n_assns ];
     229
     230                                ++stats.by_name[ decl->get_name() ];
     231
     232                                analyzeFunc( fnTy, stats, stats.params, stats.returns );
     233                        }
     234                }
     235
     236                void previsit( UntypedExpr *expr ) {
     237                        visit_children = false;
    244238                        analyzeExpr( expr, 0 );
    245239                }
     
    366360
    367361        void printDeclStats( std::list< Declaration * > &translationUnit ) {
    368                 DeclStats stats;
     362                PassVisitor<DeclStats> stats;
    369363                acceptAll( translationUnit, stats );
    370                 stats.print();
     364                stats.pass.print();
    371365        }
    372366
Note: See TracChangeset for help on using the changeset viewer.