V0findingPerformanceEvaluationModule.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 <tracking/modules/trackingPerformanceEvaluation/PerformanceEvaluationBaseClass.h>
#include <tracking/dataobjects/V0ValidationVertex.h>
#include <mdst/dataobjects/MCParticle.h>
 
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
 
 
// forward declarations
namespace Belle2 {

  class V0findingPerformanceEvaluationModule : public Module, PerformanceEvaluationBaseClass {
 
  public:

    V0findingPerformanceEvaluationModule();

    ~V0findingPerformanceEvaluationModule();

    void initialize() override;

    void beginRun() override;

    void event() override;

    void endRun() override;

    void terminate() override;
 
  private:
 
    bool isV0(const MCParticle& the_mcParticle); 
 
    int nMatchedDaughters(const MCParticle& the_mcParticle); 
 
    /* user-defined parameters */
    std::string m_MCParticlesName; 
    std::string m_V0sName; 
 
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<V0ValidationVertex> m_V0ValidationVertices;  
 
    /* list of histograms filled per MCParticle found in the event */
 
    TH1F* m_multiplicityV0s = nullptr; 
    TH1F* m_multiplicityMCParticles = nullptr;  
    TH1F* m_MCParticlePDGcode = nullptr;  
 
    //error on track parameters
    TH1F* m_h1_vtxX_err = nullptr; 
    TH1F* m_h1_vtxY_err = nullptr; 
    TH1F* m_h1_vtxZ_err = nullptr; 
    TH2F* m_h2_vtxTvsR_err = nullptr; 
    //    TH1F* m_h1_mom_err; /**< mom error*/
    //    TH1F* m_h1_mass_err; /**< mom error*/
    //residuals on track parameters
    TH1F* m_h1_vtxX_res = nullptr; 
    TH1F* m_h1_vtxY_res = nullptr; 
    TH1F* m_h1_vtxZ_res = nullptr; 
    TH1F* m_h1_mom_res = nullptr; 
    TH2F* m_h2_mom = nullptr; 
    TH1F* m_h1_mass_res = nullptr; 
    TH2F* m_h2_mass = nullptr; 
    //pulls on track parameters
    TH1F* m_h1_vtxX_pll = nullptr; 
    TH1F* m_h1_vtxY_pll = nullptr; 
    TH1F* m_h1_vtxZ_pll = nullptr; 
    //    TH1F* m_h1_mom_pll; /**< mom pull*/
    //    TH1F* m_h1_mass_pll; /**< mom pull*/
 
    TH1F* m_h1_ChiSquare = nullptr; 
 
    TH1F* m_h1_nMatchedDau = nullptr; 
 
    //histograms used for efficiency plots
    TH3F* m_h3_MCParticle = nullptr; 
    TH3F* m_h3_V0sPerMCParticle = nullptr; 
    TH1F* m_h1_MCParticle_R = nullptr; 
    TH1F* m_h1_V0sPerMCParticle_R = nullptr; 
 
    //histograms used for purity plots
    TH3F* m_h3_V0s = nullptr; 
    TH3F* m_h3_MCParticlesPerV0 = nullptr; 
 
  };

} // end of namespace