ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeService/src/TreeService.cc
(Generate patch)

Comparing UserCode/MitProd/TreeService/src/TreeService.cc (file contents):
Revision 1.3 by loizides, Wed Jun 11 13:10:27 2008 UTC vs.
Revision 1.7 by loizides, Tue Jul 1 16:33:55 2008 UTC

# Line 7 | Line 7
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;
# Line 14 | Line 15 | using namespace mithep;
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()
# Line 38 | Line 67 | TreeService::TreeService(const Parameter
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));

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines