TrackingTools/TrackRefitter/interface/TrackTransformerForCosmicMuons.h - master - githubmirror/cmssw

TrackingTools/TrackRefitter/interface/TrackTransformerForCosmicMuons.h
126 строк5 KB
Shahzad Malik Muzaffar
[RECONSTRUCTION] Drop Geometry/CommonDetUnit package
28 май 2026, 12:57
28 май 2026, 12:57685f036
#ifndef TrackingTools_TrackRefitter_TrackTransformerForCosmicMuons_H
#define TrackingTools_TrackRefitter_TrackTransformerForCosmicMuons_H

/** \class TrackTransformer
 *  This class takes a reco::Track and refits the rechits inside it.
 *  The final result is a Trajectory refitted and smoothed.
 *  To make the refitting (and the smoothing) the usual KF tools are used.
 *
 *  CAVEAT: till now (it will be changed in the near future) the class stores the
 *  pointers to the services, therefore EACH event the setServices(const edm::EventSetup&)
 *  method MUST be called in the code in which the TrackTransformer is used.
 *
 *  Rec hits are ordered in rank of increasing |z| assuing muons are coming from the 
 *  top of the detector.
 *
 *
 *  $Date: 2009/01/15
 *  \original author R. Bellan - CERN <riccardo.bellan@cern.ch>
 *  modified for zed ordering by N. Kypreos - UF <nicholas.theodore.kypreos@cern.ch> 
 */

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/CommonTopologies/interface/GlobalTrackingGeometry.h"
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "TrackingTools/TrackRefitter/interface/RefitDirection.h"
#include "TrackingTools/TrackRefitter/interface/RefitDirection.h"
#include "TrackingTools/TrackRefitter/interface/TrackTransformerBase.h"
#include "TrackingTools/TrackRefitter/interface/TrackTransformerBase.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"
#include "TrackingTools/Records/interface/TransientRecHitRecord.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"
#include "TrackingTools/PatternTools/interface/TrajectoryFwd.h"

namespace edm {
  class ParameterSet;
  class EventSetup;
}  // namespace edm
namespace reco {
  class TransientTrack;
}

class TrajectoryFitter;
class TrajectorySmoother;
class Propagator;
class TransientTrackingRecHitBuilder;

class TrackTransformerForCosmicMuons : public TrackTransformerBase {
public:
  /// Constructor
  TrackTransformerForCosmicMuons(const edm::ParameterSet&, edm::ConsumesCollector);

  /// Destructor
  ~TrackTransformerForCosmicMuons() override;

  // Operations

  /// Convert a reco::Track into Trajectory
  std::vector<Trajectory> transform(const reco::Track&) const override;

  /// the magnetic field
  const MagneticField* magneticField() const { return &*theMGField; }

  /// the tracking geometry
  edm::ESHandle<GlobalTrackingGeometry> trackingGeometry() const { return theTrackingGeometry; }

  /// set the services needed by the TrackTransformer
  void setServices(const edm::EventSetup&) override;

  ///calculate the sum of slopes for the track
  bool SlopeSum(const TransientTrackingRecHit::ConstRecHitContainer&) const;
  float SumDy(const TransientTrackingRecHit::ConstRecHitContainer&) const;

  /// the refitter used to refit the reco::Track
  edm::ESHandle<TrajectoryFitter> fitter(bool, int, float) const;

  /// the smoother used to smooth the trajectory which came from the refitting step
  edm::ESHandle<TrajectorySmoother> smoother(bool, int, float) const;

  TransientTrackingRecHit::ConstRecHitContainer getTransientRecHits(const reco::TransientTrack& track) const;

protected:
private:
  const edm::ESGetToken<Propagator, TrackingComponentsRecord> theIOpropToken;
  const edm::ESGetToken<Propagator, TrackingComponentsRecord> theOIpropToken;
  const edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> thGlobTrackGeoToken;
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theMFToken;
  const edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> theIOFitterToken;
  const edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> theOIFitterToken;
  const edm::ESGetToken<TrajectorySmoother, TrajectoryFitter::Record> theIOSmootherToken;
  const edm::ESGetToken<TrajectorySmoother, TrajectoryFitter::Record> theOISmootherToken;
  const edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theTkRecHitBuildToken;
  const edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theMuonRecHitBuildToken;

  edm::ESHandle<Propagator> thePropagatorIO;
  edm::ESHandle<Propagator> thePropagatorOI;

  edm::ESHandle<Propagator> propagator(bool, int, float) const;

  unsigned long long theCacheId_TC;
  unsigned long long theCacheId_GTG;
  unsigned long long theCacheId_MG;
  unsigned long long theCacheId_TRH;

  bool theRPCInTheFit;

  edm::ESHandle<GlobalTrackingGeometry> theTrackingGeometry;
  edm::ESHandle<MagneticField> theMGField;

  edm::ESHandle<TrajectoryFitter> theFitterIO;
  edm::ESHandle<TrajectoryFitter> theFitterOI;

  edm::ESHandle<TrajectorySmoother> theSmootherIO;
  edm::ESHandle<TrajectorySmoother> theSmootherOI;

  std::string theTrackerRecHitBuilderName;
  edm::ESHandle<TransientTrackingRecHitBuilder> theTrackerRecHitBuilder;

  std::string theMuonRecHitBuilderName;
  edm::ESHandle<TransientTrackingRecHitBuilder> theMuonRecHitBuilder;
};
#endif