ECLBhabhaTCollectorModule.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
 
/* ECL headers. */
#include <ecl/mapper/ECLChannelMapper.h>
#include <ecl/utility/ECLTimingUtilities.h>
 
/* Basf2 headers. */
#include <calibration/CalibrationCollectorModule.h>
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <framework/dataobjects/EventMetaData.h>
#include <framework/dataobjects/EventT0.h>
#include <mdst/dataobjects/SoftwareTriggerResult.h>
 
 
class TTree;
 
namespace Belle2 {
  class ECLDigit;
  class ECLCalDigit;
  class Track;
  class ECLCrystalCalib;
  class ECLReferenceCrystalPerCrateCalib;
  class ECLCluster;
  class ECLChannelMapper;
 
  const static double realNaN = std::numeric_limits<double>::quiet_NaN();  
  const static int    intNaN  = std::numeric_limits<int>::quiet_NaN();     
  class ECLBhabhaTCollectorModule : public CalibrationCollectorModule {
 
  public:
 
    ECLBhabhaTCollectorModule();
 
    virtual ~ECLBhabhaTCollectorModule();
 
    void inDefineHisto() override;
 
    void prepare() override;
 
    void collect() override;
 
  private:
 
    bool m_saveTree;
 
 
    StoreArray<Track> tracks; 
    std::unique_ptr< Belle2::ECL::ECLChannelMapper> m_crystalMapper =
      std::make_unique<Belle2::ECL::ECLChannelMapper>();
 
 
    StoreObjPtr<SoftwareTriggerResult> m_TrgResult; 
    StoreObjPtr<EventT0> m_eventT0;
 
    StoreObjPtr<EventMetaData> m_EventMetaData;
 
    DBObjPtr<ECLCrystalCalib> m_ElectronicsDB; 
    std::vector<float> m_Electronics; 
    DBObjPtr<ECLCrystalCalib> m_ElectronicsTimeDB; 
    std::vector<float> m_ElectronicsTime; 
    DBObjPtr<ECLCrystalCalib> m_FlightTimeDB; 
    std::vector<float> m_FlightTime; 
    DBObjPtr<ECLCrystalCalib> m_PreviousCrystalTimeDB; 
    std::vector<float> m_PreviousCrystalTime; 
    std::vector<float> m_PreviousCrystalTimeUnc; 
    DBObjPtr<ECLCrystalCalib> m_CrateTimeDB; 
    std::vector<float> m_CrateTime; 
    std::vector<float> m_CrateTimeUnc; 
    DBObjPtr<ECLReferenceCrystalPerCrateCalib> m_RefCrystalsCalibDB; 
    std::vector<short> m_RefCrystalsCalib; 
    DBObjPtr<Belle2::ECLChannelMap> m_channelMapDB; 
    TTree* m_dbgTree_electrons = nullptr;       
    TTree* m_dbgTree_tracks = nullptr;          
    TTree* m_dbgTree_crystals = nullptr;        
    TTree* m_dbgTree_event = nullptr;           
    TTree* m_dbgTree_allCuts = nullptr;         
    TTree* m_dbgTree_evt_allCuts = nullptr;     
    TTree* m_dbgTree_crys_allCuts = nullptr;    
    /*** tree branches ***/
    /*** See inDefineHisto method for branches description ***/
    int m_tree_evtNum = intNaN;    
    int m_tree_cid = intNaN;     
    int m_tree_amp = intNaN;     
    double m_tree_en = realNaN;     
    double m_tree_E1Etot = realNaN;     
    double m_tree_E1E2 = realNaN;     
    double m_tree_E1p = realNaN;     
    int m_tree_quality = intNaN; 
    double m_tree_timeF = realNaN; 
    double m_tree_time = realNaN; 
    double m_tree_timetsPreviousTimeCalibs = realNaN; 
    double m_tree_t0 = realNaN;   
    double m_tree_t0_unc = realNaN;   
    double m_tree_t0_ECLclosestCDC = realNaN;   
    double m_tree_t0_ECL_minChi2 = realNaN;   
    double m_tree_d0 = realNaN;    
    double m_tree_z0 = realNaN;    
    double m_tree_p = realNaN;    
    double m_tree_nCDChits = realNaN;    
    double m_tree_clustCrysE_DIV_maxEcrys = realNaN;    
    double m_tree_clustCrysE = realNaN;    
    double m_tree_enPlus = realNaN;     
    double m_tree_enNeg = realNaN;     
    double m_tree_tClustPos = realNaN;
    double m_tree_tClustNeg = realNaN;
 
    double m_tree_maxEcrystPosClust = realNaN;     
    double m_tree_maxEcrystNegClust = realNaN;     
    double m_tree_tClust = realNaN;     
    double m_tree_ECLCalDigitTime = realNaN;   
    double m_tree_ECLCalDigitE = realNaN;   
    double m_tree_ECLDigitAmplitude = realNaN;   
    int m_charge = intNaN;        
    double m_E_DIV_p = realNaN;    
    double m_massInvTracks = realNaN;    
    /*** tree branches END ***/
 
 
    StoreArray<ECLDigit> m_eclDigitArray; 
    StoreArray<ECLCalDigit> m_eclCalDigitArray; 
    StoreArray<ECLCluster> m_eclClusterArray; 
    std::vector<float> m_EperCrys; 
    std::vector<int> m_eclCalDigitID; 
    std::vector<int> m_eclDigitID; 
    /****** Parameters for cuts ******/
    short m_timeAbsMax; 
    int m_minCrystal = intNaN; 
    int m_maxCrystal = intNaN; 
    /* d0 and z0 values of the loose and tight tracks*/
    double m_looseTrkZ0 = realNaN;   
    double m_tightTrkZ0 = realNaN;   
    double m_looseTrkD0 = realNaN;   
    double m_tightTrkD0 = realNaN;   
    int m_crystalCrate = intNaN;    
    int m_runNum = intNaN;   
    bool m_storeCalib = true;   
    // For the energy dependence correction to the time
    // t-t0 = p1 + pow( (p3/(amplitude+p2)), p4 ) + p5*exp(-amplitude/p6)      ("Energy dependence equation")
 
    std::unique_ptr< Belle2::ECL::ECLTimingUtilities > m_ECLTimeUtil =
      std::make_unique<Belle2::ECL::ECLTimingUtilities>(); 
    double m_hadronEventT0_TO_bhabhaEventT0_correction;
 
    bool skipTrgSel; 
  };
}