source: src/GenPoly/PolyMutator.cc @ 186fd86

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsdeferred_resndemanglerenumforall-pointer-decayjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newwith_gc
Last change on this file since 186fd86 was d9fa60a, checked in by Rob Schluntz <rschlunt@…>, 7 years ago

moved substituion into MemberExpr? constructor, change generated tuple structs to generic structs, tuples containing type variables almost works

  • Property mode set to 100644
File size: 5.5 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// PolyMutator.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Mon May 18 07:44:20 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Aug  4 11:26:22 2016
13// Update Count     : 16
14//
15
16#include "PolyMutator.h"
17#include "SynTree/Declaration.h"
18#include "SynTree/Type.h"
19#include "SynTree/Expression.h"
20#include "SynTree/Statement.h"
21#include "SynTree/Mutator.h"
22#include "SynTree/Initializer.h"
23
24namespace GenPoly {
25        namespace {
26                const std::list<Label> noLabels;
27        }
28
29        PolyMutator::PolyMutator() : scopeTyVars( (TypeDecl::Kind)-1 ), env( 0 ) {}
30
31        void PolyMutator::mutateStatementList( std::list< Statement* > &statements ) {
32                SemanticError errors;
33
34                for ( std::list< Statement* >::iterator i = statements.begin(); i != statements.end(); ++i ) {
35                        if ( ! stmtsToAddAfter.empty() ) {
36                                statements.splice( i, stmtsToAddAfter );
37                        } // if
38                        try {
39                                *i = (*i)->acceptMutator( *this );
40                        } catch ( SemanticError &e ) {
41                                errors.append( e );
42                        } // try
43                        if ( ! stmtsToAdd.empty() ) {
44                                statements.splice( i, stmtsToAdd );
45                        } // if
46                } // for
47                if ( ! stmtsToAddAfter.empty() ) {
48                        statements.splice( statements.end(), stmtsToAddAfter );
49                } // if
50                if ( ! errors.isEmpty() ) {
51                        throw errors;
52                }
53        }
54
55        Statement * PolyMutator::mutateStatement( Statement *stmt ) {
56                Statement *newStmt = maybeMutate( stmt, *this );
57                if ( ! stmtsToAdd.empty() || ! stmtsToAddAfter.empty() ) {
58                        CompoundStmt *compound = new CompoundStmt( noLabels );
59                        compound->get_kids().splice( compound->get_kids().end(), stmtsToAdd );
60                        compound->get_kids().push_back( newStmt );
61                        compound->get_kids().splice( compound->get_kids().end(), stmtsToAddAfter );
62                        // doEndScope();
63                        return compound;
64                } else {
65                        return newStmt;
66                }
67        }
68
69        Expression * PolyMutator::mutateExpression( Expression *expr ) {
70                if ( expr ) {
71                        if ( expr->get_env() ) {
72                                env = expr->get_env();
73                        }
74                        // xxx - should env be cloned (or moved) onto the result of the mutate?
75                        return expr->acceptMutator( *this );
76                } else {
77                        return expr;
78                }
79        }
80
81        CompoundStmt * PolyMutator::mutate(CompoundStmt *compoundStmt) {
82                doBeginScope();
83                mutateStatementList( compoundStmt->get_kids() );
84                doEndScope();
85                return compoundStmt;
86        }
87
88        Statement * PolyMutator::mutate(IfStmt *ifStmt) {
89                ifStmt->set_thenPart(  mutateStatement( ifStmt->get_thenPart() ) );
90                ifStmt->set_elsePart(  mutateStatement( ifStmt->get_elsePart() ) );
91                ifStmt->set_condition(  mutateExpression( ifStmt->get_condition() ) );
92                return ifStmt;
93        }
94
95        Statement * PolyMutator::mutate(WhileStmt *whileStmt) {
96                whileStmt->set_body(  mutateStatement( whileStmt->get_body() ) );
97                whileStmt->set_condition(  mutateExpression( whileStmt->get_condition() ) );
98                return whileStmt;
99        }
100
101        Statement * PolyMutator::mutate(ForStmt *forStmt) {
102                forStmt->set_body(  mutateStatement( forStmt->get_body() ) );
103                mutateAll( forStmt->get_initialization(), *this );
104                forStmt->set_condition(  mutateExpression( forStmt->get_condition() ) );
105                forStmt->set_increment(  mutateExpression( forStmt->get_increment() ) );
106                return forStmt;
107        }
108
109        Statement * PolyMutator::mutate(SwitchStmt *switchStmt) {
110                mutateStatementList( switchStmt->get_statements() );
111                switchStmt->set_condition( mutateExpression( switchStmt->get_condition() ) );
112                return switchStmt;
113        }
114
115        Statement * PolyMutator::mutate(CaseStmt *caseStmt) {
116                mutateStatementList( caseStmt->get_statements() );
117                caseStmt->set_condition(  mutateExpression( caseStmt->get_condition() ) );
118                return caseStmt;
119        }
120
121        Statement * PolyMutator::mutate(TryStmt *tryStmt) {
122                tryStmt->set_block(  maybeMutate( tryStmt->get_block(), *this ) );
123                mutateAll( tryStmt->get_catchers(), *this );
124                return tryStmt;
125        }
126
127        Statement * PolyMutator::mutate(CatchStmt *cathStmt) {
128                cathStmt->set_body(  mutateStatement( cathStmt->get_body() ) );
129                cathStmt->set_decl(  maybeMutate( cathStmt->get_decl(), *this ) );
130                return cathStmt;
131        }
132
133        Statement * PolyMutator::mutate(ReturnStmt *retStmt) {
134                retStmt->set_expr( mutateExpression( retStmt->get_expr() ) );
135                return retStmt;
136        }
137
138        Statement * PolyMutator::mutate(ExprStmt *exprStmt) {
139                exprStmt->set_expr( mutateExpression( exprStmt->get_expr() ) );
140                return exprStmt;
141        }
142
143
144        Expression * PolyMutator::mutate(UntypedExpr *untypedExpr) {
145                for ( std::list< Expression* >::iterator i = untypedExpr->get_args().begin(); i != untypedExpr->get_args().end(); ++i ) {
146                        *i = mutateExpression( *i );
147                } // for
148                return untypedExpr;
149        }
150
151        Expression *PolyMutator::mutate( StmtExpr * stmtExpr ) {
152                // don't want statements from outer CompoundStmts to be added to this StmtExpr
153                ValueGuard< std::list< Statement* > > oldStmtsToAdd( stmtsToAdd );
154                ValueGuard< std::list< Statement* > > oldStmtsToAddAfter( stmtsToAddAfter );
155
156                // xxx - not sure if these are needed, along with appropriate assignments, but I don't think so...
157                // ValueGuard< TyVarMap > oldScopeTyVars;
158                // ValueGuard< TypeSubstitution * > oldEnv;
159
160                stmtsToAdd.clear();
161                stmtsToAddAfter.clear();
162
163                stmtExpr->set_result( maybeMutate( stmtExpr->get_result(), *this ) );
164                stmtExpr->set_statements( maybeMutate( stmtExpr->get_statements(), *this ) );
165                return stmtExpr;
166        }
167
168        Initializer *PolyMutator::mutate( SingleInit *singleInit ) {
169                singleInit->set_value( mutateExpression( singleInit->get_value() ) );
170                return singleInit;
171        }
172} // namespace GenPoly
173
174// Local Variables: //
175// tab-width: 4 //
176// mode: c++ //
177// compile-command: "make install" //
178// End: //
Note: See TracBrowser for help on using the repository browser.