KLMStripEfficiencyCollectorModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Vitaliy Popov, Dmytro Minchenko                          *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/bklm/geometry/GeometryPar.h>
#include <klm/dataobjects/eklm/EKLMElementNumbers.h>
#include <klm/dataobjects/KLMDigit.h>
#include <klm/dataobjects/KLMElementNumbers.h>
#include <klm/dataobjects/KLMPlaneArrayIndex.h>
#include <klm/dbobjects/KLMChannelStatus.h>
 
/* Belle 2 headers. */
#include <analysis/dataobjects/ParticleList.h>
#include <calibration/CalibrationCollectorModule.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
 
/* ROOT headers. */
#include <TFile.h>
#include <TH1F.h>
#include <TTree.h>
 
/* C++ headers. */
#include <map>
#include <string>
 
namespace Belle2 {
  class KLMStripEfficiencyCollectorModule : public CalibrationCollectorModule {
 
  private:
 
    struct HitData {
 
      int subdetector;
 
      int section;
 
      int layer;
 
      int sector;
 
      int plane;
 
      int strip;
 
      double localPosition;
 
      const ExtHit* hit;
 
      const KLMDigit* digit;
 
    };
 
  public:
 
    KLMStripEfficiencyCollectorModule();
 
    virtual ~KLMStripEfficiencyCollectorModule();
 
    void prepare() override;
 
    void finish() override;
 
    void collect() override;
 
    void startRun() override;
 
    void closeRun() override;
 
  private:
 
    void addHit(std::map<uint16_t, struct HitData>& hitMap,
                uint16_t planeGlobal, struct HitData* hitData);
 
    void findMatchingDigit(struct HitData* hitData);
 
    bool collectDataTrack(const Particle* muon, TH1F* matchedDigitsInPlane,
                          TH1F* allExtHitsInPlane);
 
    std::string m_MuonListName;
 
    double m_AllowedDistance1D;
 
    int m_MinimalMatchingDigits;
 
    int m_MinimalMatchingDigitsOuterLayers;
 
    double m_MinimalMomentumNoOuterLayers;
 
    bool m_RemoveUnusedMuons;
 
    bool m_IgnoreBackwardPropagation;
 
    DBObjPtr<KLMChannelStatus> m_ChannelStatus;
 
    StoreArray<KLMDigit> m_Digits;
 
    StoreArray<Track> m_tracks;
 
    StoreArray<ExtHit> m_extHits;
 
    StoreObjPtr<ParticleList> m_MuonList;
 
    const KLMElementNumbers* m_ElementNumbers;
 
    const EKLMElementNumbers* m_eklmElementNumbers;
 
    const bklm::GeometryPar* m_GeometryBKLM;
 
    const KLMPlaneArrayIndex* m_PlaneArrayIndex;
 
    bool m_Debug;
 
    std::string m_MatchingFileName;
 
    TFile* m_MatchingFile;
 
    TTree* m_MatchingTree;
 
    struct HitData m_MatchingHitData;
 
    int m_MatchedStrip;
 
  };
 
}