SVDClusterEvaluationTrueInfoModule.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 SVDClusterEvaluationTrueInfo_H
#define SVDClusterEvaluationTrueInfo_H
 
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreObjPtr.h>
#include <svd/dataobjects/SVDTrueHit.h>
#include <svd/dataobjects/SVDEventInfo.h>
#include <TList.h>
#include <TFile.h>
#include <TH1F.h>
#include <TH2F.h>
 
 
namespace Belle2 {
  class SVDClusterEvaluationTrueInfoModule : public Module {
 
  public:
 
    SVDClusterEvaluationTrueInfoModule();
 
    virtual ~SVDClusterEvaluationTrueInfoModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
 
    static const int m_Nsets =
      6; 
    static const int m_NsetsRed = 3; 
    //  private:
    TFile* m_outputFile = nullptr; 
    std::string m_outputFileName = ""; 
    std::string m_svdEventInfoName;
 
    int indexForHistosAndGraphs = 0;
 
    TString NameOfHisto = "";
    TString  TitleOfHisto = "";
 
    TH1F* m_histo_StripTimeResolution[m_Nsets] = {nullptr};
    TH1F* m_histo_ClusterTimeResolution[m_Nsets] = {nullptr};
    TH1F* m_histo_ClusterTimeResolution_bin1[m_Nsets] = {nullptr}; 
    TH1F* m_histo_ClusterTimeResolution_bin2[m_Nsets] = {nullptr}; 
    TH1F* m_histo_ClusterTimeResolution_bin3[m_Nsets] = {nullptr}; 
    TH1F* m_histo_ClusterTimeResolution_bin4[m_Nsets] = {nullptr}; 
    TH1F* m_histo_ClusterTimePull[m_Nsets] = {nullptr};
    TH1F* m_histo_ClusterUPositionResolution[m_NsetsRed] = {nullptr};
    TH1F* m_histo_ClusterVPositionResolution[m_NsetsRed] = {nullptr};
    TH1F* m_histo_ClusterUPositionPull[m_NsetsRed] = {nullptr};
    TH1F* m_histo_ClusterVPositionPull[m_NsetsRed] = {nullptr};
    TH2F* m_histo2D_TresVsPosres[m_Nsets] = {nullptr};
    TH1F* m_histo_PurityInsideTMCluster[m_Nsets] = {nullptr};
    TH2F* m_histo2D_PurityInsideTMCluster[m_Nsets] = {nullptr};
    TH1F* m_histo_PurityInsideNOTMCluster[m_Nsets] = {nullptr};
    TH1F* m_histo_THinCluster[m_Nsets] = {nullptr};
    TH1F* m_histo_THinClusterTM[m_Nsets] = {nullptr};
    TH1F* m_histo_GoodTHinClusterTM[m_Nsets] = {nullptr};
    TH1F* m_histo_GoodTHinClusterTMGood[m_Nsets] = {nullptr};
 
    float m_mean_StripTimeResolution[m_Nsets] = {0}; 
    float m_RMS_StripTimeResolution[m_Nsets] = {0}; 
    float m_mean_ClusterTimeResolution[m_Nsets] = {0}; 
    float m_RMS_ClusterTimeResolution[m_Nsets] = {0}; 
    float m_mean_ClusterUPositionResolution[m_Nsets] = {0}; 
    float m_RMS_ClusterUPositionResolution[m_Nsets] = {0}; 
    float m_mean_ClusterVPositionResolution[m_Nsets] = {0}; 
    float m_RMS_ClusterVPositionResolution[m_Nsets] = {0}; 
    float m_mean_PurityInsideTMCluster[m_Nsets] = {0}; 
    float m_RMS_PurityInsideTMCluster[m_Nsets] = {0}; 
    float m_mean_THinCluster[m_Nsets] = {0}; 
    float m_RMS_THinCluster[m_Nsets] = {0};
    float m_mean_THinClusterTM[m_Nsets] = {0}; 
    float m_RMS_THinClusterTM[m_Nsets] = {0}; 
    float m_mean_GoodTHinClusterTM[m_Nsets] = {0}; 
    float m_RMS_GoodTHinClusterTM[m_Nsets] = {0}; 
    float m_mean_GoodTHinClusterTMGood[m_Nsets] = {0}; 
    float m_RMS_GoodTHinClusterTMGood[m_Nsets] = {0}; 
    float m_OrderingVec[m_Nsets] = {1., 2., 3., 4., 5., 6.};
    float m_NullVec[m_Nsets] = {0.}; 
    TList* m_histoList_StripTimeResolution = nullptr; 
    TList* m_histoList_ClusterTimeResolution = nullptr; 
    TList* m_histoList_ClusterTimePull = nullptr; 
    TList* m_histoList_ClusterPositionResolution = nullptr; 
    TList* m_histoList_ClusterPositionPull = nullptr; 
    TList* m_histo2DList_TresVsPosres = nullptr; 
    TList* m_histoList_PurityInsideTMCluster = nullptr; 
    TList* m_histo2DList_PurityInsideTMCluster = nullptr; 
    TList* m_histoList_PurityInsideNOTMCluster = nullptr; 
    TList* m_histoList_THinCluster = nullptr; 
    TList* m_histoList_THinClusterTM = nullptr; 
    TList* m_histoList_GoodTHinClusterTM = nullptr; 
    TList* m_histoList_GoodTHinClusterTMGood = nullptr; 
    TList* m_graphList = nullptr; 
    int m_NumberOfShaperDigit[m_Nsets] = {0};
    int m_NumberOfRecoDigit[m_Nsets] = {0}; 
    int m_NumberOfTH[m_Nsets] = {0}; 
    int m_NumberOfClustersRelatedToTH[m_Nsets] = {0};  
    int m_NumberOfClusters[m_Nsets] = {0}; 
    int m_NumberOfTMClusters[m_Nsets] = {0}; 
    int m_NumberOfTMRecoInTMCluster = 0; 
    int m_NumberOfTMRecoInNOTMCluster = 0; 
    TH1F* m_histoControl_MCcharge = nullptr; 
    TH1F* m_histoControl_MCisPrimary = nullptr; 
    TH1F* m_histoControl_THToMCsize = nullptr; 
    TList* m_histoList_Control = nullptr; 
  protected:
    StoreObjPtr<SVDEventInfo> m_storeSVDEvtInfo;
 
    int indexFromLayerSensorSide(int LayerNumber, int SensorNumber, int UVNumber);
 
    TString IntExtFromIndex(int idx);
 
    TString FWFromIndex(int idx);
 
    TString UVFromIndex(int idx);
 
    TH1F* createHistogram1D(const char* name, const char* title,
                            Int_t nbins, Double_t min, Double_t max,
                            const char* xtitle, TList* histoList);
 
    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);
 
    void createEfficiencyGraph(const char* name, const char* title, int vNum[m_Nsets], int vDen[m_Nsets], TString xTitle,
                               TString yTitle, TList* list);
 
    void createArbitraryGraphErrorChooser(const char* name, const char* title, float x[m_Nsets], float xErr[m_Nsets], float y[m_Nsets],
                                          float yErr[m_Nsets], TString xTitle, TString yTitle, TList* list, int len);
 
    void createArbitraryGraphError_Std(const char* name, const char* title, float x[m_Nsets], float xErr[m_Nsets], float y[m_Nsets],
                                       float yErr[m_Nsets], TString xTitle, TString yTitle, TList* list);
 
    void createArbitraryGraphError_Red(const char* name, const char* title, float x[m_NsetsRed], float xErr[m_NsetsRed],
                                       float y[m_NsetsRed], float yErr[m_NsetsRed], TString xTitle, TString yTitle, TList* list);
 
    bool goodTrueHit(const SVDTrueHit* thino);
 
  };
}
 
 
#endif