|
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
// -*- C++ -*-
//
// Package: CondFormats/SiPixelObjects
// Class: SiPixelQualityProbabilitiesTestWriter
//
/**\class SiPixelQualityProbabilitiesTestWriter SiPixelQualityProbabilitiesTestWriter.cc CondFormats/SiPixelObjects/plugins/SiPixelQualityProbabilitiesTestWriter.cc
Description: class to build the SiPixel Quality probabilities
*/
//
// Original Author: Marco Musich
// Created: Wed, 30 Nov 2018 13:22:00 GMT
//
//
// system include files
#include <memory>
#include <fstream>
// user include files
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/SiPixelObjects/interface/SiPixelQuality.h"
#include "CondFormats/SiPixelObjects/interface/SiPixelQualityProbabilities.h"
#include "CondFormats/DataRecord/interface/SiPixelQualityFromDbRcd.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
//
// class declaration
//
class SiPixelQualityProbabilitiesWriteFromASCII : public edm::one::EDAnalyzer<> {
public:
explicit SiPixelQualityProbabilitiesWriteFromASCII(const edm::ParameterSet&);
~SiPixelQualityProbabilitiesWriteFromASCII() override;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
void analyze(const edm::Event&, const edm::EventSetup&) override;
void endJob() override;
// ----------member data ---------------------------
const std::string m_ProbInputs;
const std::string m_record;
const bool printdebug_;
std::unique_ptr<SiPixelQualityProbabilities> myProbabilities;
};
//
// constructors and destructor
//
SiPixelQualityProbabilitiesWriteFromASCII::SiPixelQualityProbabilitiesWriteFromASCII(const edm::ParameterSet& iConfig)
: m_ProbInputs(iConfig.getParameter<std::string>("probabilities")),
m_record(iConfig.getParameter<std::string>("record")),
printdebug_(iConfig.getUntrackedParameter<bool>("printDebug", false)) {
//now do what ever initialization is needed
myProbabilities = std::make_unique<SiPixelQualityProbabilities>();
}
SiPixelQualityProbabilitiesWriteFromASCII::~SiPixelQualityProbabilitiesWriteFromASCII() = default;
//
// member functions
//
// ------------ method called for each event ------------
void SiPixelQualityProbabilitiesWriteFromASCII::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
using namespace edm;
std::ifstream myfile(m_ProbInputs);
std::string line;
SiPixelQualityProbabilities::probabilityVec myProbVector;
if (myfile.is_open()) {
while (getline(myfile, line)) {
if (printdebug_) {
edm::LogInfo("SiPixelQualityProbabilitiesWriteFromASCII") << line << std::endl;
}
std::istringstream iss(line);
int pileupBinId, nEntries;
iss >> pileupBinId >> nEntries;
edm::LogInfo("SiPixelQualityProbabilitiesWriteFromASCII")
<< "PILEUP BIN/ENTRIES: " << pileupBinId << " " << nEntries << std::endl;
std::vector<std::string> ids(nEntries, "");
std::vector<float> probs(nEntries, 0.0);
for (int i = 0; i < nEntries; ++i) {
iss >> ids.at(i) >> probs.at(i);
if (printdebug_) {
edm::LogInfo("SiPixelQualityProbabilitiesWriteFromASCII") << ids.at(i) << " " << probs.at(i) << std::endl;
}
auto idAndProb = std::make_pair(ids.at(i), probs.at(i));
myProbVector.push_back(idAndProb);
}
if (nEntries > 0) {
myProbabilities->setProbabilities(pileupBinId, myProbVector);
}
myProbVector.clear();
}
myfile.close();
}
if (printdebug_) {
edm::LogInfo("SiPixelQualityProbabilitiesWriteFromASCII") << "Content of SiPixelQualityProbabilities " << std::endl;
// use buil-in method in the CondFormat
myProbabilities->printAll();
}
}
// ------------ method called once each job just after ending the event loop ------------
void SiPixelQualityProbabilitiesWriteFromASCII::endJob() {
edm::LogInfo("SiPixelQualityProbabilitiesWriteFromASCII")
<< "Size of SiPixelQualityProbabilities object " << myProbabilities->size() << std::endl
<< std::endl;
// Form the data here
edm::Service<cond::service::PoolDBOutputService> poolDbService;
if (poolDbService.isAvailable()) {
cond::Time_t valid_time = poolDbService->currentTime();
// this writes the payload to begin in current run defined in cfg
poolDbService->writeOneIOV(*myProbabilities, valid_time, m_record);
}
}
// ------------ method fills 'descriptions' with the allowed parameters for the module ------------
void SiPixelQualityProbabilitiesWriteFromASCII::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.setComment("Writes payloads of type SiPixelQualityProbabilities");
desc.addUntracked<bool>("printDebug", true);
desc.add<std::string>("record", "SiPixelStatusScenarioProbabilityRcd");
desc.add<std::string>("probabilities", "");
descriptions.add("SiPixelQualityProbabilitiesWriteFromASCII", desc);
}
//define this as a plug-in
DEFINE_FWK_MODULE(SiPixelQualityProbabilitiesWriteFromASCII);
|