Changes in src/Validate/ImplementEnumFunc.cpp [acb33f15:c92bdcc]

File:

• src/Validate/ImplementEnumFunc.cpp (modified) (5 diffs)

src/Validate/ImplementEnumFunc.cpp

@@ -2 +2 @@
 #include "AST/Pass.hpp"
 #include "AST/TranslationUnit.hpp"
-#include "CodeGen/OperatorTable.h"  // for isCtorDtor, isCtorDtorAssign
-#include "InitTweak/InitTweak.h"    // for isAssignment, isCopyConstructor
+#include "CodeGen/OperatorTable.hpp"  // for isCtorDtor, isCtorDtorAssign
+#include "InitTweak/InitTweak.hpp"    // for isAssignment, isCopyConstructor
 namespace Validate {
 
@@ -72 +72 @@
                 ast::DeclWithType* decl = declPtr.get_and_mutate();
                 decl->attributes.push_back(new ast::Attribute("unused"));
+        }
+
+        ast::ObjectDecl* dstParam() const {
+                return new ast::ObjectDecl(getLocation(), "_dst",
+                                           new ast::ReferenceType(new ast::EnumAttrType(
+                                               ast::deepCopy(instType))));
+        }
+
+        ast::ObjectDecl* srcParam() const {
+                return new ast::ObjectDecl(
+                        getLocation(), "_src",
+                        new ast::EnumAttrType(ast::deepCopy(instType)));
         }
 
@@ -317 +329 @@
 }
 
+inline ast::EnumAttrType * getPosnType( const ast::EnumDecl * decl ) {
+        return new ast::EnumAttrType(new ast::EnumInstType(decl), ast::EnumAttribute::Posn);
+}
+
 ast::ObjectDecl* EnumAttrFuncGenerator::genAttrArrayProto(
         const ast::EnumAttribute attr, const CodeLocation& location,
@@ -344 +360 @@
                         func->location,
                         new ast::VariableExpr( func->location, func->params.front() ),
-                        new ast::BasicType( ast::BasicKind::UnsignedInt ),
-                        ast::GeneratedFlag::ExplicitCast
-                )});
+                        new ast::EnumAttrType( new ast::EnumInstType(decl),
+                                ast::EnumAttribute::Posn))});
         func->stmts = new ast::CompoundStmt(
                 func->location, {new ast::ReturnStmt(func->location, untyped)});
@@ -355 +370 @@
                 func->location,
                 new ast::VariableExpr(func->location, func->params.front()),
-                new ast::BasicType( ast::BasicKind::UnsignedInt ),
-                        ast::GeneratedFlag::ExplicitCast);
+                new ast::EnumAttrType(new ast::EnumInstType(decl),
+                                                          ast::EnumAttribute::Posn));
         func->stmts = new ast::CompoundStmt(
                 func->location, {new ast::ReturnStmt(func->location, castExpr)});