#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 & );

      virtual ~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::ostrstream 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:
*/