source: src/CodeGen/CodeGenerator.h@ 78a0b88

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// CodeGenerator.h --
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Andrew Beach
// Last Modified On : Fri Aug 18 15:40:00 2017
// Update Count     : 56
//

#pragma once

#include <list>                   // for list
#include <ostream>                // for ostream, operator<<
#include <string>                 // for string

#include "Common/Indenter.h"      // for Indenter
#include "SynTree/Declaration.h"  // for DeclarationWithType (ptr only), Fun...
#include "SynTree/Visitor.h"      // for Visitor
#include "SynTree/SynTree.h"      // for Visitor Nodes

namespace CodeGen {
        class CodeGenerator : public Visitor {
          public:
                static int tabsize;

                CodeGenerator( std::ostream &os, bool pretty = false, bool genC = false, bool lineMarks = false );
                CodeGenerator( std::ostream &os, std::string, int indent = 0, bool infun = false );
                CodeGenerator( std::ostream &os, char *, int indent = 0, bool infun = false );

                //*** Declaration
                virtual void visit( StructDecl * );
                virtual void visit( FunctionDecl * );
                virtual void visit( ObjectDecl * );
                virtual void visit( UnionDecl *aggregateDecl );
                virtual void visit( EnumDecl *aggregateDecl );
                virtual void visit( TraitDecl *aggregateDecl );
                virtual void visit( TypedefDecl *typeDecl );
                virtual void visit( TypeDecl *typeDecl );

                //*** Initializer
                virtual void visit( Designation * );
                virtual void visit( SingleInit * );
                virtual void visit( ListInit * );
                virtual void visit( ConstructorInit * );

                //*** Constant
                virtual void visit( Constant * );

                //*** Expression
                virtual void visit( ApplicationExpr *applicationExpr );
                virtual void visit( UntypedExpr *untypedExpr );
                virtual void visit( RangeExpr * rangeExpr );
                virtual void visit( NameExpr *nameExpr );
                virtual void visit( AddressExpr *addressExpr );
                virtual void visit( CastExpr *castExpr );
                virtual void visit( VirtualCastExpr *castExpr );
                virtual void visit( UntypedMemberExpr *memberExpr );
                virtual void visit( MemberExpr *memberExpr );
                virtual void visit( VariableExpr *variableExpr );
                virtual void visit( ConstantExpr *constantExpr );
                virtual void visit( SizeofExpr *sizeofExpr );
                virtual void visit( AlignofExpr *alignofExpr );
                virtual void visit( UntypedOffsetofExpr *offsetofExpr );
                virtual void visit( OffsetofExpr *offsetofExpr );
                virtual void visit( OffsetPackExpr *offsetPackExpr );
                virtual void visit( LogicalExpr *logicalExpr );
                virtual void visit( ConditionalExpr *conditionalExpr );
                virtual void visit( CommaExpr *commaExpr );
                virtual void visit( CompoundLiteralExpr *compLitExpr );
                virtual void visit( UniqueExpr * );
                virtual void visit( TupleAssignExpr * tupleExpr );
                virtual void visit( UntypedTupleExpr *tupleExpr );
                virtual void visit( TupleExpr *tupleExpr );
                virtual void visit( TupleIndexExpr * tupleExpr );
                virtual void visit( TypeExpr *typeExpr );
                virtual void visit( AsmExpr * );
                virtual void visit( StmtExpr * );

                //*** Statements
                virtual void visit( CompoundStmt * );
                virtual void visit( ExprStmt * );
                virtual void visit( AsmStmt * );
                virtual void visit( AsmDecl * );                                // special: statement in declaration context
                virtual void visit( IfStmt * );
                virtual void visit( SwitchStmt * );
                virtual void visit( CaseStmt * );
                virtual void visit( BranchStmt * );
                virtual void visit( ReturnStmt * );
                virtual void visit( ThrowStmt * );
                virtual void visit( WhileStmt * );
                virtual void visit( ForStmt * );
                virtual void visit( NullStmt * );
                virtual void visit( DeclStmt * );

                void genAttributes( std::list< Attribute * > & attributes );

                template< class Iterator > void genCommaList( Iterator begin, Iterator end );

                struct LabelPrinter {
                        LabelPrinter(CodeGenerator &cg) : cg(cg), labels( 0 ) {}
                        LabelPrinter & operator()( std::list< Label > & l );
                        CodeGenerator & cg;
                        std::list< Label > * labels;
                };

                void asmName( DeclarationWithType *decl );

                void extension( Expression *expr );
                void extension( Declaration *decl );

                void updateLocation( BaseSyntaxNode const * to );
          private:
                Indenter indent;
                bool insideFunction;
                std::ostream &output;
                LabelPrinter printLabels;
                bool pretty = false;  // pretty print
                bool genC = false;    // true if output has to be C code
                bool lineMarks = false;

                CodeLocation currentLocation;
                void updateLocation( CodeLocation const & to );
                void nextLine();

                void handleStorageClass( DeclarationWithType *decl );
                void handleAggregate( AggregateDecl *aggDecl, const std::string & kind );
                void handleTypedef( NamedTypeDecl *namedType );
                std::string mangleName( DeclarationWithType * decl );
        }; // CodeGenerator

        template< class Iterator >
        void CodeGenerator::genCommaList( Iterator begin, Iterator end ) {
          if ( begin == end ) return;
                for ( ;; ) {
                        (*begin++)->accept( *this );
                  if ( begin == end ) break;
                        output << ", ";                                                         // separator
                } // for
        } // genCommaList

        inline bool doSemicolon( Declaration* decl ) {
                if ( FunctionDecl* func = dynamic_cast< FunctionDecl* >( decl ) ) {
                        return ! func->get_statements();
                } // if
                return true;
        } // doSemicolon

        /// returns C-compatible name of declaration
        std::string genName( DeclarationWithType * decl );
} // namespace CodeGen

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //