Track.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2012 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Heck, Marco Staric, Nils Braun                    *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
#include <framework/gearbox/Const.h>
#include <framework/datastore/RelationsObject.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <algorithm>
 
namespace Belle2 {
  class Track : public RelationsObject {
  public:
 
    typedef std::pair< Const::ChargedStable, const TrackFitResult*> ChargedStableTrackFitResultPair;
 
    explicit Track(float qualityIndicator = 0.) : m_qualityIndicator(qualityIndicator)
    {
      std::fill(m_trackFitIndices, m_trackFitIndices + Const::chargedStableSet.size(), -1);
    }
 
    const TrackFitResult* getTrackFitResult(const Const::ChargedStable& chargedStable) const;
 
    const TrackFitResult* getTrackFitResultWithClosestMass(const Const::ChargedStable& requestedType) const;
 
    std::vector<ChargedStableTrackFitResultPair> getTrackFitResults() const;
 
    void setTrackFitResultIndex(const Const::ChargedStable& chargedStable, short index)
    {
      m_trackFitIndices[chargedStable.getIndex()] = index;
    }
 
    unsigned int getNumberOfFittedHypotheses() const;
 
    float getQualityIndicator() const
    {
      return m_qualityIndicator;
    }
 
    virtual std::string getInfoHTML() const override;
 
  private:
    short int m_trackFitIndices[Const::ChargedStable::c_SetSize];
 
    std::vector < short int > getValidIndices() const;
 
    float const m_qualityIndicator;
 
    ClassDefOverride(Track, 4); 
  };
}