Changes in src/CodeGen/CodeGenerator.cc [a5f0529:0dd18fd]

File:

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

src/CodeGen/CodeGenerator.cc

@@ -10 +10 @@
 // Created On       : Mon May 18 07:44:20 2015
 // Last Modified By : Andrew Beach
-// Last Modified On : Tus Jul 25 15:29:00 2017
-// Update Count     : 486
+// Last Modified On : Fri Aug 18 15:34:00 2017
+// Update Count     : 488
 //
 #include "CodeGenerator.h"
@@ -64 +64 @@
         } // extension
 
-        ostream & CodeGenerator::Indenter::operator()( ostream & output ) const {
-          return output << string( cg.cur_indent, ' ' );
-        }
-
-        ostream & operator<<( ostream & output, const CodeGenerator::Indenter &indent ) {
-                return indent( output );
-        }
-
         CodeGenerator::LabelPrinter & CodeGenerator::LabelPrinter::operator()( std::list< Label > & l ) {
                 labels = &l;
@@ -87 +79 @@
         }
 
-        CodeGenerator::LineMarker::LineMarker(
-                        CodeLocation const & loc, bool toPrint) :
-                loc(loc), toPrint(toPrint)
-        {}
-
-        CodeGenerator::LineMarker CodeGenerator::lineDirective(
-                        BaseSyntaxNode const * node) {
-                return LineMarker(node->location, lineMarks);
-        }
-
-        std::ostream & operator<<(std::ostream & out,
-                        CodeGenerator::LineMarker const & marker) {
-                if (marker.toPrint && marker.loc.isSet()) {
-                        return out << "\n# " << marker.loc.linenumber << " \""
-                                << marker.loc.filename << "\"\n";
-                } else if (marker.toPrint) {
-                        return out << "\n/* Missing CodeLocation */\n";
-                } else {
-                return out;
+        /* Using updateLocation at the beginning of a node and nextLine
+         * within a node should become the method of formating.
+         */
+        void CodeGenerator::updateLocation( CodeLocation const & to ) {
+                if ( !lineMarks ) {
+                        return;
+                } else if ( currentLocation.followedBy( to, 0 ) ) {
+                        return;
+                } else if ( currentLocation.followedBy( to, 1 ) ) {
+                        output << "\n" << indent;
+                        currentLocation.linenumber += 1;
+                } else if ( currentLocation.followedBy( to, 2 ) ) {
+                        output << "\n\n" << indent;
+                        currentLocation.linenumber += 2;
+                } else {
+                        output << "\n# " << to.linenumber << " \"" << to.filename
+                               << "\"\n" << indent;
+                        currentLocation = to;
                 }
-        }
-
-        CodeGenerator::CodeGenerator( std::ostream & os, bool pretty, bool genC, bool lineMarks ) : indent( *this), cur_indent( 0 ), insideFunction( false ), output( os ), printLabels( *this ), pretty( pretty ), genC( genC ), lineMarks( lineMarks ) {}
+                output << std::flush;
+        }
+
+        void CodeGenerator::updateLocation( BaseSyntaxNode const * to ) {
+                updateLocation( to->location );
+        }
+
+        void CodeGenerator::nextLine() {
+                if ( !lineMarks ) {
+                        output << "\n" << indent << std::flush;
+                }
+        }
+
+        CodeGenerator::CodeGenerator( std::ostream & os, bool pretty, bool genC, bool lineMarks ) : indent( CodeGenerator::tabsize ), insideFunction( false ), output( os ), printLabels( *this ), pretty( pretty ), genC( genC ), lineMarks( lineMarks ) {}
 
         string CodeGenerator::mangleName( DeclarationWithType * decl ) {
@@ -200 +202 @@
                         output << " {" << endl;
 
-                        cur_indent += CodeGenerator::tabsize;
+                        ++indent;
                         for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end(); i++ ) {
-                                output << lineDirective( *i ) << indent;
+                                updateLocation( *i );
                                 (*i)->accept( *this );
                                 output << ";" << endl;
                         } // for
 
-                        cur_indent -= CodeGenerator::tabsize;
+                        --indent;
 
                         output << indent << "}";
@@ -225 +227 @@
         void CodeGenerator::visit( EnumDecl * enumDecl ) {
                 extension( enumDecl );
-                output << lineDirective ( enumDecl );
+                updateLocation( enumDecl );
                 output << "enum ";
                 genAttributes( enumDecl->get_attributes() );
@@ -237 +239 @@
                         output << " {" << endl;
 
-                        cur_indent += CodeGenerator::tabsize;
+                        ++indent;
                         for ( std::list< Declaration* >::iterator i = memb.begin(); i != memb.end();  i++) {
                                 ObjectDecl * obj = dynamic_cast< ObjectDecl* >( *i );
                                 assert( obj );
-                                output << lineDirective( obj ) << indent << mangleName( obj );
+                                updateLocation( obj );
+                                output << mangleName( obj );
                                 if ( obj->get_init() ) {
                                         output << " = ";
@@ -249 +252 @@
                         } // for
 
-                        cur_indent -= CodeGenerator::tabsize;
+                        --indent;
 
                         output << indent << "}";
@@ -259 +262 @@
         void CodeGenerator::visit( TypedefDecl * typeDecl ) {
                 assertf( ! genC, "Typedefs are removed and substituted in earlier passes." );
-                output << lineDirective( typeDecl );
+                updateLocation( typeDecl );
                 output << "typedef ";
                 output << genType( typeDecl->get_base(), typeDecl->get_name(), pretty, genC ) << endl;
@@ -760 +763 @@
         void CodeGenerator::visit( StmtExpr * stmtExpr ) {
                 std::list< Statement * > & stmts = stmtExpr->get_statements()->get_kids();
-                output << lineDirective( stmtExpr) << "({" << std::endl;
-                cur_indent += CodeGenerator::tabsize;
+                updateLocation( stmtExpr );
+                output << "({" << std::endl;
+                ++indent;
                 unsigned int numStmts = stmts.size();
                 unsigned int i = 0;
                 for ( Statement * stmt : stmts ) {
-                        output << lineDirective( stmt ) << indent;
+                        updateLocation( stmt );
             output << printLabels( stmt->get_labels() );
                         if ( i+1 == numStmts ) {
@@ -784 +788 @@
                         ++i;
                 }
-                cur_indent -= CodeGenerator::tabsize;
+                --indent;
                 output << indent << "})";
         }
@@ -793 +797 @@
                 output << "{" << endl;
 
-                cur_indent += CodeGenerator::tabsize;
+                ++indent;
 
                 for ( std::list<Statement *>::iterator i = ks.begin(); i != ks.end();  i++ ) {
@@ -804 +808 @@
                         } // if
                 } // for
-                cur_indent -= CodeGenerator::tabsize;
+                --indent;
 
                 output << indent << "}";
@@ -852 +856 @@
 
         void CodeGenerator::visit( IfStmt * ifStmt ) {
-                output << lineDirective( ifStmt );
+                updateLocation( ifStmt );
                 output << "if ( ";
                 ifStmt->get_condition()->accept( *this );
@@ -866 +870 @@
 
         void CodeGenerator::visit( SwitchStmt * switchStmt ) {
-                output << lineDirective( switchStmt );
+                updateLocation( switchStmt );
                 output << "switch ( " ;
                 switchStmt->get_condition()->accept( *this );
@@ -872 +876 @@
 
                 output << "{" << std::endl;
-                cur_indent += CodeGenerator::tabsize;
+                ++indent;
                 acceptAll( switchStmt->get_statements(), *this );
-                cur_indent -= CodeGenerator::tabsize;
+                --indent;
                 output << indent << "}";
         }
 
         void CodeGenerator::visit( CaseStmt * caseStmt ) {
-                output << lineDirective( caseStmt );
-                output << indent;
+                updateLocation( caseStmt );
                 if ( caseStmt->isDefault()) {
                         output << "default";
@@ -891 +894 @@
                 std::list<Statement *> sts = caseStmt->get_statements();
 
-                cur_indent += CodeGenerator::tabsize;
+                ++indent;
                 for ( std::list<Statement *>::iterator i = sts.begin(); i != sts.end();  i++) {
                         output << indent << printLabels( (*i)->get_labels() )  ;
@@ -897 +900 @@
                         output << endl;
                 } // for
-                cur_indent -= CodeGenerator::tabsize;
+                --indent;
         }