INTERFACE:

#include "ptree_visitor.h"
#include "downcast.h"

// Inherits "virtual"ly from Ptree_visitor<Returntype> to allow mixin
// implementations of its interface
template <typename Returntype>
class Ptree_program_visitor : virtual public Ptree_visitor<Returntype>
{
  typedef Ptree_visitor<Returntype> Super;
};


IMPLEMENTATION:

PUBLIC
template <typename Returntype>
Ptree_program_visitor<Returntype>::Ptree_program_visitor (Source* s)
  :  Ptree_visitor<Returntype> (s)
{
}

// 
// Detect type of declarations and call appropriate visit function
// 

// Overloaded version of Ptree_visitor's visit_declaration
template <typename Returntype>
virtual Returntype 
Ptree_program_visitor<Returntype>::visit_declaration (PtreeDeclaration *p) 
{
  Ptree* storagespec = p->First();
  Ptree* typespec = p->Second();

  // It is a function definition?
  PtreeDeclarator* decl = dynamic_cast<PtreeDeclarator*> (p->Third());

  if (decl)
    return visit_function (storagespec,
			   typespec,
			   decl,
			   downcast<PtreeBlock*> (p->Nth(3)));
  
  // Not a function definition

  if (p->Third()->IsLeaf())
    {
      assert (* p->Third()->ToString() == ';');
      return visit_type_declaration (storagespec, typespec);
    }

  // Must be variable or function-prototype declaration
  return visit_name_declaration(storagespec, typespec, p->Third());
}

PUBLIC
template <typename Returntype>
virtual Returntype 
Ptree_program_visitor<Returntype>::visit_function (Ptree* storagespec, 
						   Ptree* returntype, 
						   PtreeDeclarator* decl, 
						   PtreeBlock* block)
{
  visit(storagespec);
  visit(returntype);
  visit(decl);
  return visit(block);
}

PUBLIC
template <typename Returntype>
virtual Returntype 
Ptree_program_visitor<Returntype>::visit_type_declaration
  (Ptree* storagespec, Ptree* type)
{
  visit(storagespec);
  return visit(type);
}

PUBLIC
template <typename Returntype>
virtual Returntype 
Ptree_program_visitor<Returntype>::visit_name_declaration (Ptree* storagespec, 
						   Ptree* type, 
						   Ptree* decllist)
{
  visit(storagespec);
  visit(type);
  return visit(decllist);
}