Changes in src/Common/Examine.cc [c6b4432:4f6dda0]

File:

• src/Common/Examine.cc (modified) (3 diffs)

src/Common/Examine.cc

@@ -19 +19 @@
 #include "CodeGen/OperatorTable.h"
 #include "InitTweak/InitTweak.h"
+
+DeclarationWithType * isMainFor( FunctionDecl * func, AggregateDecl::Aggregate kind ) {
+        if (func->name != "main") return nullptr;
+        if (func->type->parameters.size() != 1) return nullptr;
+
+        auto param = func->type->parameters.front();
+
+        auto type = dynamic_cast<ReferenceType * >(param->get_type());
+        if (!type) return nullptr;
+
+        auto obj = dynamic_cast<StructInstType *>(type->base);
+        if (!obj) return nullptr;
+
+        if (kind != obj->baseStruct->kind) return nullptr;
+
+        return param;
+}
 
 namespace {
@@ -52 +69 @@
 
 namespace {
+        Type * getDestructorParam( FunctionDecl * func ) {
+                if ( !CodeGen::isDestructor( func->name ) ) return nullptr;
+
+                auto params = func->type->parameters;
+                if ( 1 != params.size() ) return nullptr;
+
+                auto ref = dynamic_cast<ReferenceType *>( params.front()->get_type() );
+                if ( ref ) {
+                        return ref->base;
+                }
+                return nullptr;
+        }
 
 const ast::Type * getDestructorParam( const ast::FunctionDecl * func ) {
@@ -59 +88 @@
 }
 
+}
+
+bool isDestructorFor( FunctionDecl * func, StructDecl * type_decl ) {
+        if ( Type * type = getDestructorParam( func ) ) {
+                auto stype = dynamic_cast<StructInstType *>( type );
+                return stype && stype->baseStruct == type_decl;
+        }
+        return false;
 }