ECLTrackClusterMatchingPerformanceModule.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/dataobjects/EventMetaData.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <tracking/modules/standardTrackingPerformance/ParticleProperties.h>
 
// forward declarations
class TTree;
class TFile;
 
namespace Belle2 {


  class ECLTrackClusterMatchingPerformanceModule : public Module {
  public:
    ECLTrackClusterMatchingPerformanceModule();

    void initialize() override;

    void event() override;

    void terminate() override;
 
  private:
    std::string m_outputFileName; 
    double m_minClusterEnergy; 
    double m_minWeight; 
    std::string m_trackClusterRelationName; 
 
    // Required input
    StoreArray<ECLCluster> m_eclClusters; 
    StoreArray<MCParticle> m_mcParticles; 
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 

    StoreObjPtr<EventMetaData> m_EventMetaData;
 
    TFile* m_outputFile{nullptr}; 
    TTree* m_tracksTree{nullptr}; 
    TTree* m_clusterTree{nullptr}; 

    ParticleProperties m_trackProperties;

    int m_iExperiment = 0;

    int m_iRun = 0;

    int m_iEvent = 0;

    double m_pValue = 0.0;

    int m_charge = 0;

    double m_d0 = 0.0;

    double m_z0 = 0.0;

    int m_lastCDCLayer = 0;

    int m_matchedPhotonHypothesisClusterDetectorRegion = 0;

    double m_matchedPhotonHypothesisClusterTheta = 0.0;

    double m_matchedPhotonHypothesisClusterPhi = 0.0;

    double m_matchedPhotonHypothesisClusterMinTrkDistance = 0.0;

    double m_matchedPhotonHypothesisClusterDeltaL = 0.0;

    int m_matchedHadronHypothesisClusterDetectorRegion = 0;

    double m_matchedHadronHypothesisClusterTheta = 0.0;

    double m_matchedHadronHypothesisClusterPhi = 0.0;

    double m_matchedHadronHypothesisClusterMinTrkDistance = 0.0;

    double m_matchedHadronHypothesisClusterDeltaL = 0.0;

    int m_mcparticle_cluster_detectorregion = 0;

    double m_mcparticle_cluster_theta = 0.0;

    double m_mcparticle_cluster_phi = 0.0;

    double m_mcparticle_cluster_energy = 0.0;

    int m_mcparticle_cluster_match = 0;

    int m_matchedToPhotonHypothesisECLCluster = 0;

    int m_matchedToHadronHypothesisECLCluster = 0;

    int m_sameclusters = 0;

    double m_clusterPhi = 0.0;

    double m_clusterTheta = 0.0;

    int m_clusterHypothesis = 0;

    int m_clusterIsTrack = 0;

    int m_clusterIsPhoton = 0;

    int m_clusterIsChargedStable = 0;

    double m_clusterEnergy = 0.0;

    double m_photonEnergy = 0.0;

    double m_clusterE1E9 = 0.0;

    int m_clusterDetectorRegion = 0;

    double m_clusterErrorTiming = 0.0;

    double m_clusterMinTrkDistance = 0.0;

    double m_clusterDeltaL = 0.0;

    void setVariablesToDefaultValue();

    void setClusterVariablesToDefaultValue();

    void setupTree();

    void writeData();

    void addVariableToTree(const std::string& varName, double& varReference, TTree* tree);

    void addVariableToTree(const std::string& varName, int& varReference, TTree* tree);

    bool isPrimaryMcParticle(const MCParticle& mcParticle);

    bool isChargedStable(const MCParticle& mcParticle);
  };

} // end of namespace