7 |
|
#include "FWCore/ServiceRegistry/interface/Service.h" |
8 |
|
#include "FWCore/MessageLogger/interface/JobReport.h" |
9 |
|
#include "MitAna/DataUtil/interface/TreeWriter.h" |
10 |
+ |
#include "MitAna/DataTree/interface/Names.h" |
11 |
|
|
12 |
|
using namespace edm; |
13 |
|
using namespace std; |
15 |
|
|
16 |
|
//-------------------------------------------------------------------------------------------------- |
17 |
|
TreeService::TreeService(const ParameterSet &cfg, ActivityRegistry &r) : |
18 |
< |
tws_(0), |
18 |
< |
treeNames_ (cfg.getUntrackedParameter<vector<string> >("treeNames")), |
19 |
< |
fileNames_ (cfg.getUntrackedParameter<vector<string> >("fileNames")), |
20 |
< |
pathNames_ (cfg.getUntrackedParameter<vector<string> >("pathNames")), |
21 |
< |
maxSizes_ (cfg.getUntrackedParameter<vector<unsigned> >("maxSizes")), |
22 |
< |
compLevels_ (cfg.getUntrackedParameter<vector<unsigned> >("compLevels")), |
23 |
< |
splitLevels_(cfg.getUntrackedParameter<vector<unsigned> >("splitLevels")), |
24 |
< |
brSizes_ (cfg.getUntrackedParameter<vector<unsigned> >("brSizes")) |
18 |
> |
tws_(0) |
19 |
|
{ |
20 |
+ |
if (cfg.exists("treeNames")) |
21 |
+ |
treeNames_=cfg.getUntrackedParameter<vector<string> >("treeNames"); |
22 |
+ |
else |
23 |
+ |
treeNames_.push_back(Names::gkEvtTreeName); |
24 |
+ |
|
25 |
+ |
if (cfg.exists("fileNames")) |
26 |
+ |
fileNames_=cfg.getUntrackedParameter<vector<string> >("fileNames"); |
27 |
+ |
else |
28 |
+ |
fileNames_.push_back("mit-test"); |
29 |
+ |
|
30 |
+ |
if (cfg.exists("pathNames")) |
31 |
+ |
pathNames_=cfg.getUntrackedParameter<vector<string> >("pathNames"); |
32 |
+ |
else |
33 |
+ |
pathNames_.push_back("."); |
34 |
+ |
|
35 |
+ |
if (cfg.exists("maxSizes")) |
36 |
+ |
maxSizes_=cfg.getUntrackedParameter<vector<unsigned> >("maxSizes"); |
37 |
+ |
else |
38 |
+ |
maxSizes_.push_back(1024); |
39 |
+ |
|
40 |
+ |
if (cfg.exists("compLevels")) |
41 |
+ |
compLevels_=cfg.getUntrackedParameter<vector<unsigned> >("compLevels"); |
42 |
+ |
else |
43 |
+ |
compLevels_.push_back(9); |
44 |
+ |
|
45 |
+ |
if (cfg.exists("splitLevels")) |
46 |
+ |
splitLevels_=cfg.getUntrackedParameter<vector<unsigned> >("splitLevels"); |
47 |
+ |
else |
48 |
+ |
splitLevels_.push_back(99); |
49 |
+ |
|
50 |
+ |
if (cfg.exists("brSizes")) |
51 |
+ |
brSizes_=cfg.getUntrackedParameter<vector<unsigned> >("brSizes"); |
52 |
+ |
else |
53 |
+ |
brSizes_.push_back(32*1024); |
54 |
+ |
|
55 |
|
if (treeNames_.size()!=fileNames_.size()) { |
56 |
|
throw edm::Exception(edm::errors::Configuration, "TreeService::TreeService()\n") |
57 |
|
<< "Number of main trees should match number of files " << treeNames_.size() |
67 |
|
for (unsigned int i=0; i<treeNames_.size(); ++i) { |
68 |
|
|
69 |
|
TreeWriter *t = new TreeWriter(treeNames_.at(i).c_str(),1); |
41 |
– |
|
70 |
|
t->SetPrefix(fileNames_.at(i).c_str()); |
71 |
|
|
72 |
< |
if (i<pathNames_.size()) |
72 |
> |
if (i<pathNames_.size()) |
73 |
|
t->SetBaseURL(pathNames_.at(i).c_str()); |
74 |
|
else if (pathNames_.size()>0) |
75 |
|
t->SetBaseURL(pathNames_.at(0).c_str()); |
76 |
|
|
77 |
< |
if (i<maxSizes_.size()) |
78 |
< |
t->SetMaxSize(maxSizes_.at(i)*1024*1024); |
77 |
> |
if (i<maxSizes_.size()) |
78 |
> |
t->SetMaxSize((Long64_t)maxSizes_.at(i)*1024*1024); |
79 |
|
else if (maxSizes_.size()>0) |
80 |
< |
t->SetMaxSize(maxSizes_.at(0)*1024*1024); |
80 |
> |
t->SetMaxSize((Long64_t)maxSizes_.at(0)*1024*1024); |
81 |
|
|
82 |
< |
if (i<compLevels_.size()) |
82 |
> |
if (i<compLevels_.size()) |
83 |
|
t->SetCompressLevel(compLevels_.at(i)); |
84 |
|
else if (compLevels_.size()>0) |
85 |
|
t->SetCompressLevel(compLevels_.at(0)); |
86 |
|
|
87 |
< |
if (i<splitLevels_.size()) |
87 |
> |
if (i<splitLevels_.size()) |
88 |
|
t->SetDefaultSL(splitLevels_.at(i)); |
89 |
|
else if (splitLevels_.size()>0) |
90 |
|
t->SetDefaultSL(splitLevels_.at(0)); |
91 |
|
|
92 |
< |
if (i<brSizes_.size()) |
92 |
> |
if (i<brSizes_.size()) |
93 |
|
t->SetDefaultBrSize(brSizes_.at(i)); |
94 |
|
else if (brSizes_.size()>0) |
95 |
|
t->SetDefaultBrSize(brSizes_.at(0)); |