|
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
|
// -*- C++ -*-
//
// Package: Calo
// Class : FWHGTowerSliceSelector
//
// Implementation:
// [Notes on implementation]
//
// Original Author: Alja Mrak-Tadel
// Created: Wed Jun 2 17:39:44 CEST 2010
//
// system include files
// user include files
#include "TEveVector.h"
#include "TEveCaloData.h"
#include "TH2F.h"
#include "Fireworks/Calo/plugins/FWHGTowerSliceSelector.h"
#include "Fireworks/Core/interface/FWModelChangeManager.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/fwLog.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
//
// member functions
//
void FWHGTowerSliceSelector::doSelect(const TEveCaloData::CellId_t& iCell) {
if (!m_item)
return;
const HGCRecHitCollection* hits = nullptr;
m_item->get(hits);
assert(nullptr != hits);
int index = 0;
FWChangeSentry sentry(*(m_item->changeManager()));
for (HGCRecHitCollection::const_iterator it = hits->begin(); it != hits->end(); ++it, ++index) {
DetId id((*it).detid());
if (findBinFromId(id, iCell.fTower) && m_item->modelInfo(index).m_displayProperties.isVisible() &&
!m_item->modelInfo(index).isSelected()) {
// std::cout <<" doSelect "<<index<<std::endl;
m_item->select(index);
}
}
}
void FWHGTowerSliceSelector::doUnselect(const TEveCaloData::CellId_t& iCell) {
if (!m_item)
return;
const HGCRecHitCollection* hits = nullptr;
m_item->get(hits);
assert(nullptr != hits);
int index = 0;
FWChangeSentry sentry(*(m_item->changeManager()));
for (HGCRecHitCollection::const_iterator it = hits->begin(); it != hits->end(); ++it, ++index) {
DetId id((*it).detid());
if (findBinFromId(id, iCell.fTower) && m_item->modelInfo(index).m_displayProperties.isVisible() &&
m_item->modelInfo(index).isSelected()) {
// std::cout <<" doUnselect "<<index<<std::endl;
m_item->unselect(index);
}
}
}
bool FWHGTowerSliceSelector::findBinFromId(DetId& detId, int tower) const {
TEveCaloData::vCellId_t cellIds;
const float* corners = m_item->getGeom()->getCorners(detId);
if (corners == nullptr) {
fwLog(fwlog::kInfo) << "FWHGTowerSliceSelector cannot get geometry for DetId: " << detId.rawId() << ". Ignored.\n";
return false;
}
std::vector<TEveVector> front(4);
float eta = 0, phi = 0;
int j = 0;
for (int i = 0; i < 4; ++i) {
front[i] = TEveVector(corners[j], corners[j + 1], corners[j + 2]);
j += 3;
eta += front[i].Eta();
phi += front[i].Phi();
}
eta /= 4;
phi /= 4;
const TEveCaloData::CellGeom_t& cg = m_vecData->GetCellGeom()[tower];
if ((eta >= cg.fEtaMin && eta <= cg.fEtaMax) && (phi >= cg.fPhiMin && phi <= cg.fPhiMax)) {
return true;
}
return false;
}
|