/
FWTrackProxyBuilder.cc
87 lines (68 loc) · 2.53 KB
/
FWTrackProxyBuilder.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// -*- C++ -*-
//
// Package: Tracks
// Class : FWTrackProxyBuilder
//
// Implementation:
// <Notes on implementation>
//
// Original Author: Chris Jones
// Created: Tue Nov 25 14:42:13 EST 2008
//
// system include files
#include "TEveTrack.h"
// user include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWProxyBuilderConfiguration.h"
#include "Fireworks/Core/interface/FWMagField.h"
#include "Fireworks/Tracks/interface/TrackUtils.h"
#include "Fireworks/Tracks/interface/estimate_field.h"
#include "DataFormats/TrackReco/interface/Track.h"
class FWTrackProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::Track> {
public:
FWTrackProxyBuilder();
virtual ~FWTrackProxyBuilder();
REGISTER_PROXYBUILDER_METHODS();
virtual void setItem(const FWEventItem* iItem);
private:
FWTrackProxyBuilder(const FWTrackProxyBuilder&); // stop default
const FWTrackProxyBuilder& operator=(const FWTrackProxyBuilder&); // stop default
using FWSimpleProxyBuilderTemplate<reco::Track>::build;
void build(const reco::Track& iData, unsigned int iIndex,TEveElement& oItemHolder, const FWViewContext*) override;
};
FWTrackProxyBuilder::FWTrackProxyBuilder()
{
}
FWTrackProxyBuilder::~FWTrackProxyBuilder()
{
}
void
FWTrackProxyBuilder::setItem(const FWEventItem* iItem)
{
FWProxyBuilderBase::setItem(iItem);
if (iItem) {
iItem->getConfig()->assertParam("LineWidth", long(1), long(1), long(4));
}
}
void
FWTrackProxyBuilder::build( const reco::Track& iData, unsigned int iIndex,TEveElement& oItemHolder , const FWViewContext*)
{
if( context().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 ) context().getField()->guessField( estimate );
}
}
TEveTrackPropagator* propagator = ( !iData.extra().isAvailable() ) ? context().getTrackerTrackPropagator() : context().getTrackPropagator();
TEveTrack* trk = fireworks::prepareTrack( iData, propagator );
trk->MakeTrack();
// Line width can be cached as a member. Set in virtual builder::itemChanged()
int width = item()->getConfig()->value<long>("LineWidth");
trk->SetLineWidth(width);
setupAddElement(trk, &oItemHolder);
}
//
// static member functions
//
REGISTER_FWPROXYBUILDER(FWTrackProxyBuilder, reco::Track, "Tracks", FWViewType::kAll3DBits | FWViewType::kAllRPZBits);