|
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
|
/*
* FWBeamSpotProxyBuilder.cc
* FWorks
*
* Created by Ianna Osborne on 7/29/10.
*
*/
#include "TEveStraightLineSet.h"
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
class FWBeamSpotProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::BeamSpot> {
public:
FWBeamSpotProxyBuilder(void) {}
~FWBeamSpotProxyBuilder(void) override {}
REGISTER_PROXYBUILDER_METHODS();
// Disable default copy constructor
FWBeamSpotProxyBuilder(const FWBeamSpotProxyBuilder&) = delete;
// Disable default assignment operator
const FWBeamSpotProxyBuilder& operator=(const FWBeamSpotProxyBuilder&) = delete;
private:
void localModelChanges(const FWModelId& iId,
TEveElement* parent,
FWViewType::EType viewType,
const FWViewContext* vc) override;
using FWSimpleProxyBuilderTemplate<reco::BeamSpot>::build;
void build(const reco::BeamSpot& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext*) override;
};
void FWBeamSpotProxyBuilder::localModelChanges(const FWModelId& iId,
TEveElement* parent,
FWViewType::EType viewType,
const FWViewContext* vc) {
if (TEveStraightLineSet* ls = dynamic_cast<TEveStraightLineSet*>(*parent->BeginChildren())) {
Color_t c = FWProxyBuilderBase::item()->modelInfo(iId.index()).displayProperties().color();
for (TEveProjectable::ProjList_i j = ls->BeginProjecteds(); j != ls->EndProjecteds(); ++j) {
if (TEveStraightLineSet* pls = dynamic_cast<TEveStraightLineSet*>(*j)) {
pls->SetMarkerColor(c);
pls->ElementChanged();
}
}
ls->SetMarkerColor(c);
ls->ElementChanged();
}
}
void FWBeamSpotProxyBuilder::build(const reco::BeamSpot& bs,
unsigned int iIndex,
TEveElement& oItemHolder,
const FWViewContext*) {
TEveStraightLineSet* ls = new TEveStraightLineSet();
double pos[3] = {bs.x0(), bs.y0(), bs.z0()};
double e[3] = {bs.x0Error(), bs.y0Error(), bs.z0Error()};
const Int_t N = 32;
const Float_t S = 2 * TMath::Pi() / N;
Float_t a = e[0], b = e[1];
for (Int_t i = 0; i < N; i++)
ls->AddLine(
a * TMath::Cos(i * S), b * TMath::Sin(i * S), 0, a * TMath::Cos(i * S + S), b * TMath::Sin(i * S + S), 0);
a = e[0];
b = e[2];
for (Int_t i = 0; i < N; i++)
ls->AddLine(
a * TMath::Cos(i * S), 0, b * TMath::Sin(i * S), a * TMath::Cos(i * S + S), 0, b * TMath::Sin(i * S + S));
a = e[1];
b = e[2];
for (Int_t i = 0; i < N; i++)
ls->AddLine(
0, a * TMath::Cos(i * S), b * TMath::Sin(i * S), 0, a * TMath::Cos(i * S + S), b * TMath::Sin(i * S + S));
ls->AddLine(0, 0, 0, 0, 0, 0);
ls->AddMarker(0, 0, 0);
ls->SetMarkerStyle(21);
const FWDisplayProperties& dp = FWProxyBuilderBase::item()->defaultDisplayProperties();
ls->SetMarkerColor(dp.color());
ls->RefMainTrans().SetPos(pos);
setupAddElement(ls, &oItemHolder);
}
REGISTER_FWPROXYBUILDER(FWBeamSpotProxyBuilder,
reco::BeamSpot,
"Beam Spot",
FWViewType::kAll3DBits | FWViewType::kAllRPZBits);
|