FWPFTrackUtils Class Reference

#include <FWPFTrackUtils.h>

Public Member Functions
	FWPFTrackUtils ()
TEvePointSet *	getCollisionMarkers (const TEveTrack *)
TEveStraightLineSet *	setupLegoTrack (const reco::Track &)
TEveTrack *	setupTrack (const reco::Track &)
virtual	~FWPFTrackUtils ()

Private Member Functions
	FWPFTrackUtils (const FWPFTrackUtils &)=delete
TEveTrack *	getTrack (const reco::Track &)
const FWPFTrackUtils &	operator= (const FWPFTrackUtils &)=delete

Private Attributes
FWPFTrackSingleton *	m_singleton

Detailed Description

Definition at line 68 of file FWPFTrackUtils.h.

Constructor & Destructor Documentation

FWPFTrackUtils::FWPFTrackUtils

(

)

Definition at line 44 of file FWPFTrackUtils.cc.

References FWPFTrackSingleton::Instance().

{ m_singleton = FWPFTrackSingleton::Instance(); }

virtual FWPFTrackUtils::~FWPFTrackUtils ( )

inlinevirtual

Definition at line 72 of file FWPFTrackUtils.h.

References getTrack(), and FWPFTrackSingleton::operator=().

{}

FWPFTrackUtils::FWPFTrackUtils ( const FWPFTrackUtils &  )

privatedelete

Member Function Documentation

TEvePointSet * FWPFTrackUtils::getCollisionMarkers

(

const TEveTrack *

trk

)

Definition at line 246 of file FWPFTrackUtils.cc.

References FWPFGeom::caloR1(), FWPFGeom::caloR2(), FWPFGeom::caloR3(), FWPFGeom::caloZ1(), FWPFGeom::caloZ2(), FWPFMaths::checkIntersect(), ECAL, mps_fire::i, dqmiolumiharvest::j, FWPFMaths::linearInterpolation(), FWPFMaths::lineCircleIntersect(), FWPFMaths::lineLineIntersect(), p1, p2, HLT_2018_cff::points, y, and z.

Referenced by FWPFTrackRPZProxyBuilder::build(), FWPFTrack3DProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().

                                                                      {
  using namespace FWPFGeom;

  bool ECAL = false;
  const Float_t *points = trk->GetP();
  TEvePointSet *ps = new TEvePointSet();

  for (Int_t i = 1; i < trk->GetN(); ++i) {
    int j = i * 3;
    TEveVector v1 = TEveVector(points[j], points[j + 1], points[j + 2]);

    if (!ECAL) {
      if (FWPFMaths::checkIntersect(v1, caloR1())) {
        TEveVector v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
        TEveVector xyPoint = FWPFMaths::lineCircleIntersect(v1, v2, caloR1());
        TEveVector zPoint;
        if (v1.fZ < 0)
          zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ - 50.f);
        else
          zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ + 50.f);

        TEveVector vec = FWPFMaths::lineLineIntersect(v1, v2, xyPoint, zPoint);
        ps->SetNextPoint(vec.fX, vec.fY, vec.fZ);

        ECAL = true;
      } else if (fabs(v1.fZ) >= caloZ1()) {
        TEveVector p1, p2;
        TEveVector vec, v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
        float z, y = FWPFMaths::linearInterpolation(v2, v1, caloZ1());

        if (v2.fZ > 0)
          z = caloZ1();
        else
          z = caloZ1() * -1;

        p1 = TEveVector(v2.fX + 50, y, z);
        p2 = TEveVector(v2.fX - 50, y, z);
        vec = FWPFMaths::lineLineIntersect(v1, v2, p1, p2);

        ps->SetNextPoint(vec.fX, vec.fY, vec.fZ);
        ECAL = true;
      }
    } else if (FWPFMaths::checkIntersect(v1, caloR2())) {
      TEveVector v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
      TEveVector xyPoint = FWPFMaths::lineCircleIntersect(v1, v2, caloR2());
      TEveVector zPoint;
      if (v1.fZ < 0)
        zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ - 50.f);
      else
        zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ + 50.f);

      TEveVector vec = FWPFMaths::lineLineIntersect(v1, v2, xyPoint, zPoint);
      ps->SetNextPoint(vec.fX, vec.fY, vec.fZ);
      break;  // ECAL and HCAL collisions found so stop looping
    }
  }

  // HCAL collisions (outer radius and endcap)
  int i = (trk->GetN() - 1) * 3;
  TEveVector vec = TEveVector(points[i], points[i + 1], points[i + 2]);

  if (FWPFMaths::checkIntersect(vec, caloR3() - 1))
    ps->SetNextPoint(vec.fX, vec.fY, vec.fZ);
  else if (fabs(vec.fZ) >= caloZ2())
    ps->SetNextPoint(vec.fX, vec.fY, vec.fZ);

  return ps;
}

TEveTrack * FWPFTrackUtils::getTrack ( const reco::Track &  iData )

private

Definition at line 47 of file FWPFTrackUtils.cc.

References reco::TrackBase::charge(), reco::Track::extra(), edm::Ref< C, T, F >::isAvailable(), TrackCandidateProducer_cfi::propagator, reco::TrackBase::px(), reco::TrackBase::py(), reco::TrackBase::pz(), OrderedSet::t, and reco::TrackBase::vertex().

                                                          {
  TEveTrackPropagator *propagator =
      (!iData.extra().isAvailable()) ? m_singleton->getTrackerTrackPropagator() : m_singleton->getTrackPropagator();

  TEveRecTrack t;
  t.fBeta = 1;
  t.fP = TEveVector(iData.px(), iData.py(), iData.pz());
  t.fV = TEveVector(iData.vertex().x(), iData.vertex().y(), iData.vertex().z());
  t.fSign = iData.charge();
  TEveTrack *trk = new TEveTrack(&t, propagator);
  trk->MakeTrack();

  return trk;
}

const FWPFTrackUtils& FWPFTrackUtils::operator= ( const FWPFTrackUtils &  )

privatedelete

TEveStraightLineSet * FWPFTrackUtils::setupLegoTrack

(

const reco::Track &

iData

)

Definition at line 63 of file FWPFTrackUtils.cc.

References FWPFGeom::caloR1(), FWPFGeom::caloR2(), FWPFGeom::caloR3(), FWPFGeom::caloZ1(), FWPFGeom::caloZ2(), FWPFMaths::checkIntersect(), ECAL, end, fw::estimate_field(), reco::TrackBase::eta(), getTrack(), mps_fire::i, dqmiolumiharvest::j, FWMagField::kNone, FWPFMaths::linearInterpolation(), FWPFMaths::lineCircleIntersect(), FWPFMaths::lineLineIntersect(), visualization-live-secondInstance_cfg::m, p1, p2, HLT_2018_cff::points, reco::TrackBase::pt(), groupFilesInBlocks::temp, TwoPi, y, and z.

Referenced by FWPFTrackLegoProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().

                                                                          {
  using namespace FWPFGeom;

  // Declarations
  int wraps[3] = {-1, -1, -1};
  bool ECAL = false;
  TEveTrack *trk = getTrack(iData);
  std::vector<TEveVector> trackPoints(trk->GetN() - 1);
  const Float_t *points = trk->GetP();
  TEveStraightLineSet *legoTrack = new TEveStraightLineSet();

  if (m_singleton->getField()->getSource() == FWMagField::kNone) {
    if (fabs(iData.eta()) < 2.0 && iData.pt() > 0.5 && iData.pt() < 30) {
      double estimate = fw::estimate_field(iData, true);
      if (estimate >= 0)
        m_singleton->getField()->guessField(estimate);
    }
  }

  // Convert to Eta/Phi and store in vector
  for (Int_t i = 1; i < trk->GetN(); ++i) {
    int j = i * 3;
    TEveVector temp = TEveVector(points[j], points[j + 1], points[j + 2]);
    TEveVector vec = TEveVector(temp.Eta(), temp.Phi(), 0.001);

    trackPoints[i - 1] = vec;
  }

  // Add first point to ps if necessary
  for (Int_t i = 1; i < trk->GetN(); ++i) {
    int j = i * 3;
    TEveVector v1 = TEveVector(points[j], points[j + 1], points[j + 2]);

    if (!ECAL) {
      if (FWPFMaths::checkIntersect(v1, caloR1())) {
        TEveVector v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
        TEveVector xyPoint = FWPFMaths::lineCircleIntersect(v1, v2, caloR1());
        TEveVector zPoint;
        if (v1.fZ < 0)
          zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ - 50.f);
        else
          zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ + 50.f);

        TEveVector vec = FWPFMaths::lineLineIntersect(v1, v2, xyPoint, zPoint);
        legoTrack->AddMarker(vec.Eta(), vec.Phi(), 0.001, 0);

        wraps[0] = i;  // There is now a chance that the track will also reach the HCAL radius
        ECAL = true;
      } else if (fabs(v1.fZ) >= caloZ1()) {
        TEveVector p1, p2;
        TEveVector vec, v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
        float z, y = FWPFMaths::linearInterpolation(v2, v1, caloZ1());

        if (v2.fZ > 0)
          z = caloZ1();
        else
          z = caloZ1() * -1;

        p1 = TEveVector(v2.fX + 50, y, z);
        p2 = TEveVector(v2.fX - 50, y, z);
        vec = FWPFMaths::lineLineIntersect(v1, v2, p1, p2);

        legoTrack->AddMarker(vec.Eta(), vec.Phi(), 0.001, 0);
        wraps[0] = i;
        ECAL = true;
      }
    } else if (FWPFMaths::checkIntersect(v1, caloR2())) {
      TEveVector v2 = TEveVector(points[j - 3], points[j - 2], points[j - 1]);
      TEveVector xyPoint = FWPFMaths::lineCircleIntersect(v1, v2, caloR2());
      TEveVector zPoint;
      if (v1.fZ < 0)
        zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ - 50.f);
      else
        zPoint = TEveVector(xyPoint.fX, xyPoint.fY, v1.fZ + 50.f);

      TEveVector vec = FWPFMaths::lineLineIntersect(v1, v2, xyPoint, zPoint);
      legoTrack->AddMarker(vec.Eta(), vec.Phi(), 0.001, 1);

      wraps[1] = i;  // There is now a chance that the track will also reach the HCAL radius
      break;
    }
  }

  if (wraps[0] != -1)  //if( ECAL )
  {
    int i = (trk->GetN() - 1) * 3;
    int j = trk->GetN() - 2;  // This is equal to the last index in trackPoints vector
    TEveVector vec = TEveVector(points[i], points[i + 1], points[i + 2]);

    if (FWPFMaths::checkIntersect(vec, caloR3() - 1)) {
      legoTrack->AddMarker(vec.Eta(), vec.Phi(), 0.001, 2);
      wraps[2] = j;
    } else if (fabs(vec.fZ) >= caloZ2()) {
      legoTrack->AddMarker(vec.Eta(), vec.Phi(), 0.001, 2);
      wraps[2] = j;
    }
  }

  /* Handle phi wrapping */
  for (int i = 0; i < static_cast<int>(trackPoints.size() - 1); ++i) {
    if ((trackPoints[i + 1].fY - trackPoints[i].fY) > 1) {
      trackPoints[i + 1].fY -= TMath::TwoPi();
      if (i == wraps[0]) {
        TEveChunkManager::iterator mi(legoTrack->GetMarkerPlex());
        mi.next();  // First point
        TEveStraightLineSet::Marker_t &m = *(TEveStraightLineSet::Marker_t *)mi();
        m.fV[0] = trackPoints[i + 1].fX;
        m.fV[1] = trackPoints[i + 1].fY;
        m.fV[2] = 0.001;
      } else if (i == wraps[1]) {
        TEveChunkManager::iterator mi(legoTrack->GetMarkerPlex());
        mi.next();
        mi.next();  // Second point
        TEveStraightLineSet::Marker_t &m = *(TEveStraightLineSet::Marker_t *)mi();
        m.fV[0] = trackPoints[i + 1].fX;
        m.fV[1] = trackPoints[i + 1].fY;
        m.fV[2] = 0.001;
      }
    }

    if ((trackPoints[i].fY - trackPoints[i + 1].fY) > 1) {
      trackPoints[i + 1].fY += TMath::TwoPi();
      if (i == wraps[0]) {
        TEveChunkManager::iterator mi(legoTrack->GetMarkerPlex());
        mi.next();  // First point
        TEveStraightLineSet::Marker_t &m = *(TEveStraightLineSet::Marker_t *)mi();
        m.fV[0] = trackPoints[i + 1].fX;
        m.fV[1] = trackPoints[i + 1].fY;
        m.fV[2] = 0.001;
      } else if (i == wraps[1]) {
        TEveChunkManager::iterator mi(legoTrack->GetMarkerPlex());
        mi.next();
        mi.next();  // Second point
        TEveStraightLineSet::Marker_t &m = *(TEveStraightLineSet::Marker_t *)mi();
        m.fV[0] = trackPoints[i + 1].fX;
        m.fV[1] = trackPoints[i + 1].fY;
        m.fV[2] = 0.001;
      }
    }
  }

  int end = static_cast<int>(trackPoints.size() - 1);
  if (wraps[2] == end) {
    TEveChunkManager::iterator mi(legoTrack->GetMarkerPlex());
    mi.next();
    mi.next();
    mi.next();  // Third point
    TEveStraightLineSet::Marker_t &m = *(TEveStraightLineSet::Marker_t *)mi();
    m.fV[0] = trackPoints[end].fX;
    m.fV[1] = trackPoints[end].fY;
    m.fV[2] = 0.001;
  }

  // Set points on TEveLineSet object ready for displaying
  for (unsigned int i = 0; i < trackPoints.size() - 1; ++i)
    legoTrack->AddLine(trackPoints[i], trackPoints[i + 1]);

  legoTrack->SetDepthTest(false);
  legoTrack->SetMarkerStyle(4);
  legoTrack->SetMarkerSize(1);
  legoTrack->SetRnrMarkers(true);

  delete trk;  // Release memory that is no longer required

  return legoTrack;
}

TEveTrack * FWPFTrackUtils::setupTrack

(

const reco::Track &

iData

)

Definition at line 231 of file FWPFTrackUtils.cc.

References fw::estimate_field(), reco::TrackBase::eta(), getTrack(), FWMagField::kNone, and reco::TrackBase::pt().

Referenced by FWPFTrackRPZProxyBuilder::build(), FWPFTrack3DProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().

                                                            {
  if (m_singleton->getField()->getSource() == FWMagField::kNone) {
    if (fabs(iData.eta()) < 2.0 && iData.pt() > 0.5 && iData.pt() < 30) {
      double estimate = fw::estimate_field(iData, true);
      if (estimate >= 0)
        m_singleton->getField()->guessField(estimate);
    }
  }

  TEveTrack *trk = getTrack(iData);

  return trk;
}

Member Data Documentation

FWPFTrackSingleton* FWPFTrackUtils::m_singleton

private

Definition at line 85 of file FWPFTrackUtils.h.