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 "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"

class TSGForOI : public edm::global::EDProducer<> {
public:
  explicit TSGForOI(const edm::ParameterSet& iConfig);
  ~TSGForOI() 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 numOfMaxSeedsParam_;

  const unsigned int numOfLayersToTry_;

  const unsigned int numOfHitsToTry_;

  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 bool useStereoLayersInTEC_;

  const std::unique_ptr<TrajectoryStateUpdator> updator_;

  const edm::EDGetTokenT<MeasurementTrackerEvent> measurementTrackerTag_;

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

  const double tsosDiff_;

  const std::string propagatorName_;
  const std::string theCategory;

  void findSeedsOnLayer(const TrackerTopology* tTopo,
                        const GeometricSearchDet& layer,
                        const TrajectoryStateOnSurface& tsosAtIP,
                        const Propagator& propagatorAlong,
                        const Propagator& propagatorOpposite,
                        const reco::TrackRef l2,
                        edm::ESHandle<Chi2MeasurementEstimatorBase>& estimator_,
                        edm::Handle<MeasurementTrackerEvent>& measurementTrackerH,
                        unsigned int& numSeedsMade,
                        unsigned int& numOfMaxSeeds,
                        unsigned int& layerCount,
                        bool& analysedL2,
                        std::unique_ptr<std::vector<TrajectorySeed> >& out) const;

  double calculateSFFromL2(const reco::TrackRef track) const;

  int makeSeedsFromHits(const TrackerTopology* tTopo,
                        const GeometricSearchDet& layer,
                        const TrajectoryStateOnSurface& tsosAtIP,
                        std::vector<TrajectorySeed>& out,
                        const Propagator& propagatorAlong,
                        const MeasurementTrackerEvent& measurementTracker,
                        edm::ESHandle<Chi2MeasurementEstimatorBase>& estimator_,
                        unsigned int& numSeedsMade,
                        const double errorSF,
                        const double l2Eta) const;

  //Find compatability between two TSOSs
  double match_Chi2(const TrajectoryStateOnSurface& tsos1, const TrajectoryStateOnSurface& tsos2) const;
};

#endif