RecoTracker/PixelSeeding/interface/CAHitTripletGenerator.h - master - githubmirror/cmssw

RecoTracker/PixelSeeding/interface/CAHitTripletGenerator.h
136 строк5 KB
mmusich
fix RecoTracker/Pixel* subsystems inclusions guards to comply withc cmssw rule 4.1
11 апр 2023, 10:15
11 апр 2023, 10:15e88c626
#ifndef RecoTracker_PixelSeeding_CAHitTripletGenerator_h
#define RecoTracker_PixelSeeding_CAHitTripletGenerator_h

#include "RecoTracker/TkSeedingLayers/interface/SeedComparitorFactory.h"
#include "RecoTracker/TkSeedingLayers/interface/SeedComparitor.h"
#include "RecoTracker/PixelTrackFitting/interface/RZLine.h"

#include "RecoTracker/TkMSParametrization/interface/PixelRecoUtilities.h"
#include "RecoTracker/TkMSParametrization/interface/LongitudinalBendingCorrection.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "RecoTracker/TkHitPairs/interface/HitPairGeneratorFromLayerPair.h"
#include "RecoTracker/TkHitPairs/interface/LayerHitMapCache.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"

#include "RecoTracker/TkHitPairs/interface/IntermediateHitDoublets.h"

#include "RecoTracker/PixelSeeding/interface/OrderedHitSeeds.h"
#include "RecoTracker/PixelSeeding/interface/CACut.h"
class TrackingRegion;
class SeedingLayerSetsHits;

namespace edm {
  class Event;
  class EventSetup;
  class ParameterSetDescription;
}  // namespace edm

class CAHitTripletGenerator {
public:
  typedef LayerHitMapCache LayerCacheType;

  static constexpr unsigned int minLayers = 3;
  typedef OrderedHitSeeds ResultType;

public:
  CAHitTripletGenerator(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC) : CAHitTripletGenerator(cfg, iC) {}
  CAHitTripletGenerator(const edm::ParameterSet& cfg, edm::ConsumesCollector& iC);

  ~CAHitTripletGenerator() = default;

  static void fillDescriptions(edm::ParameterSetDescription& desc);
  static const char* fillDescriptionsLabel() { return "caHitTriplet"; }

  void initEvent(const edm::Event& ev, const edm::EventSetup& es);

  void hitNtuplets(const IntermediateHitDoublets& regionDoublets,
                   std::vector<OrderedHitSeeds>& result,
                   const SeedingLayerSetsHits& layers);

private:
  LayerCacheType theLayerCache;

  std::unique_ptr<SeedComparitor> theComparitor;

  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theFieldToken;
  const MagneticField* theField = nullptr;

  class QuantityDependsPtEval {
  public:
    QuantityDependsPtEval(float v1, float v2, float c1, float c2)
        : value1_(v1), value2_(v2), curvature1_(c1), curvature2_(c2) {}

    float value(float curvature) const {
      if (value1_ == value2_)  // not enabled
        return value1_;

      if (curvature1_ < curvature)
        return value1_;
      if (curvature2_ < curvature && curvature <= curvature1_)
        return value2_ + (curvature - curvature2_) / (curvature1_ - curvature2_) * (value1_ - value2_);
      return value2_;
    }

  private:
    const float value1_;
    const float value2_;
    const float curvature1_;
    const float curvature2_;
  };

  // Linear interpolation (in curvature) between value1 at pt1 and
  // value2 at pt2. If disabled, value2 is given (the point is to
  // allow larger/smaller values of the quantity at low pt, so it
  // makes more sense to have the high-pt value as the default).

  class QuantityDependsPt {
  public:
    explicit QuantityDependsPt(const edm::ParameterSet& pset)
        : value1_(pset.getParameter<double>("value1")),
          value2_(pset.getParameter<double>("value2")),
          pt1_(pset.getParameter<double>("pt1")),
          pt2_(pset.getParameter<double>("pt2")),
          enabled_(pset.getParameter<bool>("enabled")) {
      if (enabled_ && pt1_ >= pt2_)
        throw cms::Exception("Configuration") << "CAHitTripletGenerator::QuantityDependsPt: pt1 (" << pt1_
                                              << ") needs to be smaller than pt2 (" << pt2_ << ")";
      if (pt1_ <= 0)
        throw cms::Exception("Configuration")
            << "CAHitTripletGenerator::QuantityDependsPt: pt1 needs to be > 0; is " << pt1_;
      if (pt2_ <= 0)
        throw cms::Exception("Configuration")
            << "CAHitTripletGenerator::QuantityDependsPt: pt2 needs to be > 0; is " << pt2_;
    }

    QuantityDependsPtEval evaluator(const MagneticField& field) const {
      if (enabled_) {
        return QuantityDependsPtEval(value1_,
                                     value2_,
                                     PixelRecoUtilities::curvature(1.f / pt1_, field),
                                     PixelRecoUtilities::curvature(1.f / pt2_, field));
      }
      return QuantityDependsPtEval(value2_, value2_, 0.f, 0.f);
    }

  private:
    const float value1_;
    const float value2_;
    const float pt1_;
    const float pt2_;
    const bool enabled_;
  };

  const float extraHitRPhitolerance;

  const QuantityDependsPt maxChi2;
  const bool useBendingCorrection;

  CACut caThetaCut;
  CACut caPhiCut;
  const float caHardPtCut = 0.f;
};

#endif