|
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
|
#include "Fireworks/ParticleFlow/interface/FWPFClusterRPZUtils.h"
//______________________________________________________________________________
TEveScalableStraightLineSet *FWPFClusterRPZUtils::buildRhoPhiClusterLineSet(const reco::PFCluster &cluster,
const FWViewContext *vc,
float r) {
float energy, et;
TEveVector centre = TEveVector(cluster.x(), cluster.y(), cluster.z());
energy = cluster.energy();
et = FWPFMaths::calculateEt(centre, energy);
return buildRhoPhiClusterLineSet(cluster, vc, energy, et, r);
}
//______________________________________________________________________________
TEveScalableStraightLineSet *FWPFClusterRPZUtils::buildRhoPhiClusterLineSet(
const reco::PFCluster &cluster, const FWViewContext *vc, float e, float et, float r) {
TEveScalableStraightLineSet *ls = new TEveScalableStraightLineSet("rhophiCluster");
TEveVector vec;
float size = 1.f; // Stored in scale
double phi;
vec = TEveVector(cluster.x(), cluster.y(), cluster.z());
phi = vec.Phi();
FWViewEnergyScale *energyScale = vc->getEnergyScale();
ls->SetLineWidth(4);
ls->SetScaleCenter(r * cos(phi), r * sin(phi), 0);
ls->AddLine(r * cos(phi), r * sin(phi), 0, (r + size) * cos(phi), (r + size) * sin(phi), 0);
ls->SetScale(energyScale->getScaleFactor3D() * (energyScale->getPlotEt() ? et : e));
return ls;
}
//______________________________________________________________________________
TEveScalableStraightLineSet *FWPFClusterRPZUtils::buildRhoZClusterLineSet(
const reco::PFCluster &cluster, const FWViewContext *vc, float caloTransAngle, float r, float z) {
float energy, et;
TEveVector centre = TEveVector(cluster.x(), cluster.y(), cluster.z());
energy = cluster.energy();
et = FWPFMaths::calculateEt(centre, energy);
return buildRhoZClusterLineSet(cluster, vc, caloTransAngle, energy, et, r, z);
}
//______________________________________________________________________________
TEveScalableStraightLineSet *FWPFClusterRPZUtils::buildRhoZClusterLineSet(
const reco::PFCluster &cluster, const FWViewContext *vc, float caloTransAngle, float e, float et, float r, float z) {
float size = 1.f; // Stored in scale
float offr = 4;
float ecalZ = z + offr / tan(caloTransAngle);
double theta, phi;
double rad(0);
TEveVector vec;
TEveScalableStraightLineSet *ls = new TEveScalableStraightLineSet("rhoZCluster");
vec = TEveVector(cluster.x(), cluster.y(), cluster.z());
phi = vec.Phi();
theta = vec.Theta();
FWViewEnergyScale *caloScale = vc->getEnergyScale();
ls->SetLineWidth(4);
if (theta < caloTransAngle || TMath::Pi() - theta < caloTransAngle)
rad = ecalZ / fabs(cos(theta));
else
rad = r / sin(theta);
ls->SetScaleCenter(0., (phi > 0 ? rad * fabs(sin(theta)) : -rad * fabs(sin(theta))), rad * cos(theta));
ls->AddLine(0.,
(phi > 0 ? rad * fabs(sin(theta)) : -rad * fabs(sin(theta))),
rad * cos(theta),
0.,
(phi > 0 ? (rad + size) * fabs(sin(theta)) : -(rad + size) * fabs(sin(theta))),
(rad + size) * cos(theta));
ls->SetScale(caloScale->getScaleFactor3D() * (caloScale->getPlotEt() ? et : e));
return ls;
}
|