tracking/trackingQA/comp_recoqa.C

Go to the documentation of this file.

#include "TString.h"
#include "TFile.h"
#include "TCanvas.h"
#include "TKey.h"
#include <iostream>
#include "TH1F.h"
#include "TRandom.h"

using std::cout;
using std::endl;

TCanvas* createCanvas(int picpercan){
        double division = TMath::Sqrt((double)picpercan);
        int first = ceil(division);
        int second = round(division);
        cout << "createCanvas: " << picpercan << " : " << first << "/" << second << endl;
        TCanvas* newCan = new TCanvas();
        newCan->Divide(first, second);
        return newCan;
}

int comp_recoqa(TString fn = "QA_histograms.root", TString fn2 =
                "QA_histograms_test.root", int picpercan = 9, double minP = 0.03, int minev = 3,
                int maxfail = 3) {
        // fn   -> red
        // fn2 -> blue
        std::vector<TCanvas*> canvasses;
        TCanvas *c1 = new TCanvas();
        c1->Divide(6, 6);
        bool fTest = kFALSE;

        TFile *f = new TFile(fn, "READ");
        if (!f->IsZombie()) {
                TFile *f2 = new TFile(fn2, "READ");

                TKey *key;
                TIter next(f->GetListOfKeys());

                int failcount = 0;
                Int_t histoCount = 0;
                Int_t actualCanvas = 0;
                Int_t actualPad = 0;
                while ((key = (TKey*) next())) {
                        actualCanvas = TMath::Floor((double)histoCount / picpercan);
                        actualPad = (histoCount % picpercan) + 1;
                        cout << histoCount << " : " << actualCanvas << "/" << actualPad << endl;
                        if (!(canvasses.size() > actualCanvas)){
                                cout << "New canvas" << endl;
                                canvasses.push_back(createCanvas(picpercan));
                        }
                        canvasses[actualCanvas]->cd(actualPad);
                        TObject *obj = key->ReadObj();

                        // only check TH1Fs
                        if (!obj->InheritsFrom("TH1F"))
                                continue;

                        TString name = obj->GetName();
                        TH1F* h = (TH1F*) obj;
                        TH1F* h2 = (TH1F*) f2->Get(name);
                        h->SetLineColor(2);
                        h->SetLineWidth(2);
                        h2->SetLineWidth(2);
                        if (h->GetMaximum() < h2->GetMaximum()) {
                                h2->Draw();
                                h->Draw("same");
                        } else {
                                h->Draw();
                                h2->Draw("same");
                        }
                        if (h->GetEntries() < minev) {
                                cout << "Histogram (almost) empty : " << name << " \""
                                                << h2->GetTitle() << "\":  N = " << h->GetEntries()
                                                << endl;
                                failcount++;
                        } else {
                                double P = h2->KolmogorovTest(h);
                                if (P < minP) {
                                        cout << "Incompatible distribution: " << name << " \""
                                                        << h2->GetTitle() << "\":  P = " << P << endl;
                                        failcount++;
                                }
                        }

                        auto legend = new TLegend(0.1,0.7,0.48,0.9);
                        legend->SetHeader("File Comparison","C"); // option "C" allows to center the header
                        legend->AddEntry(h,fn.Data(),"f");
                        legend->AddEntry(h2,fn2.Data(),"f");
                        legend->Draw();
                        histoCount++;
                }

                if (failcount < maxfail)
                        fTest = kTRUE;

        }

        if (fTest) {
                cout << " Test passed" << endl;
                cout << " All ok " << endl;
        } else {
                cout << " Test Failed" << endl;
                cout << " Not Ok " << endl;
        }
        return 0;
}