V0.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2012 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Heck, Markus Prim                                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
#include <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/Track.h>
#include <framework/datastore/RelationsObject.h>
#include <framework/datastore/StoreArray.h>
#include <framework/gearbox/Const.h>
#include <utility>
 
namespace Belle2 {
  class V0 : public RelationsObject {
  public:
    V0();
 
    V0(const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairPositive,
       const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairNegative);
 
    std::pair<Track*, Track*> getTracks() const
    {
      StoreArray<Track> tracks;
      return std::make_pair(tracks[m_trackIndexPositive], tracks[m_trackIndexNegative]);
    }
 
    std::pair<short, short> getTrackIndices() const
    {
      return std::make_pair(m_trackIndexPositive, m_trackIndexNegative);
    }
 
    std::pair<TrackFitResult*, TrackFitResult*> getTrackFitResults() const
    {
      StoreArray<TrackFitResult> trackFitResults;
      return std::make_pair(trackFitResults[m_trackFitResultIndexPositive], trackFitResults[m_trackFitResultIndexNegative]);
    }
 
    std::pair<short, short> getTrackFitResultIndices() const
    {
      return std::make_pair(m_trackFitResultIndexPositive, m_trackFitResultIndexNegative);
    }
 
    Const::ParticleType getV0Hypothesis() const;
 
  private:
    short m_trackIndexPositive;
 
    short m_trackIndexNegative;
 
    short m_trackFitResultIndexPositive;
 
    short m_trackFitResultIndexNegative;
 
    ClassDef(V0, 3);
  };
}