Changes in src/CodeGen/CodeGenerator.cc [68cd1ce:cda48b6]

File:

• src/CodeGen/CodeGenerator.cc (modified) (16 diffs)

src/CodeGen/CodeGenerator.cc

@@ -9 +9 @@
 // Author           : Richard C. Bilson
 // Created On       : Mon May 18 07:44:20 2015
-// Last Modified By : Peter A. Buhr
-// Last Modified On : Sat Jun 13 07:12:27 2015
-// Update Count     : 129
+// Last Modified By : Rob Schluntz
+// Last Modified On : Thu Jun 11 13:22:39 2015
+// Update Count     : 137
 //
 
@@ -19 +19 @@
 #include <list>
 
-#include "Parser/ParseNode.h"
-
 #include "SynTree/Type.h"
+#include "SynTree/Declaration.h"
+#include "SynTree/Statement.h"
 #include "SynTree/Expression.h"
 #include "SynTree/Initializer.h"
-#include "SynTree/Statement.h"
 
 #include "utility.h"
@@ -44 +43 @@
         }
 
-        CodeGenerator::CodeGenerator( std::ostream &os ) : cur_indent( 0 ), insideFunction( false ), output( os ) { }
-
-        CodeGenerator::CodeGenerator( std::ostream &os, std::string init, int indent, bool infunp )
-                        : cur_indent( indent ), insideFunction( infunp ), output( os ) {
+        ostream & CodeGenerator::Indenter::operator()( ostream & output ) {
+          return output << string( cg.cur_indent, ' ' );
+        }
+
+        ostream & operator<<( ostream & output, CodeGenerator::Indenter &indent ) {
+                return indent( output );
+        }
+
+        CodeGenerator::CodeGenerator( std::ostream &os ) : indent(*this), cur_indent( 0 ), insideFunction( false ), output( os ) { }
+
+        CodeGenerator::CodeGenerator( std::ostream &os, std::string init, int indentation, bool infunp )
+                        : indent(*this), cur_indent( indentation ), insideFunction( infunp ), output( os ) {
                 //output << std::string( init );
         }
 
-        CodeGenerator::CodeGenerator( std::ostream &os, char *init, int indent, bool infunp )
-                        : cur_indent( indent ), insideFunction( infunp ), output( os ) {
+        CodeGenerator::CodeGenerator( std::ostream &os, char *init, int indentation, bool infunp )
+                        : indent(*this), cur_indent( indentation ), insideFunction( infunp ), output( os ) {
                 //output << std::string( init );
         }
@@ -63 +70 @@
                 } // if
         }
-  
+ 
         //*** Declarations
         void CodeGenerator::visit( FunctionDecl *functionDecl ) {
@@ -105 +112 @@
 
                 if ( ! memb.empty() ) {
-                        output << endl << string( cur_indent, ' ' ) << "{" << endl;
+                        output << endl << indent << "{" << endl;
 
                         cur_indent += CodeGenerator::tabsize; 
                         for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end();  i++) {
-                                output << string( cur_indent, ' ' ); 
+                                output << indent; 
                                 (*i)->accept(*this );
                                 output << ";" << endl;
@@ -116 +123 @@
                         cur_indent -= CodeGenerator::tabsize; 
 
-                        output << string( cur_indent, ' ' ) << "}";
+                        output << indent << "}";
                 } // if
         }
@@ -139 +146 @@
 
                 if ( ! memb.empty() ) {
-                        output << endl << "{" << endl;
+                        output << " {" << endl;
 
                         cur_indent += CodeGenerator::tabsize; 
@@ -145 +152 @@
                                 ObjectDecl *obj = dynamic_cast< ObjectDecl* >( *i );
                                 assert( obj );
-                                output << string( cur_indent, ' ' ) << mangleName( obj ); 
+                                output << indent << mangleName( obj ); 
                                 if ( obj->get_init() ) {
                                         output << " = ";
@@ -155 +162 @@
                         cur_indent -= CodeGenerator::tabsize; 
 
-                        output << "}" << endl;
+                        output << indent << "}";
                 } // if
         }
@@ -445 +452 @@
 
                 for ( std::list<Statement *>::iterator i = ks.begin(); i != ks.end();  i++) {
-                        output << string( cur_indent, ' ' ) << printLabels( (*i)->get_labels() );
+                        output << indent << printLabels( (*i)->get_labels() );
                         (*i)->accept(*this );
 
@@ -455 +462 @@
                 cur_indent -= CodeGenerator::tabsize; 
 
-                output << string( cur_indent, ' ' ) << "}";
+                output << indent << "}";
         }
 
@@ -495 +502 @@
                 cur_indent -= CodeGenerator::tabsize;
 
-                output << string( cur_indent, ' ' ) << "}";
+                output << indent << "}";
         }
 
         void CodeGenerator::visit( CaseStmt *caseStmt ) {
-                output << string( cur_indent, ' ' );
+                output << indent;
                 if ( caseStmt->isDefault()) {
                         output << "default";
@@ -512 +519 @@
                 cur_indent += CodeGenerator::tabsize;
                 for ( std::list<Statement *>::iterator i = sts.begin(); i != sts.end();  i++) {
-                        output << string( cur_indent, ' ' ) << printLabels( (*i)->get_labels() )  ;
+                        output << indent << printLabels( (*i)->get_labels() )  ;
                         (*i)->accept(*this );
                         output << endl;
@@ -565 +572 @@
                 whileStmt->get_body()->accept( *this );
 
-                output << string( cur_indent, ' ' );
+                output << indent;
 
                 if ( whileStmt->get_isDoWhile() ) {
@@ -597 +604 @@
 
         void CodeGenerator::visit( NullStmt *nullStmt ) {
-                //output << string( cur_indent, ' ' ) << CodeGenerator::printLabels( nullStmt->get_labels() );
+                //output << indent << CodeGenerator::printLabels( nullStmt->get_labels() );
                 output << "/* null statement */ ;";
         }
@@ -621 +628 @@
         void CodeGenerator::handleStorageClass( Declaration *decl ) {
                 switch ( decl->get_storageClass() ) {
-                  case DeclarationNode::Extern:
+                  case Declaration::NoStorageClass:
+                        break;
+                  case Declaration::Extern:
                         output << "extern ";
                         break;
-                  case DeclarationNode::Static:
+                  case Declaration::Static:
                         output << "static ";
                         break;
-                  case DeclarationNode::Auto:
+                  case Declaration::Auto:
                         // silently drop storage class
                         break;
-                  case DeclarationNode::Register:
+                  case Declaration::Register:
                         output << "register ";
                         break;
-                  case DeclarationNode::Inline:
+                  case Declaration::Inline:
                         // handled as special via isInline flag (FIX)
                         break;
-                  case DeclarationNode::Fortran:
+                  case Declaration::Fortran:
                         // not handled
-                        output << "fortran ";
-                        break;
-                  case DeclarationNode::Noreturn:
-                        // not handled
-                        output << "_Noreturn ";
-                        break;
-                  case DeclarationNode::Threadlocal:
-                        // not handled
-                        output << "_Thread_local ";
-                        break;
-                  case DeclarationNode::NoStorageClass:
                         break;
                 } // switch