release-05-01-25/doxygen/SVDPerformanceModule_8h_source.html

/**************************************************************************

 * BASF2 (Belle Analysis Framework 2)                                     *

 * Copyright(C) 2011 - Belle II Collaboration                             *

 *                                                                        *

 * Author: The Belle II Collaboration                                     *

 * Contributor: Giulia Casarosa                                           *

 *                                                                        *

 * This software is provided "as is" without any warranty.                *

 **************************************************************************/


#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;

    virtual void terminate() 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 = NULL);

    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 = NULL);


  };

}