TrajectoryFitter.h

Go to the documentation of this file.

#ifndef _TrajectoryFitter_H_
#define _TrajectoryFitter_H_
 
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
 
#include <memory>
 
class TrajectorySeed;
class TrajectoryStateOnSurface;
class TkCloner;
 
#include "TrackingTools/TrackFitters/interface/TrajectoryFitterRecord.h"
 
class TrajectoryFitter {
public:
  enum fitType { standard = 0, looper = 1 };
 
  typedef TrajectoryFitterRecord Record;
  typedef Trajectory::RecHitContainer RecHitContainer;
 
  virtual ~TrajectoryFitter() {}
  virtual std::unique_ptr<TrajectoryFitter> clone() const = 0;
 
  // FIXME a prototype: final inplementaiton may differ
  virtual void setHitCloner(TkCloner const*) = 0;
 
  // new interface return one trajectory: if fit fails trajectory is invalid...
  virtual Trajectory fitOne(const Trajectory& traj, fitType type = standard) const = 0;
  virtual Trajectory fitOne(const TrajectorySeed& seed,
                            const RecHitContainer& hits,
                            fitType typee = standard) const = 0;
  virtual Trajectory fitOne(const TrajectorySeed& seed,
                            const RecHitContainer& hits,
                            const TrajectoryStateOnSurface& tsos,
                            fitType type = standard) const = 0;
 
  // backward compatible interface...
  std::vector<Trajectory> fit(const Trajectory& traj, fitType type = standard) const {
    return makeVect(fitOne(traj, type));
  }
 
  std::vector<Trajectory> fit(const TrajectorySeed& seed, const RecHitContainer& hits, fitType type = standard) const {
    return makeVect(fitOne(seed, hits, type));
  }
  std::vector<Trajectory> fit(const TrajectorySeed& seed,
                              const RecHitContainer& hits,
                              const TrajectoryStateOnSurface& tsos,
                              fitType type = standard) const {
    return makeVect(fitOne(seed, hits, tsos, type));
  }
 
private:
  static std::vector<Trajectory> makeVect(Trajectory&& outTraj) {
    if (outTraj.isValid())
      return std::vector<Trajectory>(1, outTraj);
    return std::vector<Trajectory>();
  }
};
 
#endif