getClass.h

Go to the documentation of this file.

#ifndef GETCLASS_H__
#define GETCLASS_H__

#include "PHNodeIterator.h"
#include "PHIODataNode.h"
#include "PHDataNode.h"
#include "PHNode.h"

#include <TObject.h>

#include <string>

class PHCompositeNode;

namespace findNode
{
  template <class T>
    T* getClass(PHCompositeNode *top, const std::string &name)
    {
      PHNodeIterator iter(top);
      PHNode *FoundNode = iter.findFirst(name.c_str()); // returns pointer to PHNode
      if (!FoundNode)
        {
          return NULL;
        }
      // first test if it is a PHDataNode
      PHDataNode<T> *DNode = dynamic_cast<PHDataNode<T>*>(FoundNode);
      if (DNode)
        {
          T* object = dynamic_cast<T*>(DNode->getData());
          if (object)
            {
              return object;
            }
        }

      // We make a static cast for PHIODataNode<TObject> since all
      // PHIODataNodes have to contain a TObject (otherwise it cannot be
      // written out and it should be a PHDataNode. dynamic cast does not 
      // work (see some explanation in PHTypedNodeIterator.h)

      PHIODataNode<TObject> *IONode = static_cast<PHIODataNode<TObject>*>(FoundNode);
      if (IONode)
        {
          T* object = dynamic_cast<T*>(IONode->getData());
          if (!object)
            {
              return NULL;
            }
          else
            {
              return object;
            }
        }

    return NULL;
  }
}

#endif /* GETCLASS_H */