DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2D.h - master - githubmirror/cmssw

DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2D.h
95 строк4 KB
Mario Masciovecchio
Strip overlap check for different product ID
25 июн 2025, 14:47
25 июн 2025, 14:47457caf3
#ifndef SiStripMatchedRecHit2D_H
#define SiStripMatchedRecHit2D_H

#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"

#include "TkCloner.h"

class SiStripMatchedRecHit2D final : public BaseTrackerRecHit {
public:
  typedef BaseTrackerRecHit Base;

  SiStripMatchedRecHit2D() {}
  ~SiStripMatchedRecHit2D() override {}

  SiStripMatchedRecHit2D(const LocalPoint& pos,
                         const LocalError& err,
                         GeomDet const& idet,
                         const SiStripRecHit2D* rMono,
                         const SiStripRecHit2D* rStereo)
      : BaseTrackerRecHit(pos, err, idet, trackerHitRTTI::match),
        clusterMono_(rMono->omniClusterRef()),
        clusterStereo_(rStereo->omniClusterRef()) {}

  // by value, as they will not exists anymore...
  SiStripRecHit2D stereoHit() const { return SiStripRecHit2D(stereoId(), stereoClusterRef()); }
  SiStripRecHit2D monoHit() const { return SiStripRecHit2D(monoId(), monoClusterRef()); }

  unsigned int stereoId() const { return rawId() + 1; }
  unsigned int monoId() const { return rawId() + 2; }

  // (to be improved)
  OmniClusterRef const& firstClusterRef() const override { return monoClusterRef(); }

  OmniClusterRef const& stereoClusterRef() const { return clusterStereo_; }
  OmniClusterRef const& monoClusterRef() const { return clusterMono_; }
  // Non const variants needed for cluster re-keying
  OmniClusterRef& stereoClusterRef() { return clusterStereo_; }
  OmniClusterRef& monoClusterRef() { return clusterMono_; }

  SiStripCluster const& stereoCluster() const { return stereoClusterRef().stripCluster(); }
  SiStripCluster const& monoCluster() const { return monoClusterRef().stripCluster(); }

  SiStripMatchedRecHit2D* clone() const override { return new SiStripMatchedRecHit2D(*this); }
#ifndef __GCCXML__
  RecHitPointer cloneSH() const override { return std::make_shared<SiStripMatchedRecHit2D>(*this); }
#endif

  int dimension() const override { return 2; }
  void getKfComponents(KfComponentsHolder& holder) const override { getKfComponents2D(holder); }

  bool sharesInput(const TrackingRecHit* other, SharedInputType what) const override;

  bool sharesInput(TrackerSingleRecHit const& other) const;

  std::vector<const TrackingRecHit*> recHits() const override;

  std::vector<TrackingRecHit*> recHits() override;

  bool canImproveWithTrack() const override { return true; }

private:
  // double dispatch
  SiStripMatchedRecHit2D* clone_(TkCloner const& cloner, TrajectoryStateOnSurface const& tsos) const override {
    return cloner(*this, tsos).release();
  }
#ifndef __GCCXML__
  RecHitPointer cloneSH_(TkCloner const& cloner, TrajectoryStateOnSurface const& tsos) const override {
    return cloner.makeShared(*this, tsos);
  }
#endif

private:
  OmniClusterRef clusterMono_, clusterStereo_;
};

inline bool sharesClusters(SiStripMatchedRecHit2D const& h1,
                           SiStripMatchedRecHit2D const& h2,
                           TrackingRecHit::SharedInputType what) {
  auto const& thisMonoClus = h1.monoClusterRef();
  auto const& otherMonoClus = h2.monoClusterRef();
  auto const& thisStereoClus = h1.stereoClusterRef();
  auto const& otherStereoClus = h2.stereoClusterRef();

  if (thisMonoClus.id() == otherMonoClus.id()) {
    const bool monoIdentity = thisMonoClus == otherMonoClus;
    const bool stereoIdentity = thisStereoClus == otherStereoClus;
    return (what == TrackingRecHit::all) ? (monoIdentity && stereoIdentity) : (monoIdentity || stereoIdentity);
  } else {
    bool monoOverlap = (h1.monoId() == h2.monoId()) ? otherMonoClus.stripOverlap(thisMonoClus) : false;
    bool stereoOverlap = (h1.stereoId() == h2.stereoId()) ? otherStereoClus.stripOverlap(thisStereoClus) : false;
    return (what == TrackingRecHit::all) ? (monoOverlap && stereoOverlap) : (monoOverlap || stereoOverlap);
  }
}

#endif