PHDataNodeIterator.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHDataNodeIterator.h

#ifndef PHOOL_PHDATANODEITERATOR_H
#define PHOOL_PHDATANODEITERATOR_H

#include "PHIODataNode.h"
#include "PHNodeIterator.h"

#include <cstddef>

class PHDataNodeIterator : public PHNodeIterator
{
 public:
  PHDataNodeIterator(PHCompositeNode* node);

  ~PHDataNodeIterator() override {}
  template <class T>
  PHIODataNode<T>* FindIODataNode(PHIODataNode<T>* node,
                                  const char* name);

  template <class T>
  PHBoolean AddIODataNode(T* data, const char* name);
};

inline PHDataNodeIterator::PHDataNodeIterator(PHCompositeNode* node)
  : PHNodeIterator(node)
{
}

template <class T>
PHIODataNode<T>*
PHDataNodeIterator::FindIODataNode(PHIODataNode<T>* node,
                                   const char* name)
{
  // TODO:  also check that "name" is not a null string!
  if (!name)
  {
    return 0;
  }
  // Can't do dynamic_cast here; it fails if node was created as
  // PHIODataNode<X> instead of PHIODataNode<T>, even if T is a
  // derived class of X!
  // In general, T -> X does not imply A<T> -> A<X>.
  // ("->" denotes "derives from", and "A" is any template class)
  node = static_cast<PHIODataNode<T>*>(findFirst("PHIODataNode",
                                                 name));
  return node;
}

template <class T>
PHBoolean
PHDataNodeIterator::AddIODataNode(T* data, const char* name)
{
  // TODO:  also check that "name" is not a null string!
  if (!name)
  {
    return false;
  }
  // For IODataNode, ought to check (if possible) that T derives
  // from TObject.  Will typeid() give us this info?

  PHIODataNode<T>* n = new PHIODataNode<T>(data, name);
  if (!n)
  {
    return false;  // problem creating node?
  }
  return addNode(n);
}

#endif