ReattachCDCWireHitsToRecoTracksModule.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 <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/dbobjects/DAFConfiguration.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;

      short m_trackFitType = DAFConfiguration::c_CDConly;
    };
  }
}