Merge pull request #4 from kornilova-l/struct-alloc-helper

Add struct alloc helper

Author: mrRosset

SimpleTypeTests.cpp

@@ -66,6 +66,19 @@ TEST_CASE("struct pointer", "[Type]"){
     REQUIRE(answ == Translate(code));
 }
 
+TEST_CASE("struct helper methods") {
+    std::string code = "struct listing { int a; int b; };";
+    Translate(code);
+    std::string answ = "\timplicit class struct_listing_ops(val p: native.Ptr[struct_listing]) extends AnyVal {\n"
+                       "\t\tdef a: native.CInt = !p._1\n"
+                       "\t\tdef a_=(value: native.CInt):Unit = !p._1 = value\n"
+                       "\t\tdef b: native.CInt = !p._2\n"
+                       "\t\tdef b_=(value: native.CInt):Unit = !p._2 = value\n"
+                       "\t}\n\n"
+                       "\tdef struct_listing()(implicit z: native.Zone): native.Ptr[struct_listing] = native.alloc[struct_listing]\n\n";
+    REQUIRE(answ == helpers);
+}
+
 TEST_CASE("func no args", "[Func]"){
     std::string code = "int foo();";
     std::string answ = "\tdef foo(): native.CInt = native.extern\n";

TreeVisitor.cpp

@@ -112,7 +112,7 @@ bool TreeVisitor::VisitRecordDecl(clang::RecordDecl *record){
         helpers += helpersFunc;
         helpers += "\t}\n\n";
 
-        return true;        
+        return true;
 
     } else if (record->isStruct() && record->isThisDeclarationADefinition() && !record->isAnonymousStructOrUnion() && name != ""){
 
@@ -165,6 +165,8 @@ bool TreeVisitor::VisitRecordDecl(clang::RecordDecl *record){
             helpers += "\timplicit class " + newName + "_ops(val p: native.Ptr[struct_" + name + "]) extends AnyVal {\n";
             helpers += helpersFunc;
             helpers += "\t}\n\n";
+
+            helpers += "\tdef " + newName + "()(implicit z: native.Zone): native.Ptr[" + newName + "] = native.alloc[" + newName + "]\n\n";
         }
 
         return true;