BmnHistZDC.cxx

Go to the documentation of this file.

#include "BmnHistZDC.h"
 
#include "TGaxis.h"
 
BmnHistZDC::BmnHistZDC(TString title, TString path)
    : BmnHist()
{
    fTitle = title;
    fName = title + "_cl";
    TGaxis::SetMaxDigits(2);
    TString name;
    name = fTitle + "_h2d_Grid";
    h2d_grid = new TH2D(name, "ZDC Grid", 20, 0, 20, 20, 0, 20);
    h2d_grid->GetXaxis()->SetTitle("iX");
    h2d_grid->GetXaxis()->SetTitleColor(kOrange + 10);
    h2d_grid->GetYaxis()->SetTitle("iY");
    h2d_grid->GetYaxis()->SetTitleColor(kOrange + 10);
    h2d_grid->GetZaxis()->SetNoExponent(kFALSE);
    name = fTitle + "_h2d_Profile";
    h2d_profile = new TH2D(name, "ZDC Profile", 200, -700, 700, 200, -700, 700);
    h2d_profile->GetXaxis()->SetTitle("X");
    h2d_profile->GetXaxis()->SetTitleColor(kOrange + 10);
    h2d_profile->GetYaxis()->SetTitle("Y");
    h2d_profile->GetYaxis()->SetTitleColor(kOrange + 10);
    name = fTitle + "_X_Amplitude";
    hx = new TH1D(name, "ZDC X Amplitude", 200, -700, 700);
    name = fTitle + "_Y_Amplitude";
    hy = new TH1D(name, "ZDC Y Amplitude", 200, -700, 700);
    name = fTitle + "CanvasAmplitudes";
    canAmps = new TCanvas(name, name, PAD_WIDTH * ZDC_ROWS, PAD_HEIGHT * ZDC_COLS);
    canAmps->Divide(ZDC_ROWS, ZDC_COLS);
    canAmpsPads.resize(ZDC_ROWS * ZDC_COLS);
    NamesAmps.resize(ZDC_ROWS * ZDC_COLS);
 
    PadInfo* p = new PadInfo();
    p->current = hx;
    canAmpsPads[0] = p;
    PadInfo* p1 = new PadInfo();
    p1->current = hy;
    canAmpsPads[1] = p1;
 
    for (Int_t rowIndex = 0; rowIndex < ZDC_ROWS; rowIndex++) {
        for (Int_t colIndex = 0; colIndex < ZDC_COLS; colIndex++) {
            Int_t iPad = rowIndex * ZDC_COLS + colIndex;
            canAmps->GetPad(iPad + 1)->SetGrid();
            NamesAmps[iPad] = canAmpsPads[iPad]->current->GetName();
            TH1* h = canAmpsPads[iPad]->current;
            h->SetTitleSize(0.06, "XY");
            h->SetLabelSize(0.08, "XY");
            h->GetXaxis()->SetTitle("Coordinate");
            h->GetXaxis()->SetTitleColor(kOrange + 10);
            h->GetYaxis()->SetTitle("Amplitude");
            h->GetYaxis()->SetTitleColor(kOrange + 10);
            h->GetYaxis()->SetTitleOffset(1.4);
            h->GetYaxis()->SetNoExponent(kFALSE);
        }
    }
}
 
BmnHistZDC::~BmnHistZDC()
{
    if (fDir != NULL)
        return;
    delete h2d_grid;
    delete h2d_profile;
    delete hx;
    delete hy;
    if (fDir)
        return;
    for (auto pad : canAmpsPads)
        delete pad;
}
 
void BmnHistZDC::Register(THttpServer* serv)
{
    fServer = serv;
    fServer->Register("/", this);
    TString path = "/" + fTitle + "/";
    fServer->Register(path, h2d_grid);
    fServer->Register(path, h2d_profile);
    fServer->Register(path, canAmps);
    TString cmd = "/" + fName + "/->Reset()";
    fServer->SetItemField(path.Data(), "_monitoring", "2000");
    fServer->SetItemField(path.Data(), "_layout", "grid3x3");
    TString cmdTitle = path + "Reset";
    fServer->RegisterCommand(cmdTitle.Data(), cmd.Data(), "button;");
    fServer->Restrict(cmdTitle, "visible=shift");
    fServer->Restrict(cmdTitle, "allow=shift");
    cmd = "/" + fName + "/->SetRefRun(%arg1%)";
    cmdTitle = path + "SetRefRun";
    fServer->RegisterCommand(cmdTitle.Data(), cmd.Data(), "button;");
}
 
void BmnHistZDC::SetDir(TFile* outFile = NULL, TTree* recoTree = NULL)
{
    frecoTree = recoTree;
    fDir = NULL;
    if (outFile != NULL)
        fDir = outFile->mkdir(fTitle + "_hists");
    h2d_grid->SetDirectory(fDir);
    h2d_profile->SetDirectory(fDir);
    hx->SetDirectory(fDir);
    hy->SetDirectory(fDir);
}
 
void BmnHistZDC::DrawBoth()
{
    BmnHist::DrawRef(canAmps, &canAmpsPads);
    return;
}
 
void BmnHistZDC::FillFromDigi(DigiArrays* fDigiArrays)
{
    TClonesArray* digits = fDigiArrays->zdc;
    if (!digits)
        return;
    Double_t xAmp = 0;
    Double_t yAmp = 0;
    Double_t Amp = 0;
    for (Int_t iDig = 0; iDig < digits->GetEntriesFast(); iDig++) {
        BmnZDCDigit* dig = (BmnZDCDigit*)digits->At(iDig);
        Int_t ix = dig->GetIX();
        Int_t iy = dig->GetIY();
        Double_t amp = dig->GetAmp();
        Amp += amp;
        h2d_grid->Fill(ix, iy, dig->GetAmp());
        xAmp += amp * dig->GetX();
        yAmp += amp * dig->GetY();
    }
    if (Amp == 0)
        return;
    xAmp /= Amp;
    yAmp /= Amp;
    h2d_profile->Fill(xAmp, yAmp, Amp);
    hx->Fill(xAmp, Amp);
    hy->Fill(yAmp, Amp);
}
 
BmnStatus BmnHistZDC::SetRefRun(Int_t id)
{
    TString FileName = Form("bmn_run%04d_hist.root", id);
    printf("SetRefRun: %s\n", FileName.Data());
    if (refRunName != FileName) {
        refRunName = FileName;
        refID = id;
        BmnHist::LoadRefRun(refID, refPath + FileName, fTitle, canAmpsPads, NamesAmps);
        DrawBoth();
    }
    return kBMNSUCCESS;
}
 
void BmnHistZDC::ClearRefRun()
{
    for (auto pad : canAmpsPads) {
        if (pad->ref)
            delete pad->ref;
        pad->ref = NULL;
    }
    refID = 0;
}
 
void BmnHistZDC::Reset()
{
    h2d_grid->Reset();
    h2d_profile->Reset();
    hx->Reset();
    hy->Reset();
}