TRGGRLMatchModule.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 <trg/cdc/dataobjects/CDCTriggerTrack.h>
#include <trg/ecl/dataobjects/TRGECLCluster.h>
#include <trg/klm/dataobjects/KLMTrgSummary.h>
#include <trg/cdc/dataobjects/CDCTriggerSegmentHit.h>
#include <trg/grl/dataobjects/TRGGRLShortTrack.h>
#include <trg/grl/dataobjects/TRGGRLInnerTrack.h>
#include <trg/grl/dataobjects/TRGGRLInfo.h>
#include <string>
#include <vector>
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
 
 
namespace Belle2 {
  class TRGGRLMatchModule : public Module {
 
  public:
 
    TRGGRLMatchModule();
 
    virtual ~TRGGRLMatchModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    void calculationdistance(CDCTriggerTrack* track, TRGECLCluster* cluster, double* ds, int _match3D);
 
    void calculationphiangle(CDCTriggerTrack* track, TRGECLCluster* cluster, int& dphi_d, std::vector<bool>& track_phimap,
                             std::vector<bool>& track_phimap_i);
 
    void sectormatching_klm(CDCTriggerTrack* track, StoreObjPtr<KLMTrgSummary> klmtrgsummary, double& dphi, int& klmtrack_ind_phi);
 
    bool photon_cluster(TRGECLCluster* cluster, std::vector<bool> track_phimap, double e_threshold);
 
    int N64(int x);
 
    int N36(int x);
 
    void fill_pattern_base2(std::vector< std::vector<int> >& patt);
 
    void make_veto_map(StoreArray<CDCTriggerTrack> track2Dlist, std::vector<bool>&  map_veto);
 
    void make_eecl_map(StoreArray<TRGECLCluster> clusterlist, std::vector<bool>& ecl_phimap, std::vector<bool>& ecl_phimap_fwd,
                       std::vector<bool>& ecl_phimap_bwd, std::vector<bool>& ecl_sectormap_fwd, std::vector<bool>& ecl_sectormap_bwd);
 
    void make_eklm_map(StoreObjPtr<KLMTrgSummary> klmtrgsummary,
                       std::vector<bool>& eklm_sectormap, std::vector<bool>& eklm_sectormap_fwd, std::vector<bool>& eklm_sectormap_bwd);
 
    void short_tracking(StoreArray<CDCTriggerSegmentHit> tslist, std::vector<bool>  map_veto, std::vector<bool>  phimap_i,
                        std::vector<bool>  ecl_phimap_fwd, std::vector<bool>  ecl_phimap_bwd,
                        std::vector<bool>  klm_sectormap_fwd, std::vector<bool>  klm_sectormap_bwd,
                        std::vector< std::vector<int> >& pattern_base0, std::vector< std::vector<int> >& pattern_base2,
                        StoreArray<TRGGRLShortTrack> grlst,
                        StoreObjPtr<TRGGRLInfo> trgInfo);
 
    void inner_tracking(StoreArray<CDCTriggerSegmentHit> tslist, std::vector<bool>  phimap_i,
                        std::vector<bool>  ecl_phimap, std::vector<bool>  klm_sectormap,
                        StoreArray<TRGGRLInnerTrack> grlit,
                        StoreObjPtr<TRGGRLInfo> trgInfo);
 
    void matching_eecl_eklm(std::vector<bool>  eecl_sectormap_fw,
                            std::vector<bool>  eecl_setormap_bw,
                            std::vector<bool>  eklm_sectormap_fw,
                            std::vector<bool>  eklm_sectormap_bw,
                            StoreObjPtr<TRGGRLInfo> trgInfo);
 
    void extrapolation(int pattern, int& l, int& r, int& ec);
 
  private:
    StoreObjPtr<TRGGRLInfo> m_TRGGRLInfo; 
    int m_simulationMode;
 
    int m_fastSimulationMode;
 
    int m_firmwareSimulationMode;
 
    double m_dr_threshold;
    double m_dz_threshold;
    int m_dphi_d_threshold;
    double m_e_threshold;
    double m_dphi_klm_threshold;
    std::vector<bool> track_phimap;
    std::vector<bool> track_phimap_i;
    std::vector<bool> eecl_phimap;
    std::vector<bool> eecl_phimap_fwd;
    std::vector<bool> eecl_phimap_bwd;
    std::vector<bool> eecl_sectormap_fwd;
    std::vector<bool> eecl_sectormap_bwd;
    std::vector<bool> eklm_sectormap;
    std::vector<bool> eklm_sectormap_fwd;
    std::vector<bool> eklm_sectormap_bwd;
    std::string m_2d_tracklist;
    std::string m_3d_tracklist;
    std::string m_clusterlist;
    std::string m_klmtrgsummarylist;
    //double m_dr;
    //double m_dz;
    std::string m_2dmatch_tracklist;
    std::string m_phimatch_tracklist;
    std::string m_3dmatch_tracklist;
    std::string m_klmmatch_tracklist;
    std::string m_grlphotonlist;
    std::string m_hitCollectionName;
    std::string m_grlstCollectionName;
    std::string m_grlitCollectionName;
    std::string m_TrgGrlInformationName;
    std::vector< std::vector<int> > patterns_base0;
    std::vector< std::vector<int> > patterns_base2;
  };
}