CDCMCCloneLookUp.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/trackingUtilities/eventdata/tracks/CDCTrack.h>
#include <tracking/trackFindingCDC/mclookup/ITrackType.h>
 
#include <tracking/trackFindingCDC/mclookup/CDCMCTrackLookUp.h>
#include <tracking/trackFindingCDC/mclookup/CDCMCHitLookUp.h>
 
#include <map>
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {

    class CDCMCCloneLookUp {
 
    public:

      static CDCMCCloneLookUp& getInstance();

      CDCMCCloneLookUp(CDCMCCloneLookUp&) = delete;

      CDCMCCloneLookUp& operator=(const CDCMCCloneLookUp&) = delete;

      void fill(std::vector<TrackingUtilities::CDCTrack>& cdcTracks);

      void clear();

      bool isTrackClone(const TrackingUtilities::CDCTrack& cdcTrack);
 
    private:
      CDCMCCloneLookUp() = default;

      std::map<const ITrackType, std::vector<TrackingUtilities::CDCTrack*>> getMatchedCDCTracksByMCID(
            std::vector<TrackingUtilities::CDCTrack>& cdcTracks);

      TrackingUtilities::CDCTrack* findBestMatchedTrack(std::vector<TrackingUtilities::CDCTrack*> matchedTrackPtrs);

      std::map<const TrackingUtilities::CDCTrack*, bool> m_cdcTrackIsCloneMap;
    };

    struct CompareCDCTracks {
      operator TrackingUtilities::FunctorTag();

      CompareCDCTracks(const CDCMCTrackLookUp& cdcMCTrackLookUp,
                       const CDCMCHitLookUp& cdcMCHitLookUp)
        : m_CDCMCTrackLookUp(cdcMCTrackLookUp)
        , m_CDCMCHitLookUp(cdcMCHitLookUp) {};
 

      bool operator()(const TrackingUtilities::CDCTrack* ptrCDCTrack1, const TrackingUtilities::CDCTrack* ptrCDCTrack2) const;
 
    private:
      unsigned int getNumberOfCorrectHits(const TrackingUtilities::CDCTrack* ptrCDCTrack) const;

      const CDCMCTrackLookUp& m_CDCMCTrackLookUp;
      const CDCMCHitLookUp& m_CDCMCHitLookUp;
    };
  } // namespace TrackFindingCDC
} // namespace Belle2