PropagateToMuon.h

Go to the documentation of this file.

#ifndef MuonAnalysis_MuonAssociators_interface_PropagateToMuon_h
#define MuonAnalysis_MuonAssociators_interface_PropagateToMuon_h
//
//

#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/GeometrySurface/interface/BoundCylinder.h"
#include "DataFormats/GeometrySurface/interface/BoundDisk.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "RecoMuon/DetLayers/interface/MuonDetLayerGeometry.h"
struct DetLayer; // #include "TrackingTools/DetLayers/interface/DetLayer.h" // forward declaration can suffice
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"


class PropagateToMuon {
    public:
        explicit PropagateToMuon(const edm::ParameterSet & iConfig) ;
        ~PropagateToMuon() ;

        void init(const edm::EventSetup &iSetup) ;

        TrajectoryStateOnSurface extrapolate(const reco::Track &tk)     const { return extrapolate(startingState(tk)); }

        TrajectoryStateOnSurface extrapolate(const reco::Candidate &tk) const { return extrapolate(startingState(tk)); }

        //  don't ask me why SimVertexContainer is in edm namespace
        TrajectoryStateOnSurface extrapolate(const SimTrack &tk, const edm::SimVertexContainer &vtxs) const { return extrapolate(startingState(tk, vtxs)); }

        TrajectoryStateOnSurface extrapolate(const FreeTrajectoryState &state) const ;
    private:
        enum WhichTrack { None, TrackerTk, MuonTk, GlobalTk };
        enum WhichState { AtVertex, Innermost, Outermost };

        bool useSimpleGeometry_;

        bool useMB2_;

        bool fallbackToME1_;

        WhichTrack whichTrack_;
        WhichState whichState_;

        bool cosmicPropagation_;

        // needed services for track propagation
        edm::ESHandle<MagneticField> magfield_;
        edm::ESHandle<Propagator> propagator_, propagatorAny_, propagatorOpposite_;
        edm::ESHandle<MuonDetLayerGeometry> muonGeometry_;
        // simplified geometry for track propagation
        const  BoundCylinder *barrelCylinder_;
        const  BoundDisk *endcapDiskPos_[3], *endcapDiskNeg_[3];
        double barrelHalfLength_;
        std::pair<float,float> endcapRadii_[3];

        FreeTrajectoryState startingState(const reco::Candidate &reco) const ;

        FreeTrajectoryState startingState(const reco::Track &tk) const ;

        FreeTrajectoryState startingState(const SimTrack &tk, const edm::SimVertexContainer &vtxs) const ;

        TrajectoryStateOnSurface getBestDet(const TrajectoryStateOnSurface &tsos, const DetLayer *station) const;
};

#endif