CDCSimHitLookUp.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/numerics/ERightLeft.h>
 
#include <tracking/trackingUtilities/utilities/MayBePtr.h>
 
#include <map>
#include <vector>
 
namespace Belle2 {
 
  class CDCHit;
  class CDCSimHit;
 
  namespace TrackingUtilities {
    class CDCWireHit;
    class Vector3D;
    class CDCRecoHit3D;
    class CDCRecoHit2D;
    class CDCRLWireHit;
  }
 
  namespace TrackFindingCDC {
    class CDCMCMap;


    class CDCSimHitLookUp {
 
    public:
      static const CDCSimHitLookUp& getInstance();

      CDCSimHitLookUp(CDCSimHitLookUp&) = delete;

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

      CDCSimHitLookUp() = default;

      void clear();

      void fill(const CDCMCMap* ptrMCMap);
 
    private:
      void fillPrimarySimHits();

      TrackingUtilities::MayBePtr<const CDCSimHit> getClosestPrimarySimHit(const CDCSimHit* ptrSimHit) const;
 
    public:
      const CDCSimHit* getClosestPrimarySimHit(const CDCHit* ptrHit) const;

      TrackingUtilities::Vector3D getDirectionOfFlight(const CDCHit* ptrHit);
 
    private:
      void fillRLInfo();
 
    public:
      TrackingUtilities::ERightLeft getRLInfo(const CDCHit* ptrHit) const;

      TrackingUtilities::Vector3D getRecoPos3D(const CDCHit* ptrHit) const;

      double getDriftLength(const CDCHit* ptrHit) const;

      TrackingUtilities::Vector3D getClosestPrimaryRecoPos3D(const CDCHit* ptrHit) const;

      double getClosestPrimaryDriftLength(const CDCHit* ptrHit) const;
 
    public:
      const TrackingUtilities::CDCWireHit* getWireHit(const CDCHit* ptrHit,
                                                      const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;

      TrackingUtilities::CDCRLWireHit getRLWireHit(const CDCHit* ptrHit,
                                                   const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;

      TrackingUtilities::CDCRecoHit3D getRecoHit3D(const CDCHit* ptrHit,
                                                   const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;

      TrackingUtilities::CDCRecoHit3D getClosestPrimaryRecoHit3D(const CDCHit* ptrHit,
                                                                 const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;

      TrackingUtilities::CDCRecoHit2D getRecoHit2D(const CDCHit* ptrHit,
                                                   const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;

      TrackingUtilities::CDCRecoHit2D getClosestPrimaryRecoHit2D(const CDCHit* ptrHit,
                                                                 const std::vector<TrackingUtilities::CDCWireHit>& wireHits) const;
 
    private:
      const CDCMCMap* m_ptrMCMap;

      std::map<const CDCHit*, TrackingUtilities::MayBePtr<const CDCSimHit> >  m_primarySimHits;

      std::map<const CDCHit*, TrackingUtilities::ERightLeft> m_rightLeftInfos;
 
    };
  }
}