RecoTracker/TrackProducer/interface/TrackProducerBase.h - master - githubmirror/cmssw

RecoTracker/TrackProducer/interface/TrackProducerBase.h
120 строк4 KB
mmusich
add fillDescriptions to {Gsf}Track{Producer|Refitter}
15 янв 2025, 21:43
15 янв 2025, 21:43a02520a
#ifndef TrackProducerBase_h
#define TrackProducerBase_h

/** \class TrackProducerBase
 *  Base Class To Produce Tracks
 *
 *  \author cerati
 */

#include "AlgoProductTraits.h"

#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/TrackerRecHit2D/interface/ClusterRemovalInfo.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "RecoTracker/MeasurementDet/interface/MeasurementTracker.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"

#include <tuple>

class Propagator;
class TrajectoryStateUpdator;
class MeasurementEstimator;
class TrackerGeometry;
class TrajectoryFitter;
class TransientTrackingRecHitBuilder;
class NavigationSchool;
class TrackerDigiGeometryRecord;
class IdealMagneticFieldRecord;
class TransientRecHitRecord;
class TrackingComponentsRecord;
class NavigationSchoolRecord;
class CkfComponentsRecord;

template <class T>
class TrackProducerBase : public AlgoProductTraits<T> {
public:
  using Base = AlgoProductTraits<T>;
  using TrackView = typename Base::TrackView;
  using TrackCollection = typename Base::TrackCollection;
  using AlgoProductCollection = typename Base::AlgoProductCollection;

public:
  /// Constructor
  TrackProducerBase(bool trajectoryInEvent = false) : trajectoryInEvent_(trajectoryInEvent) {}

  /// Destructor
  virtual ~TrackProducerBase() noexcept(false);

  /// Get needed services from the Event Setup
  virtual void getFromES(const edm::EventSetup&,
                         edm::ESHandle<TrackerGeometry>&,
                         edm::ESHandle<MagneticField>&,
                         edm::ESHandle<TrajectoryFitter>&,
                         edm::ESHandle<Propagator>&,
                         edm::ESHandle<MeasurementTracker>&,
                         edm::ESHandle<TransientTrackingRecHitBuilder>&);

  /// Get TrackCandidateCollection from the Event (needed by TrackProducer)
  virtual void getFromEvt(edm::Event&, edm::Handle<TrackCandidateCollection>&, reco::BeamSpot&);
  /// Get TrackCollection from the Event (needed by TrackRefitter)
  virtual void getFromEvt(edm::Event&, edm::Handle<TrackView>&, reco::BeamSpot&);

  /// Method where the procduction take place. To be implemented in concrete classes
  virtual void produce(edm::Event&, const edm::EventSetup&) = 0;

  /// Call this method in inheriting class' constructor
  void initTrackProducerBase(const edm::ParameterSet& conf, edm::ConsumesCollector cc, const edm::EDGetToken& src);

  /// set the aliases of produced collections
  void setAlias(std::string alias) {
    alias.erase(alias.size() - 6, alias.size());
    alias_ = alias;
  }

  void setSecondHitPattern(Trajectory* traj,
                           T& track,
                           const Propagator* prop,
                           const MeasurementTrackerEvent* measTk,
                           const TrackerTopology* ttopo);

  const edm::ParameterSet& getConf() const { return conf_; }

  static void fillPSetDescription(edm::ParameterSetDescription& desc);

protected:
  edm::ParameterSet conf_;
  edm::EDGetToken src_;

protected:
  std::string alias_;
  bool trajectoryInEvent_;
  edm::OrphanHandle<TrackCollection> rTracks_;
  edm::EDGetTokenT<reco::BeamSpot> bsSrc_;
  edm::EDGetTokenT<MeasurementTrackerEvent> mteSrc_;

  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> trackGeomSrc_;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> mfSrc_;
  edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> fitterSrc_;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> propagatorSrc_;
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> builderSrc_;
  edm::ESGetToken<MeasurementTracker, CkfComponentsRecord> measTkSrc_;
  edm::ESGetToken<NavigationSchool, NavigationSchoolRecord> schoolSrc_;

  edm::ESHandle<NavigationSchool> theSchool;
  bool useSchool_ = false;
};

#include "RecoTracker/TrackProducer/interface/TrackProducerBase.icc"

#endif