|
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
|
#ifndef _FWCANDIDATEHGCALLEGOPROXYBUILDER_H_
#define _FWCANDIDATEHGCALLEGOPROXYBUILDER_H_
// -*- C++ -*-
//
// Package: Candidates
// Class : FWCandidateHGCalLegoProxyBuilder
//
//
// User include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/Context.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "DataFormats/ParticleFlowReco/interface/HGCalMultiCluster.h"
#include "Fireworks/Candidates/interface/FWLegoCandidate.h"
#include <cmath>
//-----------------------------------------------------------------------------
// FWCandidateHGCalLegoProxyBuilder
//-----------------------------------------------------------------------------
class FWCandidateHGCalLegoProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::HGCalMultiCluster> {
public:
// ---------------- Constructor(s)/Destructor ----------------------
FWCandidateHGCalLegoProxyBuilder() {}
~FWCandidateHGCalLegoProxyBuilder() override {}
// --------------------- Member Functions --------------------------
bool havePerViewProduct(FWViewType::EType) const override { return true; }
void scaleProduct(TEveElementList *, FWViewType::EType, const FWViewContext *) override;
void localModelChanges(const FWModelId &, TEveElement *, FWViewType::EType, const FWViewContext *) override;
REGISTER_PROXYBUILDER_METHODS();
// ----------------------- Data Members ----------------------------
FWCandidateHGCalLegoProxyBuilder(const FWCandidateHGCalLegoProxyBuilder &) = delete;
const FWCandidateHGCalLegoProxyBuilder &operator=(const FWCandidateHGCalLegoProxyBuilder &) = delete;
private:
// --------------------- Member Functions --------------------------
using FWSimpleProxyBuilderTemplate<reco::HGCalMultiCluster>::build;
void build(const reco::HGCalMultiCluster &, unsigned int, TEveElement &, const FWViewContext *) override;
};
#endif
//=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_
//______________________________________________________________________________
void FWCandidateHGCalLegoProxyBuilder::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
FWLegoCandidate *candidate = dynamic_cast<FWLegoCandidate *>(el);
candidate->updateScale(vc, context());
}
}
}
//______________________________________________________________________________
void FWCandidateHGCalLegoProxyBuilder::localModelChanges(const FWModelId &iId,
TEveElement *parent,
FWViewType::EType type,
const FWViewContext *vc) {
// Line set marker is nto the same color as line, have to fix it here
if ((parent)->HasChildren()) {
TEveElement *el = (parent)->FirstChild(); // There is only one child
FWLegoCandidate *candidate = dynamic_cast<FWLegoCandidate *>(el);
candidate->SetMarkerColor(item()->modelInfo(iId.index()).displayProperties().color());
candidate->ElementChanged();
}
}
//______________________________________________________________________________
void FWCandidateHGCalLegoProxyBuilder::build(const reco::HGCalMultiCluster &iData,
unsigned int iIndex,
TEveElement &oItemHolder,
const FWViewContext *vc) {
const auto &clusters = iData.clusters();
for (const auto &c : clusters) {
auto pt = c->energy() / std::cosh(c->eta());
FWLegoCandidate *candidate = new FWLegoCandidate(vc, context(), c->energy(), pt, pt, c->eta(), c->phi());
candidate->SetMarkerColor(item()->defaultDisplayProperties().color());
context().voteMaxEtAndEnergy(pt, c->energy());
setupAddElement(candidate, &oItemHolder);
}
}
//______________________________________________________________________________
REGISTER_FWPROXYBUILDER(FWCandidateHGCalLegoProxyBuilder,
reco::HGCalMultiCluster,
"HGCal Multiclusters Lego",
FWViewType::kLegoBit | FWViewType::kLegoHFBit);
|