BmnDrawOnline.cxx

Go to the documentation of this file.

#include "BmnDrawOnline.h"
 
#include "BmnDrawHist.h"
#include "TGaxis.h"
#include "TH1.h"
#include "TH1D.h"
#include "TH2.h"
#include "TLegend.h"
#include "TMath.h"
#include "TPad.h"
#include "TPaveLabel.h"
#include "TStyle.h"
 
#include <cassert>
#include <iostream>
#include <limits>
#include <sstream>
#include <string>
 
using std::stringstream;
 
BmnDrawOnline::BmnDrawOnline(TString param, TString storageName)
{
    // For param use:
    // RECREATE Create a new file, if the file already exists it will be overwritten.
    // UPDATE   Open an existing file for writing. If no file exists, it is created.
    // canvStorage = TFile::Open("canvStorage.root",param);
    canvStorage = new TFile(storageName + ".root", param);
}
 
BmnDrawOnline::BmnDrawOnline(TFile* file, Int_t port)
    : fServer(nullptr)
{
    InitServer(port);
    RegisterCanvases(file);
}
 
BmnDrawOnline::~BmnDrawOnline()
{
    if (fServer)
        delete fServer;
}
 
void BmnDrawOnline::DrawH1(TCanvas* canvas,
                           TH1* hist,
                           HistScale logx,
                           HistScale logy,
                           const string& drawOpt,
                           Int_t lineColor,
                           Int_t lineWidth,
                           Int_t lineStyle,
                           Float_t markerSize,
                           Int_t markerStyle,
                           Int_t fillColor)
{
    Double_t textSize = BmnDrawingOptions::TextSize();
    hist->SetLineColor(lineColor);
    hist->SetLineWidth(lineWidth);
    hist->SetLineStyle(lineStyle);
    hist->SetMarkerColor(lineColor);
    hist->SetMarkerSize(markerSize);
    hist->SetMarkerStyle(markerStyle);
    if (logx == kLog) {
        gPad->SetLogx();
    }
    if (logy == kLog) {
        gPad->SetLogy();
    }
    hist->GetXaxis()->SetLabelSize(textSize);
    hist->GetXaxis()->SetNdivisions(505, kTRUE);
    hist->GetYaxis()->SetLabelSize(textSize);
    hist->GetXaxis()->SetTitleSize(textSize);
    hist->GetYaxis()->SetTitleSize(textSize);
    hist->GetXaxis()->SetTitleOffset(1.0);
    hist->GetYaxis()->SetTitleOffset(1.3);
 
    gPad->SetTopMargin(0.20);
    gPad->SetLeftMargin(0.17);
    gPad->SetBottomMargin(0.15);
    gPad->SetTicks(1, 1);
    if (fillColor > 0)
        hist->SetFillColor(fillColor);
 
    if (canvName != canvas->GetName()) {
        canvName = canvas->GetName();
        canvVect.push_back(canvas);
        maxHeight += canvas->GetWh();
    }
 
    hist->Draw(drawOpt.c_str());
    gPad->SetGrid(true, true);
    hist->SetStats(false);
}
 
/* Draw TH2 histogram.*/
void BmnDrawOnline::DrawH2(TCanvas* canvas,
                           TH2* hist,
                           HistScale logx,
                           HistScale logy,
                           HistScale logz,
                           const string& drawOpt)
{
    Double_t textSize = BmnDrawingOptions::TextSize();
    if (logx == kLog) {
        gPad->SetLogx();
    }
    if (logy == kLog) {
        gPad->SetLogy();
    }
    if (logz == kLog) {
        gPad->SetLogz();
    }
    hist->GetXaxis()->SetLabelSize(textSize);
    hist->GetXaxis()->SetNdivisions(505, kTRUE);
    hist->GetYaxis()->SetLabelSize(textSize);
    hist->GetYaxis()->SetNdivisions(505, kTRUE);
    hist->GetZaxis()->SetLabelSize(textSize);
    // hist->GetZaxis()->SetNdivisions(505, kTRUE);
    hist->GetXaxis()->SetTitleSize(textSize);
    hist->GetYaxis()->SetTitleSize(textSize);
    hist->GetZaxis()->SetTitleSize(textSize);
    hist->GetXaxis()->SetTitleOffset(1.0);
    hist->GetYaxis()->SetTitleOffset(1.3);
    hist->GetZaxis()->SetTitleOffset(1.5);
 
    gPad->SetTopMargin(0.20);
    gPad->SetLeftMargin(0.17);
    gPad->SetRightMargin(0.30);
    gPad->SetBottomMargin(0.15);
    gPad->SetTicks(1, 1);
 
    if (canvName != canvas->GetName()) {
        canvName = canvas->GetName();
        canvVect.push_back(canvas);
        maxHeight += canvas->GetWh();
    }
 
    hist->Draw(drawOpt.c_str());
    gPad->SetGrid(true, true);
    hist->SetStats(false);
}
 
/* Draw several TH1 histograms. */
void BmnDrawOnline::DrawH1(TCanvas* canvas,
                           const vector<TH1*>& histos,
                           const vector<string>& histLabels,
                           HistScale logx,
                           HistScale logy,
                           Bool_t drawLegend,
                           Double_t x1,
                           Double_t y1,
                           Double_t x2,
                           Double_t y2,
                           const string& drawOpt,
                           Bool_t outputMeanValue01)
{
    assert(histos.size() != 0 && histLabels.size() == histos.size());
    Double_t max = std::numeric_limits<Double_t>::min();
    Int_t nofHistos = histos.size();
    TLegend* legend = new TLegend(x1, y1, x2, y2);
    legend->SetFillColor(kWhite);
    for (Int_t iHist = 0; iHist < nofHistos; iHist++) {
        TH1* hist = histos[iHist];
        string opt = (iHist == 0) ? drawOpt : (iHist == nofHistos - 1) ? "SAME" + drawOpt : "SAME" + drawOpt;
        // string opt = drawOpt;
        DrawH1(canvas, hist, logx, logy, opt, BmnDrawingOptions::Color(iHist), BmnDrawingOptions::LineWidth(),
               BmnDrawingOptions::LineStyle(0), BmnDrawingOptions::MarkerSize(), BmnDrawingOptions::MarkerStyle(iHist));
        max = std::max(max, hist->GetMaximum());
        Int_t nonZeroBins = 0;
        for (Int_t i = 0; i < hist->GetNbinsX(); ++i) {
            if (hist->GetBinContent(i) != 0.0)
                nonZeroBins++;
        }
 
        Double_t xMax = hist->GetBinCenter(hist->GetNbinsX());
        Double_t xGev = 1.0;   // 1 GeV
        Int_t nGev = Int_t(hist->GetNbinsX() * xGev / xMax);
        Double_t effAll = hist->Integral() / nonZeroBins;
        Double_t eff01 = hist->Integral(0.0, nGev) / nGev;   // efficiency from 0 to 1 GeV
        TString legText;
        if (outputMeanValue01)
            legText = TString::Format("%s | mean =  %3.1f | mean01 =  %3.1f", histLabels[iHist].c_str(), effAll, eff01)
                          .Data();
        else
            legText = TString::Format("%s | mean =  %3.1f", histLabels[iHist].c_str(), effAll).Data();
        legend->AddEntry(hist, legText, "lp");
        // To choose the best combination of parameters
        FILE* f = fopen("test.dat", "a+");
        if (((TString)hist->GetName()).Contains("vs_P"))
            fprintf(f, "%s %s %f %f\n", hist->GetName(), histLabels[iHist].c_str(), effAll, eff01);
        fclose(f);
    }
 
    histos[0]->SetMaximum(max * 1.10);
 
    if (drawLegend) {
        legend->Draw();
    }
}
 
void BmnDrawOnline::DrawH1(TCanvas* canvas, TEfficiency* hist)
{
 
    gPad->SetTopMargin(0.20);
    gPad->SetLeftMargin(0.17);
    gPad->SetBottomMargin(0.15);
    gPad->SetTicks(1, 1);
 
    if (canvName != canvas->GetName()) {
        canvName = canvas->GetName();
        canvVect.push_back(canvas);
        maxHeight += canvas->GetWh();
    }
 
    hist->Draw();
    gPad->SetGrid(true, true);
}
 
void BmnDrawOnline::InitServer(Int_t port)
{
 
    const Char_t* httpStr = Form("http:%d;cors;noglobal", port);
 
    // Start http-server ...
    fServer = new THttpServer(httpStr);
}
 
/* Принцип работы DrawMainCanvas():
1.Создается холст с высотой, равной суммарной высоте всех холстов с гистограммами. 2.Размечаем его так, чтобы он состоял
из одного столбца из числа строк, равному общему числу холстов. 3.Для каждого холста создаем TPad и TPaveLabel  - будем
использовать их, чтобы подписывать холсты. 4.Запускаем цикл, с первой строчки общего холста, до последней, в котором: С
помощью метода DrawClonePad() на нужную строку холста добавляем набор   гистограмм из вектора canvVect. Поверх
нарисованных гистограмм добавляем панель  TPad, на этой панели, с помощью   TPaveLabel подписываем заголовок для нужного
подхолста. 5.Полученный холст записывается в root-файл canvStorage.root 6.Зануляются значения  общей высоты, вектора
холстов и имени холста.
*/
 
void BmnDrawOnline::DrawMainCanvas(TString nameOfCanv)
{
 
    Int_t height = maxHeight;
    TCanvas* mainCanvas = new TCanvas(nameOfCanv, "title", 1280, height);
    Int_t numOfCanvases = (canvVect.size());
    mainCanvas->Divide(1, numOfCanvases, 0.01, 0.002);
    TPad* newpad[numOfCanvases];
    TPaveLabel* title[numOfCanvases];
 
    for (Int_t i = 0; i < numOfCanvases; ++i) {
        mainCanvas->cd(i + 1);
        canvVect[i]->DrawClonePad();
 
        TString newpadName = "newpad" + std::to_string(i);
        newpad[i] = new TPad(newpadName, "a transparent pad", 0.0, 0.93, 1.0, 1.0);
        newpad[i]->SetFillStyle(4000);
        newpad[i]->Draw();
        newpad[i]->cd();
        title[i] = new TPaveLabel(0.0, 0.0, 1.0, 1.0, canvVect[i]->GetName());
        title[i]->SetFillColor(16);
        title[i]->SetTextFont(52);
        title[i]->Draw();
    }
    mainCanvas->Write();
    maxHeight = 0;
    canvName = "null";
    vector<TCanvas*>().swap(canvVect);
}
 
void BmnDrawOnline::RegisterCanvases(TFile* file)
{
    TString namesOfCanv[] = {"inner", "TOF", "CSC", "DCH", "pid", "He3", "pi+", "e-", "proton", "K+", "D", "T", "He4"};
    Int_t size = sizeof(namesOfCanv) / sizeof(namesOfCanv[0]);
    for (Int_t i = 0; i < size; ++i) {
        TCanvas* canv = (TCanvas*)file->Get(namesOfCanv[i]);
        if (!canv)
            continue;
        fServer->Register("/Objects", canv);
    }
}