ReattachCDCWireHitsToRecoTracksModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Cyrille Praz, Tracking group                                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/trackFindingCDC/eventdata/hits/CDCWireHit.h>
#include <tracking/trackFindingCDC/rootification/StoreWrappedObjPtr.h>
#include <tracking/trackFindingCDC/numerics/ERightLeft.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
 
    class ReattachCDCWireHitsToRecoTracksModule : public Module {
 
    public:
 
      ReattachCDCWireHitsToRecoTracksModule();
 
      void initialize() override;
 
      void event() override;
 
    private:
 
      struct ReconstructionResults {
        double arcLength = 0.0;
        double z = 0.0;
        double distanceToTrack = 0.0;
        ERightLeft rlInfo = ERightLeft::c_Unknown;
        bool isValid = false;
      };
 
      struct HitToAddInfo {
        CDCWireHit* hit = nullptr;
        ERightLeft rlInfo = ERightLeft::c_Unknown;
      };
 
      void findHits();
 
      void addHits();
 
      ReconstructionResults reconstruct(const CDCWireHit& wireHit,
                                        const RecoTrack& recoTrack,
                                        const RecoHitInformation* recoHitInformation) const;
 
      RecoHitInformation::RightLeftInformation rightLeftInformationTranslator(ERightLeft rlInfo) const;
 
    private:
 
      std::string m_CDCWireHitsStoreArrayName = "CDCWireHitVector";
      std::string m_inputRecoTracksStoreArrayName = "CDCRecoTracks";
      std::string m_outputRecoTracksStoreArrayName = "ExtendedCDCRecoTracks";
      double m_maximumDistance = 0.1;
      int m_minimumADC = 1;
      int m_minimumTOT = 1;
      double m_maximumAbsD0 = 13.5;
      double m_maximumAbsZ0 = 35.0;
 
      StoreWrappedObjPtr<std::vector<CDCWireHit> > m_CDCWireHits;
      StoreArray<RecoTrack> m_inputRecoTracks;
      StoreArray<RecoTrack> m_outputRecoTracks;
 
      std::unordered_map<RecoTrack*, std::vector<CDCHit*> > m_mapToHitsOnTrack;
      std::unordered_map<RecoTrack*, std::vector<HitToAddInfo> > m_mapToHitsToAdd;
    };
  }
}