chains.cxx File Reference

#include <iostream>
#include <string>
#include "TFile.h"
#include "TDirectory.h"
#include "TList.h"
#include "TKey.h"
#include "TH1D.h"

Include dependency graph for chains.cxx:

Go to the source code of this file.

Functions
void	descend (TDirectory *d, int level, int maxdepth, const std::vector< std::string > &labels)
int	usage (std::ostream &s, int argc, char **argv)
int	main (int argc, char **argv)
double	hsum (TH1 *h)
	sum the bin contents including the over and underflow bins More...

Variables
bool	verbose = false
	very lazy!!! should not use unnessecary global variables !!!! More...
bool	always_descend = false
std::string	histogram = ""
bool	fast = false

Detailed Description

Author

mark sutton

Date

Fri 11 Jan 2019 07:41:26 CET

Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration

Definition in file chains.cxx.

Function Documentation

descend()

void descend	(	TDirectory *	d,
		int	level,
		int	maxdepth,
		const std::vector< std::string > &	labels
	)

Definition at line 107 of file chains.cxx.

                                                                                           { 
 
  if ( verbose ) std::cout << "level " << level << "\tmaxdepth " << maxdepth << std::endl;
 
  if ( d==0 ) return;
 
  TList* tl  = d->GetListOfKeys();
 
  if ( verbose ) std::cout << "list of keys " << tl << " " << ( tl ? tl->Capacity() : 0 ) << std::endl; 
 
  if ( tl==0 ) return;
 
  std::string name = d->GetName();
 
  int plevel = level;
 
  if ( !verbose ) if ( plevel>0 ) plevel--;  
 
  for ( int ip=plevel ; ip-- ; ) std::cout << "\t";
  
  bool displayed = false;
 
  int dsize = 0;
 
  if ( verbose )  std::cout << "Directory: ";
  if ( verbose || level>0 ) { 
    std::cout<< name;
    displayed = true;
    dsize = name.size();
    if ( dsize<60 ) std::cout << "\t";
  }
 
  //    struct timeval tv = simpletimer_start();
  
  if ( verbose ) std::cout << "level " << level << "\tmaxdepth " << maxdepth << std::endl;
 
  for ( int i=tl->GetSize() ; i-- ; ) { 
    
    TKey* tobj = (TKey*)tl->At(i);
    
    if ( tobj==0 ) continue;
  
 
    //    for ( int ip=level+1 ; ip-- ; ) std::cout << "\t";
    //    std::cout << "tobj " << tobj << " : \t" << tobj->GetName() << std::endl; 
    
    if ( std::string(tobj->GetClassName()).find("TDirectory")!=std::string::npos ) { 
      //      std::cout << ns << "Directory " << tobj->GetName() << std::endl;
      TDirectory* tnd = (TDirectory*)tobj->ReadObj();
 
      if ( tnd==0 ) continue;
 
      std::string dname = tnd->GetName();
 
      
      //      if ( dname.find("L2_")!=std::string::npos || 
      //       dname.find("EF_")!=std::string::npos || 
      //       dname.find("HLT_")!=std::string::npos )
      if ( level<maxdepth ) { 
    bool dscnd = true;
    if ( !always_descend && labels.size() ) { 
      dscnd = false;
      for ( unsigned il=labels.size() ; il-- ; ) { 
        if ( dname.find(labels[il])!=std::string::npos) dscnd = true;
      } 
    }
    if ( dscnd ) descend( tnd, level+1, maxdepth, labels );
      }
    }
    else { 
      if ( histogram != "" ){ 
        if ( std::string(tobj->GetClassName()).find("TH1")!=std::string::npos ||
             //  std::string(tobj->GetClassName()).find("TH2")!=std::string::npos ||
             std::string(tobj->GetClassName()).find("TProfile")!=std::string::npos ) {
          
          TH1* th = (TH1*)tobj->ReadObj();
          
          if ( th && std::string(th->GetName()) == histogram ) { 
 
            double d = hsum( th );
        
        std::cout << "\t" << d << std::endl;
 
        displayed = false;
 
          }
        }
      }
    }
  }
  if ( displayed ) std::cout << std::endl;
}

hsum()

double hsum

(

TH1 *

h

)

sum the bin contents including the over and underflow bins

Definition at line 101 of file chains.cxx.

                      { 
  double d=0;
  for ( int i=0 ; i<=h->GetNbinsX()+1 ; i++ ) d += h->GetBinContent(i);
  return d;
}

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 55 of file chains.cxx.

                                 { 
 
  std::vector<std::string> args;
 
  int maxdepth = 2;
 
 
  for ( int i=1 ; i<argc ; i++ ) { 
    std::string arg = argv[i];
    if      ( arg=="-h" || arg=="--help" )    return usage( std::cout, argc, argv );
    else if ( arg=="-v" || arg=="--verbose" ) verbose        = true;
    else if ( arg=="-f" || arg=="--fast" )    fast = true;
    else if ( arg=="-a" || arg=="--always" )  always_descend = true;
    else if ( arg=="-d" || arg=="--depth" )  { 
      if ( ++i<argc ) maxdepth = std::atoi(argv[i]);
      else            return usage( std::cout, argc, argv );
    }
    else if ( arg=="-c" || arg=="--contents" )  { 
      if ( ++i<argc ) histogram = argv[i];
      else            return usage( std::cout, argc, argv );
    }
 
    else args.push_back(arg);
  } 
 
  std::vector<std::string> labels;
  labels.push_back("L2_");
  labels.push_back("EF_");
  labels.push_back("HLT_");
 
  for ( unsigned i=0 ; i<args.size() ; i++ ) {
    TFile f( args[i].c_str() );
 
    if ( !f.IsOpen() || f.IsZombie() ) {
      std::cerr << "Error: could not open input file: " << args[i] << std::endl;
      std::exit(1);
    }
 
    //    gDirectory->ls();
    descend( gDirectory, 0, maxdepth, labels );
  }
 
  return 0;
}

usage()

int usage	(	std::ostream &	s,
		int	argc,
		char **	argv
	)

should never be the case

Definition at line 29 of file chains.cxx.

                                                { 
  if ( argc<1 ) return -1; 
  s << "Usage: " << argv[0] << " [OPTIONS] <file1>  <file2>  ... \n\n";
  s << "  writes names of chains from files\n\n";
  s << "Options: \n";
  s << "    -a, --always             \t always descend, \n";
  s << "    -d, --depth    maxdepth  \t maximum depth to descend, \n";
  s << "    -c, --contents histogram \t display number of entries in histogram, \n";
  s << "    -f, --fast               \t bomb out as soon as a single chain has been found, \n";
  s << "    -v, --verbose            \t verbose output, \n";
  s << "    -h, --help               \t this help,\n";
  //  s << "\nSee " << PACKAGE_URL << " for more details\n"; 
  //  s << "\nReport bugs to <" << PACKAGE_BUGREPORT << ">";
  s << std::endl;
  return 0;
}

Variable Documentation

always_descend

bool always_descend = false

Definition at line 51 of file chains.cxx.

fast

bool fast = false

Definition at line 53 of file chains.cxx.

histogram

std::string histogram = ""

Definition at line 52 of file chains.cxx.

verbose

bool verbose = false

very lazy!!! should not use unnessecary global variables !!!!

Definition at line 50 of file chains.cxx.