TransientInitialStateEstimator.h

Go to the documentation of this file.

#ifndef TransientInitialStateEstimator_H
#define TransientInitialStateEstimator_H

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

#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkClonerImpl.h"

#include <utility>

class Propagator;
class GeomDet;
class Trajectory;
class TrackingComponentsRecord;
namespace edm {
  class EventSetup;
}


class TransientInitialStateEstimator {
public:
  typedef TrajectoryStateOnSurface TSOS;

  TransientInitialStateEstimator(const edm::ParameterSet& conf, edm::ConsumesCollector iC);
  void setEventSetup(const edm::EventSetup& es, const TkClonerImpl& hc);

  std::pair<TrajectoryStateOnSurface, const GeomDet*> innerState(const Trajectory& traj, bool doBackFit = true) const;

private:
  const edm::ESGetToken<Propagator, TrackingComponentsRecord> thePropagatorAlongToken;
  const edm::ESGetToken<Propagator, TrackingComponentsRecord> thePropagatorOppositeToken;
  const Propagator* thePropagatorAlong;
  const Propagator* thePropagatorOpposite;  // not used? can we remove it?
  TkClonerImpl theHitCloner;
  const int theNumberMeasurementsForFit;
};

#endif