TRGGRLMatchModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2014 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Chunhua LI, Yun-Tsung Lai, Junhao Yin                    *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <trg/cdc/dataobjects/CDCTriggerTrack.h>
#include <trg/ecl/dataobjects/TRGECLCluster.h>
#include <trg/klm/dataobjects/KLMTriggerTrack.h>
#include <trg/cdc/dataobjects/CDCTriggerSegmentHit.h>
#include <trg/grl/dataobjects/TRGGRLShortTrack.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, KLMTriggerTrack* klmtrack, double& dphi);
 
    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 short_tracking(StoreArray<CDCTriggerSegmentHit> tslist, std::vector<bool>  map_veto, std::vector<bool>  phimap_i,
                        std::vector< std::vector<int> >& pattern_base0, std::vector< std::vector<int> >& pattern_base2,
                        StoreArray<TRGGRLShortTrack> grlst,
                        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::string m_2d_tracklist;
    std::string m_3d_tracklist;
    std::string m_clusterlist;
    std::string m_klmtracklist;
    //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_TrgGrlInformationName;
    std::vector< std::vector<int> > patterns_base0;
    std::vector< std::vector<int> > patterns_base2;
  };
}