RT2SPTCConverterModuleDev.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 <tracking/spacePointCreation/SpacePoint.h>
#include <tracking/spacePointCreation/SpacePointTrackCand.h>
#include <tracking/trackFindingVXD/sectorMapTools/NoKickRTSel.h>
#include <svd/dataobjects/SVDCluster.h>
#include <mdst/dataobjects/MCParticle.h>
#include <svd/dataobjects/SVDTrueHit.h>
#include <pxd/dataobjects/PXDTrueHit.h>
 
#include <bitset>
 
namespace Belle2 {
  class RT2SPTCConverterModule : public Module {
 
  public:

    RT2SPTCConverterModule();

    ~RT2SPTCConverterModule() override;

    RT2SPTCConverterModule(const RT2SPTCConverterModule&) = delete;

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

    void initialize() override;

    void event() override;

    void endRun() override;

    void terminate() override;
 
  protected:

    enum conversionFlags {
      c_singleCluster = 0,
      c_undefinedError = 1
    };
 
    using ConversionState = std::bitset<2>; 

    std::pair<std::vector<const SpacePoint*>, ConversionState>
    getSpacePointsFromRecoHitInformations(std::vector<RecoHitInformation*> hitInfos);

    std::pair<std::vector<const SpacePoint*>, ConversionState>
    getSpacePointsFromRecoHitInformationViaTrueHits(const std::vector<RecoHitInformation*>& hitInfos);

    void initializeCounters()
    {
      m_noFailCtr = 0;
      m_singleClusterUseCtr = 0;
      m_undefinedErrorCtr = 0;
      m_missingTrueHitCtr = 0;
      m_minSPCtr = 0;
      m_mcParticlesPresent = false;
    }
 
    std::string m_SVDClusterName; 
 
    std::optional<std::string> m_pxdSpacePointsStoreArrayName; 
    std::optional<std::string> m_svdSpacePointsStoreArrayName; 
 
    std::string m_SVDSingleClusterSPName; 
 
    std::string m_RecoTracksName; 
 
    std::string m_SPTCName; 
 
    StoreArray<SpacePoint> m_PXDSpacePoints; 
    StoreArray<SpacePoint> m_SVDSpacePoints; 
    StoreArray<RecoTrack> m_RecoTracks; 
    StoreArray<SpacePointTrackCand> m_SpacePointTrackCands; 
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<SVDTrueHit> m_SVDTrueHit; 
    StoreArray<SVDCluster> m_SVDClusters; 
 
    // parameters
    bool m_ignorePXDHits; 
 
    int m_minSP; 
    bool m_useTrueHits; 
    bool m_skipProblematicCluster; 
    bool m_useSingleClusterSP; 
    bool m_markRecoTracks; 

    bool m_convertFittedOnly = false;

    NoKickRTSel* m_trackSel; 
    std::string m_noKickCutsFile; 
    bool m_noKickOutput; 
 
 
    int m_ncut = 0; 
    int m_npass = 0; 
 
 
    // state variables
    bool m_mcParticlesPresent; 
    unsigned int m_minSPCtr; 
    unsigned int m_noFailCtr; 
    unsigned int m_singleClusterUseCtr; 
    unsigned int m_undefinedErrorCtr; 
    unsigned int m_missingTrueHitCtr; 
  };

}