ThirdHitRZPrediction.h

Go to the documentation of this file.

#ifndef ThirdHitRZPrediction_H
#define ThirdHitRZPrediction_H

#include <algorithm>

#include "RecoPixelVertexing/PixelTriplets/interface/ThirdHitRZPredictionBase.h"

template<class Propagator>
class ThirdHitRZPrediction : public ThirdHitRZPredictionBase {
public:

  ThirdHitRZPrediction() : ThirdHitRZPredictionBase(), thePropagator(nullptr) {}
  ThirdHitRZPrediction(const Propagator *propagator, float tolerance, const DetLayer* layer = nullptr) :
      ThirdHitRZPredictionBase(tolerance, layer), thePropagator(propagator) {}

  inline Range operator()(const DetLayer *layer = nullptr);
  inline Range operator()(float rORz) const { return (*this)(rORz, *thePropagator); }
  inline Range operator()(float rORz, const Propagator &propagator) const;

  void initPropagator(const Propagator *propagator) { thePropagator = propagator; }

private:
  float transform( const Propagator &propagator, float rOrZ) const
  { return theBarrel ? propagator.zAtR(rOrZ) : propagator.rAtZ(rOrZ); }

  const Propagator *thePropagator;
};

template<class Propagator>
typename ThirdHitRZPrediction<Propagator>::Range
ThirdHitRZPrediction<Propagator>::operator()(const DetLayer *layer)
{
  if (layer) initLayer(layer);
  if (!theBarrel && !theForward) return Range(0., 0.);
  float v1 = transform(*thePropagator, theDetRange.min());
  float v2 = transform(*thePropagator, theDetRange.max());
  if (v1 > v2) std::swap(v1, v2);
  return Range(v1 - theTolerance.left(), v2 + theTolerance.right());
}

template<class Propagator>
typename ThirdHitRZPrediction<Propagator>::Range
ThirdHitRZPrediction<Propagator>::operator()(float rOrZ, const Propagator &propagator) const
{
  float v = transform(propagator, rOrZ);
  return Range(v - theTolerance.left(), v + theTolerance.right());
}

#endif