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/Framework/interface/ESHandle.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 "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 edm {class Event;}
namespace reco {class TransientTrack;}

class TrajectoryStateOnSurface;
class TrackerTopology;

class MuonDetLayerMeasurements;
class MuonServiceProxy;
class Trajectory;

class TrajectoryFitter;

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;

    std::string theFitterName;
    std::unique_ptr<TrajectoryFitter> theFitter;

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

    std::string theMuonRecHitBuilderName;
    edm::ESHandle<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