TrackingRegion.h

Go to the documentation of this file.

#ifndef TrackingRegion_H
#define TrackingRegion_H

#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"

#include "RecoTracker/TkMSParametrization/interface/PixelRecoRange.h"
#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "RecoTracker/TkTrackingRegions/interface/HitRZCompatibility.h"
#include "RecoTracker/TkTrackingRegions/interface/HitEtaCheck.h"
#include "RecoTracker/TkTrackingRegions/interface/HitRCheck.h"
#include "RecoTracker/TkTrackingRegions/interface/HitZCheck.h"

#include "TrackingTools/TransientTrackingRecHit/interface/SeedingLayerSetsHits.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include <utility>

#include <sstream>
#include <memory>
#include <vector>
#include <string>

class DetLayer;
class HitRZCompatibility;
class BarrelDetLayer;
class ForwardDetLayer;

namespace edm {
  class Event;
}

class TrackingRegion {
public:
public:
  virtual ~TrackingRegion() {}
  typedef PixelRecoRange<float> Range;
  typedef SeedingLayerSetsHits::ConstRecHitPointer Hit;
  typedef SeedingLayerSetsHits::Hits Hits;

public:
  TrackingRegion() {}

  TrackingRegion(const GlobalVector& direction,
                 const GlobalPoint& originPos,
                 const Range& invPtRange,
                 const float& originRBound,
                 const float& originZBound)
      : theDirection(direction),
        theUnitDirection(direction.unit()),
        theVertexPos(originPos),
        theInvPtRange(invPtRange),
        thePhi(direction.barePhi()),
        thePtMin(1.f / std::max(std::abs(invPtRange.max()), std::abs(invPtRange.min()))),
        theVertexRBound(originRBound),
        theVertexZBound(originZBound) {}

  GlobalVector const& direction() const { return theDirection; }
  GlobalVector const& unitDirection() const { return theUnitDirection; }

  float phiDirection() const { return thePhi; }

  GlobalPoint const& origin() const { return theVertexPos; }

  float originRBound() const { return theVertexRBound; }

  float originZBound() const { return theVertexZBound; }

  float ptMin() const { return thePtMin; }

  Range invPtRange() const { return theInvPtRange; }

  virtual std::unique_ptr<HitRZCompatibility> checkRZ(const DetLayer* layer,
                                                      const Hit& outerHit,
                                                      const DetLayer* outerlayer = nullptr,
                                                      float lr = 0,
                                                      float gz = 0,
                                                      float dr = 0,
                                                      float dz = 0) const = 0;

  virtual Hits hits(const SeedingLayerSetsHits::SeedingLayer& layer) const = 0;

  virtual void checkTracks(reco::TrackCollection const& tracks, std::vector<bool>& mask) const = 0;

  std::vector<bool> checkTracks(reco::TrackCollection const& tracks) const {
    std::vector<bool> region_mask(tracks.size(), false);
    checkTracks(tracks, region_mask);
    return region_mask;
  }
  std::unique_ptr<TrackingRegion> restrictedRegion(const GlobalPoint& originPos,
                                                   const float& originRBound,
                                                   const float& originZBound) const {
    auto restr = clone();
    restr->theVertexPos = originPos;
    restr->theVertexRBound = originRBound;
    restr->theVertexZBound = originZBound;
    return restr;
  }

  virtual std::unique_ptr<TrackingRegion> clone() const = 0;

  virtual std::string name() const { return "TrackingRegion"; }
  virtual std::string print() const {
    std::ostringstream str;
    str << name() << " dir:" << theDirection << " vtx:" << theVertexPos << " dr:" << theVertexRBound
        << " dz:" << theVertexZBound << " pt:" << 1. / theInvPtRange.max();
    return str.str();
  }

  // void setDirection(const GlobalVector & dir ) { theDirection = dir; }

private:
  GlobalVector theDirection;      // this is not direction is momentum...
  GlobalVector theUnitDirection;  // the real direction
  GlobalPoint theVertexPos;
  Range theInvPtRange;
  float thePhi;
  float thePtMin;
  float theVertexRBound;
  float theVertexZBound;
};

using TrackingRegionBase = TrackingRegion;

#endif