DecayTree.h

/**************************************************************************
* BASF2 (Belle Analysis Framework 2)                                     *
* Copyright(C) 2010 - Belle II Collaboration                             *
*                                                                        *
* Author: The Belle II Collaboration                                     *
* Contributors: Christian Oswald                                         *
*                                                                        *
* This software is provided "as is" without any warranty.                *
**************************************************************************/
#pragma once
#include <vector>
 
namespace Belle2 {
  template<class T>
  class DecayTree {
  private:
    T* m_myObject;
    std::vector< DecayTree<T>* > m_daughters;
  public:
    DecayTree();
    explicit DecayTree(T* obj);
    ~DecayTree();
    std::vector< DecayTree<T>* > getDaughters();
    T* getObj();
    void setObj(T* obj);
    void clear();
    void append(DecayTree<T>* t);
  };
 
  template <class T>
  DecayTree<T>::DecayTree() : m_myObject(0), m_daughters() {};
 
  template <class T>
  DecayTree<T>::DecayTree(T* obj) : m_myObject(obj), m_daughters() {};
 
  template <class T>
  DecayTree<T>::~DecayTree()
  {
    m_myObject = 0;
    while (!m_daughters.empty()) {
      delete m_daughters.back();
      m_daughters.pop_back();
    }
  };
 
  template <class T>
  std::vector< DecayTree<T>* > DecayTree<T>::getDaughters()
  {
    return m_daughters;
  }
 
  template <class T>
  T* DecayTree<T>::getObj()
  {
    return m_myObject;
  }
 
  template <class T>
  void DecayTree<T>::setObj(T* obj)
  {
    m_myObject = obj;
  }
 
  template <class T>
  void DecayTree<T>::clear()
  {
    delete m_myObject;
    m_myObject = 0;
    while (!m_daughters.empty()) {
      m_daughters.back().clear();
      m_daughters.pop_back();
    }
  }
 
  template <class T>
  void DecayTree<T>::append(DecayTree<T>* t)
  {
    m_daughters.push_back(t);
  }
} // namespace Belle2