WatchersUtilities.C

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (c) IPNL, IN2P3, CNRS                                       *
 *   Contibutor(s) :                                                       *
 *      Jeremie Dudouet  dudouet(AT)ipnl.in2p3.fr [2014]                   *
 *      Olivier Stezowski stezow(AT)ipnl.in2p3.fr [2014]                   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include <fstream>

#include "TFolder.h"
#include "TList.h"
#include "TClassMenuItem.h"
#include "TClass.h"
#include "TVirtualPad.h"
#include "TSocket.h"

#include "PadManager.h"

namespace {
    // wheel color use to display spectra on the same pad
    const Int_t nb_color_wheel = 20;
    Color_t color_wheel[nb_color_wheel] = {
        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 Count(const char *include="* {TH*}", const char *exclude="- {TH2*}", const char *drawopt="clone !wheel", int searchdepth = kMaxInt, TObject *topfolder = 0x0);

//void ZeroOnFolder(const char *include="* {TH*}", const char *exclude="* {TH2*}", int searchdepth = kMaxInt, TObject *topfolder = 0x0);
void ZeroOnFolder(const char *include="* {TH*}", const char *exclude="- {TH2*}", const char *drawopt="ICE", int searchdepth = kMaxInt, TObject *topfolder = 0x0);

void DisplayFolder(const char *include="* {TH*}", const char *exclude="- {TH2*}", const char *drawopt="clone !wheel", int searchdepth = kMaxInt, TObject *topfolder = 0x0);

void DisplayFolderSUM(const char *include="* {TH1*}", const char *exclude="- {TH2*}", const char *drawopt="", int searchdepth = kMaxInt, TObject *topfolder = 0x0);

void WatchersUtilities(const Char_t * /*WhatMethod*/ = "first")
{
    // Add some method to the folder
    TClass *cl = TFolder::Class(); TList *l = cl->GetMenuList(); TClassMenuItem *n = 0x0;


    n = new TClassMenuItem(TClassMenuItem::kPopupUserFunction,cl,
                           "Zero","ZeroOnFolder",0,"char*,char*,char *,int,int,int,TObject*",4);
    l->AddFirst(n);
    n = new TClassMenuItem(TClassMenuItem::kPopupUserFunction,cl,
                           "DisplaySUM","DisplayFolderSUM",0,"char*,char*,char *,int,int,int,TObject*",4);
    l->AddFirst(n);
    n = new TClassMenuItem(TClassMenuItem::kPopupUserFunction,cl,
                           "Display","DisplayFolder",0,"char*,char*,char *,int,int,int,TObject*",4);
    l->AddFirst(n);
 //   n = new TClassMenuItem(TClassMenuItem::kPopupUserFunction,cl,
 //                         "Count","Count",0,"char*,char*,char *,int,int,int,TObject*",4);
 //   l->AddFirst(n);
}
/*
void Myls(const char *include, const char *exclude, TString drawopt, int searchdepth, TObject *topfolder)
{
    // First collect all the spectra
    TString include = inc, exclude = exc, include_name = "*", include_class = "*", exlude_name = "", exclude_class = "", tmp_obj = "", tmp_token = "",
                        draw_option = draw, h_draw_opt = "";

    TList collected;
    TObject *obj = 0x0, *obj_to_draw = 0x0 ;
    TFolder *topfolder =
        (TFolder *)folder;
    TVirtualPad *top_pad = TVirtualPad::Pad();

    if ( ! folder->InheritsFrom("TFolder") || TVirtualPad::Pad() == 0x0 ) {
        return;
    }

    // separation filter on name and classname
    Ssiz_t begin_ = include.Index("{"), end_ = include.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = include.Length();
        }
        if ( begin_ < end_ ) {
            include_class = include(begin_+1,end_-begin_-1);
            include.Remove(begin_,end_-begin_+1);
        }
    }
    begin_ = exclude.Index("{"), end_ = exclude.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = exclude.Length();
        }
        if ( begin_ < end_ ) {
            exclude_class = exclude(begin_+1,end_-begin_-1);
            exclude.Remove(begin_,end_-begin_+1);
        }
    }

    // collect according to includes/excludes
    Gw::PadManager::Collect<TObject>(&collected,topfolder,include,exclude,include_class,exclude_class,depth);

    std::cout <<
}
*/

void ZeroOnFolder(const char *inc, const char *exc, const char *ropt, int depth, TObject *folder)
{
    // First collect all the spectra
    TString include = inc, exclude = exc, include_name = "*", include_class = "*", exlude_name = "", exclude_class = "", tmp_obj = "", tmp_token = "";
    TList collected;
    TObject *obj = 0x0, *obj_to_draw = 0x0 ;
    TFolder *topfolder =
    (TFolder *)folder;
    TVirtualPad *top_pad = TVirtualPad::Pad();
    
    if ( ! folder->InheritsFrom("TFolder") || TVirtualPad::Pad() == 0x0 ) {
        return;
    }
    
    // separation filter on name and classname
    Ssiz_t begin_ = include.Index("{"), end_ = include.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = include.Length();
        }
        if ( begin_ < end_ ) {
            include_class = include(begin_+1,end_-begin_-1);
            include.Remove(begin_,end_-begin_+1);
        }
    }
    begin_ = exclude.Index("{"), end_ = exclude.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = exclude.Length();
        }
        if ( begin_ < end_ ) {
            exclude_class = exclude(begin_+1,end_-begin_-1);
            exclude.Remove(begin_,end_-begin_+1);
        }
    }
    
    // collect according to includes/excludes
    Gw::PadManager::Collect<TObject>(&collected,topfolder,include,exclude,include_class,exclude_class,depth);
    
        printf("From TFolder::Drawall, %d object to be drawn \n",collected.GetSize());
        collected.ls();
    
    //
    if ( collected.GetSize() ) {
        
        TIter next(&collected);
        while ( (obj = next()) ) {
            
            if ( obj->InheritsFrom("TH1") ) {
                ((TH1 *)obj)->Reset(ropt);
            }
            
        }
    }
}



void DisplayFolder(const char *inc, const char *exc, const char *draw, int depth, TObject *folder)
{
    // First collect all the spectra
    TString include = inc, exclude = exc, include_name = "*", include_class = "*", exlude_name = "", exclude_class = "", tmp_obj = "", tmp_token = "",
                        draw_option = draw, h_draw_opt = "";
    TList collected;
    TObject *obj = 0x0, *obj_to_draw = 0x0 ;
    TFolder *topfolder =
        (TFolder *)folder;
    TVirtualPad *top_pad = TVirtualPad::Pad();
    
    if ( ! folder->InheritsFrom("TFolder") || TVirtualPad::Pad() == 0x0 ) {
        return;
    }
    
    // separation filter on name and classname
    Ssiz_t begin_ = include.Index("{"), end_ = include.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = include.Length();
        }
        if ( begin_ < end_ ) {
            include_class = include(begin_+1,end_-begin_-1);
            include.Remove(begin_,end_-begin_+1);
        }
    }
    begin_ = exclude.Index("{"), end_ = exclude.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = exclude.Length();
        }
        if ( begin_ < end_ ) {
            exclude_class = exclude(begin_+1,end_-begin_-1);
            exclude.Remove(begin_,end_-begin_+1);
        }
    }
    
    // collect according to includes/excludes
    Gw::PadManager::Collect<TObject>(&collected,topfolder,include,exclude,include_class,exclude_class,depth);
    
//    printf("From TFolder::Drawall, %d object to be drawn \n");
//    collected.ls();
    
    //
    if ( collected.GetSize() ) {
        
        // treat drawing options
        Bool_t do_clone = false, do_wheel = false;
        //
        h_draw_opt = draw_option;
        if ( h_draw_opt.Contains("clone") ) {
            if ( h_draw_opt.Contains("!clone") ) {
                h_draw_opt.ReplaceAll("!clone","");
            }
            else { h_draw_opt.ReplaceAll("clone",""); do_clone = true; }
        }
        if ( h_draw_opt.Contains("wheel") ) {
            if ( h_draw_opt.Contains("!wheel") ) {
                h_draw_opt.ReplaceAll("!wheel","");
            }
            else {
                h_draw_opt.ReplaceAll("!wheel","");
                if ( do_clone )
                    do_wheel = true;
            }
        }
        
        TIter next(&collected); Int_t pad_number = top_pad->GetPad(1) ? 1 : 0; Int_t color = 0;
        while ( (obj = next()) ) {
            
            if ( top_pad->GetPad(pad_number) ) {
                top_pad->cd(pad_number++);
            }
            else { // same pad
                if ( !h_draw_opt.Contains("same") ) {
                    h_draw_opt += " same ";
                }
            }
            
            obj_to_draw = obj;
            if ( draw_option.Contains("clone") && !draw_option.Contains("!clone") ) {
 
            }
//            printf("PAD # %d %s\n",pad_number,obj_to_draw->GetName());

            if ( h_draw_opt.IsWhitespace() ) {
                obj_to_draw->Draw();
            }
            else obj_to_draw->Draw(h_draw_opt);

            //
            if ( obj_to_draw->InheritsFrom("TAttLine") && do_wheel ) {
                TAttLine *line = dynamic_cast <TAttLine*> (obj_to_draw);
                line->SetLineColor( color_wheel[color % nb_color_wheel] ); color++;
            }

        }
        
        top_pad->Update(); top_pad->cd();
    }
}

void DisplayFolderSUM(const char *inc, const char *exc, const char *draw, int depth, TObject *folder)
{
    // First collect all the spectra
    TString include = inc, exclude = exc, include_name = "*", include_class = "*", exlude_name = "", exclude_class = "", tmp_obj = "", tmp_token = "",
    draw_option = draw, h_draw_opt = "";
    TList collected;
    TFolder *topfolder =
        (TFolder *)folder;
    TVirtualPad *top_pad = TVirtualPad::Pad();
    
    if ( ! folder->InheritsFrom("TFolder") || TVirtualPad::Pad() == 0x0 ) {
        return;
    }
    
    // separation filter on name and classname
    Ssiz_t begin_ = include.Index("{"), end_ = include.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = include.Length();
        }
        if ( begin_ < end_ ) {
            include_class = include(begin_+1,end_-begin_-1);
            include.Remove(begin_,end_-begin_+1);
        }
    }
    begin_ = exclude.Index("{"), end_ = exclude.Index("}");
    if ( begin_ != (const Ssiz_t)(-1) ) {
        if ( end_ == (const Ssiz_t)(-1) ) {
            end_ = exclude.Length();
        }
        if ( begin_ < end_ ) {
            exclude_class = exclude(begin_+1,end_-begin_-1);
            exclude.Remove(begin_,end_-begin_+1);
        }
    }
    
    // collect according to includes/excludes
    Gw::PadManager::Collect<TH1>(&collected,topfolder,include,exclude,include_class,exclude_class,depth);
    
    //    printf("From TFolder::Drawall, %d object to be drawn \n");
    //    collected.ls();
    
    //
    if ( collected.GetSize() ) {
        
        TH1 *hsum, *h;
        
        hsum = (TH1 *)collected.First()->Clone(Form("SUM_OF_%s",topfolder->GetName()));
        hsum->SetTitle( Form("SUM of histograms from %s with pattern: %s and %s",topfolder->GetName(),inc,exc) );
        //hsum->SetDirectory(0x0);
        
        TIter next(&collected);
        while ( (h = (TH1 *)next()) ) {
            hsum->Add(h);
        }
        hsum->Draw(draw);
        top_pad->Update(); top_pad->cd();
    }
}


namespace  { TList PingNames; char buffer[4*1024*1024]; }

/*
*/
void LoadPing( const char * filename, Option_t * opt = "+" )
{
    TString option = opt;
    std::ifstream inputfile(filename);
    TList listName_tmp;
    std::string machineName;
    std::string portName;
    bool test_is;
    TNamed * tmp_name = NULL;
    while(1){
        machineName.clear();
        portName.clear();
        inputfile >> machineName >> portName;
        if( inputfile.eof() ) break;
        std::cerr << machineName << " " << portName << std::endl;
        listName_tmp.Add(new TNamed( machineName.c_str(),
                                    portName.c_str() ) );

    };
    std::cerr << listName_tmp.GetSize() << std::endl;
    if( PingNames.GetSize() == 0 && option.Contains("+") ){
        for( Int_t i = 0; i < listName_tmp.GetSize(); i++)
            PingNames.Add( listName_tmp.At(i) );
        listName_tmp.Clear("nodelete");
    }
    else if( option.Contains("+") ){
        for( Int_t i = 0; i<listName_tmp.GetSize(); i++){
            test_is = false;
            for( Int_t j = 0; j<PingNames.GetSize(); j++){
                if(std::string(PingNames.At(j)->GetName()) ==
                   std::string(listName_tmp.At(i)->GetName()) &&
                   std::string(PingNames.At(j)->GetTitle()) ==
                   std::string(listName_tmp.At(i)->GetTitle()) ) test_is = true;
            }
            if( test_is == false ) PingNames.Add(listName_tmp.At(i));
        }
        listName_tmp.Clear("nodelete");
    }

    if( option.Contains("-") ){
        for( Int_t i = 0; i<listName_tmp.GetSize(); i++ ){
            test_is = false;
            for( Int_t j = 0; j<PingNames.GetSize(); j++){
                if(std::string(PingNames.At(j)->GetName()) ==
                   std::string(listName_tmp.At(i)->GetName()) &&
                   std::string(PingNames.At(j)->GetTitle()) ==
                   std::string(listName_tmp.At(i)->GetTitle()) ){
                    test_is = true;
                    tmp_name = (TNamed*)PingNames.At(j);
                }
            }
            if( test_is == true ){
                PingNames.Remove(tmp_name);
                delete tmp_name;
            }
        }
        listName_tmp.Clear("delete");
    }

    PingNames.Print();
    for( Int_t ii = 0; ii < PingNames.GetSize(); ii++ ){
        std::cerr << PingNames.At(ii)->GetName() << " " << PingNames.At(ii)->GetTitle() << std::endl;}
}

/*
 */
void Ping( UInt_t nbtimes = 5, bool dodump = false)
{
    TIter next(&PingNames);
    TObject * obj;
    while ( (obj = next()) ) {

        TString option = obj->GetTitle(), ifdump;
        cout << "Ping on : " << obj->GetName() << " with Port # " << obj->GetTitle() << endl;

        TSocket *fSocket = new TSocket(obj->GetName(),option.Atoi());
        if ( fSocket ) {
            if ( fSocket->IsValid() ) {
                Int_t bsize, ret;
                cout << "Select with 1 millisecond " << endl;
                for (UInt_t i = 0u; i < nbtimes; i++) {
                    cout << " - try # " << i << endl;
                    ifdump = Form("%s_%s_%d",obj->GetName(),obj->GetTitle(),i);

                    switch ( fSocket->Select(TSocket::kRead,1000) )
                    {
                        case 0:
                            cout << "--> timeout " << endl;
                            break;
                        case -1:
                            cout << "--> Error " << endl;
                            break;
                        default:
                            fSocket->RecvRaw(&bsize, sizeof(Int_t));
                            cout << " Size expected " << bsize << endl;
                            ret = fSocket->RecvRaw(buffer, bsize);
                            cout << " Size received " <<  ret << endl;
                            if ( dodump ) {
                                ofstream file(ifdump.Data(),ios_base::app);
                                file << bsize;
                                file.write(buffer,bsize);
                            }
                            break;
                    }
                }
            }
            else cout << "  *** Invalid connection *** " << endl;
        }
        else cout << "  *** Invalid connection *** " << endl;
        delete fSocket;
    }
}

void Ping(const char *machine, UInt_t port, UInt_t nbtimes = 5, Option_t *opt = "")
{
    TString option = opt, sport = Form("%d",port);
    bool test_is;
    TNamed * tmp_name = NULL;
    // add or removed from the list
    if ( option.Contains("+") ) {
        test_is = false;
        for( Int_t i = 0; i < PingNames.GetSize(); i++ )
            if( std::string(PingNames.At(i)->GetName()) == std::string(machine) &&
                std::string(PingNames.At(i)->GetTitle()) == std::string(sport) ) test_is = true;
        if( test_is == false )
            PingNames.Add ( new TNamed(machine,sport.Data()) );
    }
    if ( option.Contains("-") ){
        test_is = false;
        for( Int_t i = 0; i < PingNames.GetSize(); i++ )
            if( std::string(PingNames.At(i)->GetName()) == std::string(machine) &&
               std::string(PingNames.At(i)->GetTitle()) == std::string(sport) ) {
                test_is = true;
                tmp_name = (TNamed*)PingNames.At(i);
            }
        if ( test_is == true ) {
            PingNames.Remove(tmp_name);
            delete tmp_name;
        }
    }

    Ping(nbtimes);
}

/*
 */
void SSHPing( UInt_t nbtimes = 1, bool dodump = false)
{
//      TIter next(&PingNames);
//      TObject * obj;
//      while ( (obj = next()) ) {

    TNamed *obj = new TNamed("localhost","10001");

        TString option = obj->GetTitle(), ifdump;
        cout << "Ping on : " << obj->GetName() << " with Port # " << obj->GetTitle() << endl;


        TSocket *fSocket = new TSocket(obj->GetName(),option.Atoi());
        if ( fSocket ) {
            if ( fSocket->IsValid() ) {
                Int_t bsize, ret;
                cout << "Select with 1 millisecond " << endl;
                for (UInt_t i = 0u; i < nbtimes; i++) {
                    cout << " - try # " << i << endl;
                    ifdump = Form("%s_%s_%d",obj->GetName(),obj->GetTitle(),i);

                    switch ( fSocket->Select(TSocket::kRead,1000) )
                    {
                        case 0:
                            cout << "--> timeout " << endl;
                            break;
                        case -1:
                            cout << "--> Error " << endl;
                            break;
                        default:
                            fSocket->RecvRaw(&bsize, sizeof(Int_t));
                            cout << " Size expected " << bsize << endl;
                            ret = fSocket->RecvRaw(buffer, bsize);
                            cout << " Size received " <<  ret << endl;
                            if ( dodump ) {
                                ofstream file(ifdump.Data(),ios_base::app);
                                file << bsize;
                                file.write(buffer,bsize);
                            }
                            break;
                    }
                }
            }
            else cout << "  *** Invalid connection *** " << endl;
        }
        else cout << "  *** Invalid connection *** " << endl;
        delete fSocket;

    delete obj;
//      }
}