TSGForOIFromL2.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackerSeedGenerator_TSGForOIFromL2_H
#define RecoMuon_TrackerSeedGenerator_TSGForOIFromL2_H

#include "DataFormats/TrackReco/interface/Track.h"
#include "FWCore/Framework/interface/global/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"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"

class TSGForOIFromL2 : public edm::global::EDProducer<> {
 
  public:
 
    explicit TSGForOIFromL2(const edm::ParameterSet & iConfig);
    ~TSGForOIFromL2() override;
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
    void produce(edm::StreamID sid, edm::Event& iEvent, const edm::EventSetup& iSetup) const override;

  private:
 
    const edm::EDGetTokenT<reco::TrackCollection> src_;

    const unsigned int maxSeeds_;

    const unsigned int maxHitlessSeeds_;

    const unsigned int maxHitSeeds_;

    const unsigned int numOfLayersToTry_;

    const unsigned int numOfHitsToTry_;

    const unsigned int numL2ValidHitsCutAllEta_;
    const unsigned int numL2ValidHitsCutAllEndcap_;

    const double fixedErrorRescalingForHits_;
    const double fixedErrorRescalingForHitless_;

    const bool adjustErrorsDynamicallyForHits_;
    const bool adjustErrorsDynamicallyForHitless_;

    const std::string estimatorName_;

    const double minEtaForTEC_;

    const double maxEtaForTOB_;

    const bool useHitLessSeeds_;

    const std::unique_ptr<TrajectoryStateUpdator> updator_;

    const edm::EDGetTokenT<MeasurementTrackerEvent> measurementTrackerTag_;

    const double pT1_,pT2_,pT3_;
    const double eta1_,eta2_,eta3_,eta4_,eta5_,eta6_,eta7_;
    const double SF1_,SF2_,SF3_,SF4_,SF5_,SF6_;

    const double tsosDiff1_;
    const double tsosDiff2_;

    const std::string propagatorName_;
    const std::string theCategory_;

    void makeSeedsWithoutHits(
        const GeometricSearchDet& layer,
        const TrajectoryStateOnSurface& tsos,
        const Propagator& propagatorAlong,
        edm::ESHandle<Chi2MeasurementEstimatorBase>& estimator,
        double errorSF,
        unsigned int& hitlessSeedsMade,
        unsigned int& numSeedsMade,
        std::vector<TrajectorySeed>& out) const;

    void makeSeedsFromHits(
        const GeometricSearchDet& layer,
        const TrajectoryStateOnSurface& tsos,
        const Propagator& propagatorAlong,
        edm::ESHandle<Chi2MeasurementEstimatorBase>& estimator,
        edm::Handle<MeasurementTrackerEvent>& measurementTracker,
        double errorSF,
        unsigned int& hitSeedsMade,
        unsigned int& numSeedsMade,
        unsigned int& layerCount,
        std::vector<TrajectorySeed>& out) const;
    
    double calculateSFFromL2(const reco::TrackRef track) const;
 
    double match_Chi2(const TrajectoryStateOnSurface& tsos1,
                      const TrajectoryStateOnSurface& tsos2) const;
 
};

#endif