source: translator/InitTweak/InitExpander.h @ 51587aa

//
// 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.
//
// XXX.cc -- 
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : 
// Last Modified On : 
// Update Count     : 0
//
#ifndef _INIT_EXPANDER_H_
#define _INIT_EXPANDER_H_

#include <string>

#include "utility.h"
#include "SynTree/Mutator.h"
#include "SymTab/Indexer.h"

#include "SynTree/Statement.h"
#include "SynTree/Declaration.h"
#include "SynTree/Type.h"

namespace InitTweak {

  class InitExpander : public Mutator
  {
    typedef Mutator Parent;

  public:
    InitExpander();
    ~InitExpander();

    virtual ObjectDecl *mutate( ObjectDecl * );

    // indexer runs
    virtual FunctionDecl   *mutate( FunctionDecl *functionDecl )
    {
      functionDecl->set_functionType( maybeMutate( functionDecl->get_functionType(), *this ) );
      mutateAll( functionDecl->get_oldDecls(), *this );
      functionDecl->set_statements( maybeMutate( functionDecl->get_statements(), *this ) );

      index.visit( functionDecl );
      return functionDecl;
    }

    virtual TypeDecl       *mutate( TypeDecl *typeDecl )
    { index.visit( typeDecl ); return typeDecl; }
    virtual TypedefDecl    *mutate( TypedefDecl *typeDecl )
    { index.visit( typeDecl ); return typeDecl; }
    virtual StructDecl     *mutate( StructDecl *aggregateDecl )
    { index.visit( aggregateDecl ); return aggregateDecl; }
    virtual UnionDecl      *mutate( UnionDecl *aggregateDecl )
    { index.visit( aggregateDecl ); return aggregateDecl; }
    virtual EnumDecl       *mutate( EnumDecl *aggregateDecl )
    { index.visit( aggregateDecl ); return aggregateDecl; }

    virtual Type           *mutate( StructInstType *aggrInst )
    { index.visit( aggrInst ); return aggrInst; }
    virtual Type           *mutate( UnionInstType *aggrInst )
    { index.visit( aggrInst ); return aggrInst; }

  private:
    SymTab::Indexer index;
  };  // class InitExpander

} // namespace InitTweak


#endif // #ifndef _INIT_EXPANDER_H_

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