KlongValidationModule.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 <framework/datastore/StoreArray.h>
 
#include <TFile.h>
#include <string>
#include <TH1F.h>
#include <TGraph.h>
 
namespace Belle2 {
  class KLMCluster;
  class MCParticle;

  class KlongValidationModule : public Module {
 
  public:

    KlongValidationModule();

    virtual ~KlongValidationModule();

    virtual void initialize() override;

    virtual void beginRun() override;

    virtual void event() override;

    virtual void endRun() override;

    virtual void terminate() override;
 
  protected:
 
  private:

    double m_momentum{ -1.};
    double m_theta{ -1.};
    double m_phi{ -1.};
    double m_isKl{ -1.};
    bool   m_passed{false};
    bool   m_faked{false};
    bool   m_reconstructedAsKl{false};
    float m_KlIdCut{ -1};

    bool m_isBeamBKG{false};
    // use TH1F histogramms to calculate efficiency
    TH1F* m_effPhi{nullptr};
    TH1F* m_effTheta{nullptr};
    TH1F* m_effMom{nullptr};
    TH1F* m_fakePhi{nullptr};
    TH1F* m_fakeTheta{nullptr};
    TH1F* m_fakeMom{nullptr};
    TH1F* m_Phi_Pass{nullptr};
    TH1F* m_Theta_Pass{nullptr};
    TH1F* m_Mom_Pass{nullptr};

    TH1F* m_Phi_all{nullptr};
    TH1F* m_Theta_all{nullptr};
    TH1F* m_Mom_all{nullptr};
    TH1F* m_Mom_all_plot{nullptr};
    TH1F* m_fakePhi_Pass{nullptr};
    TH1F* m_fakeTheta_Pass{nullptr};
    TH1F* m_fakeMom_Pass{nullptr};
    TH1F* m_time{nullptr};
    TH1F* m_trackSep{nullptr};
    TH1F* m_nLayer{nullptr};
    TH1F*  m_innermostLayer{nullptr};
    TH1F*  m_trackFlag{nullptr};
    TH1F*  m_ECLFlag{nullptr};
    TH1F*  m_bkgPhi{nullptr};
    TH1F*  m_bkgTheta{nullptr};
    TH1F*  m_bkgMom{nullptr};
    TH1F*  m_klidFake{nullptr};
    TH1F*  m_klidTrue{nullptr};
    TH1F*  m_klidAll{nullptr};
    TGraph*  m_ROC{nullptr};
    TGraph*  m_backRej{nullptr};
 

    StoreArray<KLMCluster> m_klmClusters;
    StoreArray<MCParticle> m_mcParticles;

    const std::vector<double> m_xbins =
    {0, 0.001, 0.01, 0.1, 0.15, 0.175, 0.2, 0.3, 0.4, 0.5, 1};

    std::string m_outputName = "K_long_full_validation_sample.root";

    std::string m_contact = "";

    TFile* m_f = nullptr;
 
  }; // end class

} // end namespace Belle2