SVDChargeSharingAnalysisModule.h

/*************************************************************************
* BASF2 (Belle Analysis Framework 2)                                     *
* Copyright(C) 2018 - Belle II Collaboration                             *
*                                                                        *
* Author: The Belle II Collaboration                                     *
* Contributors: Mateusz Kaleta                                           *
*                                                                        *
* 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 <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TFile.h>
#include <TList.h>
#include <TCanvas.h>
 
namespace Belle2 {
  class SVDChargeSharingAnalysisModule : public Module {
 
  public:
 
    SVDChargeSharingAnalysisModule();
 
    virtual ~SVDChargeSharingAnalysisModule() override;
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
 
  private:
 
    // Input, output.
    TFile* m_outputRootFile = NULL; 
    std::string m_outputDirName; 
    std::string m_outputRootFileName; 
    // Module parameters.
    bool m_useTrackInfo = true; 
    // Store arrays.
    StoreArray<SVDCluster> m_svdClusters; 
    StoreArray<Track> m_Tracks; 
    StoreArray<RecoTrack> m_recoTracks; 
    StoreArray<TrackFitResult> m_tfr; 
    // Constants related to SVD geometry.
    static const int m_nLayers = 4; 
    static const int m_nSides = 2; 
    static const int m_nClSizes = 3; 
    static const int m_nSensorTypes = 4; 
    std::string m_nameSensorTypes[m_nSensorTypes] = {"L3", "Origami", "BWD", "FWD"}; 
    int m_nSensorsOnLayer[m_nLayers] = {2, 3, 4, 5}; 
    // Histogram related parameters.
    const Double_t m_ADUEquivalent = 375.; 
    const Int_t m_nBins = 270; 
    const Double_t m_minCharge = 0.; 
    const Double_t m_maxCharge = m_nBins * m_ADUEquivalent; 
    // tracks.
    TH1F* h_nTracks = NULL; 
    TH1F* h_TracksPvalue = NULL; 
    TH1F* h_TracksMomentum = NULL; 
    TH1F* h_TracksnSVDhits = NULL; 
    // Cluster charge.
    TH1F* h_clCharge[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    TH2F* h_clChargeVsMomentum[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    TH2F* h_clChargeVsIncidentAngle[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    TH2F* h_clChargeVsSNR[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    // Cluster size.
    TH1F* h_clSize[m_nSensorTypes][m_nSides] = {NULL}; 
    TH2F* h_clSizeVsMomentum[m_nSensorTypes][m_nSides] = {NULL}; 
    TH2F* h_clSizeVsIncidentAngle[m_nSensorTypes][m_nSides] = {NULL}; 
    TH2F* h_clSizeVsSNR[m_nSensorTypes][m_nSides] = {NULL}; 
    // cluster SNR.
    TH1F* h_clSNR[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    TH2F* h_clSNRVsMomentum[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    TH2F* h_clSNRVsIncidentAngle[m_nSensorTypes][m_nSides][m_nClSizes] = {NULL}; 
    // histogram lists.
    TList* m_histoList_Tracks = NULL;
 
    TList* m_histoList_clCharge[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clChargeVsMomentum[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clChargeVsIncidentAngle[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clChargeVsSNR[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSize[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSizeVsMomentum[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSizeVsIncidentAngle[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSizeVsSNR[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSNR[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSNRVsMomentum[m_nSensorTypes] = {NULL}; 
    TList* m_histoList_clSNRVsIncidentAngle[m_nSensorTypes] = {NULL}; 
    // utility methods
    TH1F* createHistogram1D(const char* name, const char* title,
                            Int_t nbins, Double_t min, Double_t max,
                            const char* xtitle, const char* ytitle, 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);
    TCanvas* comparisonPlot(TH1F* h1, TH1F* h2, TH1F* h3);
  };
}