SVDPerformanceModule.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 <svd/dataobjects/SVDCluster.h>
#include <svd/dataobjects/SVDShaperDigit.h>
#include <svd/dataobjects/SVDRecoDigit.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <svd/calibration/SVDPulseShapeCalibrations.h>
#include <svd/calibration/SVDNoiseCalibrations.h>
#include <svd/dataobjects/SVDEventInfo.h>
 
#include <string>
#include <TTree.h>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TList.h>
 
namespace Belle2 {


  class SVDPerformanceModule : public Module {
 
  public:

    SVDPerformanceModule();

    virtual ~SVDPerformanceModule() {};
 
    virtual void initialize() override; 
    virtual void beginRun() override; 
    virtual void event() override; 
    virtual void endRun() override; 
 
    std::string m_ShaperDigitName = "SVDShaperDigits"; 
    std::string m_RecoDigitName = "SVDRecoDigits"; 
    std::string m_ClusterName = "SVDClusters"; 
    std::string m_TrackFitResultName = "TrackFitResults"; 
    std::string m_TrackName = "Tracks"; 
    bool m_is2017TBanalysis = false; 
    bool m_isSimulation = false; 
 
    float m_debugLowTime = - 100; 
 
 
    /* user-defined parameters */
    std::string m_rootFileName = "";   
 
    /* ROOT file related parameters */
    TFile* m_rootFilePtr = nullptr; 
 
  private:
 
    int m_nEvents = 0; 
 
    SVDNoiseCalibrations m_NoiseCal; 
    SVDPulseShapeCalibrations m_PulseShapeCal; 
 
    StoreArray<SVDShaperDigit> m_svdShapers; 
    StoreArray<SVDRecoDigit> m_svdRecos; 
    StoreArray<SVDCluster> m_svdClusters; 
    StoreArray<RecoTrack> m_recoTracks; 
    StoreArray<Track> m_Tracks; 
    StoreArray<TrackFitResult> m_tfr; 

    StoreObjPtr<SVDEventInfo> m_storeSVDEvtInfo;

    std::string m_svdEventInfoName;
 
    int m_ntracks = 0; 
 
    static const int m_nLayers = 4; 
    static const int m_nSensors = 5; 
    static const int m_nSides = 2; 
 
    unsigned int sensorsOnLayer[4] = {0}; 
 
    TList* m_histoList_track = nullptr;  
    TList* m_histoList_corr = nullptr; 
    TList* m_histoList_clTRK[m_nLayers] = {nullptr}; 
    TList* m_histoList_cluster[m_nLayers] = {nullptr}; 
    TList* m_histoList_shaper[m_nLayers] = {nullptr}; 
    TList* m_histoList_reco[m_nLayers] = {nullptr}; 
 
    //TRACKS
    TH1F* m_nTracks = nullptr; 
    TH1F* m_Pvalue = nullptr; 
    TH1F* m_mom = nullptr; 
    TH1F* m_nSVDhits = nullptr; 
 
    //SHAPER
    TH1F* h_nShaper[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
 
    //RECO
    TH1F* h_nReco[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_recoCharge[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_recoEnergy[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_stripNoise[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH1F* h_recoTime[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
 
    //CLUSTERS NOT RELATED TO TRACKS
    TH1F* h_nCl[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clCharge[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clEnergy[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clSeedMaxbin[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSMaxbin[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSCoorU[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSCoorV[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clNuVSNv[m_nLayers][m_nSensors] = {{nullptr}}; 
    TH2F* h_clCoorUVSCoorV[m_nLayers][m_nSensors] = {{nullptr}}; 
    TH1F* h_clCoor1VSCoor2[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergy12VSdelta[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clCellID1VSCellID2[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyUVSEnergyV[m_nLayers][m_nSensors] = {{nullptr}}; 
    TH1F* h_clSN[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_clTime[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH2F* h_clChargeVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSSize_mb12[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSSize_mb345[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clEnergyVSSize_mb6[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clSNVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clTimeVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_clTimeVSTrueTime[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
 
    //CLUSTERS RELATED TO TRACKS
    TH1F* h_nCltrk[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_cltrkSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_cltrkCharge[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_cltrkEnergy[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_cltrkSN[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_cltrkTime[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH1F* h_cltrkTime_TB1[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH1F* h_cltrkTime_TB2[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH1F* h_cltrkTime_TB3[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH1F* h_cltrkTime_TB4[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
    TH2F* h_cltrkChargeVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_cltrkSNVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_cltrkTimeVSSize[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH2F* h_cltrkTimeVSTrueTime[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}};  
 
    //1-STRIP CLUSTERS
    TH1F* h_1cltrkCharge[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_1cltrkSN[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
 
    //2-STRIP CLUSTERS
    TH1F* h_2cltrkCharge[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
    TH1F* h_2cltrkSN[m_nLayers][m_nSensors][m_nSides] = {{{nullptr}}}; 
 
    //CORRELATIONS
    TH1F* h_cltrk_UU = nullptr; 
    TH1F* h_cltrk_VV = nullptr; 
    TH1F* h_cltrk_UV = nullptr; 
 
    TH2F* h_cltrkTime_L4uL5u = nullptr; 
    TH2F* h_cltrkTime_L4vL5v = nullptr; 
    TH2F* h_cltrkTime_L5uL5v = nullptr; 

    int getSensor(int layer, int sensor, bool isTB)
    {
      int result = 0;
      if (isTB) {
        if (layer == 0)
          result = sensor - 1;
        else if (layer == 1 || layer == 2)
          result = sensor - 2;
        else if (layer == 3)
          result = sensor - 3;
      } else result = sensor - 1;
 
      return result;
    }

    TH1F* createHistogram1D(const char* name, const char* title,
                            Int_t nbins, Double_t min, Double_t max,
                            const char* xtitle, TList* histoList = nullptr);
    TH2F* createHistogram2D(const char* name, const char* title,
                            Int_t nbinsX, Double_t minX, Double_t maxX, const char* titleX,
                            Int_t nbinsY, Double_t minY, Double_t maxY, const char* titleY,
                            TList* histoList = nullptr);
 
  };

}