BKLMTrackingModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Yinghui GUAN                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/dataobjects/bklm/BKLMHit2d.h>
#include <klm/dataobjects/bklm/BKLMTrack.h>
#include <klm/bklm/geometry/GeometryPar.h>
 
/* Belle 2 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 BKLMTrackingModule: public Module {
 
  public:
 
    BKLMTrackingModule();
 
    virtual ~BKLMTrackingModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    bool sameSector(BKLMHit2d* hit1, BKLMHit2d* hit2);
 
    bool findClosestRecoTrack(BKLMTrack* bklmTrk, RecoTrack*& closestTrack);
 
 
  protected:
 
    bool m_studyEffi;
 
    bool m_MatchToRecoTrack;
 
    double m_maxAngleRequired = 10;
 
    bool m_globalFit;
 
    std::string m_outPath = "bklmEffi.root";
 
  private:
 
    bklm::GeometryPar* m_GeoPar = nullptr;
 
    TFile* m_file = nullptr;
 
    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;
    StoreArray<BKLMTrack> m_storeTracks;
 
    StoreArray<BKLMHit2d> hits2D;
 
    StoreArray<RecoTrack> recoTracks;
 
    StoreArray<RecoHitInformation> recoHitInformation;
 
    void runTracking(int mode, int section, int sector, int layer);
 
    void generateEffi(int section, int sector, int layer);
 
    static bool sortByLayer(BKLMHit2d* hit1, BKLMHit2d* hit2);
 
    bool isLayerUnderStudy(int section, int iSector, int iLayer, BKLMHit2d* hit);
 
    bool isSectorUnderStudy(int section, int iSector, BKLMHit2d* hit);
 
    double distanceToHit(BKLMTrack* track, BKLMHit2d* 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