SVDROIFinderAnalysisDataModule.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 <TH1F.h>
#include <TH2F.h>
 
#include <string>
#include <TFile.h>
 
namespace Belle2 {
  class ROIid;
  class SVDIntercept;
  class SVDShaperDigit;
  class Track;
  class RecoTrack;


  class SVDROIFinderAnalysisDataModule : public Module {
 
  public:

    SVDROIFinderAnalysisDataModule();

    ~SVDROIFinderAnalysisDataModule();

    void initialize() override;

    void event() override;

    void terminate() override;
 
  private:
 
    StoreArray<SVDShaperDigit> m_shapers; 
    StoreArray<ROIid> m_ROIs; 
    StoreArray<RecoTrack> m_recoTracks;
    StoreArray<Track> m_tracks;
    StoreArray<SVDIntercept> m_SVDIntercepts; 
 
    float m_edgeU; 
    float m_edgeV; 
    float m_minPVal; 
 
    std::string m_shapersName; 
    std::string m_recoTrackListName; 
    std::string m_SVDInterceptListName; 
    std::string m_ROIListName; 
 
    TFile* m_rootFilePtr; 
    std::string m_rootFileName; 
    bool m_writeToRoot; 
 
    int m_rootEvent;   
 
    //graphs & histos - results
 
    TH1F* m_h1Track; 
    TH1F* m_h1Track_pt; 
    TH1F* m_h1Track_phi; 
    TH1F* m_h1Track_lambda; 
    TH1F* m_h1Track_cosTheta; 
    TH1F* m_h1Track_pVal; 
 
    //Tracks with attached ROI
    TH1F* m_h1ROItrack; 
    TH1F* m_h1ROItrack_pt; 
    TH1F* m_h1ROItrack_phi; 
    TH1F* m_h1ROItrack_lambda; 
    TH1F* m_h1ROItrack_cosTheta; 
    TH1F* m_h1ROItrack_pVal; 
 
    //Tracks with attached Good ROI
    TH1F* m_h1GoodROItrack; 
    TH1F* m_h1GoodROItrack_pt; 
    TH1F* m_h1GoodROItrack_phi; 
    TH1F* m_h1GoodROItrack_lambda; 
    TH1F* m_h1GoodROItrack_cosTheta; 
    TH1F* m_h1GoodROItrack_pVal; 
 
    //Tracks with attached ROI containing at least one SVDShaperDigit
    TH1F* m_h1FullROItrack; 
    TH1F* m_h1FullROItrack_pt; 
    TH1F* m_h1FullROItrack_phi; 
    TH1F* m_h1FullROItrack_lambda; 
    TH1F* m_h1FullROItrack_cosTheta; 
    TH1F* m_h1FullROItrack_pVal; 
 
 
    //fill digits inside ROIs
    TH1F* m_h1PullU; 
    TH1F* m_h1PullV; 
    TH2F* m_h2sigmaUphi; 
    TH2F* m_h2sigmaVphi; 
    TH1F* m_h1ResidU; 
    TH1F* m_h1ResidV; 
    TH1F* m_h1SigmaU; 
    TH1F* m_h1SigmaV; 
    TH1F* m_h1GlobalTime; 
 
    //ROI stuff
    TH2F* m_h2ROIuMinMax; 
    TH2F* m_h2ROIvMinMax;  
    TH2F* m_h2ROIcenters; 
    TH2F* m_h2GoodROIcenters; 
    TH2F* m_h2FullROIcenters; 
    TH1F* m_h1totROIs; 
    TH1F* m_h1goodROIs; 
    TH1F* m_h1okROIs; 
    TH1F* m_h1effROIs; 
 
    unsigned int n_rois; 
    unsigned int m_nGoodROIs; 
    unsigned int m_nOkROIs; 
    unsigned int n_intercepts; 
    unsigned int n_tracks; 
 
  };

}