TSGForOI.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackerSeedGenerator_TSGForOI_H
#define RecoMuon_TrackerSeedGenerator_TSGForOI_H

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoTracker/MeasurementDet/interface/MeasurementTrackerEvent.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/KalmanUpdators/interface/Chi2MeasurementEstimator.h"
#include "TrackingTools/KalmanUpdators/interface/KFUpdator.h"
#include "TrackingTools/MeasurementDet/interface/MeasurementDet.h"
#include "TrackingTools/PatternTools/interface/TrajMeasLessEstim.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/GeomPropagators/interface/StateOnTrackerBound.h"

class TSGForOI : public edm::stream::EDProducer<> {
public:
    explicit TSGForOI(const edm::ParameterSet & iConfig);
    virtual ~TSGForOI();
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
    virtual void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;
private:
    const edm::EDGetTokenT<reco::TrackCollection> src_;

    unsigned int numOfMaxSeeds_;

    const unsigned int numOfLayersToTry_;

    const unsigned int numOfHitsToTry_;

    const double fixedErrorRescalingForHits_;
    const double fixedErrorRescalingForHitless_;

    const bool adjustErrorsDyanmicallyForHits_;
    const bool adjustErrorsDyanmicallyForHitless_;

    const std::string estimatorName_;

    const edm::EDGetTokenT<MeasurementTrackerEvent> measurementTrackerTag_;

    const double minEtaForTEC_;

    const double maxEtaForTOB_;

    const bool useHitlessSeeds_;

    const bool useHybridSeeds_;

    Plane::PlanePointer dummyPlane_;

    std::unique_ptr<TrajectoryStateUpdator> updator_;

    const double pT1_,pT2_,pT3_;
    const double eta1_,eta2_;
    const double SF1_,SF2_,SF3_,SF4_,SF5_;

    const double tsosDiffDeltaR_;

    bool foundCompatibleDet_;
    bool analysedL2_;
    bool useHitsInHybrid_;
    unsigned int numSeedsMade_;
    unsigned int layerCount_;

    std::string theCategory;

    edm::ESHandle<MagneticField>          magfield_;
    edm::ESHandle<Propagator>             propagatorAlong_;
    edm::ESHandle<Propagator>             propagatorOpposite_;
    edm::ESHandle<GlobalTrackingGeometry> geometry_;
    edm::Handle<MeasurementTrackerEvent>  measurementTracker_;

    void findSeedsOnLayer(const GeometricSearchDet &layer,
            const TrajectoryStateOnSurface &tsosAtIP,
            const TrajectoryStateOnSurface &tsosAtMuonSystem,
            const Propagator& propagatorAlong,
            const Propagator& propagatorOpposite,
            const reco::TrackRef l2,
            std::unique_ptr<std::vector<TrajectorySeed> >& seeds);

    double calculateSFFromL2(const GeometricSearchDet& layer,
            const TrajectoryStateOnSurface &tsosAtMuonSystem,
            const TrajectoryStateOnSurface &tsosOnLayer,
            const Propagator& propagatorOpposite,
            const reco::TrackRef track);

    int makeSeedsFromHits(const GeometricSearchDet &layer,
            const TrajectoryStateOnSurface &state,
            std::vector<TrajectorySeed> &out,
            const Propagator& propagatorAlong,
            const MeasurementTrackerEvent &mte,
            double errorSF);

    edm::ESHandle<Chi2MeasurementEstimatorBase>   estimator_;
};

#endif