TOPCommonT0CalibratorModule.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 <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
#include <top/dataobjects/TOPDigit.h>
#include <top/dataobjects/TOPRecBunch.h>
#include <top/utilities/TrackSelector.h>
#include <top/utilities/Chi2MinimumFinder1D.h>
#include <top/reconstruction_cpp/PDFConstructor.h>
 
#include <framework/database/DBObjPtr.h>
#include <top/dbobjects/TOPCalCommonT0.h>
 
#include <string>
 
#include <TFile.h>
#include <TTree.h>
#include <TH1F.h>
#include <TH2F.h>
 
namespace Belle2 {

  class TOPCommonT0CalibratorModule : public Module {
 
  public:

    TOPCommonT0CalibratorModule();

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;
 
  private:

    bool isRunningOffsetSubtracted();

    enum {c_numModules = 16, 
          c_numSets = 32,  
         };
 
    // steering parameters
    int m_numBins;      
    double m_timeRange; 
    double m_sigmaSmear;  
    std::string m_sample; 
    double m_minMomentum; 
    double m_deltaEcms; 
    double m_dr; 
    double m_dz; 
    double m_minZ; 
    double m_maxZ; 
    std::string m_outFileName; 
    std::string m_pdfOption;   
 
    // procedure
    TOP::TrackSelector m_selector; 
    TOP::Chi2MinimumFinder1D m_finders[c_numSets]; 
    TOP::PDFConstructor::EPDFOption m_PDFOption = TOP::PDFConstructor::c_Rough; 
    double m_bunchTimeSep = 0; 
 
    // datastore objects
    StoreArray<TOPDigit> m_digits; 
    StoreArray<Track> m_tracks;    
    StoreArray<ExtHit> m_extHits;  
    StoreObjPtr<TOPRecBunch> m_recBunch; 
 
    // database
    DBObjPtr<TOPCalCommonT0> m_commonT0;   
 
    // output root file
    TFile* m_file = 0;                 
 
    // control histograms
    TH1F m_hits1D;  
    TH2F m_hits2D;  
 
    // tree and its variables
    TTree* m_tree = 0;  
    int m_moduleID = 0; 
    int m_numPhotons = 0; 
    float m_x = 0; 
    float m_y = 0; 
    float m_z = 0; 
    float m_p = 0; 
    float m_theta = 0; 
    float m_phi = 0; 
    float m_pocaR = 0; 
    float m_pocaZ = 0; 
    float m_pocaX = 0; 
    float m_pocaY = 0; 
    float m_cmsE = 0; 
    int m_charge = 0; 
    int m_PDG = 0; 
 
  };

} // Belle2 namespace