Changeset e01eb4a for src/AST

Timestamp:

Sep 22, 2022, 2:23:18 PM (22 months ago)

Author:

Andrew Beach <ajbeach@…>

Branches:

ADT, ast-experimental, master

Children:

5d8dae7

Parents:

0bd46fd

Message:

Moved some functions from InitTweak? to Inspect.

Location:

src/AST

Files:

• Expr.cpp (modified) (6 diffs)
• Inspect.cpp (modified) (2 diffs)
• Inspect.hpp (modified) (2 diffs)
• SymbolTable.cpp (modified) (2 diffs)
• Type.cpp (modified) (2 diffs)

src/AST/Expr.cpp

@@ -22 +22 @@
 #include "Copy.hpp"                // for shallowCopy
 #include "GenericSubstitution.hpp"
+#include "Inspect.hpp"
 #include "LinkageSpec.hpp"
 #include "Stmt.hpp"
@@ -29 +30 @@
 #include "Common/SemanticError.h"
 #include "GenPoly/Lvalue.h"        // for referencesPermissable
-#include "InitTweak/InitTweak.h"   // for getFunction, getPointerBase
 #include "ResolvExpr/typeops.h"    // for extractResultType
 #include "Tuples/Tuples.h"         // for makeTupleType
@@ -58 +58 @@
 
 bool ApplicationExpr::get_lvalue() const {
-        if ( const DeclWithType * func = InitTweak::getFunction( this ) ) {
+        if ( const DeclWithType * func = getFunction( this ) ) {
                 return func->linkage == Linkage::Intrinsic && lvalueFunctionNames.count( func->name );
         }
@@ -67 +67 @@
 
 bool UntypedExpr::get_lvalue() const {
-        std::string fname = InitTweak::getFunctionName( this );
+        std::string fname = getFunctionName( this );
         return lvalueFunctionNames.count( fname );
 }
@@ -76 +76 @@
         UntypedExpr * ret = createCall( loc, "*?", { arg } );
         if ( const Type * ty = arg->result ) {
-                const Type * base = InitTweak::getPointerBase( ty );
+                const Type * base = getPointerBase( ty );
                 assertf( base, "expected pointer type in dereference (type was %s)", toString( ty ).c_str() );
 
@@ -335 +335 @@
         // first argument
         assert( callExpr );
-        const Expr * arg = InitTweak::getCallArg( callExpr, 0 );
+        const Expr * arg = getCallArg( callExpr, 0 );
         assert( arg );
         result = arg->result;

src/AST/Inspect.cpp

@@ -10 +10 @@
 // Created On       : Fri Jun 24 13:16:31 2022
 // Last Modified By : Andrew Beach
-// Last Modified On : Mon Jun 27 15:35:00 2022
-// Update Count     : 1
+// Last Modified On : Wed Sep 22 13:50:00 2022
+// Update Count     : 2
 //
 
-#include "AST/Decl.hpp"
-#include "AST/Type.hpp"
+#include "Inspect.hpp"
 
 #include <iostream>
-#include <AST/Print.hpp>
+
+#include "AST/Decl.hpp"
+#include "AST/Expr.hpp"
+#include "AST/Print.hpp"
+#include "AST/Stmt.hpp"
+#include "AST/Type.hpp"
+#include "CodeGen/OperatorTable.h"
 
 namespace ast {
+
+const Type * getPointerBase( const Type * t ) {
+        if ( const auto * p = dynamic_cast< const PointerType * >( t ) ) {
+                return p->base;
+        } else if ( const auto * a = dynamic_cast< const ArrayType * >( t ) ) {
+                return a->base;
+        } else if ( const auto * r = dynamic_cast< const ReferenceType * >( t ) ) {
+                return r->base;
+        } else {
+                return nullptr;
+        }
+}
+
+template<typename CallExpr>
+static const Expr * callArg( const CallExpr * call, unsigned int pos ) {
+        assertf( pos < call->args.size(),
+                "getCallArg for argument that doesn't exist: (%u); %s.",
+                pos, toString( call ).c_str() );
+        for ( const Expr * arg : call->args ) {
+                if ( 0 == pos ) return arg;
+                --pos;
+        }
+        assert( false );
+}
+
+template<typename CallExpr, typename Ret>
+static Ret throughDeref( const CallExpr * expr, Ret(*func)( const Expr * ) ) {
+        // In `(*f)(x)` the function we want is `f`.
+        std::string name = getFunctionName( expr );
+        assertf( name == "*?", "Unexpected untyped expression: %s", name.c_str() );
+        assertf( !expr->args.empty(), "Cannot get function name from dereference with no arguments" );
+        return func( expr->args.front() );
+}
+
+static const DeclWithType * getCalledFunction( const Expr * expr ) {
+        assert( expr );
+        if ( const ast::VariableExpr * varExpr = dynamic_cast< const ast::VariableExpr * >( expr ) ) {
+                return varExpr->var;
+        } else if ( const ast::MemberExpr * memberExpr = dynamic_cast< const ast::MemberExpr * >( expr ) ) {
+                return memberExpr->member;
+        } else if ( const ast::CastExpr * castExpr = dynamic_cast< const ast::CastExpr * >( expr ) ) {
+                return getCalledFunction( castExpr->arg );
+        } else if ( const ast::UntypedExpr * untypedExpr = dynamic_cast< const ast::UntypedExpr * >( expr ) ) {
+                return throughDeref( untypedExpr, getCalledFunction );
+        } else if ( const ast::ApplicationExpr * appExpr = dynamic_cast< const ast::ApplicationExpr * > ( expr ) ) {
+                return throughDeref( appExpr, getCalledFunction );
+        } else if ( const ast::AddressExpr * addrExpr = dynamic_cast< const ast::AddressExpr * >( expr ) ) {
+                return getCalledFunction( addrExpr->arg );
+        } else if ( const ast::CommaExpr * commaExpr = dynamic_cast< const ast::CommaExpr * >( expr ) ) {
+                return getCalledFunction( commaExpr->arg2 );
+        }
+        return nullptr;
+}
+
+const DeclWithType * getFunction( const Expr * expr ) {
+        if ( auto app = dynamic_cast< const ApplicationExpr * >( expr ) ) {
+                return getCalledFunction( app->func );
+        } else if ( auto untyped = dynamic_cast< const UntypedExpr * >( expr ) ) {
+                return getCalledFunction( untyped->func );
+        }
+        assertf( false, "getFunction received unknown expression: %s", toString( expr ).c_str() );
+}
+
+// There is a lot of overlap with getCalledFunction. Ideally it would use
+// it as a helper function and return the name of the DeclWithType. But the
+// NameExpr and UntypedMemberExpr only work on this version.
+static std::string funcName( const Expr * func ) {
+        assert( func );
+        if ( const ast::NameExpr * nameExpr = dynamic_cast< const ast::NameExpr * >( func ) ) {
+                return nameExpr->name;
+        } else if ( const ast::VariableExpr * varExpr = dynamic_cast< const ast::VariableExpr * >( func ) ) {
+                return varExpr->var->name;
+        } else if ( const ast::CastExpr * castExpr = dynamic_cast< const ast::CastExpr * >( func ) ) {
+                return funcName( castExpr->arg );
+        } else if ( const ast::MemberExpr * memberExpr = dynamic_cast< const ast::MemberExpr * >( func ) ) {
+                return memberExpr->member->name;
+        } else if ( const ast::UntypedMemberExpr * memberExpr = dynamic_cast< const ast::UntypedMemberExpr * > ( func ) ) {
+                return funcName( memberExpr->member );
+        } else if ( const ast::UntypedExpr * untypedExpr = dynamic_cast< const ast::UntypedExpr * >( func ) ) {
+                return throughDeref( untypedExpr, funcName );
+        } else if ( const ast::ApplicationExpr * appExpr = dynamic_cast< const ast::ApplicationExpr * >( func ) ) {
+                return throughDeref( appExpr, funcName );
+        } else if ( const ast::ConstructorExpr * ctorExpr = dynamic_cast< const ast::ConstructorExpr * >( func ) ) {
+                return funcName( getCallArg( ctorExpr->callExpr, 0 ) );
+        } else {
+                assertf( false, "Unexpected expression type being called as a function in call expression: %s", toString( func ).c_str() );
+        }
+}
+
+std::string getFunctionName( const Expr * expr ) {
+        // There's some unforunate overlap here with getCalledFunction. Ideally
+        // this would be able to use getCalledFunction and return the name of the
+        // DeclWithType, but this needs to work for NameExpr and UntypedMemberExpr,
+        // where getCalledFunction can't possibly do anything reasonable.
+        if ( auto app = dynamic_cast< const ApplicationExpr * >( expr ) ) {
+                return funcName( app->func );
+        } else if ( auto untyped = dynamic_cast< const UntypedExpr * >( expr ) ) {
+                return funcName( untyped->func );
+        } else {
+                assertf( false, "Unexpected expression type passed to getFunctionName: %s", toString( expr ).c_str() );
+        }
+}
+
+const Expr * getCallArg( const Expr * call, unsigned int pos ) {
+        if ( auto app = dynamic_cast< const ApplicationExpr * >( call ) ) {
+                return callArg( app, pos );
+        } else if ( auto untyped = dynamic_cast< const UntypedExpr * >( call ) ) {
+                return callArg( untyped, pos );
+        } else if ( auto tupleAssn = dynamic_cast< const TupleAssignExpr * >( call ) ) {
+                const std::list<ptr<Stmt>>& stmts = tupleAssn->stmtExpr->stmts->kids;
+                assertf( !stmts.empty(), "TupleAssignExpr missing statements." );
+                auto stmt  = strict_dynamic_cast< const ExprStmt * >( stmts.back().get() );
+                auto tuple = strict_dynamic_cast< const TupleExpr * >( stmt->expr.get() );
+                assertf( !tuple->exprs.empty(), "TupleAssignExpr has empty tuple expr." );
+                return getCallArg( tuple->exprs.front(), pos );
+        } else if ( auto ctor = dynamic_cast< const ImplicitCopyCtorExpr * >( call ) ) {
+                return getCallArg( ctor->callExpr, pos );
+        } else {
+                assertf( false, "Unexpected expression type passed to getCallArg: %s",
+                        toString( call ).c_str() );
+        }
+}
 
 bool structHasFlexibleArray( const ast::StructDecl * decl ) {
@@ -33 +160 @@
 }
 
+const ApplicationExpr * isIntrinsicCallExpr( const Expr * expr ) {
+        auto appExpr = dynamic_cast< const ApplicationExpr * >( expr );
+        if ( !appExpr ) return nullptr;
+
+        const DeclWithType * func = getCalledFunction( appExpr->func );
+        assertf( func,
+                "getCalledFunction returned nullptr: %s", toString( appExpr->func ).c_str() );
+
+        return func->linkage == ast::Linkage::Intrinsic ? appExpr : nullptr;
+}
+
 } // namespace ast

src/AST/Inspect.hpp

@@ -10 +10 @@
 // Created On       : Fri Jun 24 13:16:31 2022
 // Last Modified By : Andrew Beach
-// Last Modified On : Mon Jun 27 15:35:00 2022
-// Update Count     : 1
+// Last Modified On : Thr Sep 22 13:44:00 2022
+// Update Count     : 2
 //
 
@@ -18 +18 @@
 namespace ast {
 
-// Does the structure end in a flexable array declaration?
-bool structHasFlexibleArray( const ast::StructDecl * );
+/// Returns the base type of an pointer/array/reference type,
+/// if the argument is not one of those types, return null.
+const Type * getPointerBase( const Type * );
+
+/// Get the declaration of the function called (ApplicationExpr or UntypedExpr).
+const DeclWithType * getFunction( const Expr * expr );
+
+/// Get the name of the function being called.
+std::string getFunctionName( const Expr * expr );
+
+/// Returns the argument to a call expression in position N, indexed from 0.
+const Expr * getCallArg( const Expr * call, unsigned pos );
+
+/// Does the structure end in a flexable array declaration?
+bool structHasFlexibleArray( const StructDecl * );
+
+/// If the expression is an application whose target function is an
+/// intrinsic, then returns a pointer to that application.
+const ApplicationExpr * isIntrinsicCallExpr( const Expr * expr );
 
 }

src/AST/SymbolTable.cpp

@@ -20 +20 @@
 #include "Decl.hpp"
 #include "Expr.hpp"
+#include "Inspect.hpp"
 #include "Type.hpp"
 #include "CodeGen/OperatorTable.h"  // for isCtorDtorAssign
@@ -466 +467 @@
                 assert( ! params.empty() );
                 // use base type of pointer, so that qualifiers on the pointer type aren't considered.
-                const Type * base = InitTweak::getPointerBase( params.front() );
+                const Type * base = ast::getPointerBase( params.front() );
                 assert( base );
                 if (stripParams) {

src/AST/Type.cpp

@@ -22 +22 @@
 #include "Decl.hpp"
 #include "Init.hpp"
+#include "Inspect.hpp"
 #include "Common/utility.h"      // for copy, move
-#include "InitTweak/InitTweak.h" // for getPointerBase
 #include "Tuples/Tuples.h"       // for isTtype
 
@@ -36 +36 @@
         const Type * t;
         const Type * a;
-        for ( t = this; (a = InitTweak::getPointerBase( t )); t = a );
+        for ( t = this; (a = ast::getPointerBase( t )); t = a );
         return t;
 }