RtGraphs Class Reference

Inheritance diagram for RtGraphs:

Collaboration diagram for RtGraphs:

Public Member Functions
	RtGraphs (char *, const string &, bool)

Public Attributes
TH2F *	rthist {}
TGraphErrors *	rtgraph {}
TGraphErrors *	trgraph {}
TF1 *	oldrtfunc {}

Detailed Description

Definition at line 2053 of file TRTCalib_makeplots.cxx.

Constructor & Destructor Documentation

RtGraphs()

RtGraphs::RtGraphs	(	char *	infile,
		const string &	folder,
		bool	isAr = false
	)

Definition at line 2064 of file TRTCalib_makeplots.cxx.

{
 
    this->SetName(Form("Rt_%s", folder.c_str()));
 
    TFile *file = new TFile(infile);
    cout << "in RtGraphs. Infile: " << string(infile) << endl;
    file->cd();
    TDirectory *trt = nullptr;
 
    map<string, string> titlemap;
    if (!isAr)
    {
 
        trt = (TDirectory *)file->FindKey("TRT_all")->ReadObj();
        titlemap["WholeBarrel_1"] = "whole barrel";
        titlemap["Detector_-1"] = "barrel C";
        titlemap["Detector_1"] = "barrel A";
        titlemap["Detector_-2"] = "end-cap C";
        titlemap["Detector_2"] = "end-cap A";
        if (trt)
            cout << " Xenon. Found TRT_all " << endl;
        if (!trt)
            cout << " Xenon. Did not find TRT_all " << endl;
    }
 
    else
    {
 
        trt = (TDirectory *)file->FindKey("TRT_Ar_all")->ReadObj();
        titlemap["WholeBarrel_Ar_1"] = "whole Ar barrel";
        titlemap["Detector_Ar_-1"] = "barrel Ar C";
        titlemap["Detector_Ar_1"] = "barrel Ar A";
        titlemap["Detector_Ar_-2"] = "end-cap Ar C";
        titlemap["Detector_Ar_2"] = "end-cap Ar A";
        if (trt)
            cout << " Argon. Found TRT_Ar_all " << endl;
        if (!trt)
            cout << " Argon. Did not find TRT_Ar_all " << endl;
    }
 
    if (trt and (not folder.empty()))
    {
        cout << "  PLOT FOR " << folder << endl;
        TDirectory *det = (TDirectory *)trt->FindKey(folder.c_str())->ReadObj();
        if (det->FindKey("rt-relation"))
        {
            cout << "  found rt-relation " << endl;
            rthist = (TH2F *)det->FindKey("rt-relation")->ReadObj();
            this->cd();
            rthist->GetXaxis()->SetRangeUser(0, 40);
            char name[500];
            sprintf(name, "rt (%s)", titlemap[folder].c_str());
            rthist->SetTitle(name);
            if (det->FindKey("rtgraph"))
            {
                rtgraph = (TGraphErrors *)det->FindKey("rtgraph")->ReadObj();
                cout << "  found also rtgraph " << endl;
            }
            else
            {
                rtgraph = nullptr;
                cout << "  did not found rtgraph " << endl;
            }
            if (det->FindKey("trgraph"))
            {
                trgraph = (TGraphErrors *)det->FindKey("trgraph")->ReadObj();
                cout << "  found also trgraph " << endl;
            }
            else
            {
                trgraph = nullptr;
                cout << "  did not found trgraph " << endl;
            }
            if (det->FindKey("oldrtfunc"))
            {
                oldrtfunc = (TF1 *)det->FindKey("oldrtfunc")->ReadObj();
                cout << "  found also oldrtfunc " << endl;
            }
            else
            {
                oldrtfunc = nullptr;
                cout << "  did not find oldrtfunc " << endl;
            }
        }
        else
        {
            throw string("  NO RT OBJECTS FOUND");
        }
    }
    else if (trt and trt->FindKey("rt-relation"))
    {
        cout << "   No folder. Found rt-relation. PLOT FOR TRT" << endl;
        rthist = (TH2F *)trt->FindKey("rt-relation")->ReadObj();
        this->cd();
        rthist->GetXaxis()->SetRangeUser(0, 40);
        rthist->SetTitle("r(t) for whole TRT");
        if (trt->FindKey("rtgraph"))
        {
            rtgraph = (TGraphErrors *)trt->FindKey("rtgraph")->ReadObj();
            cout << "  found also rtgraph " << endl;
        }
        else
        {
            rtgraph = nullptr;
            cout << "  did not find rtgraph " << endl;
        }
        if (trt->FindKey("trgraph"))
        {
            trgraph = (TGraphErrors *)trt->FindKey("trgraph")->ReadObj();
            cout << "  found also trgraph " << endl;
        }
        else
        {
            trgraph = nullptr;
            cout << "  did not find trgraph " << endl;
        }
        if (trt->FindKey("oldrtfunc"))
        {
            oldrtfunc = (TF1 *)trt->FindKey("oldrtfunc")->ReadObj();
            cout << "  found also oldtrtfunc " << endl;
        }
        else
        {
            oldrtfunc = nullptr;
            cout << "  did not find oldtrtfunc " << endl;
        }
    }
    else
    {
        cout << "   No folder. No rt-relation." << endl;
        rthist = nullptr;
        rtgraph = nullptr;
        oldrtfunc = nullptr;
    }
 
    this->cd();
    this->Divide(2, 1, 0.01, 0.01);
    this->cd(1);
 
    TPad *c1_3 = new TPad("c1_3", "newpad", 0.01, 0.33, 0.99, 0.99);
    c1_3->Draw();
    c1_3->cd();
    c1_3->SetTopMargin(0.1);
    c1_3->SetBottomMargin(0.01);
    c1_3->SetRightMargin(0.1);
    c1_3->SetFillStyle(0);
    c1_3->SetGrid();
 
    if (rtgraph)
    {
        if (folder == "")
        {
            cout << "  found rtgraph for entire TRT " << endl;
            rtgraph->SetTitle("r(t) fit (whole TRT)");
        }
        else
        {
            cout << "  found rtgraph for folder: " << folder << endl;
            rtgraph->SetTitle(titlemap[folder].c_str());
        }
        cout << "  set rtgraph X title t-T0/ns and Y title |r|_{track}/mm" << endl;
 
        rtgraph->GetXaxis()->SetTitle("t-T0/ns");
        rtgraph->GetYaxis()->SetTitle("|r|_{track}/mm");
        // rtgraph->GetYaxis()->SetRangeUser(0,2.7);
        rtgraph->GetXaxis()->SetTitleSize(0.06);
        rtgraph->GetXaxis()->SetLabelSize(0.06);
 
        // Blue line too wide, extract and plot again:
        rtgraph->SetMarkerStyle(20);
        rtgraph->SetMarkerSize(0.5);
        rtgraph->GetYaxis()->SetRangeUser(-0.05, 2.6);
        cout << "    Draw rtgraph " << endl;
        rtgraph->Draw("ap");
        TF1 *newRT;
        if (rtgraph->GetFunction("rtfunc"))
            newRT = rtgraph->GetFunction("rtfunc");
        else
            newRT = rtgraph->GetFunction("rtfunc2");
        if (newRT)
        {
            cout << "    Draw new fit " << endl;
            newRT->SetLineWidth(1);
            newRT->Draw("same");
        }
        if (oldrtfunc)
        {
            oldrtfunc->SetLineColor(1);
            oldrtfunc->SetLineWidth(1);
            oldrtfunc->SetLineStyle(2);
            cout << "    Draw old fit " << endl;
            oldrtfunc->Draw("same");
        }
        // Legend
 
        TLegend *leg = new TLegend(0.7155172, 0.1038136, 0.8965517, 0.2542373, nullptr, "brNDC");
        leg->AddEntry(rtgraph, "Data", "p");
        leg->AddEntry(newRT, "Fit", "l");
        leg->AddEntry(oldrtfunc, "Fit old rt", "l");
        leg->SetTextSize(0.03);
        leg->Draw();
 
        // Plot the difference between old and new rt:
        if (oldrtfunc && newRT)
        {
            this->cd();
            this->cd(1);
            TPad *c1_4 = new TPad("c1_4", "newpad", 0.01, 0, 0.99, 0.32);
            c1_4->Draw();
            c1_4->cd();
            c1_4->SetTopMargin(0.01);
            c1_4->SetBottomMargin(0.3);
            c1_4->SetRightMargin(0.1);
            c1_4->SetFillStyle(0);
            c1_4->SetGrid();
 
            Double_t Xmax = rtgraph->GetXaxis()->GetXmax();
            Double_t Xmin = rtgraph->GetXaxis()->GetXmin();
 
            const Int_t n = 300;
            Double_t x[n], y[n], dx;
            dx = (Xmax - Xmin) / n;
 
            Double_t Ymax = -10;
            Double_t Ymin = 10;
 
            for (Int_t i = 0; i < n; i++)
            {
                x[i] = Xmin + i * dx;
                y[i] = (oldrtfunc->Eval(x[i]) - newRT->Eval(x[i]));
                if (y[i] > Ymax)
                    Ymax = y[i];
                if (y[i] < Ymin)
                    Ymin = y[i];
            }
 
            TGraph *gr1 = new TGraph(n, x, y);
 
            gr1->SetTitle("");
            gr1->GetXaxis()->SetTitle("t-T0/ns");
            gr1->GetXaxis()->SetRangeUser(Xmin, Xmax);
            gr1->GetYaxis()->SetTitle("Old r(t) - New r(t) /mm");
            gr1->GetYaxis()->SetRangeUser(-0.1, 0.1);
            // gr1->GetYaxis()->SetRangeUser(1.1*Ymin,1.1*Ymax);
            gr1->GetYaxis()->SetNdivisions(5);
            gr1->GetXaxis()->SetTitleSize(0.06);
            gr1->GetXaxis()->SetLabelSize(0.06);
            gr1->GetYaxis()->SetLabelSize(0.08);
            gr1->GetYaxis()->SetTitleOffset(0.8);
            gr1->GetYaxis()->SetTitleSize(0.06);
            gr1->SetLineStyle(2);
 
            // Also add the difference between the fit and the points of the fit
 
            const Int_t n1 = rtgraph->GetN();
            std::vector<Double_t> x1(n1, 0), y1(n1, 0), ex1(n1, 0), ey1(n1, 0);
            Double_t X, Y;
 
            for (Int_t i = 0; i < n1; i++)
            {
                X = 0;
                Y = 0;
                rtgraph->GetPoint(i, X, Y);
                x1.at(i) = X;
                y1.at(i) = Y - newRT->Eval(X);
                ex1.at(i) = 0;
                ey1.at(i) = rtgraph->GetErrorY(i);
                if (y1.at(i) > Ymax)
                {
                    Ymax = y1.at(i);
                }
                if (y1.at(i) < Ymin)
                {
                    Ymin = y1.at(i);
                }
            }
 
            TGraphErrors *gr2 = new TGraphErrors(n1, &x1.at(0), &y1.at(0), &ex1.at(0), &ey1.at(0));
            gr2->SetMarkerColor(2);
            gr2->SetMarkerStyle(20);
            gr2->SetTitle("");
            gr2->GetYaxis()->SetTitle("Old r(t) - New r(t) /mm");
 
            gr2->GetYaxis()->SetRangeUser(-0.2, 0.2);
            gr2->GetXaxis()->SetRangeUser(-5, 50);
            cout << "    Draw difference " << endl;
            gr2->Draw("ap");
            gr1->Draw("l");
 
            TLine *line0 = new TLine(Xmin, 0, Xmax, 0);
            line0->SetLineColor(4);
            line0->Draw();
        }
    }
    else
    {
        cout << " nullptr rtgraph " << endl;
    }
 
    this->cd(2);
    if (trgraph)
    {
 
        if (folder == "")
            trgraph->SetTitle("t(r) fit (whole TRT)");
        else
        {
            char name[500];
            //don't move 'folder'; it is passed by const ref
            sprintf(name, "t(r) fit (%s)", titlemap[string(folder)].c_str());
            trgraph->SetTitle(name);
        }
 
        trgraph->GetYaxis()->SetTitle("t-T0/ns");
        trgraph->GetXaxis()->SetTitle("|r|_{track}/mm");
        // trgraph->GetYaxis()->SetRangeUser(0,32);
        trgraph->SetMarkerStyle(20);
        trgraph->SetMarkerSize(0.5);
        cout << "    Draw trgraph " << endl;
        trgraph->Draw("ap");
    }
    else
    {
        cout << " nullptr trgraph " << endl;
    }
 
    gStyle->SetOptStat(1);
}

Member Data Documentation

oldrtfunc

TF1* RtGraphs::oldrtfunc {}

Definition at line 2061 of file TRTCalib_makeplots.cxx.

rtgraph

TGraphErrors* RtGraphs::rtgraph {}

Definition at line 2059 of file TRTCalib_makeplots.cxx.

rthist

TH2F* RtGraphs::rthist {}

Definition at line 2058 of file TRTCalib_makeplots.cxx.

trgraph

TGraphErrors* RtGraphs::trgraph {}

Definition at line 2060 of file TRTCalib_makeplots.cxx.

---

The documentation for this class was generated from the following file:

• TRTCalib_makeplots.cxx