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

DataFormats/TrackerRecHit2D/interface/BaseTrackerRecHit.h
165 строк6 KB
Francesco Tenchini
Merge pull request #50968 from makortel/evoIntegrationDataFormats
05 июн 2026, 23:21
05 июн 2026, 23:219eaeb93
#ifndef BaseTrackerRecHit_H
#define BaseTrackerRecHit_H

#include "DataFormats/TrackerRecHit2D/interface/trackerHitRTTI.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitGlobalState.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/ErrorFrameTransformer.h"
#include "Geometry/CommonTopologies/interface/TrackerGeomDet.h"
#include "DataFormats/GeometrySurface/interface/Surface.h"

//#define DO_INTERNAL_CHECKS_BTR
//#define VI_DEBUG

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

namespace io_v1 {
  class BaseTrackerRecHit : public TrackingRecHit {
  public:
    BaseTrackerRecHit() : qualWord_(0) {}

    // fake TTRH interface
    BaseTrackerRecHit const* hit() const final { return this; }

    ~BaseTrackerRecHit() override {}

    // no position (as in persistent)
    BaseTrackerRecHit(DetId id, trackerHitRTTI::RTTI rt) : TrackingRecHit(id, (unsigned int)(rt)), qualWord_(0) {}
    BaseTrackerRecHit(const GeomDet& idet, trackerHitRTTI::RTTI rt)
        : TrackingRecHit(idet, (unsigned int)(rt)), qualWord_(0) {}

    BaseTrackerRecHit(const LocalPoint& p, const LocalError& e, GeomDet const& idet, trackerHitRTTI::RTTI rt)
        : TrackingRecHit(idet, (unsigned int)(rt)), pos_(p), err_(e), qualWord_(0) {
      auto trackerDet = static_cast<TrackerGeomDet const*>(det());
      LocalError lape = trackerDet->localAlignmentError();
      if (lape.valid())
        err_ = LocalError(err_.xx() + lape.xx(), err_.xy() + lape.xy(), err_.yy() + lape.yy());
    }

    trackerHitRTTI::RTTI rtti() const { return trackerHitRTTI::rtti(*this); }
    bool isSingle() const { return trackerHitRTTI::isSingle(*this); }
    bool isMatched() const { return trackerHitRTTI::isMatched(*this); }
    bool isProjected() const { return trackerHitRTTI::isProjected(*this); }
    bool isProjMono() const { return trackerHitRTTI::isProjMono(*this); }
    bool isProjStereo() const { return trackerHitRTTI::isProjStereo(*this); }
    bool isMulti() const { return trackerHitRTTI::isMulti(*this); }

    virtual bool isPixel() const { return false; }
    virtual bool isPhase2() const { return false; }

    // used by trackMerger (to be improved)
    virtual OmniClusterRef const& firstClusterRef() const = 0;

    // verify that hits can share clusters...
    inline bool sameDetModule(TrackingRecHit const& hit) const;

    bool hasPositionAndError() const override;

    LocalPoint localPosition() const override {
      check();
      return pos_;
    }

    LocalError localPositionError() const override {
      check();
      return err_;
    }
    const LocalPoint& localPositionFast() const {
      check();
      return pos_;
    }
    const LocalError& localPositionErrorFast() const {
      check();
      return err_;
    }

    // to be specialized for 1D and 2D
    void getKfComponents(KfComponentsHolder& holder) const override = 0;
    int dimension() const override = 0;

    void getKfComponents1D(KfComponentsHolder& holder) const;
    void getKfComponents2D(KfComponentsHolder& holder) const;

    // global coordinates
    // Extension of the TrackingRecHit interface
    const Surface* surface() const final { return &(det()->surface()); }

    GlobalPoint globalPosition() const final { return surface()->toGlobal(localPosition()); }

    GlobalError globalPositionError() const final {
      return ErrorFrameTransformer().transform(localPositionError(), *surface());
    }
    float errorGlobalR() const final { return std::sqrt(globalPositionError().rerr(globalPosition())); }
    float errorGlobalZ() const final { return std::sqrt(globalPositionError().czz()); }
    float errorGlobalRPhi() const final {
      return globalPosition().perp() * sqrt(globalPositionError().phierr(globalPosition()));
    }

    // once cache removed will obsolete the above
    TrackingRecHitGlobalState globalState() const {
      GlobalError globalError = ErrorFrameTransformer::transform(localPositionError(), *surface());
      GlobalPoint gp = globalPosition();
      float r = gp.perp();
      float errorRPhi = r * std::sqrt(float(globalError.phierr(gp)));
      float errorR = std::sqrt(float(globalError.rerr(gp)));
      float errorZ = std::sqrt(float(globalError.czz()));
      return (TrackingRecHitGlobalState){gp.basicVector(), r, gp.barePhi(), errorR, errorZ, errorRPhi};
    }

    /// cluster probability, overloaded by pixel rechits.
    virtual float clusterProbability() const { return 1.f; }

  public:
    // obsolete (for what tracker is concerned...) interface
    AlgebraicVector parameters() const override;
    AlgebraicSymMatrix parametersError() const override;
    AlgebraicMatrix projectionMatrix() const override;

  private:
#ifdef VI_DEBUG
    void check() const { assert(det()); }
#elif defined(DO_INTERNAL_CHECKS_BTR)
    void check() const;
#else
    static void check() {}
#endif

  protected:
    LocalPoint pos_;
    LocalError err_;

  protected:
    //this comes for free (padding)
    unsigned int qualWord_;
  };

  bool BaseTrackerRecHit::sameDetModule(TrackingRecHit const& hit) const {
    unsigned int myid = geographicalId().rawId();
    unsigned int mysubd = myid >> (DetId::kSubdetOffset);

    unsigned int id = hit.geographicalId().rawId();
    unsigned int subd = id >> (DetId::kSubdetOffset);

    if (mysubd != subd)
      return false;

    //Protection against invalid hits
    if (!hit.isValid())
      return false;

    const unsigned int limdet = 10;  // TIB=11

    if (mysubd > limdet) {  // strip
      // mask glue and stereo
      myid |= 3;
      id |= 3;
    }
    return id == myid;
  }

  // Comparison operators
  inline bool operator<(const BaseTrackerRecHit& one, const BaseTrackerRecHit& other) {
    return (one.geographicalId() < other.geographicalId());
  }
}  // namespace io_v1
using BaseTrackerRecHit = io_v1::BaseTrackerRecHit;
#endif  // BaseTrackerRecHit_H