KLMTrackingModule.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
 
/* KLM headers. */
#include <klm/dataobjects/KLMTrack.h>
#include <klm/dataobjects/KLMHit2d.h>
#include <klm/geometry/KLMGeometryPar.h>
 
/* Basf2 headers. */
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <tracking/dataobjects/RecoTrack.h>
 
/* ROOT headers. */
#include <TEfficiency.h>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
 
namespace Belle2 {
  class KLMTrackingModule: public Module {
 
  public:
 
    KLMTrackingModule();
 
    ~KLMTrackingModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
    void endRun() override;
 
    void terminate() override;
 
    bool sameSector(KLMHit2d* hit1, KLMHit2d* hit2);
 
    bool findClosestRecoTrack(KLMTrack* klmTrk, RecoTrack*& closestTrack);
 
 
  protected:
 
    bool m_studyEffi;
 
    bool m_MatchToRecoTrack;
 
    double m_maxAngleRequired = 10;
 
    double m_maxDistance = 10;
 
    double m_maxSigma = 5;
 
    unsigned int m_minHitList = 2;
 
    unsigned int m_maxHitList = 60;
 
    int m_minNLayer = 4;
 
    std::string m_outPath = "standaloneKLMEffi.root";
 
  private:
 
    Belle2::KLM::KLMGeometryPar* m_GeoPar;
 
    TFile* m_file = nullptr;
 
    //BKLM Efficiency Plots
 
    TH1F* m_total[2][8];
 
    TH1F* m_pass[2][8];
 
    TEfficiency* m_effiVsLayer[2][8];
 
    //TEfficiency* m_effiYX;
    TH2F* m_effiYX;
 
    //TEfficiency* m_effiYZ;
    TH2F* m_effiYZ;
 
    TH2F* m_passYX;
 
    TH2F* m_totalYX;
 
    TH2F* m_passYZ;
 
    TH2F* m_totalYZ;
 
    //EKLM Efficiency Plots
 
    TH1F* m_totalE[2][8];
 
    TH1F* m_passE[2][8];
 
    TEfficiency* m_effiVsLayerE[2][8];
 
    //TEfficiency* m_effiYX;
    TH2F* m_effiYXE;
 
    //TEfficiency* m_effiYZ;
    TH2F* m_effiYZE;
 
    TH2F* m_passYXE;
 
    TH2F* m_totalYXE;
 
    TH2F* m_passYZE;
 
    TH2F* m_totalYZE;
 
    StoreArray<KLMTrack> m_storeTracks;
 
    StoreArray<KLMHit2d> hits2D;
 
    StoreArray<RecoTrack> recoTracks;
 
    StoreArray<RecoHitInformation> recoHitInformation;
 
    void runTracking(int mode, int iSubdetector, int section, int sector, int layer);
 
    void generateEffi(int iSubdetector, int section, int sector, int layer);
 
    static bool sortByLayer(KLMHit2d* hit1, KLMHit2d* hit2);
 
    bool isLayerUnderStudy(int section, int iSector, int iLayer, KLMHit2d* hit);
 
    bool isSectorUnderStudy(int section, int iSector, KLMHit2d* hit);
 
    double distanceToHit(KLMTrack* track, KLMHit2d* hit,
                         double& error,
                         double& sigma);
 
    std::vector<int> m_runNumber;
 
    int m_runTotalEvents;
 
    std::vector<int> m_totalEvents;
 
    int m_runTotalEventsWithTracks;
 
    std::vector<int> m_totalEventsWithTracks;
  };
} // end namespace Belle2