|
|
|
#include <FWPFTrackUtils.h>
Public Types | |
| enum | Type { LEGO = 0, RPZ = 1 } |
Public Member Functions | |
| FWPFTrackUtils () | |
| TEvePointSet * | getCollisionMarkers (const TEveTrack *) |
| TEveStraightLineSet * | setupLegoTrack (const reco::Track &) |
| TEveTrack * | setupRPZTrack (const reco::Track &) |
| virtual | ~FWPFTrackUtils () |
Private Member Functions | |
| FWPFTrackUtils (const FWPFTrackUtils &) | |
| const FWPFTrackUtils & | operator= (const FWPFTrackUtils &) |
Private Attributes | |
| FWPFUtils * | m_trackUtils |
Definition at line 31 of file FWPFTrackUtils.h.
| enum FWPFTrackUtils::Type |
| FWPFTrackUtils::FWPFTrackUtils | ( | ) | [inline] |
Definition at line 37 of file FWPFTrackUtils.h.
References FWPFUtils::initPropagator(), and m_trackUtils.
{ m_trackUtils = new FWPFUtils(); m_trackUtils->initPropagator(); }
| virtual FWPFTrackUtils::~FWPFTrackUtils | ( | ) | [inline, virtual] |
| FWPFTrackUtils::FWPFTrackUtils | ( | const FWPFTrackUtils & | ) | [private] |
| TEvePointSet * FWPFTrackUtils::getCollisionMarkers | ( | const TEveTrack * | trk | ) |
Definition at line 200 of file FWPFTrackUtils.cc.
References FWPFUtils::checkIntersect(), ECAL, FWPFUtils::getCaloR1(), FWPFUtils::getCaloR2(), FWPFUtils::getCaloR3(), FWPFUtils::getCaloZ1(), FWPFUtils::getCaloZ2(), i, j, FWPFUtils::linearInterpolation(), FWPFUtils::lineCircleIntersect(), FWPFUtils::lineLineIntersect(), m_trackUtils, p1, p2, ExpressReco_HICollisions_FallBack::y, and z.
Referenced by FWPFTrackRPZProxyBuilder::build(), FWPFTrack3DProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().
{
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( m_trackUtils->checkIntersect( v1, m_trackUtils->getCaloR1() ) )
{
TEveVector v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
TEveVector xyPoint = m_trackUtils->lineCircleIntersect( v1, v2, m_trackUtils->getCaloR1() );
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 = m_trackUtils->lineLineIntersect( v1, v2, xyPoint, zPoint );
ps->SetNextPoint( vec.fX, vec.fY, vec.fZ );
ECAL = true;
}
else if( fabs( v1.fZ ) >= m_trackUtils->getCaloZ1() )
{
TEveVector p1, p2;
TEveVector vec, v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
float z, y = m_trackUtils->linearInterpolation( v2, v1, m_trackUtils->getCaloZ1() );
if( v2.fZ > 0 )
z = m_trackUtils->getCaloZ1();
else
z = m_trackUtils->getCaloZ1() * -1;
p1 = TEveVector( v2.fX + 50, y, z );
p2 = TEveVector( v2.fX - 50, y, z );
vec = m_trackUtils->lineLineIntersect( v1, v2, p1, p2 );
ps->SetNextPoint( vec.fX, vec.fY, vec.fZ );
ECAL = true;
}
}
else if ( m_trackUtils->checkIntersect( v1, m_trackUtils->getCaloR2() ) )
{
TEveVector v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
TEveVector xyPoint = m_trackUtils->lineCircleIntersect( v1, v2, m_trackUtils->getCaloR2() );
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 = m_trackUtils->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( m_trackUtils->checkIntersect( vec, m_trackUtils->getCaloR3() - 1 ) )
ps->SetNextPoint( vec.fX, vec.fY, vec.fZ );
else if( fabs( vec.fZ ) >= m_trackUtils->getCaloZ2() )
ps->SetNextPoint( vec.fX, vec.fY, vec.fZ );
return ps;
}
| const FWPFTrackUtils& FWPFTrackUtils::operator= | ( | const FWPFTrackUtils & | ) | [private] |
| TEveStraightLineSet * FWPFTrackUtils::setupLegoTrack | ( | const reco::Track & | iData | ) |
Definition at line 5 of file FWPFTrackUtils.cc.
References FWPFUtils::checkIntersect(), ECAL, end, fw::estimate_field(), reco::TrackBase::eta(), FWPFUtils::getCaloR1(), FWPFUtils::getCaloR2(), FWPFUtils::getCaloR3(), FWPFUtils::getCaloZ1(), FWPFUtils::getCaloZ2(), FWPFUtils::getField(), FWMagField::getSource(), FWPFUtils::getTrack(), FWMagField::guessField(), i, j, WDecay::kNone, FWPFUtils::linearInterpolation(), FWPFUtils::lineCircleIntersect(), FWPFUtils::lineLineIntersect(), m, m_trackUtils, p1, p2, reco::TrackBase::pt(), cond::rpcobtemp::temp, TwoPi, ExpressReco_HICollisions_FallBack::y, and z.
Referenced by FWPFTrackLegoProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().
{
// Declarations
int wraps[3] = { -1, -1, -1 };
bool ECAL = false;
TEveTrack *trk = m_trackUtils->getTrack( iData );
std::vector<TEveVector> trackPoints( trk->GetN() - 1 );
const Float_t *points = trk->GetP();
TEveStraightLineSet *legoTrack = new TEveStraightLineSet();
if( m_trackUtils->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_trackUtils->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( m_trackUtils->checkIntersect( v1, m_trackUtils->getCaloR1() ) )
{
TEveVector v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
TEveVector xyPoint = m_trackUtils->lineCircleIntersect( v1, v2, m_trackUtils->getCaloR1() );
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 = m_trackUtils->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 ) >= m_trackUtils->getCaloZ1() )
{
TEveVector p1, p2;
TEveVector vec, v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
float z, y = m_trackUtils->linearInterpolation( v2, v1, m_trackUtils->getCaloZ1() );
if( v2.fZ > 0 )
z = m_trackUtils->getCaloZ1();
else
z = m_trackUtils->getCaloZ1() * -1;
p1 = TEveVector( v2.fX + 50, y, z );
p2 = TEveVector( v2.fX - 50, y, z );
vec = m_trackUtils->lineLineIntersect( v1, v2, p1, p2 );
legoTrack->AddMarker( vec.Eta(), vec.Phi(), 0.001, 0 );
wraps[0] = i;
ECAL = true;
}
}
else if( m_trackUtils->checkIntersect( v1, m_trackUtils->getCaloR2() ) )
{
TEveVector v2 = TEveVector( points[j-3], points[j-2], points[j-1] );
TEveVector xyPoint = m_trackUtils->lineCircleIntersect( v1, v2, m_trackUtils->getCaloR2() );
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 = m_trackUtils->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( m_trackUtils->checkIntersect( vec, m_trackUtils->getCaloR3() - 1 ) )
{
legoTrack->AddMarker( vec.Eta(), vec.Phi(), 0.001, 2 );
wraps[2] = j;
}
else if( fabs( vec.fZ ) >= m_trackUtils->getCaloZ2() )
{
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( false );
delete trk; // Release memory that is no longer required*/
return legoTrack;
}
| TEveTrack * FWPFTrackUtils::setupRPZTrack | ( | const reco::Track & | iData | ) |
Definition at line 181 of file FWPFTrackUtils.cc.
References fw::estimate_field(), reco::TrackBase::eta(), FWPFUtils::getField(), FWMagField::getSource(), FWPFUtils::getTrack(), FWMagField::guessField(), WDecay::kNone, m_trackUtils, and reco::TrackBase::pt().
Referenced by FWPFTrackRPZProxyBuilder::build(), FWPFTrack3DProxyBuilder::build(), and FWPFBlockProxyBuilder::setupTrackElement().
{
if( m_trackUtils->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_trackUtils->getField()->guessField( estimate );
}
}
TEveTrack *trk = m_trackUtils->getTrack( iData );
return trk;
}
FWPFUtils* FWPFTrackUtils::m_trackUtils [private] |
Definition at line 50 of file FWPFTrackUtils.h.
Referenced by FWPFTrackUtils(), getCollisionMarkers(), setupLegoTrack(), setupRPZTrack(), and ~FWPFTrackUtils().
1.7.3