Changes in src/AST/Decl.cpp [2bb4a01:a300e4a]

File:

• src/AST/Decl.cpp (modified) (2 diffs)

src/AST/Decl.cpp

@@ -14 +14 @@
 //
 
+#include <cassert>        // for assert, strict_dynamic_cast
 #include <unordered_map>
 
 #include "Decl.hpp"
 
-#include "Fwd.hpp"   // for UniqueId
-#include "Node.hpp"  // for readonly
+#include "Fwd.hpp"        // for UniqueId
+#include "Init.hpp"
+#include "Node.hpp"       // for readonly
 
 namespace ast {
+
 // To canonicalize declarations
 static UniqueId lastUniqueId = 0;
@@ -39 +42 @@
         return {};
 }
+
+// --- EnumDecl
+
+bool EnumDecl::valueOf( Decl* enumerator, long long& value ) const {
+        if ( enumValues.empty() ) {
+                long long crntVal = 0;
+                for ( const Decl* member : members ) {
+                        const ObjectDecl* field = strict_dynamic_cast< const ObjectDecl* >( member );
+                        if ( field->init ) {
+                                const SingleInit* init = strict_dynamic_cast< const SingleInit* >( field->init );
+                                auto result = eval( init->value );
+                                if ( ! result.second ) {
+                                        SemanticError( init->location, toString( "Non-constexpr in initialization of "
+                                                "enumerator: ", field ) );
+                                }
+                                crntVal = result.first;
+                        }
+                        if ( enumValues.count( field->name ) != 0 ) {
+                                SemanticError( location, toString( "Enum ", name, " has multiple members with the "     "name ", field->name ) );
+                        }
+                        enumValues[ field->name ] = crntVal;
+                        ++crntVal;
+                }
+        }
+
+        auto it = enumValues.find( enumerator->name );
+        if ( it != enumValues.end() ) {
+                value = it->second;
+                return true;
+        }
+        return false;
+}
+
 }