|
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// -*- C++ -*-
//
// Package: ParticleFlow
// Class : FWCandidatesLegoProxyBuilder
//
// Implementation:
// <Notes on implementation>
//
// Original Author: Colin Bernet
// Created: Fri May 28 14:54:19 CEST 2010
// Edited: sharris, Wed 10 Feb 2011, 13:00
//
// User include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/Context.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "Fireworks/Candidates/interface/FWLegoCandidate.h"
#include "Fireworks/ParticleFlow/interface/setTrackTypePF.h" // NB: This has to come after FWLegoCandidate include
//-----------------------------------------------------------------------------
// FWCandidate3DProxyBuilder
//-----------------------------------------------------------------------------
class FWPFCandidatesLegoProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::PFCandidate> {
public:
FWPFCandidatesLegoProxyBuilder();
~FWPFCandidatesLegoProxyBuilder() override;
// --------------------- Member Functions --------------------------
bool havePerViewProduct(FWViewType::EType) const override { return true; }
void scaleProduct(TEveElementList* parent, FWViewType::EType, const FWViewContext* vc) override;
void localModelChanges(const FWModelId& iId,
TEveElement* iCompound,
FWViewType::EType viewType,
const FWViewContext* vc) override;
REGISTER_PROXYBUILDER_METHODS();
FWPFCandidatesLegoProxyBuilder(const FWPFCandidatesLegoProxyBuilder&) = delete; // stop default
const FWPFCandidatesLegoProxyBuilder& operator=(const FWPFCandidatesLegoProxyBuilder&) = delete; // stop default
private:
// --------------------- Member Functions --------------------------
using FWSimpleProxyBuilderTemplate<reco::PFCandidate>::build;
void build(const reco::PFCandidate&, unsigned int, TEveElement&, const FWViewContext*) override;
};
//=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_
//
// constructors and destructor
//
//______________________________________________________________________________
FWPFCandidatesLegoProxyBuilder::FWPFCandidatesLegoProxyBuilder() {}
FWPFCandidatesLegoProxyBuilder::~FWPFCandidatesLegoProxyBuilder() {}
//
// member functions
//
//______________________________________________________________________________
void FWPFCandidatesLegoProxyBuilder::build(const reco::PFCandidate& iData,
unsigned int iIndex,
TEveElement& oItemHolder,
const FWViewContext* vc) {
FWLegoCandidate* candidate =
new FWLegoCandidate(vc, context(), iData.energy(), iData.et(), iData.pt(), iData.eta(), iData.phi());
candidate->SetMarkerColor(item()->defaultDisplayProperties().color());
fireworks::setTrackTypePF(iData, candidate);
context().voteMaxEtAndEnergy(iData.et(), iData.et());
setupAddElement(candidate, &oItemHolder);
}
//______________________________________________________________________________
void FWPFCandidatesLegoProxyBuilder::scaleProduct(TEveElementList* parent,
FWViewType::EType type,
const FWViewContext* vc) {
for (TEveElement::List_i i = parent->BeginChildren(); i != parent->EndChildren(); ++i) {
if ((*i)->HasChildren()) {
TEveElement* el = (*i)->FirstChild(); // there is only one child added in this proxy builder
FWLegoCandidate* candidate = dynamic_cast<FWLegoCandidate*>(el);
candidate->updateScale(vc, context());
}
}
}
//______________________________________________________________________________
void FWPFCandidatesLegoProxyBuilder::localModelChanges(const FWModelId& iId,
TEveElement* parent,
FWViewType::EType viewType,
const FWViewContext* vc) {
// line set marker is not same color as line, have to fix it here
if ((parent)->HasChildren()) {
TEveElement* el = (parent)->FirstChild(); // we know there is only one child added in this proxy builder
FWLegoCandidate* candidate = dynamic_cast<FWLegoCandidate*>(el);
candidate->SetMarkerColor(item()->modelInfo(iId.index()).displayProperties().color());
candidate->ElementChanged();
}
}
//______________________________________________________________________________
REGISTER_FWPROXYBUILDER(FWPFCandidatesLegoProxyBuilder,
reco::PFCandidate,
"PF Candidates",
FWViewType::kLegoPFECALBit | FWViewType::kLegoBit);
|