RecoTracker/TkTrackingRegions/interface/TrackingRegion.h - master - githubmirror/cmssw

RecoTracker/TkTrackingRegions/interface/TrackingRegion.h
148 строк5 KB
cmsbuild
Merge pull request #35452 from CeliaFernandez/fix_UBSAN_issue35036_MUO-TSGFromL2Muon
07 окт 2021, 20:02
07 окт 2021, 20:0215f50df
#ifndef TrackingRegion_H
#define TrackingRegion_H

/** \class TrackingRegion
 * kinematic data common to 
 * some concreate implementations of TrackingRegion.
 */

#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) {}

  /// the direction around which region is constructed
  GlobalVector const& direction() const { return theDirection; }
  GlobalVector const& unitDirection() const { return theUnitDirection; }

  float phiDirection() const { return thePhi; }

  /** The origin (centre,vertex) of the region. <BR> 
  *  The origin with bounds is ment to constraint point of the <BR>
  *  closest approach of the track to the beam line
  */
  GlobalPoint const& origin() const { return theVertexPos; }

  /// bounds the particle vertex in the transverse plane
  float originRBound() const { return theVertexRBound; }

  /// bounds the particle vertex in the longitudinal plane
  float originZBound() const { return theVertexZBound; }

  /// minimal pt of interest
  float ptMin() const { return thePtMin; }

  /// inverse pt range
  Range invPtRange() const { return theInvPtRange; }

  /// utility to check eta/theta hit compatibility with region constraints
  /// and outer hit constraint
  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;

  /// get hits from layer compatible with region constraints
  virtual Hits hits(const SeedingLayerSetsHits::SeedingLayer& layer) const = 0;

  /// Set the elements of the mask corresponding to the tracks that are compatable with the region.
  /// Does not reset the elements corresponding to the tracks that are not compatible.
  virtual void checkTracks(reco::TrackCollection const& tracks, std::vector<bool>& mask) const = 0;

  /// return a boolean mask over the TrackCollection reflecting the compatibility of each track with the region constraints
  std::vector<bool> checkTracks(reco::TrackCollection const& tracks) const {
    std::vector<bool> region_mask(tracks.size(), false);
    checkTracks(tracks, region_mask);
    return region_mask;
  }
  /// clone region with new vertex position
  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