SeedTransformer.h

Go to the documentation of this file.

#ifndef SeedTransformer_H
#define SeedTransformer_H

// Base class header
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"

#include <vector>

namespace edm {
  class ParameterSet;
  class EventSetup;
  class ConsumesCollector;
}  // namespace edm
class Trajectory;
class TrajectorySeed;
class TrajectoryStateOnSurface;
class GlobalTrackingGeometry;
class MagneticField;
class TransientTrackingRecHitBuilder;
class Propagator;
class GlobalTrackingGeometryRecord;
class IdealMagneticFieldRecord;
class TransientRecHitRecord;
class TrackingComponentsRecord;

class SeedTransformer {
public:
  SeedTransformer(const edm::ParameterSet&, edm::ConsumesCollector);

  virtual ~SeedTransformer();

  // Operations
  void setServices(const edm::EventSetup&);

  std::vector<Trajectory> seedTransform(const TrajectorySeed&) const;
  TrajectoryStateOnSurface seedTransientState(const TrajectorySeed&) const;

protected:
private:
  const GlobalTrackingGeometry* theTrackingGeometry;
  const MagneticField* theMagneticField;
  const TrajectoryFitter* theFitter;
  const TransientTrackingRecHitBuilder* theMuonRecHitBuilder;
  const Propagator* thePropagator;

  edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> theTrackingGeometryToken;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theMagneticFieldToken;
  edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> theFitterToken;
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> theMuonRecHitBuilderToken;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> thePropagatorToken;

  unsigned int nMinRecHits;

  double errorRescale;

  bool useSubRecHits;
};
#endif