Changeset 7c3f3be

Timestamp:

Sep 27, 2017, 3:51:01 PM (8 years ago)

Author:

Rob Schluntz <rschlunt@…>

Branches:

ADT, arm-eh, ast-experimental, cleanup-dtors, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, pthread-emulation, qualifiedEnum

Children:

7ca84116

Parents:

2b7bf59

Message:

Add getThisParam and getThisType helpers for constructors and destructors

Location:

src/InitTweak

Files:

• InitTweak.cc (modified) (1 diff)
• InitTweak.h (modified) (1 diff)

src/InitTweak/InitTweak.cc

@@ -270 +270 @@
         }
 
+        Type * getThisType( FunctionType * ftype ) {
+                assertf( ftype, "getThisType: nullptr ftype" );
+                ObjectDecl * thisParam = getThisParam( ftype );
+                ReferenceType * refType = strict_dynamic_cast< ReferenceType * >( thisParam->type );
+                return refType->base;
+        }
+
+        ObjectDecl * getThisParam( FunctionType * ftype ) {
+                assertf( ftype, "getThisParam: nullptr ftype" );
+                auto & params = ftype->parameters;
+                assertf( ! params.empty(), "getThisParam: ftype with 0 parameters: %s", toString( ftype ).c_str() );
+                return strict_dynamic_cast< ObjectDecl * >( params.front() );
+        }
+
         bool tryConstruct( DeclarationWithType * dwt ) {
                 ObjectDecl * objDecl = dynamic_cast< ObjectDecl * >( dwt );

src/InitTweak/InitTweak.h

@@ -29 +29 @@
         FunctionDecl * isCopyConstructor( Declaration * decl );
         FunctionDecl * isCopyFunction( Declaration * decl, const std::string & fname );
+
+        /// returns the base type of the first parameter to a constructor/destructor/assignment function
+        Type * getThisType( FunctionType * ftype );
+
+        /// returns the first parameter of a constructor/destructor/assignment function
+        ObjectDecl * getThisParam( FunctionType * ftype );
 
         /// transform Initializer into an argument list that can be passed to a call expression