Changes in src/AST/Decl.cpp [c92bdcc:85855b0]

File:

• src/AST/Decl.cpp (modified) (1 diff)

src/AST/Decl.cpp

@@ -170 +170 @@
 
 const std::string EnumDecl::getUnmangeldArrayName( const ast::EnumAttribute attr ) const {
-                switch( attr ) {
-                        case ast::EnumAttribute::Value: return "values_" + name ;
-                        case ast::EnumAttribute::Label: return "labels_" + name;
-                        default: /* Posn does not generate array */ 
-                                return "";
+        switch( attr ) {
+                case ast::EnumAttribute::Value: return "values_" + name ;
+                case ast::EnumAttribute::Label: return "labels_" + name;
+                default: /* Posn does not generate array */ 
+                        return "";
+        }
+}
+
+unsigned EnumDecl::calChildOffset(const std::string & target) const{
+        unsigned offset = 0;
+        for (auto childEnum: inlinedDecl) {
+                auto childDecl = childEnum->base;
+                if (childDecl->name == target) {
+                        return offset;
                 }
-        }
+                offset += childDecl->members.size();
+        }
+    std::cerr << "Cannot find the target enum" << std::endl;
+        return 0;
+}
+
+unsigned EnumDecl::calChildOffset(const ast::EnumInstType * target) const{
+        return calChildOffset(target->base->name);
+}
+
+bool EnumDecl::isSubTypeOf(const ast::EnumDecl * other) const {
+        if (name == other->name) return true;
+        for (auto inlined: other->inlinedDecl) {
+                if (isSubTypeOf(inlined->base)) return true;
+        }
+        return false;
+}
 
 }