TrajectoryFactoryBase.h

Go to the documentation of this file.

#ifndef Alignment_ReferenceTrajectories_TrajectoryFactoryBase_h
#define Alignment_ReferenceTrajectories_TrajectoryFactoryBase_h

#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentAlgorithmBase.h"
#include "Alignment/ReferenceTrajectories/interface/ReferenceTrajectoryBase.h"

#include "TrackingTools/PatternTools/interface/Trajectory.h"

#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

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



namespace reco { class BeamSpot;}

class TrajectoryFactoryBase
{

public:

  typedef ReferenceTrajectoryBase::ReferenceTrajectoryPtr ReferenceTrajectoryPtr;
  typedef ReferenceTrajectoryBase::MaterialEffects MaterialEffects;
  typedef AlignmentAlgorithmBase::ConstTrajTrackPair ConstTrajTrackPair;
  typedef AlignmentAlgorithmBase::ConstTrajTrackPairCollection ConstTrajTrackPairCollection;
  typedef std::vector< ReferenceTrajectoryPtr > ReferenceTrajectoryCollection;
  typedef std::pair< TrajectoryStateOnSurface, TransientTrackingRecHit::ConstRecHitContainer > TrajectoryInput;
  typedef std::vector< TrajectoryStateOnSurface > ExternalPredictionCollection;

  TrajectoryFactoryBase( const edm::ParameterSet & config );
  virtual ~TrajectoryFactoryBase( void );

  virtual const ReferenceTrajectoryCollection trajectories(const edm::EventSetup &setup,
                               const ConstTrajTrackPairCollection &tracks,
                               const reco::BeamSpot &beamSpot) const = 0;

  virtual const ReferenceTrajectoryCollection trajectories(const edm::EventSetup &setup,
                               const ConstTrajTrackPairCollection &tracks,
                               const ExternalPredictionCollection &external,
                               const reco::BeamSpot &beamSpot) const = 0;

  virtual TrajectoryFactoryBase* clone( void ) const = 0;

  inline MaterialEffects materialEffects( void ) const { return theMaterialEffects; }
  inline PropagationDirection propagationDirection( void ) const { return thePropDir; }
  inline const edm::ParameterSet &configuration() const { return theConfig; }

protected:

  virtual const TrajectoryInput innermostStateAndRecHits( const ConstTrajTrackPair & track ) const;
  virtual const Trajectory::DataContainer orderedTrajectoryMeasurements( const Trajectory & trajectory ) const;
  bool sameSurface( const Surface& s1, const Surface& s2 ) const;
  bool useRecHit( const TransientTrackingRecHit::ConstRecHitPointer& hitPtr ) const;

private:

  MaterialEffects materialEffects( const std::string & strME ) const;
  PropagationDirection propagationDirection( const std::string & strPD ) const;

  MaterialEffects theMaterialEffects;
  PropagationDirection thePropDir;
  const edm::ParameterSet theConfig; // need to keep for possible re-use after constructor... :-(

  bool theUseWithoutDet;
  bool theUseInvalidHits;
  bool theUseProjectedHits;
  
protected:

  bool theUseBeamSpot;
};


#endif