GlobalMuonRefitter.h

Go to the documentation of this file.

#ifndef RecoMuon_GlobalTrackingTools_GlobalMuonRefitter_H
#define RecoMuon_GlobalTrackingTools_GlobalMuonRefitter_H
 
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include "RecoMuon/TrackingTools/interface/MuonTrajectoryBuilder.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
#include "TrackingTools/TrackRefitter/interface/TrackTransformer.h"
#include "DataFormats/DTRecHit/interface/DTRecHitCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/ME0SegmentCollection.h"
#include "DataFormats/DTRecHit/interface/DTRecSegment4DCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/MuonReco/interface/DYTInfo.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
 
namespace reco {
  class TransientTrack;
}
 
class TrajectoryStateOnSurface;
class TrackerTopology;
 
class MuonDetLayerMeasurements;
class MuonServiceProxy;
class Trajectory;
 
class TrajectoryFitter;
class TrajectoryFitterRecord;
class TransientRecHitRecord;
 
class GlobalMuonRefitter {
public:
  typedef TransientTrackingRecHit::RecHitContainer RecHitContainer;
  typedef TransientTrackingRecHit::ConstRecHitContainer ConstRecHitContainer;
  typedef TransientTrackingRecHit::RecHitPointer RecHitPointer;
  typedef TransientTrackingRecHit::ConstRecHitPointer ConstRecHitPointer;
 
  typedef MuonTransientTrackingRecHit::MuonRecHitPointer MuonRecHitPointer;
  typedef MuonTransientTrackingRecHit::ConstMuonRecHitPointer ConstMuonRecHitPointer;
  typedef MuonTransientTrackingRecHit::MuonRecHitContainer MuonRecHitContainer;
  typedef MuonTransientTrackingRecHit::ConstMuonRecHitContainer ConstMuonRecHitContainer;
 
  typedef std::vector<Trajectory> TC;
  typedef TC::const_iterator TI;
 
  enum subDetector { PXB = 1, PXF = 2, TIB = 3, TID = 4, TOB = 5, TEC = 6 };
 
public:
  GlobalMuonRefitter(const edm::ParameterSet&, const MuonServiceProxy*, edm::ConsumesCollector&);
 
  virtual ~GlobalMuonRefitter();
 
  virtual void setEvent(const edm::Event&);
 
  void setServices(const edm::EventSetup&);
 
  std::vector<Trajectory> refit(const reco::Track& globalTrack,
                                const int theMuonHitsOption,
                                const TrackerTopology* tTopo) const;
 
  std::vector<Trajectory> refit(const reco::Track& globalTrack,
                                const reco::TransientTrack track,
                                const TransientTrackingRecHit::ConstRecHitContainer& allRecHitsTemp,
                                const int theMuonHitsOption,
                                const TrackerTopology* tTopo) const;
 
  std::vector<Trajectory> transform(const reco::Track& newTrack,
                                    const reco::TransientTrack track,
                                    const TransientTrackingRecHit::ConstRecHitContainer& recHitsForReFit) const;
 
  // get rid of selected station RecHits
  ConstRecHitContainer getRidOfSelectStationHits(const ConstRecHitContainer& hits, const TrackerTopology* tTopo) const;
 
  // return DYT-related informations
  const reco::DYTInfo* getDYTInfo() { return dytInfo; }
 
protected:
  enum RefitDirection { insideOut, outsideIn, undetermined };
 
  void checkMuonHits(const reco::Track&, ConstRecHitContainer&, std::map<DetId, int>&) const;
 
  void getFirstHits(const reco::Track&, ConstRecHitContainer&, ConstRecHitContainer&) const;
 
  ConstRecHitContainer selectMuonHits(const Trajectory&, const std::map<DetId, int>&) const;
 
  void printHits(const ConstRecHitContainer&) const;
 
  RefitDirection checkRecHitsOrdering(const ConstRecHitContainer&) const;
 
  const MuonServiceProxy* service() const { return theService; }
 
protected:
  std::string theCategory;
  bool theTkTrajsAvailableFlag;
  float thePtCut;
 
private:
  int theMuonHitsOption;
  float theProbCut;
  int theHitThreshold;
  float theDTChi2Cut;
  float theCSCChi2Cut;
  float theRPCChi2Cut;
  float theGEMChi2Cut;
  float theME0Chi2Cut;
  bool theCosmicFlag;
 
  edm::InputTag theDTRecHitLabel;
  edm::InputTag theCSCRecHitLabel;
  edm::InputTag theGEMRecHitLabel;
  edm::InputTag theME0RecHitLabel;
  edm::Handle<DTRecHitCollection> theDTRecHits;
  edm::Handle<CSCRecHit2DCollection> theCSCRecHits;
  edm::Handle<GEMRecHitCollection> theGEMRecHits;
  edm::Handle<ME0SegmentCollection> theME0RecHits;
  edm::EDGetTokenT<DTRecHitCollection> theDTRecHitToken;
  edm::EDGetTokenT<CSCRecHit2DCollection> theCSCRecHitToken;
  edm::EDGetTokenT<GEMRecHitCollection> theGEMRecHitToken;
  edm::EDGetTokenT<ME0SegmentCollection> theME0RecHitToken;
 
  int theSkipStation;
  int theTrackerSkipSystem;
  int theTrackerSkipSection;
 
  unsigned long long theCacheId_TRH;
 
  std::string thePropagatorName;
 
  bool theRPCInTheFit;
 
  double theRescaleErrorFactor;
 
  RefitDirection theRefitDirection;
 
  std::vector<int> theDYTthrs;
  int theDYTselector;
  bool theDYTupdator;
  bool theDYTuseAPE;
  bool theDYTParThrsMode;
  edm::ParameterSet theDYTthrsParameters;
  reco::DYTInfo* dytInfo;
 
  edm::ESGetToken<TrajectoryFitter, TrajectoryFitterRecord> theFitterToken;
  std::unique_ptr<TrajectoryFitter> theFitter;
 
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theTrackerRecHitBuilderToken;
  const TransientTrackingRecHitBuilder* theTrackerRecHitBuilder;
  TkClonerImpl hitCloner;
 
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theMuonRecHitBuilderToken;
  const TransientTrackingRecHitBuilder* theMuonRecHitBuilder;
 
  const MuonServiceProxy* theService;
  const edm::Event* theEvent;
 
  edm::EDGetTokenT<CSCSegmentCollection> CSCSegmentsToken;
  edm::EDGetTokenT<DTRecSegment4DCollection> all4DSegmentsToken;
  edm::Handle<CSCSegmentCollection> CSCSegments;
  edm::Handle<DTRecSegment4DCollection> all4DSegments;
};
#endif