CDCMCHitCollectionLookUp.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 <tracking/trackFindingCDC/mclookup/CDCMCHitLookUp.h>
 
#include <tracking/trackingUtilities/numerics/EForwardBackward.h>
#include <tracking/trackingUtilities/numerics/Index.h>
 
#include <map>
 
namespace Belle2 {
  class MCParticle;
  class CDCHit;
 
  namespace TrackingUtilities {
    class CDCTrajectory3D;
  }
 
  namespace TrackFindingCDC {

    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;
 
 

      TrackingUtilities::Index getFirstInTrackId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackId(getFirstHit(ptrHits)); }

      TrackingUtilities::Index getLastInTrackId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackId(getLastHit(ptrHits)); }
 
 

      TrackingUtilities::Index getFirstInTrackSegmentId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackSegmentId(getFirstHit(ptrHits)); }

      TrackingUtilities::Index getLastInTrackSegmentId(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getInTrackSegmentId(getLastHit(ptrHits)); }
 
 

      TrackingUtilities::Index getFirstNPassedSuperLayers(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNPassedSuperLayers(getFirstHit(ptrHits)); }

      TrackingUtilities::Index getLastNPassedSuperLayers(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNPassedSuperLayers(getLastHit(ptrHits)); }

      TrackingUtilities::Index getFirstNLoops(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNLoops(getFirstHit(ptrHits)); }

      TrackingUtilities::Index getLastNLoops(const ACDCHitCollection* ptrHits) const
      { return CDCMCHitLookUp::getInstance().getNLoops(getLastHit(ptrHits)); }

      TrackingUtilities::EForwardBackward isForwardOrBackwardToMCTrack(const ACDCHitCollection* ptrHits) const;

      TrackingUtilities::EForwardBackward areAlignedInMCTrack(const ACDCHitCollection* ptrFromHits,
                                                              const ACDCHitCollection* ptrToHits) const;

      TrackingUtilities::EForwardBackward areAlignedInMCTrackWithRLCheck(const ACDCHitCollection* ptrFromHits,
          const ACDCHitCollection* ptrToHits) const;

      TrackingUtilities::CDCTrajectory3D getTrajectory3D(const ACDCHitCollection* ptrHits) const;
 
    private:
      float m_minimalMatchPurity = 0.5;

      float m_minimalRLPurity = 0.5;
    };
  }
}