#ifndef CODEGENV_H
#define CODEGENV_H

#include <strstream>
#include <list>

#include "SynTree/SynTree.h"
#include "SynTree/Visitor.h"
#include "SymTab/Indexer.h"

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

	CodeGenerator2( std::ostream &os );
	CodeGenerator2( std::ostream &os, std::string, int indent = 0, bool infun = false );
	CodeGenerator2( std::ostream &os, char *, int indent = 0, bool infun = false );

	CodeGenerator2( CodeGenerator2 & );

	//*** 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( ContextDecl *aggregateDecl );
	virtual void visit( TypedefDecl *typeDecl );
	virtual void visit( TypeDecl *typeDecl );

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

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

	//*** Expression
	virtual void visit( ApplicationExpr *applicationExpr );
	virtual void visit( UntypedExpr *untypedExpr );
	virtual void visit( NameExpr *nameExpr );
	virtual void visit( AddressExpr *addressExpr );
	virtual void visit( CastExpr *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( LogicalExpr *logicalExpr );
	virtual void visit( ConditionalExpr *conditionalExpr );
	virtual void visit( CommaExpr *commaExpr );
	virtual void visit( TupleExpr *tupleExpr );
	virtual void visit( TypeExpr *typeExpr );

	//*** Statements
	virtual void visit( CompoundStmt * );
	virtual void visit( ExprStmt * );
	virtual void visit( IfStmt * );
	virtual void visit( SwitchStmt * );
	virtual void visit( CaseStmt * );
	virtual void visit( BranchStmt * );
	virtual void visit( ReturnStmt * );
	virtual void visit( WhileStmt * );
	virtual void visit( ForStmt * );
	virtual void visit( NullStmt * );
	virtual void visit( DeclStmt * ); 

	std::string get_string( void );
	void add_string_left( std::string s ) { before << s; }
	void shift_left();
	template< class Iterator > void genCommaList( Iterator begin, Iterator end );
      private:
	int cur_indent;
	bool insideFunction;
	std::ostream &before;
	std::string after;

	static std::string printLabels ( std::list < Label > & );
	void handleStorageClass( Declaration *decl );
	void handleAggregate( AggregateDecl *aggDecl );
	void handleTypedef( NamedTypeDecl *namedType );

    };
    
    template< class Iterator >
    void CodeGenerator2::genCommaList( Iterator begin, Iterator end ) {
	if ( begin == end ) return;

	for ( ;; ) {
	    (*begin++)->accept( *this );
	    if ( begin == end ) return;
	    before << ", ";
	}
    }
  
    inline bool doSemicolon( Declaration* decl ) {
	if ( FunctionDecl* func = dynamic_cast< FunctionDecl* >( decl ) ) {
	    return !func->get_statements();
	}
	return true;
    }
} // namespace CodeGen

#endif /* #ifndef CODEGENV_H */

/*
  Local Variables:
  mode: "c++"
  End:
*/