DualReferenceTrajectory.h

Go to the documentation of this file.

#ifndef Alignment_ReferenceTrajectories_DualReferenceTrajectory_H
#define Alignment_ReferenceTrajectories_DualReferenceTrajectory_H

#include "Alignment/ReferenceTrajectories/interface/ReferenceTrajectoryBase.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

class ReferenceTrajectory;
namespace reco {
  class BeamSpot;
}

class DualReferenceTrajectory : public ReferenceTrajectoryBase {
public:
  typedef TransientTrackingRecHit::ConstRecHitContainer ConstRecHitContainer;

  DualReferenceTrajectory(const TrajectoryStateOnSurface &tsos,
                          const ConstRecHitContainer &forwardRecHits,
                          const ConstRecHitContainer &backwardRecHits,
                          const MagneticField *magField,
                          const reco::BeamSpot &beamSpot,
                          const ReferenceTrajectoryBase::Config &config);

  ~DualReferenceTrajectory() override {}

  DualReferenceTrajectory *clone() const override { return new DualReferenceTrajectory(*this); }

protected:
  DualReferenceTrajectory(unsigned int nPar, unsigned int nHits, const ReferenceTrajectoryBase::Config &config);

  virtual bool construct(const TrajectoryStateOnSurface &referenceTsos,
                         const ConstRecHitContainer &forwardRecHits,
                         const ConstRecHitContainer &backwardRecHits,
                         const MagneticField *magField,
                         const reco::BeamSpot &beamSpot);

  virtual ReferenceTrajectory *construct(const TrajectoryStateOnSurface &referenceTsos,
                                         const ConstRecHitContainer &recHits,
                                         const MagneticField *magField,
                                         const reco::BeamSpot &beamSpot,
                                         const bool revertDirection = false) const;

  virtual AlgebraicVector extractParameters(const TrajectoryStateOnSurface &referenceTsos) const;

  inline const PropagationDirection oppositeDirection(const PropagationDirection propDir) const {
    return (propDir == anyDirection) ? anyDirection : ((propDir == alongMomentum) ? oppositeToMomentum : alongMomentum);
  }

private:
  const double mass_;
  const MaterialEffects materialEffects_;
  const PropagationDirection propDir_;
  const bool useBeamSpot_;
};

#endif