HitLevelInfoWriter.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 <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <mdst/dataobjects/KLMCluster.h>
#include <mdst/dataobjects/EventLevelTriggerTimeInfo.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/database/DBObjPtr.h>
#include <framework/core/Module.h>
#include <framework/gearbox/Const.h>
 
#include <cdc/dataobjects/CDCDedxTrack.h>
#include <cdc/dbobjects/CDCDedxScaleFactor.h>
#include <cdc/dbobjects/CDCDedxWireGain.h>
#include <cdc/dbobjects/CDCDedxRunGain.h>
#include <cdc/dbobjects/CDCDedxCosineCor.h>
#include <cdc/dbobjects/CDCDedx2DCell.h>
#include <cdc/dbobjects/CDCDedx1DCell.h>
#include <cdc/dbobjects/CDCDedxInjectionTime.h>
#include <cdc/dbobjects/CDCDedxADCNonLinearity.h>
#include <cdc/dbobjects/CDCDedxCosineEdge.h>
#include <cdc/dbobjects/CDCDedxHadronCor.h>
 
#include <vector>
 
#include <TFile.h>
#include <TTree.h>
 
class TH2F;
 
namespace Belle2 {
 
  class CDCDedxTrack;

  class HitLevelInfoWriterModule : public Module {
 
  public:

    HitLevelInfoWriterModule();

    virtual ~HitLevelInfoWriterModule();

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;

    void bookOutput(std::string filename);

    void recalculateDedx(CDCDedxTrack* dedxTrack, std::map<int, std::vector<double>>& l_var,
                         double (&cdcChi)[Const::ChargedStable::c_SetSize]);

    double GetCorrection(int& adc, int layer, int wireID, double doca, double enta, double costheta, double ring, double time) const;

    void HadronCorrection(double costheta, double& dedx) const;

    double D2I(const double cosTheta, const double D) const;

    double I2D(const double cosTheta, const double I) const;

    void calculateMeans(double* mean, double* truncatedMean, double* truncatedMeanErr, const std::vector<double>& dedx) const;

    void saveChiValue(double(&chi)[Const::ChargedStable::c_SetSize], CDCDedxTrack* dedxTrack, double dedx) const;
 
  private:
 
    std::string m_strOutputBaseName; 
    std::vector<std::string> m_strParticleList; 
    std::vector<std::string> m_filename; 
    std::vector<TFile* > m_file;  
    std::vector<TTree* > m_tree;  
 
    StoreArray<CDCDedxTrack> m_dedxTracks; 
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 
    StoreArray<ECLCluster> m_eclClusters; 
    StoreArray<KLMCluster> m_klmClusters; 
    StoreObjPtr<EventLevelTriggerTimeInfo> m_TTDInfo; 

    void fillTrack(const TrackFitResult* fitResult);

    void fillDedx(CDCDedxTrack* dedxTrack);

    void clearEntries();
 
    // event level information (from emd)
    int m_expID{ -1};   
    int m_runID{ -1};   
    int m_eventID{ -1}; 
 
    double m_injring{ -1.0}; 
    double m_injtime{ -1.0}; 
 
    // track level information (from tfr)
    double m_d0{0.};     
    double m_z0{0.};     
    double m_dz{ -1.};  
    double m_dr{ -1.};  
    double m_dphi{ -1.};  
    double m_vx0{0.};    
    double m_vy0{0.};    
    double m_vz0{0.};    
    double m_tanlambda{ -1.};  
    double m_phi0{ -1.};  
    double m_chi2{ -1.}; 
 
    double m_nCDChits{ -1.};  
    int m_inCDC{ -1}; 
    int m_trackID{ -1};     
    double m_length{ -1.};  
    int m_charge{0};        
    double m_cosTheta{ -2.}; 
    double m_pCDC{ -1.};       
    double m_p{ -1.};       
    double m_pt{ -1.};       
    double m_phi{ -1.};  
    double m_ioasym{ -1.};  
    double m_theta{ -2.}; 
 
    // track level Mc
    double m_PDG{ -1.};     
    //    double m_motherPDG; /**< MC PID of mother particle */
    //    double m_pTrue;     /**< MC true momentum */
    //    double m_trackDist; /**< the total distance traveled by the track */
 
    // track level dE/dx measurements
    double m_mean{ -1.};       
    double m_trunc{ -1.};      
    double m_truncNoSat{ -1.}; 
    double m_error{ -1.};      
 
    // other dec specific information
    double m_eop{ -1.};       
    double m_e{ -1.};       
    double m_e1_9{ -1.};       
    double m_e9_21{ -1.};        
    double m_klmLayers{ -1.}; 
    double m_eclsnHits{ -1.};        
 
    // calibration constants
    double m_scale{ -1.};   
    double m_cosCor{ -1.};  
    double m_cosEdgeCor{ -1.};  
    double m_runGain{ -1.}; 
    double m_timeGain{ -1.}; 
    double m_timeReso{ -1.}; 
 
    // hadron cal and PID related variables
    double m_chieOld{ -1.};  
    double m_chimuOld{ -1.}; 
    double m_chipiOld{ -1.}; 
    double m_chikOld{ -1.};  
    double m_chipOld{ -1.};  
    double m_chidOld{ -1.};  
 
    double m_chie{ -1.};  
    double m_chimu{ -1.}; 
    double m_chipi{ -1.}; 
    double m_chik{ -1.};  
    double m_chip{ -1.};  
    double m_chid{ -1.};  
 
    double m_prese{ -1.};  
    double m_presmu{ -1.}; 
    double m_prespi{ -1.}; 
    double m_presk{ -1.};  
    double m_presp{ -1.};  
    double m_presd{ -1.};  
 
    double m_pmeane{ -1.};  
    double m_pmeanmu{ -1.}; 
    double m_pmeanpi{ -1.}; 
    double m_pmeank{ -1.};  
    double m_pmeanp{ -1.};  
    double m_pmeand{ -1.};  
 
    static const int kMaxHits = 200; 
    // layer level information
    int l_nhits{ -1};                    
    int l_nhitsused{ -1};                
    int l_nhitscombined[kMaxHits] = {};  
    int l_wirelongesthit[kMaxHits] = {}; 
    int l_layer[kMaxHits] = {};          
    double l_path[kMaxHits] = {};        
    double l_dedx[kMaxHits] = {};        
 
    // hit level information (references on nhits)
    int h_nhits{ -1};           
    int h_lwire[kMaxHits] = {}; 
    int h_wire[kMaxHits] = {};  
    int h_layer[kMaxHits] = {}; 
 
    double h_path[kMaxHits] = {};   
    double h_dedx[kMaxHits] = {};   
    double h_adcraw[kMaxHits] = {}; 
    double h_adccorr[kMaxHits] = {}; 
    double h_doca[kMaxHits] = {};   
    double h_ndoca[kMaxHits] = {};  
    double h_ndocaRS[kMaxHits] = {}; 
    double h_enta[kMaxHits] = {};     
    double h_entaRS[kMaxHits] = {};   
    double h_driftT[kMaxHits] = {};   
    double h_driftD[kMaxHits] = {};   
    double h_facnladc[kMaxHits] = {}; 
    double h_wireGain[kMaxHits] = {}; 
    double h_twodCor[kMaxHits] = {};  
    double h_onedCor[kMaxHits] = {};  
 
    //Tracking variables for extra hits
    double h_WeightPionHypo[kMaxHits] = {}; 
    double h_WeightKaonHypo[kMaxHits] = {}; 
    double h_WeightProtonHypo[kMaxHits] = {}; 
    int h_foundByTrackFinder[kMaxHits] = {}; 
 
    // parameters: calibration constants
    DBObjPtr<CDCDedxScaleFactor> m_DBScaleFactor; 
    DBObjPtr<CDCDedxWireGain> m_DBWireGains;      
    DBObjPtr<CDCDedxRunGain> m_DBRunGain;         
    DBObjPtr<CDCDedxInjectionTime> m_DBInjectTime; 
    DBObjPtr<CDCDedxCosineCor> m_DBCosineCor;     
    DBObjPtr<CDCDedx2DCell> m_DB2DCell; 
    DBObjPtr<CDCDedx1DCell> m_DB1DCell; 
    DBObjPtr<CDCDedxADCNonLinearity> m_DBNonlADC; 
    DBObjPtr<CDCDedxCosineEdge> m_DBCosEdgeCor; 
    DBObjPtr<CDCDedxHadronCor> m_DBHadronCor; 
 
    std::vector<double> m_hadronpars; 
 
    bool m_isDeadwire; 
    //Flag to enable and disable set of variables
    bool m_isHitLevel; 
    bool m_isExtraVar; 
    bool m_isRelative; 
    bool m_isCorrection; 
  };

} // Belle2 namespace