source: src/ResolvExpr/PolyCost.cc @ 1e7a765

Last change on this file since 1e7a765 was bccd70a, checked in by Andrew Beach <ajbeach@…>, 18 months ago

Removed internal code from TypeSubstitution? header. It caused a chain of include problems, which have been corrected.

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