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.11 by bendavid, Tue Mar 3 18:06:10 2009 UTC vs.
Revision 1.13 by loizides, Mon Mar 23 22:12:58 2009 UTC

# Line 6 | Line 6
6   #include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
7   #include "FWCore/ServiceRegistry/interface/Service.h"
8   #include "FWCore/MessageLogger/interface/JobReport.h"
9 + #include "FWCore/MessageLogger/interface/MessageLogger.h"
10   #include "MitAna/DataUtil/interface/TreeWriter.h"
11   #include "MitAna/DataTree/interface/Names.h"
12   #include "MitCommon/OptIO/interface/OptInt.h"
# Line 17 | Line 18 | using namespace mithep;
18   //--------------------------------------------------------------------------------------------------
19   TreeService::TreeService(const ParameterSet &cfg, ActivityRegistry &ar) :
20    tws_(0),
21 +  zipMode_(1),
22 +  bZipThres_(1),
23 +  gZipThres_(1),
24 +  lzoThres_(1),
25 +  lzmaThres_(1),
26 +  optIOVerbose_(1),
27    fDoReset_(cfg.getUntrackedParameter<bool>("doReset",1))
28   {
29    // Constructor.
# Line 54 | Line 61 | TreeService::TreeService(const Parameter
61    if (cfg.exists("brSizes"))
62      brSizes_=cfg.getUntrackedParameter<vector<unsigned> >("brSizes");
63    else
64 <    brSizes_.push_back(64*1024);
58 <
59 <  if (cfg.exists("zipMode"))
60 <    zipMode_=cfg.getUntrackedParameter<unsigned>("zipMode");
61 <  else
62 <    zipMode_ = 99;
63 <
64 <  if (cfg.exists("bZipThres"))
65 <    bZipThres_=cfg.getUntrackedParameter<double>("bZipThres");
66 <  else
67 <    bZipThres_ = -1.0;
68 <
69 <  if (cfg.exists("gZipThres"))
70 <    gZipThres_=cfg.getUntrackedParameter<double>("gZipThres");
71 <  else
72 <    gZipThres_ = 1.0;
73 <
74 <  if (cfg.exists("lzoThres"))
75 <    lzoThres_=cfg.getUntrackedParameter<double>("lzoThres");
76 <  else
77 <    lzoThres_ = -1.0;
78 <
79 <  if (cfg.exists("lzmaThres"))
80 <    lzmaThres_=cfg.getUntrackedParameter<double>("lzmaThres");
81 <  else
82 <    lzmaThres_ = 0.95;
83 <
84 <  if (cfg.exists("optIOVerbose"))
85 <    optIOVerbose_=cfg.getUntrackedParameter<unsigned>("optIOVerbose");
86 <  else
87 <    optIOVerbose_ = 0;
64 >    brSizes_.push_back(16*1024);
65  
66    if (treeNames_.size()!=fileNames_.size()) {
67      throw edm::Exception(edm::errors::Configuration, "TreeService::TreeService()\n")
# Line 93 | Line 70 | TreeService::TreeService(const Parameter
70      return;
71    }
72  
73 +  if (OptInt::IsActivated()) {
74 +    if (cfg.exists("zipMode"))
75 +      zipMode_=cfg.getUntrackedParameter<unsigned>("zipMode");
76 +    else
77 +      zipMode_ = 99;
78 +
79 +    if (cfg.exists("bZipThres"))
80 +      bZipThres_=cfg.getUntrackedParameter<double>("bZipThres");
81 +    else
82 +      bZipThres_ = -1.0;
83 +
84 +    if (cfg.exists("gZipThres"))
85 +      gZipThres_=cfg.getUntrackedParameter<double>("gZipThres");
86 +    else
87 +      gZipThres_ = 1.0;
88 +
89 +    if (cfg.exists("lzoThres"))
90 +      lzoThres_=cfg.getUntrackedParameter<double>("lzoThres");
91 +    else
92 +      lzoThres_ = -1.0;
93 +
94 +    if (cfg.exists("lzmaThres"))
95 +      lzmaThres_=cfg.getUntrackedParameter<double>("lzmaThres");
96 +    else
97 +      lzmaThres_ = 0.95;
98 +
99 +    if (cfg.exists("optIOVerbose"))
100 +      optIOVerbose_=cfg.getUntrackedParameter<unsigned>("optIOVerbose");
101 +    else
102 +      optIOVerbose_ = 0;
103 +
104 +    OptInt::SetZipMode(zipMode_);
105 +    OptInt::SetGzipFraction(gZipThres_);
106 +    OptInt::SetBzipFraction(bZipThres_);
107 +    OptInt::SetLzoFraction(lzoThres_);
108 +    OptInt::SetLzmaFraction(lzmaThres_);
109 +    OptInt::SetVerbose(optIOVerbose_);
110 +
111 +  } else {
112 +
113 +    if (cfg.exists("zipMode")   || cfg.exists("bZipThres") ||
114 +        cfg.exists("gZipThres") || cfg.exists("lzoThres")  ||
115 +        cfg.exists("lzmaThres")) {
116 +      edm::LogError("TreeService") << "OptIO interface not properly pre-loaded, "
117 +        "ignoring given settings." << std::endl;
118 +    }
119 +  }
120 +
121    // setup tw array
122    tws_.SetOwner(kTRUE);
123    tws_.Expand(treeNames_.size());
# Line 128 | Line 153 | TreeService::TreeService(const Parameter
153      else if (brSizes_.size()>0)
154        t->SetDefaultBrSize(brSizes_.at(0));
155  
131    OptInt::SetZipMode(zipMode_);
132
133    OptInt::SetGzipFraction(gZipThres_);
134
135    OptInt::SetBzipFraction(bZipThres_);
136
137    OptInt::SetLzoFraction(lzoThres_);
138
139    OptInt::SetLzmaFraction(lzmaThres_);
140
141    OptInt::SetVerbose(optIOVerbose_);
142
156      //t->Print();
157      tws_.Add(t);
158    }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines