source: src/ResolvExpr/PolyCost.cc @ 7ff3e522

arm-ehjacob/cs343-translationnew-astnew-ast-unique-expr
Last change on this file since 7ff3e522 was 7ff3e522, checked in by Andrew Beach <ajbeach@…>, 14 months ago

{pass_t Pass::pass; => core_t Pass::core;} To avoid confusion about which pass we are talking about.

  • Property mode set to 100644
File size: 2.8 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// PolyCost.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Sun May 17 09:50:12 2015
11// Last Modified By : Andrew Beach
12// Last Modified On : Wed Jun 19 10:45:00 2019
13// Update Count     : 4
14//
15
16#include "AST/SymbolTable.hpp"
17#include "AST/Type.hpp"
18#include "AST/TypeEnvironment.hpp"
19#include "Common/PassVisitor.h"
20#include "SymTab/Indexer.h"   // for Indexer
21#include "SynTree/Type.h"     // for TypeInstType, Type
22#include "TypeEnvironment.h"  // for EqvClass, TypeEnvironment
23
24namespace ResolvExpr {
25        struct PolyCost {
26                PolyCost( const TypeEnvironment &env, const SymTab::Indexer &indexer );
27
28                void previsit( TypeInstType * aggregateUseType );
29                int result;
30                const TypeEnvironment &tenv;
31                const SymTab::Indexer &indexer;
32        };
33
34        int polyCost( Type *type, const TypeEnvironment & env, const SymTab::Indexer &indexer ) {
35                PassVisitor<PolyCost> coster( env, indexer );
36                type->accept( coster );
37                return coster.pass.result;
38        }
39
40        PolyCost::PolyCost( const TypeEnvironment & env, const SymTab::Indexer & indexer ) : result( 0 ), tenv( env ), indexer( indexer ) {
41        }
42
43        void PolyCost::previsit(TypeInstType * typeInst) {
44                if ( const EqvClass *eqvClass = tenv.lookup( typeInst->name ) ) {
45                        if ( eqvClass->type ) {
46                                if ( TypeInstType * otherTypeInst = dynamic_cast< TypeInstType* >( eqvClass->type ) ) {
47                                        if ( indexer.lookupType( otherTypeInst->name ) ) {
48                                                // bound to opaque type
49                                                result += 1;
50                                        } // if
51                                } else {
52                                        // bound to concrete type
53                                        result += 1;
54                                } // if
55                        } // if
56                } // if
57        }
58
59// TODO: When the old PolyCost is torn out get rid of the _new suffix.
60class PolyCost_new {
61        const ast::SymbolTable &symtab;
62public:
63        int result;
64        const ast::TypeEnvironment &env_;
65
66        PolyCost_new( const ast::SymbolTable & symtab, const ast::TypeEnvironment & env ) 
67        : symtab( symtab ), result( 0 ), env_( env ) {}
68
69        void previsit( const ast::TypeInstType * type ) {
70                if ( const ast::EqvClass * eqv = env_.lookup( type->name ) ) /* && */ if ( eqv->bound ) {
71                        if ( const ast::TypeInstType * otherType = eqv->bound.as< ast::TypeInstType >() ) {
72                                if ( symtab.lookupType( otherType->name ) ) {
73                                        // Bound to opaque type.
74                                        result += 1;
75                                }
76                        } else {
77                                // Bound to concrete type.
78                                result += 1;
79                        }
80                }
81        }
82};
83
84int polyCost(
85        const ast::Type * type, const ast::SymbolTable & symtab, const ast::TypeEnvironment & env
86) {
87        ast::Pass<PolyCost_new> costing( symtab, env );
88        type->accept( costing );
89        return costing.core.result;
90}
91
92} // namespace ResolvExpr
93
94// Local Variables: //
95// tab-width: 4 //
96// mode: c++ //
97// compile-command: "make install" //
98// End: //
Note: See TracBrowser for help on using the repository browser.