KlongValidationModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jo-Frederik Krohn                                        *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/KLMCluster.h>
 
#include <TFile.h>
#include <string>
#include <TH1F.h>
#include <TGraph.h>
 
namespace Belle2 {
  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_energy{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";
 
    TFile* m_f = nullptr;
 
  }; // end class
 
} // end namespace Belle2