PXDROIFinderAnalysisModule.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.                  *
 **************************************************************************/
/* Additional Info:
* This Module is in an early stage of developement. The comments are mainly for temporal purposes
* and will be changed and corrected in later stages of developement. So please ignore them.
*/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/dataobjects/ROIid.h>
#include <tracking/dataobjects/PXDIntercept.h>
#include <string>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TGraphErrors.h>
 
namespace Belle2 {
  class PXDROIFinderAnalysisModule : public Module {
 
  public:
 
    PXDROIFinderAnalysisModule();
 
 
  private:
 
    void initialize() override final;
 
    void beginRun() override final;
 
    void event() override final;
 
    void terminate() override final;
 
    StoreArray<RecoTrack> m_recoTracks; 
    StoreArray<ROIid> m_ROIs; 
    StoreArray<PXDIntercept> m_PXDIntercepts; 
    StoreArray<MCParticle> m_MCParticles; 
    std::string m_recoTrackListName; 
    std::string m_PXDInterceptListName; 
    std::string m_ROIListName; 
    TFile* m_rootFilePtr = nullptr; 
    std::string m_rootFileName; 
    bool m_writeToRoot; 
    int m_rootEvent;   
    //pxd sensor number
    Int_t m_nSensorsL1; 
    Int_t m_nSensorsL2; 
    //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 = nullptr; 
    TGraphErrors* m_gEff = nullptr; 
    TH1F* m_h1DigitsPerParticle = nullptr; 
    TH1F* m_h1RecoTracksPerParticle = nullptr; 
    TH1F* m_h1digiIn = nullptr; 
    TH1F* m_h1digiOut2 = nullptr; 
    TH1F* m_h1digiOut3 = nullptr; 
    TH1F* m_h1digiOut4 = nullptr; 
    TH1F* m_h1digiOut5 = nullptr; 
    TH1F* m_h1TrackOneDigiIn = nullptr; 
    TH1F* m_h1nnotINtrack2 = nullptr; 
    TH1F* m_h1nnotINtrack3 = nullptr; 
    TH1F* m_h1nnotINtrack4 = nullptr; 
    TH1F* m_h1nnotINtrack5 = nullptr; 
    TH1F* m_h1Track = nullptr; 
    TH1F* m_h1Track_pt = nullptr; 
    TH1F* m_h1Track_phi = nullptr; 
    TH1F* m_h1Track_lambda = nullptr; 
    TH1F* m_h1Track_cosTheta = nullptr; 
    TH1F* m_h1Track_pVal = nullptr; 
    TH1F* m_h1Track_nSVDhits = nullptr; 
    TH1F* m_h1Track_nCDChits = nullptr; 
    TH1F* m_h1INtrack1 = nullptr; 
    TH1F* m_h1INtrack1_pt = nullptr; 
    TH1F* m_h1INtrack1_phi = nullptr; 
    TH1F* m_h1INtrack1_lambda = nullptr; 
    TH1F* m_h1INtrack1_cosTheta = nullptr; 
    TH1F* m_h1INtrack1_pVal = nullptr; 
    TH1F* m_h1INtrack1_nSVDhits = nullptr; 
    TH1F* m_h1INtrack1_nCDChits = nullptr; 
    TH1F* m_h1notINtrack5 = nullptr; 
    TH1F* m_h1notINtrack5_pt = nullptr; 
    TH1F* m_h1notINtrack5_phi = nullptr; 
    TH1F* m_h1notINtrack5_lambda = nullptr; 
    TH1F* m_h1notINtrack5_cosTheta = nullptr; 
    TH1F* m_h1notINtrack5_pVal = nullptr; 
    TH1F* m_h1notINtrack5_nSVDhits = nullptr; 
    TH1F* m_h1notINtrack5_nCDChits = nullptr; 
    //fill digits inside ROIs
    TH1F* m_h1PullU = nullptr; 
    TH1F* m_h1PullV = nullptr; 
    TH2F* m_h2sigmaUphi = nullptr; 
    TH2F* m_h2sigmaVphi = nullptr; 
    TH1F* m_h1ResidU = nullptr; 
    TH1F* m_h1ResidV = nullptr; 
    TH2F* m_h2ResidUV = nullptr; 
    TH1F* m_h1SigmaU = nullptr; 
    TH1F* m_h1SigmaV = nullptr; 
    TH1F* m_h1GlobalTime = nullptr; 
    TH2F* m_h2Mapglob = nullptr; 
    TH2F* m_h2MaplocL1 = nullptr; 
    TH2F* m_h2MaplocL2 = nullptr; 
    //fill digits outside2 ROIs
    TH2F* m_h2sigmaUphi_out2 = nullptr; 
    TH2F* m_h2sigmaVphi_out2 = nullptr; 
    TH1F* m_h1ResidU_out2 = nullptr; 
    TH1F* m_h1ResidV_out2 = nullptr; 
    TH2F* m_h2ResidUV_out2 = nullptr; 
    TH1F* m_h1SigmaU_out2 = nullptr; 
    TH1F* m_h1SigmaV_out2 = nullptr; 
    TH1F* m_h1GlobalTime_out2 = nullptr; 
    TH2F* m_h2Mapglob_out2 = nullptr; 
    TH2F* m_h2MaplocL1_out2 = nullptr; 
    TH2F* m_h2MaplocL2_out2 = nullptr; 
    //fill digits outside3 ROIs
    TH2F* m_h2sigmaUphi_out3 = nullptr; 
    TH2F* m_h2sigmaVphi_out3 = nullptr; 
    TH1F* m_h1ResidU_out3 = nullptr; 
    TH1F* m_h1ResidV_out3 = nullptr; 
    TH2F* m_h2ResidUV_out3 = nullptr; 
    TH1F* m_h1SigmaU_out3 = nullptr; 
    TH1F* m_h1SigmaV_out3 = nullptr; 
    TH1F* m_h1GlobalTime_out3 = nullptr; 
    TH2F* m_h2Mapglob_out3 = nullptr; 
    TH2F* m_h2MaplocL1_out3 = nullptr; 
    TH2F* m_h2MaplocL2_out3 = nullptr; 
    //fill digits outside4 ROIs
    TH2F* m_h2sigmaUphi_out4 = nullptr; 
    TH2F* m_h2sigmaVphi_out4 = nullptr; 
    TH1F* m_h1SigmaU_out4 = nullptr; 
    TH1F* m_h1SigmaV_out4 = nullptr; 
    TH1F* m_h1GlobalTime_out4 = nullptr; 
    TH2F* m_h2Mapglob_out4 = nullptr; 
    TH2F* m_h2MaplocL1_out4 = nullptr; 
    TH2F* m_h2MaplocL2_out4 = nullptr; 
    //fill digits outside5 ROIs
    TH1F* m_h1GlobalTime_out5 = nullptr; 
    TH2F* m_h2Mapglob_out5 = nullptr; 
    TH2F* m_h2MaplocL1_out5 = nullptr; 
    TH2F* m_h2MaplocL2_out5 = nullptr; 
    //ROI stuff
    TH2F* m_h2ROIbottomLeft = nullptr; 
    TH2F* m_h2ROItopRight = nullptr; 
    TH2F* m_h2ROIuMinMax = nullptr; 
    TH2F* m_h2ROIvMinMax = nullptr; 
    TH1F* m_h1totROIs = nullptr; 
    TH1F* m_h1okROIs = nullptr; 
    TH1F* m_h1okROIfrac = nullptr; 
    TH1F* m_h1redFactor = nullptr; 
    TH1F* m_h1redFactor_L1 = nullptr; 
    TH1F* m_h1redFactor_L2 = nullptr; 
    TH1F* m_h1totArea = nullptr; 
    TH1F* m_h1okArea = nullptr; 
    TH1F* m_h1effPerTrack = nullptr; 
    //variables
    double m_globalTime; 
    double m_coorU; 
    double m_coorV; 
    double m_sigmaU; 
    double m_sigmaV; 
    int m_vxdID; 
    double m_coorUmc; 
    double m_coorVmc; 
    int m_Uidmc; 
    int m_Vidmc; 
    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_intercepts; 
    unsigned int n_tracks; 
    unsigned int n_tracksWithDigits; 
    unsigned int n_tracksWithDigitsInROI; 
    unsigned int n_pxdDigit; 
    unsigned int n_pxdDigitInROI; 
    unsigned int n_notINdigit2; 
    unsigned int n_notINdigit3;
    unsigned int n_notINdigit4; 
    unsigned int n_notINdigit5; 
    unsigned int npxdDigit[6]; 
    unsigned int npxdDigitInROI[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]; 
  };
 
}