Changeset ce8c12f for src/GenPoly

Timestamp:

May 15, 2017, 11:30:26 AM (8 years ago)

Author:

Rob Schluntz <rschlunt@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

d36c117

Parents:

65aca88

Message:

initial work on references: reference types passed through the system, very simple examples work

File:

• src/GenPoly/Lvalue.cc (modified) (5 diffs)

src/GenPoly/Lvalue.cc

@@ -32 +32 @@
 #include "Common/UniqueName.h"
 #include "Common/utility.h"
+#include "InitTweak/InitTweak.h"
 
 namespace GenPoly {
@@ -40 +41 @@
                 class Pass1 : public Mutator {
                   public:
+                        typedef Mutator Parent;
                         Pass1();
 
+                        // xxx - should this happen to every expression with reference result type? probably just appexpr and varexpr?
+                        virtual Type *mutate( ReferenceType * refType );
+                        virtual Expression *mutate( VariableExpr *varExpr );
                         virtual Expression *mutate( ApplicationExpr *appExpr );
                         virtual Statement *mutate( ReturnStmt *appExpr );
@@ -105 +110 @@
                         } // if
                         return funcDecl;
+                }
+
+                Type * Pass1::mutate( ReferenceType * refType ) {
+                        Type * base = refType->get_base();
+                        refType->set_base( nullptr );
+                        delete refType;
+                        return new PointerType( Type::Qualifiers(), base );
+                }
+
+                Expression * Pass1::mutate( VariableExpr *varExpr ) {
+                        if ( ReferenceType * refType = dynamic_cast< ReferenceType * >( varExpr->get_result() ) ) {
+                                varExpr->set_result( refType->acceptMutator( *this ) );
+                                return UntypedExpr::createDeref( varExpr );
+                        }
+                        return Parent::mutate( varExpr );
                 }
 
@@ -165 +185 @@
                 }
 
+                bool isDeref( Expression * expr ) {
+                        if ( UntypedExpr * untyped = dynamic_cast< UntypedExpr * >( expr ) ) {
+                                return InitTweak::getFunctionName( untyped ) == "*?";
+                        } else if ( ApplicationExpr * appExpr = dynamic_cast< ApplicationExpr * > ( expr ) ) {
+                                return InitTweak::getFunctionName( appExpr ) == "*?";
+                        } else {
+                                return false;
+                        }
+                }
+
                 Expression * GeneralizedLvalue::mutate( AddressExpr * addrExpr ) {
                         addrExpr = safe_dynamic_cast< AddressExpr * >( Parent::mutate( addrExpr ) );
@@ -178 +208 @@
                                 delete addrExpr;
                                 return new ConditionalExpr( arg1, new AddressExpr( arg2 ), new AddressExpr( arg3 ) );
+                        } else if ( isDeref( addrExpr->get_arg() ) ) {
+                                // xxx - this doesn't belong here -- move it somewhere else
+                                Expression *& arg = InitTweak::getCallArg( addrExpr->get_arg(), 0 );
+                                Expression * inner = arg;
+                                arg = nullptr;
+                                delete addrExpr;
+                                return inner;
                         }
                         return addrExpr;