|
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#include "CondTools/Ecal/interface/EcalPFRecHitThresholdsHandler.h"
#include "FWCore/ParameterSet/interface/ParameterSetfwd.h"
#include "CondFormats/EcalObjects/interface/EcalPFRecHitThresholds.h"
#include "CondTools/Ecal/interface/EcalFloatCondObjectContainerXMLTranslator.h"
#include <iostream>
const Int_t kEBChannels = 61200, kEEChannels = 14648;
popcon::EcalPFRecHitThresholdsHandler::EcalPFRecHitThresholdsHandler(const edm::ParameterSet& ps)
: m_name(ps.getUntrackedParameter<std::string>("name", "EcalPFRecHitThresholdsHandler")) {
edm::LogInfo("EcalPFRecHitThresholds Source handler constructor\n");
m_firstRun = static_cast<unsigned int>(atoi(ps.getParameter<std::string>("firstRun").c_str()));
m_file_type = ps.getParameter<std::string>("type"); // xml/txt
m_file_name = ps.getParameter<std::string>("fileName");
}
popcon::EcalPFRecHitThresholdsHandler::~EcalPFRecHitThresholdsHandler() {}
void popcon::EcalPFRecHitThresholdsHandler::getNewObjects() {
// std::cout << "------- Ecal - > getNewObjects\n";
std::ostringstream ss;
ss << "ECAL ";
unsigned long long irun;
std::string file_ = m_file_name;
edm::LogInfo("going to open file ") << file_;
// EcalCondHeader header;
EcalPFRecHitThresholds* payload = new EcalPFRecHitThresholds;
if (m_file_type == "xml")
readXML(file_, *payload);
else
readTXT(file_, *payload);
irun = m_firstRun;
Time_t snc = (Time_t)irun;
popcon::PopConSourceHandler<EcalPFRecHitThresholds>::m_to_transfer.push_back(std::make_pair(payload, snc));
}
void popcon::EcalPFRecHitThresholdsHandler::readXML(const std::string& file_, EcalFloatCondObjectContainer& record) {
std::string dummyLine, bid;
std::ifstream fxml;
fxml.open(file_);
if (!fxml.is_open()) {
edm::LogInfo("ERROR : cannot open file ") << file_;
exit(1);
}
// header
for (int i = 0; i < 6; i++) {
getline(fxml, dummyLine); // skip first lines
// std::cout << dummyLine << std::endl;
}
fxml >> bid;
std::string stt = bid.substr(7, 5);
std::istringstream iEB(stt);
int nEB;
iEB >> nEB;
if (nEB != kEBChannels) {
edm::LogInfo("strange number of EB channels ") << nEB;
exit(-1);
}
fxml >> bid; // <item_version>0</item_version>
for (int iChannel = 0; iChannel < kEBChannels; iChannel++) {
EBDetId myEBDetId = EBDetId::unhashIndex(iChannel);
fxml >> bid;
std::size_t found = bid.find("</");
stt = bid.substr(6, found - 6);
float val = std::stof(stt);
record[myEBDetId] = val;
}
for (int i = 0; i < 5; i++) {
getline(fxml, dummyLine); // skip first lines
// std::cout << dummyLine << std::endl;
}
fxml >> bid;
stt = bid.substr(7, 5);
std::istringstream iEE(stt);
int nEE;
iEE >> nEE;
if (nEE != kEEChannels) {
edm::LogInfo("strange number of EE channels ") << nEE;
exit(-1);
}
fxml >> bid; // <item_version>0</item_version>
// now endcaps
for (int iChannel = 0; iChannel < kEEChannels; iChannel++) {
EEDetId myEEDetId = EEDetId::unhashIndex(iChannel);
fxml >> bid;
std::size_t found = bid.find("</");
stt = bid.substr(6, found - 6);
float val = std::stof(stt);
record[myEEDetId] = val;
}
}
void popcon::EcalPFRecHitThresholdsHandler::readTXT(const std::string& file_, EcalFloatCondObjectContainer& record) {
std::ifstream ftxt;
ftxt.open(file_);
if (!ftxt.is_open()) {
edm::LogInfo("ERROR : cannot open file ") << file_;
exit(1);
}
int number_of_lines = 0, eta, phi, x, y, z;
float val;
std::string line;
while (std::getline(ftxt, line)) {
if (number_of_lines < kEBChannels) { // barrel
sscanf(line.c_str(), "%i %i %i %f", &eta, &phi, &z, &val);
EBDetId ebdetid(eta, phi, EBDetId::ETAPHIMODE);
record[ebdetid] = val;
} else { // endcaps
sscanf(line.c_str(), "%i %i %i %f", &x, &y, &z, &val);
EEDetId eedetid(x, y, z, EEDetId::XYMODE);
record[eedetid] = val;
}
number_of_lines++;
}
edm::LogInfo("Number of lines in text file: ") << number_of_lines;
int kChannels = kEBChannels + kEEChannels;
if (number_of_lines != kChannels)
edm::LogInfo("wrong number of channels! Please check ");
}
|