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 TrackerRecoGeometryRecord;

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<GeometricSearchTracker> theGeometricSearchTracker;
  edm::ESGetToken<GeometricSearchTracker, TrackerRecoGeometryRecord> theGeometricSearchTrackerToken;

  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;
  std::string thePropagatorCompatibleName;
  Chi2MeasurementEstimator *theChi2Estimator;
  std::string theCategory;

  MuonErrorMatrix *theErrorMatrixAdjuster;
};

#endif