Changeset fc9153d

Timestamp:

Dec 22, 2017, 4:17:52 PM (7 years ago)

Author:

Rob Schluntz <rschlunt@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

caab997

Parents:

ddb80bd

Message:

Add helper function for determining the value of an enumerator

Location:

src/SynTree

Files:

• AggregateDecl.cc (modified) (1 diff)
• Declaration.h (modified) (1 diff)

src/SynTree/AggregateDecl.cc

@@ -86 +86 @@
 std::string TraitDecl::typeString() const { return "trait"; }
 
+namespace {
+        long long int getConstValue( Expression * expr ) {
+                if ( CastExpr * castExpr = dynamic_cast< CastExpr * > ( expr ) ) {
+                        return getConstValue( castExpr->arg );
+                } else if ( ConstantExpr * constExpr = dynamic_cast< ConstantExpr * >( expr ) ) {
+                        return constExpr->intValue();
+                } else {
+                        assertf( false, "Unhandled expression type in getConstValue for enumerators: %s", toString( expr ).c_str() );
+                }
+        }
+}
+
+bool EnumDecl::valueOf( Declaration * enumerator, long long int & value ) {
+        if ( enumValues.empty() ) {
+                long long int currentValue = 0;
+                for ( Declaration * member : members ) {
+                        ObjectDecl * field = strict_dynamic_cast< ObjectDecl * >( member );
+                        if ( field->init ) {
+                                SingleInit * init = strict_dynamic_cast< SingleInit * >( field->init );
+                                currentValue = getConstValue( init->value );
+                        }
+                        assertf( enumValues.count( field->name ) == 0, "Enum %s has multiple members with the name %s", name.c_str(), field->name.c_str() );
+                        enumValues[ field->name ] = currentValue;
+                        ++currentValue;
+                }
+        }
+        if ( enumValues.count( enumerator->name ) ) {
+                value = enumValues[ enumerator->name ];
+                return true;
+        }
+        return false;
+}
+
 // Local Variables: //
 // tab-width: 4 //

src/SynTree/Declaration.h

@@ -319 +319 @@
         EnumDecl( const EnumDecl &other ) : Parent( other ) {}
 
+        bool valueOf( Declaration * enumerator, long long int & value );
+
         virtual EnumDecl *clone() const override { return new EnumDecl( *this ); }
         virtual void accept( Visitor &v ) override { v.visit( this ); }
         virtual Declaration *acceptMutator( Mutator &m )  override { return m.mutate( this ); }
   private:
+        std::map< std::string, long long int > enumValues;
         virtual std::string typeString() const override;
 };