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

RecoTracker/TkTrackingRegions/interface/GlobalTrackingRegion.h
73 строки3 KB
Matti Kortelainen
Remove EventSetup from TrackingRegion method arguments
27 сен 2021, 16:29
27 сен 2021, 16:29c73b9b6
#ifndef GlobalTrackingRegion_H
#define GlobalTrackingRegion_H

/** \class GlobalTrackingRegion
 * An implementation of the TrackingRegion where the region of interest is
 * global, ie there are no constraints on the allowed direction of particles
 * of interest
 */

#include "RecoTracker/TkTrackingRegions/interface/TrackingRegion.h"
#include "RecoTracker/TkMSParametrization/interface/MultipleScatteringParametrisationMaker.h"
#include <vector>

class GlobalTrackingRegion final : public TrackingRegion {
public:
  /** Construct from minimal track P_t, and origin size and position.
   *  The origin region is a cylinder of radius originRadius, half length 
   *  originHalfLength, positioned at "origin".
   *  This class DOES provide the possibility to displace the origin
   *  in the transverse plane. 
   *
   * if useMS == true, msmaker needs to be given
   */
  GlobalTrackingRegion(float ptMin,
                       const GlobalPoint& origin,
                       float originRadius,
                       float originHalfLength,
                       bool precise = false,
                       bool useMS = false,
                       const MultipleScatteringParametrisationMaker* msmaker = nullptr)
      : TrackingRegionBase(GlobalVector(0, 0, 0), origin, Range(-1 / ptMin, 1 / ptMin), originRadius, originHalfLength),
        theMSMaker(msmaker),
        thePrecise(precise),
        theUseMS(useMS) {}

  // obsolete constructor
  GlobalTrackingRegion(float ptMin = 1.,
                       float originRadius = 0.2,
                       float originHalfLength = 22.7,
                       float originZPos = 0.,
                       bool precise = false)
      : TrackingRegionBase(GlobalVector(0, 0, 0),
                           GlobalPoint(0, 0, originZPos),
                           Range(-1 / ptMin, 1 / ptMin),
                           originRadius,
                           originHalfLength),
        thePrecise(precise) {}

  TrackingRegion::Hits hits(const SeedingLayerSetsHits::SeedingLayer& layer) const override;

  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 override;

  /// 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.
  void checkTracks(reco::TrackCollection const& tracks, std::vector<bool>& mask) const override;

  std::unique_ptr<TrackingRegion> clone() const override { return std::make_unique<GlobalTrackingRegion>(*this); }

  std::string name() const override { return "GlobalTrackingRegion"; }
  std::string print() const override;

private:
  const MultipleScatteringParametrisationMaker* theMSMaker = nullptr;
  bool thePrecise = false;
  bool theUseMS = false;
};
#endif