CDCMCTrackStore.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/ITrackType.h>
#include <tracking/trackFindingCDC/numerics/Index.h>
 
#include <map>
#include <vector>
 
namespace Belle2 {
  class CDCHit;
 
  namespace TrackFindingCDC {
    class CDCMCMap;
    class CDCSimHitLookUp;
 
    class CDCMCTrackStore {
 
    public:
      using CDCHitVector = std::vector<const CDCHit*>;
 
    public:
      static const CDCMCTrackStore& getInstance();
 
    public:
      CDCMCTrackStore() = default;
      CDCMCTrackStore(CDCMCTrackStore&) = delete;
      CDCMCTrackStore& operator=(const CDCMCTrackStore&) = delete;
 
    public:
      void clear();
 
      void fill(const CDCMCMap* ptrMCMap, const CDCSimHitLookUp* ptrSimHitLookUp);
 
    public:
      const std::map<ITrackType, Belle2::TrackFindingCDC::CDCMCTrackStore::CDCHitVector>& getMCTracksByMCParticleIdx() const
      { return  m_mcTracksByMCParticleIdx;}
 
      const std::map<ITrackType, std::vector<Belle2::TrackFindingCDC::CDCMCTrackStore::CDCHitVector> >& getMCSegmentsByMCParticleIdx()
      const
      { return  m_mcSegmentsByMCParticleIdx;}
 
    private:
      void fillMCTracks();
 
      void fillMCSegments();
 
      void arrangeMCTrack(CDCHitVector& mcTrack) const;
 
      void fillInTrackId();
 
      void fillInTrackSegmentId();
 
      void fillNLoopsAndNPassedSuperLayers();
 
 
      bool changedSuperLayer(const CDCHitVector& mcSegment, const CDCHitVector& nextMCSegment) const;
 
    public:
      Index getInTrackId(const CDCHit* ptrHit) const;
 
      Index getInTrackSegmentId(const CDCHit* ptrHit) const;
 
      Index getNPassedSuperLayers(const CDCHit* ptrHit) const;
 
      Index getNLoops(const CDCHit* ptrHit) const;
 
    private:
      const CDCMCMap* m_ptrMCMap;
 
      const CDCSimHitLookUp* m_ptrSimHitLookUp;
 
      std::map<ITrackType, CDCHitVector> m_mcTracksByMCParticleIdx;
 
      std::map<ITrackType, std::vector<CDCHitVector> > m_mcSegmentsByMCParticleIdx;
 
      std::map<const CDCHit*, int> m_inTrackIds;
 
      std::map<const CDCHit*, int> m_inTrackSegmentIds;
 
      std::map<const CDCHit*, int> m_nPassedSuperLayers;
 
      std::map<const CDCHit*, int> m_nLoops;
    };
  }
}