|
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
|
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "CondFormats/PhysicsToolsObjects/interface/PhysicsTFormulaPayload.h"
#include "CondFormats/PhysicsToolsObjects/test/TFormulaWriter.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <TFile.h>
#include <TFormula.h>
TFormulaWriter::TFormulaWriter(const edm::ParameterSet& cfg)
: moduleLabel_(cfg.getParameter<std::string>("@module_label")) {
edm::VParameterSet cfgJobs = cfg.getParameter<edm::VParameterSet>("jobs");
for (edm::VParameterSet::const_iterator cfgJob = cfgJobs.begin(); cfgJob != cfgJobs.end(); ++cfgJob) {
jobEntryType* job = new jobEntryType(*cfgJob);
jobs_.push_back(job);
}
}
TFormulaWriter::~TFormulaWriter() {
for (std::vector<jobEntryType*>::iterator it = jobs_.begin(); it != jobs_.end(); ++it) {
delete (*it);
}
}
void TFormulaWriter::analyze(const edm::Event&, const edm::EventSetup&) {
std::cout << "<TFormulaWriter::analyze (moduleLabel = " << moduleLabel_ << ")>:" << std::endl;
for (std::vector<jobEntryType*>::iterator job = jobs_.begin(); job != jobs_.end(); ++job) {
TFile* inputFile = new TFile((*job)->inputFileName_.data());
std::cout << "reading TFormula = " << (*job)->formulaName_ << " from ROOT file = " << (*job)->inputFileName_ << "."
<< std::endl;
const TFormula* formula = dynamic_cast<TFormula*>(inputFile->Get((*job)->formulaName_.data()));
std::cout << "the formula is " << formula->GetExpFormula("p") << std::endl;
delete inputFile;
if (!formula)
throw cms::Exception("TFormulaWriter") << " Failed to load TFormula = " << (*job)->formulaName_.data()
<< " from file = " << (*job)->inputFileName_ << " !!\n";
edm::Service<cond::service::PoolDBOutputService> dbService;
if (!dbService.isAvailable())
throw cms::Exception("TFormulaWriter") << " Failed to access PoolDBOutputService !!\n";
std::cout << " writing TFormula = " << (*job)->formulaName_
<< " to SQLlite file, record = " << (*job)->outputRecord_ << "." << std::endl;
typedef std::pair<float, float> vfloat;
std::vector<vfloat> limits;
limits.push_back(vfloat(0., 1.e+6));
std::vector<std::string> formulas;
formulas.push_back((formula->GetExpFormula("p")).Data());
PhysicsTFormulaPayload formulaPayload(limits, formulas);
delete formula;
dbService->writeOneIOV(formulaPayload, dbService->beginOfTime(), (*job)->outputRecord_);
}
std::cout << "done." << std::endl;
}
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(TFormulaWriter);
|