Changes in src/CodeTools/ResolvProtoDump.cc [4c42a5f:04bdc26]

File:

• src/CodeTools/ResolvProtoDump.cc (modified) (9 diffs)

src/CodeTools/ResolvProtoDump.cc

@@ -204 +204 @@
                 /// ensures type inst names are uppercase
                 static void ti_name( const std::string& name, std::stringstream& ss ) {
+                        // replace built-in wide character types with named types
+                        if ( name == "char16_t" || name == "char32_t" || name == "wchar_t" ) {
+                                ss << "#" << name;
+                                return;
+                        }
+
+                        // strip leading underscore
                         unsigned i = 0;
                         while ( i < name.size() && name[i] == '_' ) { ++i; }
@@ -210 +217 @@
                                 return;
                         }
-                        ss << (char)std::toupper( static_cast<unsigned char>(name[i]) )
-                           << (name.c_str() + i + 1);
+
+                        std::string stripped = name.substr(i);
+                        // strip trailing "_generic_" from autogen names (avoids some user-generation issues)
+                        char generic[] = "_generic_"; size_t n_generic = sizeof(generic) - 1;
+                        if ( stripped.size() >= n_generic 
+                                        && stripped.substr( stripped.size() - n_generic ) == generic ) {
+                                stripped.resize( stripped.size() - n_generic );
+                        }
+
+                        // uppercase first character
+                        ss << (char)std::toupper( static_cast<unsigned char>(stripped[0]) )
+                           << (stripped.c_str() + 1);
                 }
 
@@ -304 +321 @@
                         }
 
-                        // TODO support VarArgsType
+                        // TODO support variable args for functions
+                        void previsit( VarArgsType* ) {
+                                // only include varargs for top level (argument type)
+                                if ( depth == 0 ) { ss << "#$varargs"; }
+                        }
 
                         // replace 0 and 1 with int
@@ -397 +418 @@
                         }
 
+                        /// Handle already-resolved variables as type constants
+                        void previsit( VariableExpr* expr ) {
+                                PassVisitor<TypePrinter> tyPrinter{ closed, ss };
+                                expr->var->get_type()->accept( tyPrinter );
+                                visit_children = false;
+                        }
+
                         /// Calls handled as calls
                         void previsit( UntypedExpr* expr ) {
@@ -426 +454 @@
                         }
 
-                        /// Already-resolved calls skipped
-                        void previsit( ApplicationExpr* ) {
+                        /// Already-resolved calls reduced to their type constant
+                        void previsit( ApplicationExpr* expr ) {
+                                PassVisitor<TypePrinter> tyPrinter{ closed, ss };
+                                expr->result->accept( tyPrinter );
                                 visit_children = false;
                         }
@@ -532 +562 @@
                                 for ( Initializer* it : li->initializers ) {
                                         build( it, ss );
-                                        ss << ' ';
                                 }
                         }
@@ -539 +568 @@
                 /// Adds an object initializer to the list of expressions
                 void build( const std::string& name, Initializer* init, std::stringstream& ss ) {
-                        ss << "$constructor( ";
+                        ss << "$constructor( &";
                         rp_name( name, ss );
-                        ss << "() ";
+                        ss << ' ';
                         build( init, ss );
                         ss << ')';
@@ -676 +705 @@
                 }
 
+                void previsit( AsmStmt* ) {
+                        // skip asm statements
+                        visit_children = false;
+                }
+
                 void previsit( Expression* expr ) {
                         std::stringstream ss;
@@ -686 +720 @@
                 /// Print non-prelude global declarations for resolv proto
                 void printGlobals() const {
-                        std::cout << "#ptr<T> $addr T" << std::endl;  // &?
+                        std::cout << "#$ptr<T> $addr T" << std::endl;  // &?
                         int i = (int)BasicType::SignedInt;
                         std::cout << i << " $and " << i << ' ' << i << std::endl;  // ?&&?