CDCBadWireCollector.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 <cdc/dataobjects/CDCHit.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <tracking/trackFindingCDC/eventdata/tracks/CDCTrack.h>
#include <tracking/trackFindingCDC/rootification/StoreWrappedObjPtr.h>
#include <tracking/trackFindingCDC/topology/CDCWire.h>
#include <framework/datastore/StoreArray.h>
#include <string>
 
namespace Belle2 {
  namespace CDC {

    class CDCBadWireCollectorModule : public CalibrationCollectorModule {
 
    public:
      CDCBadWireCollectorModule();

      virtual ~CDCBadWireCollectorModule();

      void prepare() override;
 
      void collect() override;

      void finish() override;
 
    private:
      void buildEfficiencies(std::vector<unsigned short> wireHits, const Helix helixFit);
      const TrackFindingCDC::CDCWire& getIntersectingWire(const B2Vector3D& xyz, const TrackFindingCDC::CDCWireLayer& layer,
                                                          const Helix& helixFit) const;
 
      StoreArray<Track> m_Tracks;                   
      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_trackFitResultArrayName;                     
 
      std::string m_effTreeName;              
 
      double m_minimumPt = 0;   
      double m_minimumNDF = 0;   
      unsigned short wireID;                   
      unsigned short layerID;                  
      float z;                                 
      bool isFound;                            
    };
  }
}