RelationElement.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 **************************************************************************/
 
#pragma once
 
#include <TObject.h>
 
#include <vector>
#include <utility>
 
namespace Belle2 {
  class RelationElement: public TObject {
  public:
 
    typedef unsigned int index_type;
 
    typedef float weight_type;
 
    RelationElement();
 
 
    RelationElement(index_type from, index_type to, weight_type weight = 1.0):
      TObject(), m_from(from), m_to(1, to), m_weights(1, weight) {}
 
    RelationElement(index_type from, const std::vector<index_type>& to, const std::vector<weight_type>& weights);
 
    template <class InputIterator> RelationElement(index_type from, const InputIterator& begin, const InputIterator& end):
      TObject(), m_from(from)
    {
      setToIndices(begin, end);
    }
 
    index_type                        getFromIndex()         const { return m_from; }
 
    size_t                            getSize()              const { return m_to.size(); }
 
    std::pair<index_type, weight_type> getTo(size_t n = 0)      const { return std::make_pair(m_to[n], m_weights[n]); }
 
    index_type                        getToIndex(size_t n = 0) const { return m_to[n]; }
 
    weight_type                       getWeight(size_t n = 0)  const { return m_weights[n]; }
 
    const std::vector<index_type>&     getToIndices()         const { return m_to; }
 
    const std::vector<weight_type>&    getWeights()           const { return m_weights; }
 
    void setFromIndex(index_type from) { m_from = from; }
 
    void setToIndex(index_type to, weight_type weight = 1.0)
    {
      m_to = std::vector<index_type>(1, to);
      m_weights = std::vector<weight_type>(1, weight);
    }
 
    template<class InputIterator> void setToIndices(InputIterator begin, const InputIterator& end)
    {
      for (; begin != end; ++begin) {
        m_to.push_back(begin->first);
        m_weights.push_back(begin->second);
      }
    }
 
  protected:
 
    index_type m_from;
 
    std::vector<index_type> m_to;
 
    std::vector<weight_type> m_weights;
 
    ClassDef(RelationElement, 1); 
  };
 
}