|
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
|
#ifndef Fireworks_Calo_FWECALDetailViewBuilder_h
#define Fireworks_Calo_FWECALDetailViewBuilder_h
#include "Rtypes.h"
#include <map>
#include <vector>
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "Fireworks/Calo/interface/FWBoxRecHit.h"
namespace edm {
class EventBase;
}
class FWGeometry;
class TEveCaloDataVec;
class TEveCaloLego;
// builder class for ecal detail view
class FWECALDetailViewBuilder {
public:
// construct an ecal detail view builder
// the arguments are the event, a pointer to geometry object,
// the eta and phi position to show,
// the half width of the region (in indices, e.g. iEta) and
// the default color for the hits.
FWECALDetailViewBuilder(const edm::EventBase *event,
const FWGeometry *geom,
float eta,
float phi,
int size = 50,
Color_t defaultColor = kMagenta + 1);
// draw the ecal information with the preset colors
// (if any colors have been preset)
TEveCaloLego *build();
TEveCaloData *buildCaloData(bool xyEE);
// set colors of some predefined detids
void setColor(Color_t color, const std::vector<DetId> &detIds);
// show superclusters using two alternating colors
// to make adjacent clusters visible
void showSuperClusters(Color_t color1 = kGreen + 2, Color_t color2 = kTeal);
// show a specific supercluster in a specific color
void showSuperCluster(const reco::SuperCluster &cluster, Color_t color = kYellow);
// add legends; returns final y
double makeLegend(double x0 = 0.02,
double y0 = 0.95,
Color_t clustered1 = kGreen + 1,
Color_t clustered2 = kTeal,
Color_t supercluster = kYellow);
private:
// fill data
void fillData(TEveCaloDataVec *data);
void fillEtaPhi(const EcalRecHitCollection *hits, TEveCaloDataVec *data);
const edm::EventBase *m_event; // the event
const FWGeometry *m_geom; // the geometry
float m_eta; // eta position view centred on
float m_phi; // phi position view centred on
int m_size; // view half width in number of crystals
Color_t m_defaultColor; // default color for crystals
std::vector<FWBoxRecHit *> m_boxes;
// for keeping track of what det id goes in what slice
std::map<DetId, int> m_detIdsToColor;
TEveElement *m_towerList;
// sorting function to sort super clusters by eta.
static bool superClusterEtaLess(const reco::CaloCluster &lhs, const reco::CaloCluster &rhs) {
return (lhs.eta() < rhs.eta());
}
float sizeRad() const;
};
#endif // Fireworks_Calo_FWECALDetailViewBuilder_h
|