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/trackFindingCDC/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<CDCTrack>& cdcTracks);
 
      void clear();
 
      bool isTrackClone(const CDCTrack& cdcTrack);
 
    private:
      CDCMCCloneLookUp() = default;
 
      std::map<const ITrackType, std::vector<CDCTrack*>> getMatchedCDCTracksByMCID(
                                                        std::vector<CDCTrack>& cdcTracks);
 
      CDCTrack* findBestMatchedTrack(std::vector<CDCTrack*> matchedTrackPtrs);
 
      std::map<const CDCTrack*, bool> m_cdcTrackIsCloneMap;
    };
 
    struct CompareCDCTracks {
      operator FunctorTag();
 
      CompareCDCTracks(const CDCMCTrackLookUp& cdcMCTrackLookUp,
                       const CDCMCHitLookUp& cdcMCHitLookUp)
        : m_CDCMCTrackLookUp(cdcMCTrackLookUp)
        , m_CDCMCHitLookUp(cdcMCHitLookUp) {};
 
 
      bool operator()(const CDCTrack* ptrCDCTrack1, const CDCTrack* ptrCDCTrack2) const;
 
    private:
      unsigned int getNumberOfCorrectHits(const CDCTrack* ptrCDCTrack) const;
 
      const CDCMCTrackLookUp& m_CDCMCTrackLookUp;
      const CDCMCHitLookUp& m_CDCMCHitLookUp;
    };
  } // namespace TrackFindingCDC
} // namespace Belle2