CDCSimHitLookUp.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/numerics/ERightLeft.h>
 
#include <tracking/trackFindingCDC/utilities/MayBePtr.h>
 
#include <map>
#include <vector>
 
namespace Belle2 {
  class CDCHit;
  class CDCSimHit;
 
  namespace TrackFindingCDC {
    class CDCMCMap;
    class CDCRecoHit3D;
    class CDCRecoHit2D;
    class CDCRLWireHit;
    class CDCWireHit;
    class Vector3D;
 
 
    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();
 
      MayBePtr<const CDCSimHit> getClosestPrimarySimHit(const CDCSimHit* ptrSimHit) const;
 
    public:
      const CDCSimHit* getClosestPrimarySimHit(const CDCHit* ptrHit) const;
 
      Vector3D getDirectionOfFlight(const CDCHit* ptrHit);
 
    private:
      void fillRLInfo();
 
    public:
      ERightLeft getRLInfo(const CDCHit* ptrHit) const;
 
      Vector3D getRecoPos3D(const CDCHit* ptrHit) const;
 
      double getDriftLength(const CDCHit* ptrHit) const;
 
      Vector3D getClosestPrimaryRecoPos3D(const CDCHit* ptrHit) const;
 
      double getClosestPrimaryDriftLength(const CDCHit* ptrHit) const;
 
    public:
      const CDCWireHit* getWireHit(const CDCHit* ptrHit,
                                   const std::vector<CDCWireHit>& wireHits) const;
 
      CDCRLWireHit getRLWireHit(const CDCHit* ptrHit,
                                const std::vector<CDCWireHit>& wireHits) const;
 
      CDCRecoHit3D getRecoHit3D(const CDCHit* ptrHit,
                                const std::vector<CDCWireHit>& wireHits) const;
 
      CDCRecoHit3D getClosestPrimaryRecoHit3D(const CDCHit* ptrHit,
                                              const std::vector<CDCWireHit>& wireHits) const;
 
      CDCRecoHit2D getRecoHit2D(const CDCHit* ptrHit,
                                const std::vector<CDCWireHit>& wireHits) const;
 
      CDCRecoHit2D getClosestPrimaryRecoHit2D(const CDCHit* ptrHit,
                                              const std::vector<CDCWireHit>& wireHits) const;
 
    private:
      const CDCMCMap* m_ptrMCMap;
 
      std::map<const CDCHit*, MayBePtr<const CDCSimHit> >  m_primarySimHits;
 
      std::map<const CDCHit*, ERightLeft> m_rightLeftInfos;
 
    };
  }
}