interface/fwlite/Scanner.h

// these includes are FWLite-safe
#include "DataFormats/FWLite/interface/Handle.h"
#include "DataFormats/FWLite/interface/Event.h"
// these are from ROOT, so they're safe too
#include <TString.h>
#include <TObjString.h>
#include <TObjArray.h>
#include <TEnv.h>

#if !defined(__CINT__) && !defined(__MAKECINT__)
#include "UserCode/GPetrucc/interface/fwliteHelpers.h"
#else
// load the library that contains the dictionaries
int _load_UserCodeGPetrucc = gSystem->Load("libUserCodeGPetrucc.so");
#endif

namespace fwlite {
    template<typename T>
    class Scanner {
        public:
            typedef fwlite::Handle<std::vector<T> > HandleT;
            Scanner(fwlite::EventBase *ev, const char *label, const char *instance = "", const char *process="")  :
                //helper::ScannerBase(Reflex::Type::ByTypeInfo(typeid(T))),
                event_(ev), label_(label), instance_(instance), 
                printFullEventId_(ev->isRealData()),
                ignoreExceptions_(false),
                exprSep_(":")
            {
                Reflex::Type wrapperType = Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo());
                objType = helpme.elementType(Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo()));
            }

            
            void scan(const char *exprs, const char *cut="", int nmax=20) {
                helper::ScannerBase scanner(objType); 
                scanner.setIgnoreExceptions(ignoreExceptions_);

                TObjArray  *exprArray = TString(exprs).Tokenize(exprSep_);
                int rowline = 0;
                if (printFullEventId_) {
                    printf(" : %9s : %4s : %9s : %3s", "RUN", "LUMI", "EVENT", "#IT");
                    rowline += 3*4+9+4+9+3-1; // -1 as first char remain blank
                } else {
                    printf(" : %5s : %3s", "EVENT", "#IT");
                    rowline += 3+6+3+3-1;  // -1 as first char remain blank
                }
                for (int i = 0; i < exprArray->GetEntries(); ++i) {
                    const char *ex = ((TObjString *)(*exprArray)[i])->GetString();
                    scanner.addExpression(ex);
                    printf(" : %8s", (strlen(ex)>8 ? ex+strlen(ex)-8 : ex)); // the rightmost part is usually the more interesting one
                    rowline += 3+8;
                }
                std::cout << " :" << std::endl;
                rowline += 2;
                delete exprArray;

                TString rule('-', rowline);
                std::cout << " " << rule << " " << std::endl;

                if (strlen(cut)) scanner.setCut(cut);

                int iev = 0;
                for (event_->toBegin(); (iev < nmax) && !event_->atEnd(); ++iev, ++(*event_)) {
                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
                    const std::vector<T> & vals = *handle_;
                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
                        if (!scanner.test(&vals[j])) continue;
                        if (printFullEventId_) {
                            const edm::EventAuxiliary &id = event_->eventAuxiliary();
                            printf(" : %9d : %4d : %9d : %3d", id.run(), id.luminosityBlock(), id.event(), j);
                        } else {
                            printf(" : %5d : %3d", iev, j);
                        }
                        scanner.print(&vals[j]);
                        std::cout << " :" << std::endl;
                    }
                }
                std::cout << std::endl;
            }


            TH1 * draw(const char *expr, const char *cut = "", TString drawopt = "", TH1 *hist = 0) {
                // prep the machinery
                helper::ScannerBase scanner(objType);
                scanner.setIgnoreExceptions(ignoreExceptions_);
                scanner.addExpression(expr);
                if (strlen(cut)) scanner.setCut(cut);

                // make histo, if needed
                if (hist == 0) {
                    hist = new TH1F("htemp", 
                            (strlen(cut) ? TString(expr)+"{"+cut+"}" : TString(expr)), 
                            gEnv->GetValue("Hist.Binning.1D.x",100), 0, 0);
                    hist->SetBit(TH1::kCanRebin);
                }

                // fill histogram
                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
                    const std::vector<T> & vals = *handle_;
                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
                        scanner.fill1D(&vals[j], hist);
                    }
                }

                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
                return hist;
            }
            TH1 * draw(const char *expr, int bins, double xlow, double xhigh, const char *cut = "", const char *drawopt = "") {
                TH1 * htemp = new TH1F("htemp", (strlen(cut) ? TString(expr)+"{"+cut+"}" : TString(expr)), bins, xlow, xhigh);
                return draw(expr,cut,drawopt,htemp);
            }

            TProfile * drawProf(TString xexpr, TString yexpr, const char *cut = "", TString drawopt = "", TProfile *hist = 0) {
                // prep the machinery
                helper::ScannerBase scanner(objType);
                scanner.setIgnoreExceptions(ignoreExceptions_);
                scanner.addExpression(xexpr);
                scanner.addExpression(yexpr);
                if (strlen(cut)) scanner.setCut(cut);

                // make histo, if needed
                if (hist == 0) {
                    hist = new TProfile("htemp", 
                            (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr), 
                            gEnv->GetValue("Hist.Binning.1D.x",100), 0., 0.);
                   hist->SetBit(TH1::kCanRebin);
                }

                // fill histogram
                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
                    const std::vector<T> & vals = *handle_;
                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
                        scanner.fillProf(&vals[j], hist);
                    }
                }

                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
                return hist;
            }
            TProfile * drawProf(TString xexpr, int bins, double xlow, double xhigh, TString yexpr, const char *cut = "", const char *drawopt = "") {
                TProfile * htemp = new TProfile("htemp", (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr), bins, xlow, xhigh);
                return drawProf(xexpr,yexpr,cut,drawopt,htemp);
            }

            TH2 * draw2D(TString xexpr, TString yexpr, const char *cut = "", TString drawopt = "", TH2 *hist = 0) {
                // prep the machinery
                helper::ScannerBase scanner(objType);
                scanner.setIgnoreExceptions(ignoreExceptions_);
                scanner.addExpression(xexpr);
                scanner.addExpression(yexpr);
                if (strlen(cut)) scanner.setCut(cut);

                // make histo, if needed
                if (hist == 0) {
                    hist = new TH2F("htemp", 
                            (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr), 
                            gEnv->GetValue("Hist.Binning.2D.x",20), 0, 0,
                            gEnv->GetValue("Hist.Binning.2D.y",20), 0, 0);
                   hist->SetBit(TH1::kCanRebin);
                }

                // fill histogram
                for (event_->toBegin(); !event_->atEnd(); ++(*event_)) {
                    handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
                    const std::vector<T> & vals = *handle_;
                    for (size_t j = 0, n = vals.size(); j < n; ++j) {
                        scanner.fill2D(&vals[j], hist);
                    }
                }

                if (!strlen(hist->GetXaxis()->GetTitle())) hist->GetXaxis()->SetTitle(xexpr);
                if (!strlen(hist->GetYaxis()->GetTitle())) hist->GetYaxis()->SetTitle(yexpr);
                if (!TString(drawopt).Contains("goff",TString::kIgnoreCase)) hist->Draw(drawopt);
                return hist;
            }
            TH2 * draw2D(TString xexpr, int xbins, double xlow, double xhigh, 
                         TString yexpr, int ybins, double ylow, double yhigh,
                         const char *cut = "", const char *drawopt = "") {
                TH2 * htemp = new TH2F("htemp", (strlen(cut) ? yexpr+":"+xexpr+"{"+cut+"}" : yexpr+":"+xexpr),
                                        xbins, xlow, xhigh, ybins,ylow,yhigh);
                return draw2D(xexpr,yexpr,cut,drawopt,htemp);
            }
 
            void setPrintFullEventId(bool printIt=true) { printFullEventId_ = printIt; }
            void setExpressionSeparator(TString separator) { exprSep_ = separator; }
            void setIgnoreExceptions(bool ignoreThem) { ignoreExceptions_ = ignoreThem; }
        private:
            fwlite::EventBase *event_;
            std::string    label_, instance_, process_;
            bool printFullEventId_;
            bool ignoreExceptions_;
            TString exprSep_;
            HandleT        handle_;
            helper::Parser helpme;
            Reflex::Type   objType;

    };
}
Revision:	1.5
Committed:	Sun Jan 31 19:44:41 2010 UTC (15 years, 3 months ago) by gpetrucc
Content type:	text/plain
Branch:	MAIN
Changes since 1.4:	+7 -0 lines
Log Message:	allow ignoring exceptions when scanning
#	User	Rev	Content
1	gpetrucc	1.1	// these includes are FWLite-safe
2			#include "DataFormats/FWLite/interface/Handle.h"
3			#include "DataFormats/FWLite/interface/Event.h"
4			// these are from ROOT, so they're safe too
5			#include <TString.h>
6			#include <TObjString.h>
7			#include <TObjArray.h>
8	gpetrucc	1.4	#include <TEnv.h>
9	gpetrucc	1.1
10			#if !defined(__CINT__) && !defined(__MAKECINT__)
11			#include "UserCode/GPetrucc/interface/fwliteHelpers.h"
12			#else
13			// load the library that contains the dictionaries
14			int _load_UserCodeGPetrucc = gSystem->Load("libUserCodeGPetrucc.so");
15			#endif
16
17			namespace fwlite {
18			template<typename T>
19			class Scanner {
20			public:
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	gpetrucc	1.4	event_(ev), label_(label), instance_(instance),
25			printFullEventId_(ev->isRealData()),
26	gpetrucc	1.5	ignoreExceptions_(false),
27	gpetrucc	1.4	exprSep_(":")
28	gpetrucc	1.1	{
29			Reflex::Type wrapperType = Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo());
30			objType = helpme.elementType(Reflex::Type::ByTypeInfo(HandleT::TempWrapT::typeInfo()));
31			}
32
33
34			void scan(const char exprs, const char cut="", int nmax=20) {
35			helper::ScannerBase scanner(objType);
36	gpetrucc	1.5	scanner.setIgnoreExceptions(ignoreExceptions_);
37	gpetrucc	1.1
38	gpetrucc	1.4	TObjArray *exprArray = TString(exprs).Tokenize(exprSep_);
39	gpetrucc	1.1	int rowline = 0;
40			if (printFullEventId_) {
41			printf(" : %9s : %4s : %9s : %3s", "RUN", "LUMI", "EVENT", "#IT");
42			rowline += 3*4+9+4+9+3-1; // -1 as first char remain blank
43			} else {
44	gpetrucc	1.2	printf(" : %5s : %3s", "EVENT", "#IT");
45	gpetrucc	1.1	rowline += 3+6+3+3-1; // -1 as first char remain blank
46			}
47			for (int i = 0; i < exprArray->GetEntries(); ++i) {
48			const char ex = ((TObjString )(*exprArray)[i])->GetString();
49			scanner.addExpression(ex);
50	gpetrucc	1.3	printf(" : %8s", (strlen(ex)>8 ? ex+strlen(ex)-8 : ex)); // the rightmost part is usually the more interesting one
51	gpetrucc	1.1	rowline += 3+8;
52			}
53			std::cout << " :" << std::endl;
54			rowline += 2;
55			delete exprArray;
56
57			TString rule('-', rowline);
58			std::cout << " " << rule << " " << std::endl;
59
60			if (strlen(cut)) scanner.setCut(cut);
61
62			int iev = 0;
63			for (event_->toBegin(); (iev < nmax) && !event_->atEnd(); ++iev, ++(*event_)) {
64			handle_.getByLabel(*event_, label_.c_str(), instance_.c_str(), process_.c_str());
65			const std::vector<T> & vals = *handle_;
66			for (size_t j = 0, n = vals.size(); j < n; ++j) {
67			if (!scanner.test(&vals[j])) continue;
68			if (printFullEventId_) {
69			const edm::EventAuxiliary &id = event_->eventAuxiliary();
70			printf(" : %9d : %4d : %9d : %3d", id.run(), id.luminosityBlock(), id.event(), j);
71			} else {
72			printf(" : %5d : %3d", iev, j);
73			}
74			scanner.print(&vals[j]);
75			std::cout << " :" << std::endl;
76			}
77			}
78			std::cout << std::endl;
79			}
80	gpetrucc	1.4
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	gpetrucc	1.5	scanner.setIgnoreExceptions(ignoreExceptions_);
86	gpetrucc	1.4	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	gpetrucc	1.5	scanner.setIgnoreExceptions(ignoreExceptions_);
118	gpetrucc	1.4	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	gpetrucc	1.5	scanner.setIgnoreExceptions(ignoreExceptions_);
151	gpetrucc	1.4	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	gpetrucc	1.1	void setPrintFullEventId(bool printIt=true) { printFullEventId_ = printIt; }
187	gpetrucc	1.4	void setExpressionSeparator(TString separator) { exprSep_ = separator; }
188	gpetrucc	1.5	void setIgnoreExceptions(bool ignoreThem) { ignoreExceptions_ = ignoreThem; }
189	gpetrucc	1.1	private:
190	gpetrucc	1.2	fwlite::EventBase *event_;
191			std::string label_, instance_, process_;
192	gpetrucc	1.1	bool printFullEventId_;
193	gpetrucc	1.5	bool ignoreExceptions_;
194	gpetrucc	1.4	TString exprSep_;
195	gpetrucc	1.1	HandleT handle_;
196			helper::Parser helpme;
197	gpetrucc	1.4	Reflex::Type objType;
198	gpetrucc	1.1
199			};
200			}