SVDROIFinderAnalysisModule.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/datastore/StoreArray.h>
 
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TGraphErrors.h>
 
#include <string>
 
namespace Belle2 {
  class ROIid;
  class SVDIntercept;
  class SVDShaperDigit;
  class RecoTrack;
  class MCParticle;
 
  class SVDROIFinderAnalysisModule : public Module {
 
  public:
 
    SVDROIFinderAnalysisModule();
 
    ~SVDROIFinderAnalysisModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
    void endRun() override;
 
    void terminate() override;
 
  private:
 
    StoreArray<SVDShaperDigit> m_shapers; 
    StoreArray<ROIid> m_ROIs; 
    StoreArray<RecoTrack> m_trackList;
    StoreArray<SVDIntercept> m_SVDIntercepts; 
    StoreArray<MCParticle> m_mcParticles; 
    bool m_isSimulation; 
    std::string m_shapersName; 
    std::string m_recoTrackListName; 
    std::string m_SVDInterceptListName; 
    std::string m_ROIListName; 
    TFile* m_rootFilePtr; 
    std::string m_rootFileName; 
    bool m_writeToRoot; 
    int m_rootEvent;   
    //graphs & histos - results
    Double_t pt[6]  = {0.05, 0.15, 0.25, 0.4, 0.75, 1.5}; 
    Double_t ptErr[6] = { 0.05, 0.05, 0.05, 0.1, 0.25, 0.5}; 
    TGraphErrors* m_gEff2; 
    TGraphErrors* m_gEff; 
    TH1F* m_h1DigitsPerParticle; 
    TH1F* m_h1RecoTracksPerParticle; 
    TH1F* m_h1digiIn; 
    TH1F* m_h1digiOut2; 
    TH1F* m_h1digiOut3; 
    TH1F* m_h1digiOut4; 
    TH1F* m_h1digiOut5; 
    TH1F* m_h1TrackOneDigiIn; 
    TH1F* m_h1nnotINtrack2; 
    TH1F* m_h1nnotINtrack3; 
    TH1F* m_h1nnotINtrack4; 
    TH1F* m_h1nnotINtrack5; 
    TH1F* m_h1Track; 
    TH1F* m_h1Track_pt; 
    TH1F* m_h1Track_phi; 
    TH1F* m_h1Track_lambda; 
    TH1F* m_h1Track_cosTheta; 
    TH1F* m_h1Track_pVal; 
    TH1F* m_h1Track_nSVDhits; 
    TH1F* m_h1Track_nCDChits; 
    TH1F* m_h1INtrack1; 
    TH1F* m_h1INtrack1_pt; 
    TH1F* m_h1INtrack1_phi; 
    TH1F* m_h1INtrack1_lambda; 
    TH1F* m_h1INtrack1_cosTheta; 
    TH1F* m_h1INtrack1_pVal; 
    TH1F* m_h1INtrack1_nSVDhits; 
    TH1F* m_h1INtrack1_nCDChits; 
    TH1F* m_h1notINtrack5; 
    TH1F* m_h1notINtrack5_pt; 
    TH1F* m_h1notINtrack5_phi; 
    TH1F* m_h1notINtrack5_lambda; 
    TH1F* m_h1notINtrack5_cosTheta; 
    TH1F* m_h1notINtrack5_pVal; 
    TH1F* m_h1notINtrack5_nSVDhits; 
    TH1F* m_h1notINtrack5_nCDChits; 
    //fill digits inside ROIs
    TH1F* m_h1PullU; 
    TH1F* m_h1PullV; 
    TH2F* m_h2sigmaUphi; 
    TH2F* m_h2sigmaVphi; 
    TH1F* m_h1ResidU; 
    TH1F* m_h1ResidV; 
    TH1F* m_h1SigmaU; 
    TH1F* m_h1SigmaV; 
    TH1F* m_h1GlobalTime; 
    //fill digits outside2 ROIs
    TH2F* m_h2sigmaUphi_out2; 
    TH2F* m_h2sigmaVphi_out2; 
    TH1F* m_h1ResidU_out2; 
    TH1F* m_h1ResidV_out2; 
    TH1F* m_h1SigmaU_out2; 
    TH1F* m_h1SigmaV_out2; 
    TH1F* m_h1GlobalTime_out2; 
    //fill digits outside3 ROIs
    TH2F* m_h2sigmaUphi_out3; 
    TH2F* m_h2sigmaVphi_out3; 
    TH1F* m_h1ResidU_out3; 
    TH1F* m_h1ResidV_out3; 
    TH1F* m_h1SigmaU_out3; 
    TH1F* m_h1SigmaV_out3; 
    TH1F* m_h1GlobalTime_out3; 
    //fill digits outside4 ROIs
    TH2F* m_h2sigmaUphi_out4; 
    TH2F* m_h2sigmaVphi_out4; 
    TH1F* m_h1SigmaU_out4; 
    TH1F* m_h1SigmaV_out4; 
    TH1F* m_h1GlobalTime_out4; 
    //fill digits outside5 ROIs
    TH1F* m_h1GlobalTime_out5; 
    //ROI stuff
    TH2F* m_h2ROIbottomLeft; 
    TH2F* m_h2ROItopRight; 
    TH2F* m_h2ROIuMinMax; 
    TH2F* m_h2ROIvMinMax; 
    TH1F* m_h1totROIs; 
    TH1F* m_h1okROIs; 
    TH1F* m_h1totUstrips; 
    TH1F* m_h1totVstrips; 
    TH1F* m_h1effPerTrack; 
    //variables
    double m_globalTime; 
    double m_coorU; 
    double m_coorV; 
    double m_sigmaU; 
    double m_sigmaV; 
    int m_vxdID; 
    double m_coormc; 
    int m_idmc; 
    int m_vxdIDmc; 
    double m_pTmc; 
    double m_momXmc; 
    double m_momYmc;
    double m_momZmc; 
    double m_thetamc; 
    double m_costhetamc; 
    double m_phimc;  
    double m_lambdamc;  
    unsigned int Ntrack; 
    unsigned int NtrackHit; 
    unsigned int n_notINtrack2; 
    unsigned int n_notINtrack3; 
    unsigned int n_notINtrack4; 
    unsigned int n_notINtrack5; 
    unsigned int n_rois; 
    unsigned int n_OKrois; 
    unsigned int m_nGoodROIs; 
    unsigned int n_intercepts; 
    unsigned int n_tracks; 
    unsigned int n_tracksWithDigits; 
    unsigned int n_tracksWithDigitsInROI; 
    unsigned int n_svdDigit; 
    unsigned int n_svdDigitInROI; 
    unsigned int n_notINdigit2; 
    unsigned int n_notINdigit3;
    unsigned int n_notINdigit4; 
    unsigned int n_notINdigit5; 
    unsigned int nsvdDigit[6]; 
    unsigned int nsvdDigitInROI[6]; 
    unsigned int nnotINdigit2[6]; 
    unsigned int nnotINdigit3[6]; 
    unsigned int nnotINdigit4[6];  
    unsigned int nnotINdigit5[6];  
    unsigned int TrackOneDigiIn[6]; 
    unsigned int nnotINtrack2[6];  
    unsigned int nnotINtrack3[6];   
    unsigned int nnotINtrack4[6];   
    unsigned int nnotINtrack5[6];   
  };
 
}