GwUtilities.C

Go to the documentation of this file.

#include "TROOT.h"
#include "TFolder.h"
#include "TRegexp.h"
#include "TVirtualPad.h"
#include "TH1.h"

#include <iostream>


namespace {
        const Int_t nb_color_whell = 20;
        Color_t color_wheel[nb_color_whell] = { 
                kBlue+0, kMagenta+0, kRed+0, kGreen+0, kCyan+0, 
                kBlue+1, kMagenta+1, kRed+1, kGreen+1, kCyan+1,
                kBlue+2, kMagenta+2, kRed+2, kGreen+2, kCyan+2, 
                kBlue+3, kMagenta+3, kRed+3, kGreen+3, kCyan+3 
        };
}

//void AddSpectra(TFolder *top_folder, const Char_t *pattern, const Char_t *destination_folder = "//root/GwWatchers", TH1 *h = 0x0)
void AddSpectra(const Char_t *pattern, TH1 *&h, TFolder *top_folder = 0x0)
{
        TObject *obj; TString tmp(pattern); TFolder *f;
    
    // pattern expression
    Bool_t wildcard = false;
    if ( tmp.Contains("*") )
        wildcard = true;
        TRegexp what(pattern,wildcard);
    ;
        if ( top_folder == 0x0 ) {
                f = (TFolder *)gROOT->GetRootFolder()->FindObject("//root/GwWatchers");
        }
        else f = top_folder;
        
        // iteration over this folder
        TIter next(f->GetListOfFolders());
        while ( (obj=next()) ) {
                
                tmp = obj->GetName();
                                
                if ( obj->InheritsFrom("TFolder") ) { 
                        AddSpectra(pattern,h,(TFolder *)obj);
                        continue;
                }
                if ( tmp.Contains(what) && obj->InheritsFrom("TH1") ) { 
                        if ( h == 0x0 ) {               
                                std::cout << " clone " << f->GetName() << " " << tmp << std::endl ;
                                h = (TH1 *)obj->Clone(Form("SUM_%s",pattern));
                                h->SetTitle(Form("SUM of %s",h->GetTitle()));
                                h->SetDirectory(0);
                                h->Draw();                              
                        }
                        else {
                                h->Add((TH1 *)obj);
                                std::cout << " add " << tmp << std::endl ;
                        }
                }               
        }       
}

void CompSpectra(const Char_t *pattern, TFolder *top_folder = 0x0)
{
        TObject *obj; TString tmp(pattern); TFolder *f; TH1 *h; static Int_t color = 0;
        
    // pattern expression
    Bool_t wildcard = false;
    if ( tmp.Contains("*") )
        wildcard = true;
        TRegexp what(pattern,wildcard);
    
        if ( top_folder == 0x0 ) {
                f = (TFolder *)gROOT->GetRootFolder()->FindObject("//root/GwWatchers");
                color = 0;
        }
        else f = top_folder;
        
        // iteration over this folder
        TIter next(f->GetListOfFolders());
        while ( (obj=next()) ) {
                
                tmp = obj->GetName();
                                
                if ( obj->InheritsFrom("TFolder") ) { 
                        CompSpectra(pattern,(TFolder *)obj);
                        continue;
                }
                if ( tmp.Contains(what) && obj->InheritsFrom("TH1") ) { 
                                h = (TH1 *)obj->Clone(Form("%s_%s",f->GetName(),obj->GetName()));
                                h->SetDirectory(0);
                                if ( color == 0 ) 
                                        h->Draw("");
                                else
                                        h->Draw("same");

                                std::cout << " Show " << h->GetName() << std::endl ;
                                // change color
                                h->SetLineColor( color_wheel[color % nb_color_whell] ); color++;
                }               
        }       
}