3 * @author Christian Holm Christensen <cholm@nbi.dk>
4 * @date Thu Nov 17 12:08:04 2011
6 * @brief Class to plot QA trends
8 * @ingroup pwglf_forward_qa_scripts
15 # include <TGraphErrors.h>
16 # include <TGraphAsymmErrors.h>
17 # include <TMultiGraph.h>
30 class TGraphAsymmErrors;
41 * Class to plot QA trends
43 * @ingroup pwglf_forward_qa_scripts
45 struct QAPlotter : public QABase
50 struct Ring : public QARing
57 * @param useVar Use variance for errors (not min/max)
59 Ring(UShort_t d, Char_t r, Bool_t useVar=false)
73 fGChi2 = new TGraphAsymmErrors;
74 fGC = new TGraphAsymmErrors;
75 fGDelta = new TGraphAsymmErrors;
76 fGXi = new TGraphAsymmErrors;
77 fGSigma = new TGraphAsymmErrors;
79 fGSingles = new TGraph;
82 fGOccupancy = new TGraphAsymmErrors;
84 SetAtt(fGChi2, "chi2", "#LT#chi^{2}/#nu#GT");
85 SetAtt(fGC, "c", "Constant");
86 SetAtt(fGDelta, "delta", "#LT#Delta_{p}#GT");
87 SetAtt(fGXi, "xi", "#LT#xi#GT");
88 SetAtt(fGSigma, "sigma", "#LT#sigma#GT");
89 SetAtt(fGLow, "low", "# of low statistics bins");
90 SetAtt(fGSingles, "singles", "Fraction of single hits");
91 SetAtt(fGLoss, "loss", "Data lossed due to cuts");
92 SetAtt(fGOccupancy, "occupancy","#LTOccupancy#GT");
93 SetAtt(fGBeta, "beta", "Correlation of methods");
97 * Static member function to get the ring color
99 * @param d Detector number
100 * @param r Ring identifer
104 static Color_t RingColor(UShort_t d, Char_t r)
106 return ((d == 1 ? kRed : (d == 2 ? kGreen : kBlue))
107 + ((r == 'I' || r == 'i') ? 2 : -3));
110 * Set graph attributes
113 * @param name Name of graph
115 void SetAtt(TGraph* g, const char* name, const char* /*title=""*/)
117 Color_t c = RingColor(fD, fR);
118 g->SetName(Form("FMD%d%c_%s", fD, fR, name));
119 // g->SetTitle(Form("FMD%d%c %s", fD, fR,
120 // !title || title[0] == '\0' ? name : title));
121 Int_t marker = 20+(fD-1) + (fR == 'I' ? 0 : 4);
122 g->SetTitle(Form("FMD%d%c", fD, fR));
125 g->SetMarkerColor(c);
126 g->SetMarkerStyle(marker);
129 case 20: g->SetMarkerSize(1.2); break;
130 case 21: g->SetMarkerSize(1.2); break;
131 case 22: g->SetMarkerSize(1.3); break;
132 case 26: g->SetMarkerSize(1.1); break;
136 * Update a graph from a RingQuantity
138 * @param g Graph to update
140 * @param runNo Run number
142 void UpdateGraph(TGraphAsymmErrors* g, RingQuantity& q,
143 UInt_t runNo, UInt_t /* n */)
146 Double_t el = y-q.min;
147 Double_t eh = q.max-y;
149 //Info("UpdateGraph", "Setting errors on %s to variance",g->GetName());
153 if (TMath::Abs(y) < 1e-6) return;
155 g->SetPoint(i, runNo, y);
156 g->SetPointError(i, 0, 0, el, eh);
161 * @param n Entry number (not used)
162 * @param runNo Run number
164 * @return true on success
166 Bool_t Update(UInt_t n, UInt_t runNo)
168 UpdateGraph(fGChi2, *fChi2, runNo, n);
169 UpdateGraph(fGC, *fC, runNo, n);
170 UpdateGraph(fGDelta, *fDelta, runNo, n);
171 UpdateGraph(fGXi, *fXi, runNo, n);
172 UpdateGraph(fGSigma, *fSigma, runNo, n);
173 UpdateGraph(fGOccupancy, *fOccupancy, runNo, n);
175 fGLow->SetPoint(n, runNo, fFitStatus->nLow);
177 if (fMerge->one > 1e-6)
178 fGSingles->SetPoint(fGSingles->GetN(), runNo, fMerge->one);
179 if (fCorrelation->beta > 1e-6)
180 fGBeta->SetPoint(fGBeta->GetN(), runNo, fCorrelation->beta);
181 if (-fDataLoss->full > 1e-6)
182 fGLoss->SetPoint(fGLoss->GetN(), runNo, -fDataLoss->full);
185 TGraphAsymmErrors* fGChi2; // Graph of ELoss reduced chi-square
186 TGraphAsymmErrors* fGC; // Graph of ELoss constant
187 TGraphAsymmErrors* fGDelta; // Graph of ELoss MPV
188 TGraphAsymmErrors* fGXi; // Graph of ELoss Landau width
189 TGraphAsymmErrors* fGSigma; // Graph of ELoss Gaus width
190 TGraph* fGLow; // Graph of bins with low statistics
191 TGraph* fGSingles; // Graph of fraction of singles
192 TGraph* fGLoss; // Graph of 'lost' data
193 TGraph* fGBeta; // Graph of Poisson vs ELoss correlation
194 TGraphAsymmErrors* fGOccupancy;// Graph of mean occupancy
195 Bool_t fUseVar; // Use variance
200 QAPlotter(const TString& dataType,
202 const TString& period,
205 : QABase(dataType, year, period, pass),
210 Info("QAPlotter", "Do we use variance? %s", fUseVar ? "yes" : "no");
211 fFMD1i = new Ring(1, 'I', useVar);
212 fFMD2i = new Ring(2, 'I', useVar);
213 fFMD2o = new Ring(2, 'O', useVar);
214 fFMD3i = new Ring(3, 'I', useVar);
215 fFMD3o = new Ring(3, 'O', useVar);
216 fNAccepted = new TGraph;
217 fNAccepted->SetName("nAccepted");
218 fNAccepted->SetMarkerStyle(20);
219 fNAccepted->SetLineWidth(2);
221 fVz = new TGraphErrors;
223 fVz->SetMarkerStyle(20);
224 fVz->SetLineWidth(2);
227 * Add a file to be processed
229 * @param filename Name of file
231 void AddFile(const char* filename)
233 fFiles.Add(new TObjString(filename));
235 const char* GetUserInfo(TList* l, const char* name, const char* def) const
238 Warning("GetUserInfo", "No user information list");
242 TObject* o = l->FindObject(name);
244 Warning("GetUserInfo", "User information %s not found", name);
249 Info("GetUserInfo", "Got user information %s=%s", name, o->GetTitle());
250 return o->GetTitle();
255 * @param read If true, read from file
257 * @return True on success
259 Bool_t MakeTree(bool read)
261 if (fFiles.GetEntriesFast() <= 0) return QABase::MakeTree(read);
262 if (fFiles.GetEntriesFast() == 1 && read) {
263 TFile* file = TFile::Open(fFiles.At(0)->GetName(), "READ");
265 fTree = static_cast<TTree*>(file->Get("T"));
269 TChain* chain = new TChain("T", "T");
270 if (!chain->AddFileInfoList(&fFiles)) return false;
285 Error("Run", "No input tree");
291 UInt_t nEntries = fTree->GetEntries();
294 Info("Run", "Got %d runs", nEntries);
295 TList* l = fTree->GetUserInfo();
296 fPeriod = GetUserInfo(l, "period", "?");
297 fPass = GetUserInfo(l, "pass", "?");
298 fDataType = GetUserInfo(l, "type", "?");
301 for (UInt_t i = 0; i < nEntries; i++) {
304 UInt_t run = fGlobal->runNo;
305 UInt_t nev = fGlobal->nAccepted;
307 fFirst = TMath::Min(run, fFirst);
308 fLast = TMath::Max(run, fLast);
311 Info("Run", "Got run %d with %d accepted events", run, nev);
312 fNAccepted->SetPoint(i, run, nev);
313 fVz->SetPoint(i, run, fGlobal->meanVz);
314 fVz->SetPointError(i, 0, fGlobal->sigmaVz);
316 if (nev <= 100) continue;
317 static_cast<Ring*>(fFMD1i)->Update(j, run);
318 static_cast<Ring*>(fFMD2i)->Update(j, run);
319 static_cast<Ring*>(fFMD2o)->Update(j, run);
320 static_cast<Ring*>(fFMD3i)->Update(j, run);
321 static_cast<Ring*>(fFMD3o)->Update(j, run);
333 // fTeXName = Form("trend_%09d_%09d", fFirst, fLast);
335 if (!fPeriod.IsNull() && !fPass.IsNull())
336 title.Form("QA trends for %s/%s runs %d --- %d",
337 fPeriod.Data(), fPass.Data(), fFirst, fLast);
339 title.Form("QA trends for runs %d --- %d", fFirst, fLast);
342 CanvasTitle("# of accepted events");
343 fNAccepted->Draw("apl");
344 PutCanvasTitle("# of accepted events");
345 AddRuns(fNAccepted->GetHistogram(), "# of accepted events");
347 TLine* l = new TLine(fFirst, 100, fLast, 100);
348 l->SetLineColor(kRed+2);
351 PrintCanvas("nAccepted");
353 CanvasTitle("#LTv_{z}#GT");
355 PutCanvasTitle("Mean z coordinate of interaction point");
356 AddRuns(fVz->GetHistogram(), "#LTv_{z}#GT");
359 TMultiGraph* chi2 = new TMultiGraph;
360 TMultiGraph* c = new TMultiGraph;
361 TMultiGraph* delta = new TMultiGraph;
362 TMultiGraph* xi = new TMultiGraph;
363 TMultiGraph* sigma = new TMultiGraph;
364 TMultiGraph* low = new TMultiGraph;
365 TMultiGraph* singles = new TMultiGraph;
366 TMultiGraph* loss = new TMultiGraph;
367 TMultiGraph* occ = new TMultiGraph;
368 TMultiGraph* beta = new TMultiGraph;
369 chi2 ->SetName("chi2");
371 delta ->SetName("delta");
373 sigma ->SetName("sigma");
374 low ->SetName("low");
375 singles ->SetName("singles");
376 loss ->SetName("loss");
377 beta ->SetName("beta");
378 occ ->SetName("occupancy");
381 AddToMulti(fFMD1i,chi2, c, delta, xi, sigma, low, singles, loss, beta, occ);
382 AddToMulti(fFMD2i,chi2, c, delta, xi, sigma, low, singles, loss, beta, occ);
383 AddToMulti(fFMD2o,chi2, c, delta, xi, sigma, low, singles, loss, beta, occ);
384 AddToMulti(fFMD3i,chi2, c, delta, xi, sigma, low, singles, loss, beta, occ);
385 AddToMulti(fFMD3o,chi2, c, delta, xi, sigma, low, singles, loss, beta, occ);
387 PlotMulti(chi2, "#LT#chi^{2}/#nu#GT from #Delta fits", true);
388 PlotMulti(c, "#LTc#GT from #Delta fits");
389 PlotMulti(delta, "#LT#Delta_{p}#GT from #Delta fits");
390 PlotMulti(xi, "#LT#xi#GT from #Delta fits");
391 PlotMulti(sigma, "#LT#sigma#GT from #Delta fits");
392 PlotMulti(low, "Bins with too low statistics");
393 PlotMulti(singles, "Fraction of single hits");
394 PlotMulti(loss, "% of hits 'lost' due to merging+cuts");
395 PlotMulti(occ, "#LTOccupancy#GT [%]", true);
396 PlotMulti(beta, "Correlation of methods");
410 Close(false); // Do not delete PNGs
413 * Add graphs from a ring to the multi graphs
416 * @param chi2 ELoss reduced chi-square
417 * @param c ELoss constant
418 * @param delta ELoss MPV
419 * @param xi ELoss Landau width
420 * @param sigma ELoss Gaus width
421 * @param low bins with low statistics
422 * @param singles fraction of singles
423 * @param loss 'lost' data
424 * @param beta Poisson vs ELoss correlation
425 * @param occupancy mean occupancy
427 void AddToMulti(QARing* qr,
434 TMultiGraph* singles,
437 TMultiGraph* occupancy)
439 Ring* r = static_cast<Ring*>(qr);
440 chi2 ->Add(r->fGChi2);
442 delta ->Add(r->fGDelta);
444 sigma ->Add(r->fGSigma);
446 singles ->Add(r->fGSingles);
447 loss ->Add(r->fGLoss);
448 occupancy ->Add(r->fGOccupancy);
449 beta ->Add(r->fGBeta);
454 * @param mg Multi graph
456 * @param logy If true, make @f$\log@f$ scale
458 void PlotMulti(TMultiGraph* mg, const char* title, Bool_t logy=false)
461 // fCanvas->SetBottomMargin(.15);
462 fCanvas->SetLeftMargin(.08);
463 fCanvas->SetTopMargin(.1);
464 fCanvas->SetLogy(logy);
465 // fCanvas->SetRightMargin(.2);
468 TH1* h = mg->GetHistogram();
469 Double_t max = h->GetMaximum();
470 Double_t min = h->GetMinimum();
471 if (h->GetMinimum() == 0) {
472 min = min - .1*(max-min);
475 Int_t x1 = h->GetXaxis()->GetXmin();
476 Int_t x2 = h->GetXaxis()->GetXmax();
478 TLegend* l = new TLegend(.1, .91, .97, .95);
484 TIter next(mg->GetListOfGraphs());
485 mg->GetListOfGraphs();
488 // Get the runs we have here
489 TArrayI runs(fRuns.GetSize());
492 while ((g = static_cast<TGraph*>(next()))) {
493 l->AddEntry(g, g->GetTitle(), "lp");
494 Double_t* xs = g->GetX();
496 for (Int_t i = 0; i < n; i++) {
497 if (FindRun(runs, Int_t(xs[i])) >= 0) continue;
498 runs.SetAt(xs[i], j++);
500 Double_t ymean = g->GetMean(2);
501 Double_t xh = x2 - .03 * Double_t(x2-x1);
502 TLine* lm = new TLine(x1, ymean, xh, ymean);
503 lm->SetLineColor(g->GetLineColor());
507 TLatex* al = new TLatex(xh, ymean, g->GetTitle());
508 al->SetTextColor(g->GetLineColor());
510 al->SetTextSize(.02);
511 al->SetTextAlign(12);
518 AddRuns(h, title, &runs, &areas);
520 PrintCanvas(mg->GetName(), &areas);
525 * @param runs List of runs
526 * @param run Run to find
528 * @return Index of run in run list
530 Int_t FindRun(const TArrayI& runs, Int_t run)
532 std::sort(&(runs.fArray[0]), &(runs.fArray[runs.GetSize()]));
533 Int_t idx = TMath::BinarySearch(runs.GetSize(), runs.fArray, run);
534 if (idx >= runs.GetSize() || idx < 0 || runs[idx] != run) return -1;
538 * Add run labels at appropriate places on the plot
540 * @param h Frame histogram
542 * @param runs List of runs, if any
543 * @param areas Other areas
545 void AddRuns(TH1* h, const char* title, TArrayI* runs=0,
548 h->GetXaxis()->SetNoExponent();
549 // h->GetXaxis()->SetTitleOffset(1);
550 TString ytitle(title);
551 if (fUseVar) ytitle.Append(" (errors: variance)");
552 else ytitle.Append(" (errors: min/max)");
553 h->SetYTitle(ytitle.Data());
554 h->SetXTitle("Run #");
555 // Info("AddRuns", "%s: %s vs %s", h->GetName(),
556 // h->GetXaxis()->GetTitle(), h->GetYaxis()->GetTitle());
558 Int_t r1 = h->GetXaxis()->GetXmin();
559 Int_t r2 = h->GetXaxis()->GetXmax();
561 Double_t tx = .025; // (r2 - r1) / 18;
562 Double_t wx = 1 - fCanvas->GetLeftMargin() - fCanvas->GetRightMargin();
564 Double_t y = fCanvas->GetBottomMargin()+dy;
565 for (Int_t i = 0; i < fRuns.GetSize(); i++) {
567 Double_t x = fCanvas->GetLeftMargin() + wx*Double_t(r-r1)/(r2-r1);
569 // Skip runs out of range
570 if (r < r1 || r > r2) continue;
572 // Skip runs not in the graphs
574 if (FindRun(*runs, r) < 0) {
580 if (TMath::Abs(x - lx) < tx) y += dy;
581 else y = fCanvas->GetBottomMargin() + dy;
584 // Info(h->GetName(), "%6d (x,y)=(%12f,%12f) |lx-x|=|%f-x|=%f",
585 // r, x, y, lx, TMath::Abs(lx-x));
588 Double_t* xa = fNAccepted->GetX();
589 Int_t na = fNAccepted->GetN();
590 Int_t idx = TMath::BinarySearch(na, xa, Double_t(r));
591 Color_t color = kBlue+3;
592 if (idx >= 0 && idx < na && r == xa[idx] &&
593 fNAccepted->GetY()[idx] < 10000)
596 TLatex* ll = new TLatex(x, y, Form("%d", r));
598 ll->SetTextAlign(21);
599 ll->SetTextSize(0.02);
600 ll->SetTextColor(color);
602 // ll->SetTextAngle(90);
604 TLine* tl = new TLine(x, y, x, 1-fCanvas->GetTopMargin());
605 tl->SetBit(TLine::kLineNDC);
607 tl->SetLineColor(color);
610 if (!areas) continue;
612 TObjString* area = new TObjString;
613 TString& spec = area->String();
614 spec.Form("<span style=\"left: %d%%; bottom: %d%%;\" "
615 "onClick='window.location=\"%09d/index.html\"' "
616 "onMouseOver='this.style.cursor=\"pointer\"' "
619 UInt_t(100*x)-2, UInt_t(100*y), r, r, r);
622 spec.Form("<area shape='rect' alt='%d' title='%d' href='qa_%09d.html' "
623 "coords='%d,%d,%d,%d'>", r, r, r,
624 UInt_t(cw*(x-tx)), 0, UInt_t(cw*(x+tx)), ch);
630 * Output list of runs
632 * @param o Output stream
634 void WriteRuns(std::ostream& o)
636 o << "<div class='jobid'><!--JOBID--></div>\n"
637 << "<div class='runs'>\n"
639 for (Int_t i = 0; i < fRuns.GetSize(); i++) {
640 o << "<a href='" << Form("%09d", fRuns[i]) << "/index.html'>"
641 << fRuns[i] << "</a> " << std::flush;
644 << "</div>" << std::endl;
653 QABase::WriteFooter();
656 * Write out image footer
658 * @param o Output stream
659 * @param pngName Name of the PNG file
661 void WriteImageFooter(std::ostream& o, const char* pngName)
664 QABase::WriteImageFooter(o, pngName);
666 TGraph* fNAccepted; // Graph of number of accepted events
667 TGraphErrors* fVz; // Graph of mean vertex
668 UInt_t fFirst; // First run
669 UInt_t fLast; // Last run
670 TArrayI fRuns; // Seen runs
672 Bool_t fUseVar; // Use variance rather than min/max