CDCMCHitCollectionLookUp.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2014 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/mclookup/CDCMCHitLookUp.h>
 
#include <tracking/trackFindingCDC/numerics/EForwardBackward.h>
#include <tracking/trackFindingCDC/numerics/Index.h>
 
#include <map>
 
namespace Belle2 {
  class MCParticle;
  class CDCHit;
 
  namespace TrackFindingCDC {
    class CDCTrajectory3D;
 
    struct MCTrackIdPurityPair {
 
      MCTrackIdPurityPair(ITrackType mcTrackId, float purity, int correctRLVote)
        : m_mcTrackId(mcTrackId)
        , m_purity(purity)
        , m_correctRLVote(correctRLVote)
      {
      }
 
      ITrackType getMCTrackId() const
      {
        return m_mcTrackId;
      }
 
      float getPurity() const
      {
        return m_purity;
      }
 
      int getCorrectRLVote() const
      {
        return m_correctRLVote;
      }
 
    private:
      ITrackType m_mcTrackId;
 
      float m_purity;
 
      int m_correctRLVote;
    };
 
    template <typename ACDCHitCollection>
    class CDCMCHitCollectionLookUp  {
 
    public:
      void clear();
 
      CDCMCHitCollectionLookUp() {};
 
    private:
      std::map<ITrackType, size_t> getHitCountByMCTrackId(const ACDCHitCollection& hits) const;
 
    public:
      MCTrackIdPurityPair getHighestPurity(const ACDCHitCollection& hits) const;
 
 
      ITrackType getMCTrackId(const ACDCHitCollection* ptrHits) const;
 
      int getCorrectRLVote(const ACDCHitCollection* ptrHits) const;
 
      double getRLPurity(const ACDCHitCollection* ptrHits) const;
 
      const MCParticle* getMCParticle(const ACDCHitCollection* ptrHits) const;
 
 
      const CDCHit* getFirstHit(const ACDCHitCollection* ptrHits) const;
 
      const CDCHit* getLastHit(const ACDCHitCollection* ptrHits) const;
 
 
 
      Index getFirstInTrackId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackId(getFirstHit(ptrHits)); }
 
      Index getLastInTrackId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackId(getLastHit(ptrHits)); }
 
 
 
      Index getFirstInTrackSegmentId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackSegmentId(getFirstHit(ptrHits)); }
 
      Index getLastInTrackSegmentId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackSegmentId(getLastHit(ptrHits)); }
 
 
 
      Index getFirstNPassedSuperLayers(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNPassedSuperLayers(getFirstHit(ptrHits)); }
 
      Index getLastNPassedSuperLayers(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNPassedSuperLayers(getLastHit(ptrHits)); }
 
      Index getFirstNLoops(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNLoops(getFirstHit(ptrHits)); }
 
      Index getLastNLoops(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNLoops(getLastHit(ptrHits)); }
 
      EForwardBackward isForwardOrBackwardToMCTrack(const ACDCHitCollection* ptrHits) const;
 
      EForwardBackward areAlignedInMCTrack(const ACDCHitCollection* ptrFromHits,
                                           const ACDCHitCollection* ptrToHits) const;
 
      EForwardBackward areAlignedInMCTrackWithRLCheck(const ACDCHitCollection* ptrFromHits,
                                                      const ACDCHitCollection* ptrToHits) const;
 
      CDCTrajectory3D getTrajectory3D(const ACDCHitCollection* ptrHits) const;
 
    private:
      float m_minimalMatchPurity = 0.5;
 
      float m_minimalRLPurity = 0.5;
    };
  }
}