TrackTransformerForCosmicMuons.h

Go to the documentation of this file.

#ifndef TrackingTools_TrackRefitter_TrackTransformerForCosmicMuons_H
#define TrackingTools_TrackRefitter_TrackTransformerForCosmicMuons_H

#include "TrackingTools/TrackRefitter/interface/TrackTransformerBase.h"

#include "TrackingTools/TrackRefitter/interface/RefitDirection.h"

#include "TrackingTools/TrackRefitter/interface/TrackTransformerBase.h"

#include "TrackingTools/TrackRefitter/interface/RefitDirection.h"

#include "FWCore/Framework/interface/ESHandle.h"

#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"

#include "DataFormats/TrackReco/interface/TrackFwd.h"

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

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

class TrackTransformerForCosmicMuons: public TrackTransformerBase{

public:

  TrackTransformerForCosmicMuons(const edm::ParameterSet&);

  ~TrackTransformerForCosmicMuons() override;
  
  // Operations

  std::vector<Trajectory> transform(const reco::Track&) const override;

  const MagneticField* magneticField() const {return &*theMGField;}
  
  edm::ESHandle<GlobalTrackingGeometry> trackingGeometry() const {return theTrackingGeometry;}

  void setServices(const edm::EventSetup&) override;

  bool SlopeSum(const TransientTrackingRecHit::ConstRecHitContainer&) const;
  float SumDy(const TransientTrackingRecHit::ConstRecHitContainer&) const;

  edm::ESHandle<TrajectoryFitter> fitter(bool, int, float) const;
  
  edm::ESHandle<TrajectorySmoother> smoother(bool, int, float) const;

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

  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