ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/GPetrucc/interface/fwlite/Scanner.h
(Generate patch)

Comparing UserCode/GPetrucc/interface/fwlite/Scanner.h (file contents):
Revision 1.3 by gpetrucc, Thu Jan 28 12:00:32 2010 UTC vs.
Revision 1.5 by gpetrucc, Sun Jan 31 19:44:41 2010 UTC

# Line 5 | Line 5
5   #include <TString.h>
6   #include <TObjString.h>
7   #include <TObjArray.h>
8 + #include <TEnv.h>
9  
10   #if !defined(__CINT__) && !defined(__MAKECINT__)
11   #include "UserCode/GPetrucc/interface/fwliteHelpers.h"
# Line 20 | Line 21 | namespace fwlite {
21              typedef fwlite::Handle<std::vector<T> > HandleT;
22              Scanner(fwlite::EventBase *ev, const char *label, const char *instance = "", const char *process="")  :
23                  //helper::ScannerBase(Reflex::Type::ByTypeInfo(typeid(T))),
24 <                event_(ev), label_(label), instance_(instance), printFullEventId_(ev->isRealData())
24 >                event_(ev), label_(label), instance_(instance),
25 >                printFullEventId_(ev->isRealData()),
26 >                ignoreExceptions_(false),
27 >                exprSep_(":")
28              {
29                  Reflex::Type wrapperType = Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo());
30                  objType = helpme.elementType(Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo()));
# Line 29 | Line 33 | namespace fwlite {
33              
34              void scan(const char *exprs, const char *cut="", int nmax=20) {
35                  helper::ScannerBase scanner(objType);
36 +                scanner.setIgnoreExceptions(ignoreExceptions_);
37  
38 <                TObjArray  *exprArray = TString(exprs).Tokenize(":");
38 >                TObjArray  *exprArray = TString(exprs).Tokenize(exprSep_);
39                  int rowline = 0;
40                  if (printFullEventId_) {
41                      printf(" : %9s : %4s : %9s : %3s", "RUN", "LUMI", "EVENT", "#IT");
# Line 72 | Line 77 | namespace fwlite {
77                  }
78                  std::cout << std::endl;
79              }
80 +
81 +
82 +            TH1 * draw(const char *expr, const char *cut = "", TString drawopt = "", TH1 *hist = 0) {
83 +                // prep the machinery
84 +                helper::ScannerBase scanner(objType);
85 +                scanner.setIgnoreExceptions(ignoreExceptions_);
86 +                scanner.addExpression(expr);
87 +                if (strlen(cut)) scanner.setCut(cut);
88 +
89 +                // make histo, if needed
90 +                if (hist == 0) {
91 +                    hist = new TH1F("htemp",
92 +                            (strlen(cut) ? TString(expr)+"{"+cut+"}" : TString(expr)),
93 +                            gEnv->GetValue("Hist.Binning.1D.x",100), 0, 0);
94 +                    hist->SetBit(TH1::kCanRebin);
95 +                }
96 +
97 +                // fill histogram
98 +                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
99 +                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
100 +                    const std::vector<T> & vals = *handle_;
101 +                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
102 +                        scanner.fill1D(&vals[j], hist);
103 +                    }
104 +                }
105 +
106 +                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
107 +                return hist;
108 +            }
109 +            TH1 * draw(const char *expr, int bins, double xlow, double xhigh, const char *cut = "", const char *drawopt = "") {
110 +                TH1 * htemp = new TH1F("htemp", (strlen(cut) ? TString(expr)+"{"+cut+"}" : TString(expr)), bins, xlow, xhigh);
111 +                return draw(expr,cut,drawopt,htemp);
112 +            }
113 +
114 +            TProfile * drawProf(TString xexpr, TString yexpr, const char *cut = "", TString drawopt = "", TProfile *hist = 0) {
115 +                // prep the machinery
116 +                helper::ScannerBase scanner(objType);
117 +                scanner.setIgnoreExceptions(ignoreExceptions_);
118 +                scanner.addExpression(xexpr);
119 +                scanner.addExpression(yexpr);
120 +                if (strlen(cut)) scanner.setCut(cut);
121 +
122 +                // make histo, if needed
123 +                if (hist == 0) {
124 +                    hist = new TProfile("htemp",
125 +                            (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr),
126 +                            gEnv->GetValue("Hist.Binning.1D.x",100), 0., 0.);
127 +                   hist->SetBit(TH1::kCanRebin);
128 +                }
129 +
130 +                // fill histogram
131 +                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
132 +                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
133 +                    const std::vector<T> & vals = *handle_;
134 +                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
135 +                        scanner.fillProf(&vals[j], hist);
136 +                    }
137 +                }
138 +
139 +                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
140 +                return hist;
141 +            }
142 +            TProfile * drawProf(TString xexpr, int bins, double xlow, double xhigh, TString yexpr, const char *cut = "", const char *drawopt = "") {
143 +                TProfile * htemp = new TProfile("htemp", (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr), bins, xlow, xhigh);
144 +                return drawProf(xexpr,yexpr,cut,drawopt,htemp);
145 +            }
146 +
147 +            TH2 * draw2D(TString xexpr, TString yexpr, const char *cut = "", TString drawopt = "", TH2 *hist = 0) {
148 +                // prep the machinery
149 +                helper::ScannerBase scanner(objType);
150 +                scanner.setIgnoreExceptions(ignoreExceptions_);
151 +                scanner.addExpression(xexpr);
152 +                scanner.addExpression(yexpr);
153 +                if (strlen(cut)) scanner.setCut(cut);
154 +
155 +                // make histo, if needed
156 +                if (hist == 0) {
157 +                    hist = new TH2F("htemp",
158 +                            (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr),
159 +                            gEnv->GetValue("Hist.Binning.2D.x",20), 0, 0,
160 +                            gEnv->GetValue("Hist.Binning.2D.y",20), 0, 0);
161 +                   hist->SetBit(TH1::kCanRebin);
162 +                }
163 +
164 +                // fill histogram
165 +                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
166 +                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
167 +                    const std::vector<T> & vals = *handle_;
168 +                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
169 +                        scanner.fill2D(&vals[j], hist);
170 +                    }
171 +                }
172 +
173 +                if (!strlen(hist->GetXaxis()->GetTitle())) hist->GetXaxis()->SetTitle(xexpr);
174 +                if (!strlen(hist->GetYaxis()->GetTitle())) hist->GetYaxis()->SetTitle(yexpr);
175 +                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
176 +                return hist;
177 +            }
178 +            TH2 * draw2D(TString xexpr, int xbins, double xlow, double xhigh,
179 +                         TString yexpr, int ybins, double ylow, double yhigh,
180 +                         const char *cut = "", const char *drawopt = "") {
181 +                TH2 * htemp = new TH2F("htemp", (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr),
182 +                                        xbins, xlow, xhigh, ybins,ylow,yhigh);
183 +                return draw2D(xexpr,yexpr,cut,drawopt,htemp);
184 +            }
185 +
186              void setPrintFullEventId(bool printIt=true) { printFullEventId_ = printIt; }
187 +            void setExpressionSeparator(TString separator) { exprSep_ = separator; }
188 +            void setIgnoreExceptions(bool ignoreThem) { ignoreExceptions_ = ignoreThem; }
189          private:
190              fwlite::EventBase *event_;
191              std::string    label_, instance_, process_;
192              bool printFullEventId_;
193 +            bool ignoreExceptions_;
194 +            TString exprSep_;
195              HandleT        handle_;
196              helper::Parser helpme;
197 <            Reflex::Type objType;
197 >            Reflex::Type   objType;
198  
199      };
200   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines