GlobalTrackingRegion.cc

Go to the documentation of this file.

#include "RecoTracker/TkTrackingRegions/interface/GlobalTrackingRegion.h"
#include <cmath>
#include <sstream>
#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "RecoTracker/TkTrackingRegions/interface/HitEtaCheck.h"
#include "RecoTracker/TkTrackingRegions/interface/HitRCheck.h"
#include "RecoTracker/TkTrackingRegions/interface/HitZCheck.h"
 
#include "RecoTracker/TkMSParametrization/interface/PixelRecoUtilities.h"
#include "RecoTracker/TkMSParametrization/interface/MultipleScatteringParametrisation.h"
#include "RecoTracker/TkMSParametrization/interface/PixelRecoPointRZ.h"
#include "RecoTracker/TkMSParametrization/interface/PixelRecoLineRZ.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
 
template <class T>
T sqr(T t) {
  return t * t;
}
 
using namespace GeomDetEnumerators;
 
std::string GlobalTrackingRegion::print() const {
  std::ostringstream str;
  str << TrackingRegionBase::print() << "precise: " << thePrecise;
  return str.str();
}
 
TrackingRegion::Hits GlobalTrackingRegion::hits(const edm::EventSetup& es,
                                                const SeedingLayerSetsHits::SeedingLayer& layer) const {
  return layer.hits();
}
 
std::unique_ptr<HitRZCompatibility> GlobalTrackingRegion::checkRZ(const DetLayer* layer,
                                                                  const Hit& outerHit,
                                                                  const edm::EventSetup& iSetup,
                                                                  const DetLayer* outerlayer,
                                                                  float lr,
                                                                  float gz,
                                                                  float dr,
                                                                  float dz) const {
  bool isBarrel = layer->isBarrel();
  bool isPixel = (layer->subDetector() == PixelBarrel || layer->subDetector() == PixelEndcap);
 
  if UNLIKELY (!outerlayer) {
    GlobalPoint ohit = outerHit->globalPosition();
    lr = std::sqrt(sqr(ohit.x() - origin().x()) + sqr(ohit.y() - origin().y()));
    gz = ohit.z();
    dr = outerHit->errorGlobalR();
    dz = outerHit->errorGlobalZ();
  }
 
  PixelRecoPointRZ outerred(lr, gz);
 
  PixelRecoPointRZ vtxR = (gz > origin().z() + originZBound())
                              ? PixelRecoPointRZ(-originRBound(), origin().z() + originZBound())
                              : PixelRecoPointRZ(originRBound(), origin().z() + originZBound());
  PixelRecoPointRZ vtxL = (gz < origin().z() - originZBound())
                              ? PixelRecoPointRZ(-originRBound(), origin().z() - originZBound())
                              : PixelRecoPointRZ(originRBound(), origin().z() - originZBound());
 
  if UNLIKELY ((!thePrecise) && (isPixel)) {
    auto VcotMin = PixelRecoLineRZ(vtxR, outerred).cotLine();
    auto VcotMax = PixelRecoLineRZ(vtxL, outerred).cotLine();
    return std::make_unique<HitEtaCheck>(isBarrel, outerred, VcotMax, VcotMin);
  }
 
  constexpr float nSigmaPhi = 3.;
 
  dr *= nSigmaPhi;
  dz *= nSigmaPhi;
 
  PixelRecoPointRZ outerL, outerR;
  if (isBarrel) {
    outerL = PixelRecoPointRZ(lr, gz - dz);
    outerR = PixelRecoPointRZ(lr, gz + dz);
  } else if (gz > 0) {
    outerL = PixelRecoPointRZ(lr + dr, gz);
    outerR = PixelRecoPointRZ(lr - dr, gz);
  } else {
    outerL = PixelRecoPointRZ(lr - dr, gz);
    outerR = PixelRecoPointRZ(lr + dr, gz);
  }
 
  auto corr = isBarrel ? dz : dr;
 
  SimpleLineRZ leftLine(vtxL, outerL);
  SimpleLineRZ rightLine(vtxR, outerR);
  HitRZConstraint rzConstraint(leftLine, rightLine);
 
  if UNLIKELY (theUseMS) {
    MultipleScatteringParametrisation iSigma(layer, iSetup);
    PixelRecoPointRZ vtxMean(0., origin().z());
 
    float innerScatt = 3.f * (outerlayer ? iSigma(ptMin(), vtxMean, outerred, outerlayer->seqNum())
                                         : iSigma(ptMin(), vtxMean, outerred));
 
    float cotTheta = SimpleLineRZ(vtxMean, outerred).cotLine();
 
    if (isBarrel) {
      float sinTheta = 1 / std::sqrt(1 + sqr(cotTheta));
      corr = innerScatt / sinTheta + dz;
    } else {
      float cosTheta = 1 / std::sqrt(1 + sqr(1 / cotTheta));
      corr = innerScatt / cosTheta + dr;
    }
  }
 
  if (isBarrel) {
    return std::make_unique<HitZCheck>(rzConstraint, HitZCheck::Margin(corr, corr));
  } else {
    return std::make_unique<HitRCheck>(rzConstraint, HitRCheck::Margin(corr, corr));
  }
}
 
void GlobalTrackingRegion::checkTracks(reco::TrackCollection const& tracks, std::vector<bool>& mask) const {
  const math::XYZPoint regOrigin(origin().x(), origin().y(), origin().z());
 
  assert(mask.size() == tracks.size());
  int i = -1;
  for (auto const& track : tracks) {
    i++;
    if (mask[i])
      continue;
 
    if (track.pt() < ptMin()) {
      continue;
    }
    if (std::abs(track.dxy(regOrigin)) > originRBound()) {
      continue;
    }
    if (std::abs(track.dz(regOrigin)) > originZBound()) {
      continue;
    }
 
    mask[i] = true;
  }
}