CDCCalibrationCollector.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 <calibration/CalibrationCollectorModule.h>
 
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
 
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/trackFindingCDC/eventdata/tracks/CDCTrack.h>
#include <tracking/trackFindingCDC/rootification/StoreWrappedObjPtr.h>
#include <tracking/trackFindingCDC/topology/CDCWire.h>
 
#include <framework/dataobjects/EventT0.h>
#include <framework/datastore/StoreArray.h>
 
#include <Math/Vector3D.h>
#include <string>
 
namespace Belle2 {
  namespace CDC {

    class CDCCalibrationCollectorModule : public CalibrationCollectorModule {
 
    public:

      CDCCalibrationCollectorModule();

      virtual ~CDCCalibrationCollectorModule();

      void prepare() override;
 

 
      void collect() override;

      void finish() override;
 
    private:
 

      void harvest(Belle2::RecoTrack* track);
      void buildEfficiencies(std::vector<unsigned short> wireHits, const Helix helixFit);
      const TrackFindingCDC::CDCWire& getIntersectingWire(const ROOT::Math::XYZVector& xyz, const TrackFindingCDC::CDCWireLayer& layer,
                                                          const Helix& helixFit) const;
 
      StoreObjPtr<EventT0> m_eventTimeStoreObject;  
      StoreArray<Track> m_Tracks;                   
      StoreArray<RecoTrack> m_RecoTracks;           
      StoreArray<TrackFitResult> m_TrackFitResults; 
      StoreArray<CDCHit> m_CDCHits;                 
      TrackFindingCDC::StoreWrappedObjPtr<std::vector<TrackFindingCDC::CDCTrack>> m_CDCTracks; 
 
      std::string m_trackArrayName;           
      std::string m_cdcHitArrayName;          
      std::string m_cdcTrackVectorName = "CDCTrackVector";       
      std::string m_recoTrackArrayName ;      
      std::string m_trackFitResultArrayName;  
      std::string m_relRecoTrackTrackName;    
      std::string m_treeName;                 
      std::string m_effTreeName;              
 
      Float_t weight;            
      Float_t alpha;             
      Float_t theta;             
      UShort_t adc;              
      Float_t t;                 
      Float_t t_fit;             
      Float_t evtT0;             
      Float_t x_mea;             
      Float_t x_u;               
      Float_t x_b;               
      UChar_t lay;              
      UShort_t IWire;           
      Float_t Pval;              
      Float_t ndf;               
      Float_t d0;                
      Float_t z0;                
      Float_t phi0;              
      Float_t tanL;              
      Float_t omega;             
      double m_minimumPt = 0;   
      double m_minimumNDF = 0;   
 
      bool m_calExpectedDriftTime = true;      
      bool m_bField = true;                    
      bool m_storeTrackParams = true;          
      bool m_eventT0Extraction = true;         
      bool m_isCosmic = false;                 
      bool m_effStudy = false;                 
 
      unsigned short wireID;                   
      unsigned short layerID;                  
      float z;                                 
      bool isFound;                            
    };
  }
}