SVDClusterEvaluationModule.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.                  *
 **************************************************************************/
 
#ifndef SVD_CLUSTEREVALUATION_H_
#define SVD_CLUSTEREVALUATION_H_
 
#include <framework/core/Module.h>
#include <vxd/geometry/GeoCache.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <svd/dataobjects/SVDCluster.h>
#include <svd/dataobjects/SVDHistograms.h>
#include <tracking/dataobjects/SVDIntercept.h>
#include <mdst/dataobjects/Track.h>
#include <framework/dataobjects/EventMetaData.h>
 
 
#include <string>
#include <TTree.h>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
 
namespace Belle2 {
  class SVDClusterEvaluationModule : public Module {
 
  public:
 
    SVDClusterEvaluationModule();
 
    virtual ~SVDClusterEvaluationModule();
    virtual void initialize() override;
    virtual void beginRun() override;
    virtual void event() override;
    virtual void endRun() override;
    virtual void terminate() override;
 
    /* user-defined parameters */
    std::string m_rootFileName;   
    std::string m_ClusterName;   
    std::string m_InterceptName;   
    std::string m_TrackName;   
    /* ROOT file related parameters */
    TFile* m_rootFilePtr = nullptr; 
    TTree* m_tree = nullptr; 
    TTree* m_treeSummary = nullptr; 
    double m_UbinWidth; 
    double m_VbinWidth;  
    //branches
    TBranch* b_experiment = nullptr; 
    TBranch* b_run = nullptr; 
    TBranch* b_ladder = nullptr; 
    TBranch* b_layer = nullptr; 
    TBranch* b_sensor = nullptr; 
    TBranch* b_interU = nullptr; 
    TBranch* b_interV = nullptr; 
    TBranch* b_interErrU = nullptr; 
    TBranch* b_interErrV = nullptr; 
    TBranch* b_interUprime = nullptr; 
    TBranch* b_interVprime = nullptr; 
    TBranch* b_interErrUprime = nullptr; 
    TBranch* b_interErrVprime = nullptr; 
    TBranch* b_residU = nullptr; 
    TBranch* b_residV = nullptr; 
    TBranch* b_clUpos = nullptr; 
    TBranch* b_clVpos = nullptr; 
    TBranch* b_clUcharge = nullptr; 
    TBranch* b_clVcharge = nullptr; 
    TBranch* b_clUsnr = nullptr; 
    TBranch* b_clVsnr = nullptr; 
    TBranch* b_clUsize = nullptr; 
    TBranch* b_clVsize = nullptr; 
    TBranch* b_clUtime = nullptr; 
    TBranch* b_clVtime = nullptr; 
    //branch variables
    int m_experiment = -1; 
    int m_run = -1; 
    int m_ladder = -1; 
    int m_layer = -1; 
    int m_sensor = -1; 
    float m_interU = -1; 
    float m_interV = -1; 
    float m_interErrU = -1; 
    float m_interErrV = -1; 
    float m_interUprime = -1; 
    float m_interVprime = -1; 
    float m_interErrUprime = -1; 
    float m_interErrVprime = -1; 
    float m_residU = -1; 
    float m_residV = -1; 
    float m_clUpos = -1; 
    float m_clVpos = -1; 
    float m_clUcharge = -1; 
    float m_clVcharge = -1; 
    float m_clUsnr = -1; 
    float m_clVsnr = -1; 
    int m_clUsize = -1; 
    int m_clVsize = -1; 
    float m_clUtime = -1; 
    float m_clVtime = -1; 
    //branches summary variables
    TBranch* bs_experiment = nullptr; 
    TBranch* bs_run = nullptr; 
    TBranch* bs_ladder = nullptr; 
    TBranch* bs_layer = nullptr; 
    TBranch* bs_sensor = nullptr; 
    TBranch* bs_effU = nullptr; 
    TBranch* bs_effV = nullptr; 
    TBranch* bs_effErrU = nullptr; 
    TBranch* bs_effErrV = nullptr; 
    TBranch* bs_nIntercepts = nullptr; 
    TBranch* bs_residU = nullptr; 
    TBranch* bs_residV = nullptr; 
    TBranch* bs_misU = nullptr; 
    TBranch* bs_misV = nullptr; 
    TBranch* bs_statU = nullptr; 
    TBranch* bs_statV = nullptr; 
    //branch variables
    int ms_experiment = -1; 
    int ms_run = -1; 
    int ms_ladder = -1; 
    int ms_layer = -1; 
    int ms_sensor = -1; 
    float ms_effU = -1; 
    float ms_effV = -1; 
    float ms_effErrU = -1; 
    float ms_effErrV = -1; 
    int ms_nIntercepts = -1; 
    float ms_residU = -1; 
    float ms_residV = -1; 
    float ms_misU = -1; 
    float ms_misV = -1; 
    float ms_statU = -1; 
    float ms_statV = -1; 
  private:
 
    double m_cmTomicron = 10000; 
    int m_theLayer = -1;  
    double m_interSigmaMax = -1;  
    double m_uFiducial = -1;  
    double m_vFiducial = -1;  
    float m_nSigma = -1;  
    float m_halfWidth = -1;  
    int m_groupNstrips = -1; 
    VXD::GeoCache& m_geoCache = VXD::GeoCache::getInstance(); 
    StoreObjPtr<EventMetaData> m_eventMetaData; 
    StoreArray<SVDCluster> m_svdClusters; 
    StoreArray<Track> m_tracks; 
    StoreArray<SVDIntercept> m_svdIntercepts; 
    SVDHistograms<TH2F>* m_interCoor; 
    SVDHistograms<TH1F>* m_interSigma; 
    SVDHistograms<TH1F>* m_clsCoor; 
    SVDHistograms<TH1F>* m_clsResid; 
    SVDHistograms<TH1F>* m_clsMinResid; 
    SVDHistograms<TH2F>* m_clsResid2D; 
    float m_width_LargeS_U = 5.772; 
    float m_width_LargeS_V = 12.290; 
    float m_width_SmallS_U = 3.855; 
    float m_width_SmallS_V = m_width_LargeS_V;  
    float m_safety_margin = 0.2; 
    int m_nBins_LargeS_U = 100; 
    int m_nBins_LargeS_V = 100;  
    int m_nBins_SmallS_U = 100;  
    int m_nBins_SmallS_V = 100;  
    float m_abs_LargeS_U = 1; 
    float m_abs_LargeS_V = 1; 
    float m_abs_SmallS_U = 1; 
    float m_abs_SmallS_V = 1;  
    void create_SVDHistograms_interCoor(); 
    void create_SVDHistograms_interSigma(); 
    void create_SVDHistograms_clsCoor(); 
    void create_SVDHistograms_clsResid(); 
    double getOneSigma(TH1F* h); 
    bool isRelatedToTrack(SVDIntercept* inter); 
  };
}
 
#endif /* SVDClusterEvaluationModule_H_ */