ECLTrackClusterMatchingParametrizationExpertModule.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/dataobjects/ExtHit.h>
 
class TFile;
class TTree;
 
namespace Belle2 {
  class ECLTrackClusterMatchingParametrizationExpertModule : public Module {
  public:
    ECLTrackClusterMatchingParametrizationExpertModule();
 
    virtual ~ECLTrackClusterMatchingParametrizationExpertModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  private:
 
    bool isECLHit(const ExtHit& extHit) const;
 
    // required input
    StoreArray<ExtHit> m_extHits; 
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 
    StoreArray<ECLCluster> m_eclClusters; 
    StoreArray<MCParticle> m_mcParticles; 
    // optional input
    StoreObjPtr<EventMetaData> m_eventMetaData; 
    TFile* m_rootFilePtr{nullptr}; 
    std::string m_rootFileName; 
    bool m_writeToRoot; 
    bool m_useArray; 
    TTree* m_tree{nullptr}; 
    // variables
    int m_iExperiment; 
    int m_iRun; 
    int m_iEvent; 
    int m_trackNo; 
    float m_trackMomentum; 
    float m_pT; 
    float m_trackTheta; 
    float m_deltaPhi; 
    float m_phiCluster; 
    float m_phiHit; 
    float m_errorPhi; 
    float m_deltaTheta; 
    float m_thetaCluster; 
    float m_thetaHit; 
    float m_errorTheta; 
    int m_hitstatus; 
    int m_true_track_pdg; 
    int m_true_match; 
    std::vector<int>* m_trackNo_array = {}; 
    std::vector<float>* m_trackMomentum_array = {}; 
    std::vector<float>* m_pT_array = {}; 
    std::vector<float>* m_trackTheta_array = {}; 
    std::vector<float>* m_deltaPhi_array = {}; 
    std::vector<float>* m_phiCluster_array = {}; 
    std::vector<float>* m_phiHit_array = {}; 
    std::vector<float>* m_errorPhi_array = {}; 
    std::vector<float>* m_deltaTheta_array = {}; 
    std::vector<float>* m_thetaCluster_array = {}; 
    std::vector<float>* m_thetaHit_array = {}; 
    std::vector<float>* m_errorTheta_array = {}; 
    std::vector<int>* m_hitstatus_array = {}; 
    std::vector<int>* m_true_track_pdg_array = {}; 
    std::vector<int>* m_true_match_array = {}; 
  };
} //namespace Belle2