WatchersTools.cpp

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <vector>
#include <string>
#include <map>

#include "TString.h"
#include "TObjArray.h"
#include "BashColor.h"
#include "TROOT.h"
#include "TPaveStats.h"
#include "TTimer.h"
#include "TSystem.h"
#include "TRootBrowser.h"
#include "TGTab.h"
#include "TBrowser.h"

#include "PadManager.h"

#include "WatchersTools.h"

using namespace std;


WatchersTools::WatchersTools(const char* name, const char *title) :
    TTask(name,title),
    CanvasVisu(name,title),
    fBashColor(new BashColor())
{

}


WatchersTools::~WatchersTools()
{

}



Collector::Collector(const char* name, const char *title) :
    WatchersTools(name,title),
    fNpadX(5),
    fNpadY(5)
{

}


Collector::~Collector()
{

}



void Collector::CloseTab(const char *TabName)
{
    if(gROOT->GetListOfCanvases()->FindObject(TabName))
    {
        TBrowser *Browser = (TBrowser*)((TList*)gROOT->GetListOfBrowsers())->At(0);

        TRootBrowser *RootBrowser = (TRootBrowser *) Browser->GetBrowserImp();
        if(((TGTab*)RootBrowser->GetTabRight())->SetTab(TabName))
        {
            RootBrowser->GetTabRight()->CloseTab(RootBrowser->GetTabRight()->GetCurrent());
        }
    }
}


bool Collector::Collectistos(const char* Folder, const char *inc, const char *exc, const char *class_inc, const char* class_exc, const char *CanvasName)
{
    TFolder *top_folder = (TFolder *)gROOT->GetRootFolder()->FindObject(Folder);
    if(top_folder == 0x0)
    {
        fBashColor->SetErrorOut();
        cout<<Folder<<" not found"<<endl;
        fBashColor->ResetColor();
        return false;
    }

    if(((TString)CanvasName) == "Canvas")
    {
        CanvasName = ((TString)(TString)inc + "_" + (TString)class_inc).Data();
    }

    TList collected;

    Gw::PadManager::Collect<TObject>(&collected,top_folder,inc,exc,class_inc,class_exc);

    int PadNbr = 0;

    TString Name;
    TCanvas *c = 0x0;

    double topmarg = 0.0679612;
    double rightmarg = 0.0121951;
    double leftmarg = 0.0983588;
    double bottommarg = 0.11165;

    gPad->SetRightMargin(rightmarg);
    gPad->SetTopMargin(topmarg);
    gPad->SetLeftMargin(leftmarg);
    gPad->SetBottomMargin(bottommarg);

    TPaveStats *st;

    if(collected.GetSize()>100)
    {
        fBashColor->SetWarningOut();
        cout<<"The number of histos to plot is "<<collected.GetSize()<<" ! Plot it anyway ? (y,n)"<<endl;
        fBashColor->ResetColor();
        TString test;
        cin>>test;
        if(!test.EqualTo("y",TString::kIgnoreCase)) return false;
    }

    Int_t NpadPerCanvas = fNpadX*fNpadY;

    for(int i=0 ; i<collected.GetSize() ; i++)
    {
        if(PadNbr%NpadPerCanvas==0)
        {
            if(collected.GetSize()>NpadPerCanvas) Name = Form("%s_%d_%d",CanvasName,PadNbr+1,PadNbr+NpadPerCanvas);
            else Name = CanvasName;
            c =  CanvasVisu::NewCanvas(Name,Name);

            new LoupeOnPad(c);
            SetLoupe(false);
            c->Divide(fNpadX,fNpadY,0.001,0.001);
        }

        c->cd(PadNbr%NpadPerCanvas+1);

        TH1 *h = (TH1*)collected.At(i);

        if(h)
        {
            gPad->SetRightMargin(rightmarg);
            gPad->SetTopMargin(topmarg);
            gPad->SetLeftMargin(leftmarg);
            gPad->SetBottomMargin(bottommarg);

            h->SetBit(TH1::kNoTitle);

            h->GetXaxis()->SetLabelSize(0.05);
            h->GetXaxis()->SetTitleSize(0.06);
            h->GetXaxis()->SetTitleOffset(0.85);

            h->GetYaxis()->SetLabelSize(0.05);
            h->GetYaxis()->SetTitleSize(0.06);
            h->GetYaxis()->SetTitleOffset(0.90);

            h->SetLineColor(GetColor(0));

            if(h->InheritsFrom("TH2"))
            {
                h->Draw("col");
                gPad->SetLogz();
                if( strncmp(h->GetName(),"ADC",3) == 0 ) h->GetYaxis()->SetRangeUser(0,2000);
            }
            else
            {
                h->Draw();
            }

            gPad->Modified();
            gPad->Update();

            st = (TPaveStats*)h->FindObject("stats");
            if(st)
            {
                st->SetX2NDC(1-rightmarg);
                st->SetY2NDC(1-topmarg);
                st->SetX1NDC(0.681013);
                st->SetY1NDC(0.663962);
            }

        }
        PadNbr++;
    }
    return true;
}


WatchersRefresh::WatchersRefresh(const char* name, const char *title) :
    WatchersTools(name,title),
    fTimer(new TTimer())
{

}


WatchersRefresh::~WatchersRefresh()
{

}


void WatchersRefresh::PadRefresh(bool IsOn, Long_t RefreshTimeIn_ms)
{
    if(IsOn)
    {
        fTimer->Connect("Timeout()","WatchersRefresh",this,"DoPadRefresh()");
        fTimer->Start(RefreshTimeIn_ms);
    }
    else
    {
        fTimer->Stop();
    }
}


void WatchersRefresh::CanvasRefresh(bool IsOn, Long_t RefreshTimeIn_ms)
{
    if(IsOn)
    {
        fTimer->Connect("Timeout()","WatchersRefresh",this,"DoCanvasRefresh()");
        fTimer->Start(RefreshTimeIn_ms);
    }
    else
    {
        fTimer->Stop();
    }
}


void WatchersRefresh::DoPadRefresh()
{
    TVirtualPad *Pad = gROOT->GetSelectedPad();

    if(Pad && Pad->GetListOfPrimitives())
    {
        Pad->Modified();
        Pad->Update();
    }
}


void WatchersRefresh::DoCanvasRefresh()
{
    if(gROOT->GetSelectedPad())
    {
        TCanvas *c = gROOT->GetSelectedPad()->GetCanvas();

        if(c)
        {
            c->Paint();
            c->Update();
        }
    }
}


ManipConfig::ManipConfig(const char* name, const char *title) :
    WatchersTools(name,title),
    fNpadX(5),
    fNpadY(5)
{

}


ManipConfig::~ManipConfig()
{

}


void ManipConfig::StartManipViewLLP()
{
    Collector *TheCollector = (Collector*)gROOT->FindObjectAny("SpectraCollector");
    if(TheCollector == 0x0)
    {
        fBashColor->SetErrorOut();
        cout<<"No SpectraCollector found !"<<endl;
        fBashColor->ResetColor();
        return;
    }

    TheCollector->SetNPads(fNpadX,fNpadY);
    TString CanvasName;

    CanvasName = "TimeStamp_1h";
    if(!gROOT->GetListOfCanvases()->FindObject(CanvasName))TheCollector->Collectistos("//root/GwWatchers","CCrystal*_1hour","","TH1","",CanvasName.Data());

    CanvasName = "CoreEnergy";
    if(!gROOT->GetListOfCanvases()->FindObject(CanvasName))TheCollector->Collectistos("//root/GwWatchers","*HG*CalSpectra","","TH1","",CanvasName.Data());

    CanvasName = "EvsSegID";
    if(!gROOT->GetListOfCanvases()->FindObject(CanvasName))TheCollector->Collectistos("//root/GwWatchers","*EvsSegID","","TH2","",CanvasName.Data());
}



ClassImp(WatchersTools)
ClassImp(Collector)
ClassImp(WatchersRefresh)
ClassImp(ManipConfig)