ECLChargedPIDDataAnalysisValidationModule.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
 
// Belle 2
#include <mdst/dataobjects/MCParticle.h>
 
// ROOT
#include <TFile.h>
#include <TH1F.h>
#include <TTree.h>
 
// FRAMEWORK
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/gearbox/Const.h>
 
#include <set>
 
namespace Belle2 {

  class ECLChargedPIDDataAnalysisValidationModule : public Module {
 
  public:

    ECLChargedPIDDataAnalysisValidationModule();

    virtual ~ECLChargedPIDDataAnalysisValidationModule() override;

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;
 
  private:

    static constexpr float c_PID = 0.5;

    static constexpr unsigned int c_chargedStableHypos = 2 * Const::ChargedStable::c_SetSize;

    std::vector<int> m_inputPdgIdList;

    std::vector<unsigned int> m_mergeChargeOfPdgIds;

    std::map<Const::ChargedStable, bool> m_mergeChargeFlagByHypo;

    std::set<int> m_inputPdgIdSet;

    std::vector<TFile*> m_outputFile = std::vector<TFile*>(c_chargedStableHypos);

    std::string m_outputFileName;

    bool m_saveValidationTree;

    std::vector<TTree*> m_tree = std::vector<TTree*>(c_chargedStableHypos);

    std::vector<float> m_p = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_pt = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_trkTheta = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_trkPhi = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_clusterTheta = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_clusterReg = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_clusterPhi = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_trackClusterMatch = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_logl_sig = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_logl_bkg = std::vector<float>(c_chargedStableHypos);

    std::vector<float> m_deltalogl_sig_bkg = std::vector<float>(c_chargedStableHypos);

    std::vector<std::vector<float>> m_pids_glob = std::vector<std::vector<float>>(c_chargedStableHypos,
                                                  std::vector<float>(Const::ChargedStable::c_SetSize));

    std::vector<float> m_p_binedges = {0.0, 0.5, 0.75, 1.0, 3.0, 5.0};

    std::vector<float> m_th_binedges = {0.0, 0.2164208, 0.385, 0.561996, 1.13, 1.57, 1.88, 2.2462387, 2.47, 2.7070057, 3.1415926};

    StoreArray<MCParticle> m_MCParticles;

    static void dumpPIDVars(TTree* sampleTree, const Const::ChargedStable& sigHypo, const int sigCharge,
                            const Const::ChargedStable& bkgHypo,
                            bool mergeSigCharge = false);

    void dumpPIDEfficiencyFakeRate(TTree* sampleTree, const Const::ChargedStable& sampleHypo, const int sampleCharge,
                                   const Const::ChargedStable& sigHypo, bool mergeSampleCharge = false);

    void dumpTrkClusMatchingEfficiency(TTree* sampleTree, const Const::ChargedStable& sampleHypo, const int sampleCharge,
                                       bool mergeSampleCharge = false);

    static bool isValidChargedPdg(const int pdg)
    {
      return (Const::chargedStableSet.find(pdg) != Const::invalidParticle);
    }

    static void paintUnderOverflow(TH1F* h);
 
  };

}