TSGForRoadSearch.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackerSeedGenerator_TSGForRoadSearch_H
#define RecoMuon_TrackerSeedGenerator_TSGForRoadSearch_H

#include "RecoMuon/TrackerSeedGenerator/interface/TrackerSeedGenerator.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include <RecoTracker/MeasurementDet/interface/MeasurementTracker.h>
#include <RecoTracker/MeasurementDet/interface/MeasurementTrackerEvent.h>
#include "MagneticField/Engine/interface/MagneticField.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"

#include "TrackingTools/KalmanUpdators/interface/Chi2MeasurementEstimator.h"

#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "RecoMuon/TrackingTools/interface/MuonErrorMatrix.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

class TrackingRegion;
class MuonServiceProxy;
class TrajectoryStateUpdator;
class TrackerTopology;

class TSGForRoadSearch : public TrackerSeedGenerator {

public:
  typedef std::vector<TrajectorySeed> BTSeedCollection;  
  typedef std::pair<const Trajectory*, reco::TrackRef> TrackCand;

  TSGForRoadSearch(const edm::ParameterSet &pset,edm::ConsumesCollector& IC);

  ~TSGForRoadSearch() override;

  void init(const MuonServiceProxy *service) override;
  void setEvent(const edm::Event &event) override;

  void  trackerSeeds(const TrackCand&, const TrackingRegion&, const TrackerTopology *, BTSeedCollection&) override;

private:
  //concrete implementation
  void makeSeeds_0(const reco::Track &,std::vector<TrajectorySeed> &);
  void makeSeeds_1(const reco::Track &,std::vector<TrajectorySeed> &);
  void makeSeeds_2(const reco::Track &,std::vector<TrajectorySeed> &);
  void makeSeeds_3(const reco::Track &,std::vector<TrajectorySeed> &);
  void makeSeeds_4(const reco::Track &,std::vector<TrajectorySeed> &);

private:
  bool IPfts(const reco::Track &, FreeTrajectoryState &);
  bool notAtIPtsos(TrajectoryStateOnSurface & state);

  bool theAdjustAtIp;

  void pushTrajectorySeed(const reco::Track & muon, std::vector<DetLayer::DetWithState > & compatible, PropagationDirection direction, std::vector<TrajectorySeed>& result)const;
  edm::ParameterSet theConfig;

  edm::ESHandle<MeasurementTracker> theMeasurementTracker;
  edm::ESHandle<GeometricSearchTracker> theGeometricSearchTracker;

  edm::InputTag theMeasurementTrackerEventTag;
  edm::EDGetTokenT<MeasurementTrackerEvent> theMeasurementTrackerEventToken;
  const MeasurementTrackerEvent * theMeasurementTrackerEvent;

  TrajectoryStateUpdator * theUpdator;
  const MuonServiceProxy * theProxyService;

  unsigned int theOption;
  bool theCopyMuonRecHit;
  bool theManySeeds;
  std::string thePropagatorName;
  edm::ESHandle<Propagator> theProp;
  std::string thePropagatorCompatibleName;
  edm::ESHandle<Propagator> thePropCompatible;
  Chi2MeasurementEstimator * theChi2Estimator;
  std::string theCategory;

  MuonErrorMatrix * theErrorMatrixAdjuster;
};


#endif