|
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
|
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "DataFormats/Provenance/interface/ModuleDescription.h"
#include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/ModuleCallingContext.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/MessageLogger/interface/JobReport.h"
#include "TFile.h"
#include "TROOT.h"
#include <map>
#include <unistd.h>
const std::string TFileService::kSharedResource = "TFileService";
thread_local TFileDirectory TFileService::tFileDirectory_;
TFileService::TFileService(const edm::ParameterSet& cfg, edm::ActivityRegistry& r)
: file_(nullptr),
fileName_(cfg.getParameter<std::string>("fileName")),
fileNameRecorded_(false),
closeFileFast_(cfg.getUntrackedParameter<bool>("closeFileFast", false)) {
tFileDirectory_ = TFileDirectory("", "", TFile::Open(fileName_.c_str(), "RECREATE"), "");
file_ = tFileDirectory_.file_;
// activities to monitor in order to set the proper directory
r.watchPreModuleConstruction(this, &TFileService::setDirectoryName);
r.watchPreModuleBeginJob(this, &TFileService::setDirectoryName);
r.watchPreModuleEndJob(this, &TFileService::setDirectoryName);
r.watchPreModuleEvent(this, &TFileService::preModuleEvent);
r.watchPostModuleEvent(this, &TFileService::postModuleEvent);
r.watchPreModuleGlobalBeginRun(this, &TFileService::preModuleGlobal);
r.watchPostModuleGlobalBeginRun(this, &TFileService::postModuleGlobal);
r.watchPreModuleGlobalEndRun(this, &TFileService::preModuleGlobal);
r.watchPostModuleGlobalEndRun(this, &TFileService::postModuleGlobal);
r.watchPreModuleGlobalBeginLumi(this, &TFileService::preModuleGlobal);
r.watchPostModuleGlobalBeginLumi(this, &TFileService::postModuleGlobal);
r.watchPreModuleGlobalEndLumi(this, &TFileService::preModuleGlobal);
r.watchPostModuleGlobalEndLumi(this, &TFileService::postModuleGlobal);
// delay writing into JobReport after BeginJob
r.watchPostBeginJob(this, &TFileService::afterBeginJob);
}
TFileService::~TFileService() {
file_->Write();
if (closeFileFast_)
gROOT->GetListOfFiles()->Remove(file_);
file_->Close();
delete file_;
}
void TFileService::setDirectoryName(const edm::ModuleDescription& desc) {
tFileDirectory_.file_ = file_;
tFileDirectory_.dir_ = desc.moduleLabel();
tFileDirectory_.descr_ = tFileDirectory_.dir_ + " (" + desc.moduleName() + ") folder";
}
void TFileService::preModuleEvent(edm::StreamContext const&, edm::ModuleCallingContext const& mcc) {
setDirectoryName(*mcc.moduleDescription());
}
void TFileService::postModuleEvent(edm::StreamContext const&, edm::ModuleCallingContext const& mcc) {
edm::ModuleCallingContext const* previous_mcc = mcc.previousModuleOnThread();
if (previous_mcc) {
setDirectoryName(*previous_mcc->moduleDescription());
}
}
void TFileService::preModuleGlobal(edm::GlobalContext const&, edm::ModuleCallingContext const& mcc) {
setDirectoryName(*mcc.moduleDescription());
}
void TFileService::postModuleGlobal(edm::GlobalContext const&, edm::ModuleCallingContext const& mcc) {
edm::ModuleCallingContext const* previous_mcc = mcc.previousModuleOnThread();
if (previous_mcc) {
setDirectoryName(*previous_mcc->moduleDescription());
}
}
void TFileService::afterBeginJob() {
if (!fileName_.empty()) {
if (!fileNameRecorded_) {
std::string fullName(1024, '\0');
while (getcwd(&fullName[0], fullName.size()) == nullptr) {
if (errno != ERANGE) {
throw cms::Exception("TFileService")
<< "Failed to get current directory (errno=" << errno << "): " << strerror(errno);
}
fullName.resize(fullName.size() * 2, '\0');
}
fullName.resize(fullName.find('\0'));
fullName += "/" + fileName_;
std::map<std::string, std::string> fileData;
fileData.insert(std::make_pair("Source", "TFileService"));
edm::Service<edm::JobReport> reportSvc;
reportSvc->reportAnalysisFile(fullName, fileData);
fileNameRecorded_ = true;
}
}
}
|