DecayTree.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#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