Fireworks/Tracks/interface/TrackUtils.h - master - githubmirror/cmssw

Fireworks/Tracks/interface/TrackUtils.h
89 строк3 KB
Matti Kortelainen
Move stored types in io_v1 namespace, reset class versions to 3
22 май 2026, 18:18
22 май 2026, 18:18909f324
#ifndef Fireworks_Tracks_TrackUtils_h
#define Fireworks_Tracks_TrackUtils_h
// -*- C++ -*-
//
// Package:     Tracks
// Class  :     TrackUtils
//

// system include files
#include "TEveVSDStructs.h"
#include <set>
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "DataFormats/DetId/interface/DetIdFwd.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelClusterFwd.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

// forward declarations
class RecSegment;

class FWEventItem;
class TEveElement;
class TEveTrack;
class TEveTrackPropagator;
class FWGeometry;
class TEveStraightLineSet;

class SiStripCluster;

namespace fireworks {

  struct State {
    TEveVector position;
    TEveVector momentum;
    bool valid;
    State() : valid(false) {}
    State(const TEveVector& pos) : position(pos), valid(false) {}
    State(const TEveVector& pos, const TEveVector& mom) : position(pos), momentum(mom), valid(true) {}
  };

  class StateOrdering {
    TEveVector m_direction;

  public:
    StateOrdering(const TEveVector& momentum) {
      m_direction = momentum;
      m_direction.Normalize();
    }
    bool operator()(const State& state1, const State& state2) const {
      double product1 = state1.position.Perp() *
                        (state1.position.fX * m_direction.fX + state1.position.fY * m_direction.fY > 0 ? 1 : -1);
      double product2 = state2.position.Perp() *
                        (state2.position.fX * m_direction.fX + state2.position.fY * m_direction.fY > 0 ? 1 : -1);
      return product1 < product2;
    }
  };

  TEveTrack* prepareTrack(const reco::Track& track,
                          TEveTrackPropagator* propagator,
                          const std::vector<TEveVector>& extraRefPoints = std::vector<TEveVector>());

  float pixelLocalX(const double mpx, const float*);
  float pixelLocalY(const double mpy, const float*);

  float phase2PixelLocalX(const double mpx, const float*, const float*);
  float phase2PixelLocalY(const double mpy, const float*, const float*);

  void localSiStrip(short strip, float* localTop, float* localBottom, const float* pars, unsigned int id);

  void pushPixelHits(std::vector<TVector3>& pixelPoints, const FWEventItem& iItem, const reco::Track& t);
  void pushNearbyPixelHits(std::vector<TVector3>& pixelPoints, const FWEventItem& iItem, const reco::Track& t);
  void pushPixelCluster(
      std::vector<TVector3>& pixelPoints, const FWGeometry& geom, DetId id, const SiPixelCluster& c, const float* pars);

  void addSiStripClusters(
      const FWEventItem* iItem, const reco::Track& t, class TEveElement* tList, bool addNearbyClusters, bool master);

  // Helpers for data extraction
  const SiStripCluster* extractClusterFromTrackingRecHit(const TrackingRecHit* rh);

  // Helper functions to get human readable informationa about given DetId
  // (copied from TrackingTools/TrackAssociator)
  std::string info(const DetId&);
  std::string info(const std::set<DetId>&);
  std::string info(const std::vector<DetId>&);
}  // namespace fireworks

#endif  // Fireworks_Tracks_TrackUtils_h