computils.h File Reference

#include <iostream>
#include <string>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <exception>
#include "label.h"
#include "utils.h"
#include "DrawLabel.h"
#include "TStyle.h"
#include "TPad.h"
#include "TH1D.h"
#include "TFile.h"
#include "TH1.h"
#include "TGraphAsymmErrors.h"
#include "TLegend.h"
#include <cmath>

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	data_mismatch
class	true_mean
class	AxisInfo
	class to store information about axes, limits, whether it is log or linear scale etc More...
class	Legend
	slightly more convenient legend class More...
class	tPlotter< T >
	generic plotter class - better to have one of these - make sure it can be configured however you like, line styles, marker types, legends etc now a template so can be used for TH1D and TH2D etc More...
class	Plots
	set of generic plots More...
class	HistDetails
	details of the histogram axes etc More...
class	Panel

Typedefs
typedef tPlotter< TH1F >	Plotter

Functions
void	ATLASFORAPP_LABEL (double x, double y, int color, double size=0.06)
void	myText (Double_t x, Double_t y, Color_t color, const std::string &text, Double_t tsize)
std::string	stime ()
	return the current data and time More...
double	integral (TH1 *h)
void	Norm (TH1 *h, double scale=1)
double	Entries (TH1 *h)
bool	contains (const std::string &s, const std::string &p)
	does a string contain the substring More...
bool	contains (const std::string &s, char p) noexcept
bool	fcontains (const std::string &s, const std::string &p)
	does a string contain the substring at the beginning of the string More...
bool	exists (const std::string &filename)
	does a file exist More...
std::string	tail (std::string s, const std::string &pattern)
	tail of a string More...
std::string	head (std::string s, const std::string &pattern)
	head of a string More...
std::string	globbed (const std::string &s)
	match a file name More...
void	contents (std::vector< std::string > &keys, TDirectory *td, const std::string &directory="", const std::string &pattern="", const std::string &path="")
void	contents (std::vector< std::string > &keys, TDirectory *td, const std::string &directory="", const std::vector< std::string > &patterns=std::vector< std::string >(), const std::string &path="")
double	realmax (TH1 *h, bool include_error=true, double lo=0, double hi=0)
double	realmin (TH1 *h, bool include_error=true, double lo=0, double hi=0)
std::string	findcell (std::string name, const std::string &regex, const std::string &splitex="/")
std::string	findrun (TFile *f)
double	plotable (TH1 *h)
template<typename T >
std::ostream &	operator<< (std::ostream &s, std::vector< T > &v)
std::vector< int >	findxrange (TH1 *h, bool symmetric=false)
	automatically set the xrange on a histogram More...
std::vector< double >	findxrangeuser (TH1 *h, bool symmetric=false)
void	trim_tgraph (TH1 *h, TGraphAsymmErrors *t)
void	xrange (TH1 *h, bool symmetric=true)
void	copyReleaseInfo (TFile *finput, TFile *foutdir)
	copy the TTree of release info from one directory to another More...
std::ostream &	operator<< (std::ostream &s, const AxisInfo &a)
template<typename T >
void	setParameters (T *h, TGraphAsymmErrors *tg)
template<typename T >
void	zeroErrors (T *h)
bool	empty (TH1 *h)
void	hminus (TH1 *h)
std::ostream &	operator<< (std::ostream &s, const HistDetails &h)
std::ostream &	operator<< (std::ostream &s, const Panel &p)

Variables
bool	LINEF
bool	LINES
int	colours [6]
int	markers [6]
double	msizes [6]

Detailed Description

Author

mark sutton

Date

Sat Aug 30 2014 14:38:03 CEST

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

Definition in file computils.h.

Typedef Documentation

Plotter

typedef tPlotter<TH1F> Plotter

Definition at line 886 of file computils.h.

Function Documentation

ATLASFORAPP_LABEL()

void ATLASFORAPP_LABEL	(	double	x,
		double	y,
		int	color,
		double	size = 0.06
	)

Definition at line 188 of file computils.cxx.

{
  TLatex* lat  = new TLatex(); //lat.SetTextAlign(12); lat.SetTextSize(tsize); 
  lat->SetNDC();
  lat->SetTextFont(72);
  lat->SetTextColor(color);
  lat->SetTextSize(size);
  lat->DrawLatex(x,y,"ATLAS");
 
  TLatex* lat2 = new TLatex(); //lat.SetTextAlign(12); lat.SetTextSize(tsize);   
  lat2->SetNDC();
  lat2->SetTextFont(52);
  lat2->SetTextColor(color);
  lat2->SetTextSize(size);  // this 0.13 should really be calculated as a ratio of the width of the canvas
  lat2->DrawLatex(x+0.13,y,"For Approval"); 
}

contains() [1/2]

bool contains ( const std::string &  s, char  p  )

noexcept

Definition at line 39 of file chainparser.cxx.

                                            { 
  return s.find( c )!=std::string::npos;
}

contains() [2/2]

bool contains	(	const std::string &	s,
		const std::string &	p
	)

does a string contain the substring

does a string contain the substring

Definition at line 111 of file hcg.cxx.

{ return s.find(regx)!=std::string::npos; } 

contents() [1/2]

void contents	(	std::vector< std::string > &	keys,
		TDirectory *	td,
		const std::string &	directory = "",
		const std::string &	pattern = "",
		const std::string &	path = ""
	)

Definition at line 321 of file computils.cxx.

                                                                                         { 
  std::vector<std::string> patterns;
  patterns.push_back(pattern);
  contents( keys, td, directory, patterns, path );
}

contents() [2/2]

void contents	(	std::vector< std::string > &	keys,
		TDirectory *	td,
		const std::string &	directory = "",
		const std::vector< std::string > &	patterns = std::vector< std::string >(),
		const std::string &	path = ""
	)

directory, cd to it ...

test to see whether we are searhing for a specific directory name not a directory so include this ...

Definition at line 331 of file computils.cxx.

                                                                                                     { 
 
  bool print = false;
  
  TList* tl  = td->GetListOfKeys();
  
  for ( int i=tl->GetSize() ; i-- ; ) {
 
    TKey* tobj = (TKey*)tl->At(i);
 
    if ( tobj==0 ) continue;
    
    if ( std::string(tobj->GetClassName()).find("TDirectory")!=std::string::npos ) { 
      
      TDirectory* tnd = (TDirectory*)tobj->ReadObj();
      
      std::string dname = tnd->GetName();
      
      std::string newpath = path+dname+"/";
      contents( keys, tnd, directory, patterns, newpath );
    
    }
    else { 
      if ( directory == "" || contains( path, directory ) ) {
    
    
    bool matched = true;
    for ( size_t i=patterns.size() ; i-- ; ) { 
      const std::string& pattern = patterns[i];  
      if ( contains(std::string(tobj->GetName()), pattern ) )  matched &=true;
      else matched = false;
    }
    if ( matched ) { 
      if ( print ) std::cout << "will process " << td->GetName() << " \t:: " << tobj->GetName() << "\tpatterns: " << patterns.size() << std::endl;
      print = false;
      keys.push_back( path+tobj->GetName() );
    }
      }
    }
  }
}

copyReleaseInfo()

void copyReleaseInfo	(	TFile *	finput,
		TFile *	foutdir
	)

copy the TTree of release info from one directory to another

copy the TTree of release info from one directory to another

Definition at line 620 of file computils.cxx.

                                                      { 
 
  std::vector<std::string> release_data;
 
  if ( finput && foutdir ) { 
 
    TTree* tree  = (TTree*)finput->Get("dataTree");    
    TTree* clone = tree->CloneTree();
 
    foutdir->cd();
    clone->Write("", TObject::kOverwrite);
 
    delete clone;
 
  }
  
}

empty()

bool empty

(

TH1 *

h

)

Definition at line 296 of file computils.cxx.

                     { 
  for ( int i=h->GetNbinsX() ; i>0 ; i-- ) if ( h->GetBinContent(i)!=0 ) return false;
  return true;
}

Entries()

double Entries

(

TH1 *

h

)

Definition at line 52 of file computils.cxx.

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

exists()

bool exists

(

const std::string &

filename

)

does a file exist

Definition at line 256 of file computils.cxx.

                                         { 
  struct stat sb;
  if ( stat( filename.c_str(), &sb)==0 ) return true; // && S_ISREG(sb.st_mode ))
  else return false;
}

fcontains()

bool fcontains	(	const std::string &	s,
		const std::string &	p
	)

does a string contain the substring at the beginning of the string

does a string contain the substring at the beginning of the string

Definition at line 234 of file computils.cxx.

                                                        { 
  return (s.find(p)==0);
}

findcell()

std::string findcell	(	std::string	name,
		const std::string &	regex,
		const std::string &	splitex = "/"
	)

Definition at line 547 of file computils.cxx.

                                                                                       { 
  
  size_t posex = name.find( regex );
  
  if ( posex==std::string::npos ) return "";
 
  size_t pos = name.find_last_of( splitex );
 
  std::string duff = name;
 
  while ( pos!=std::string::npos && pos>posex+regex.size() ) { 
    name.resize(pos); //pos must be <=string length
    pos = name.find_last_of( splitex );
  }
  
  pos = name.find( regex );
 
  name = name.substr( pos, name.size() );
 
  pos = name.find( splitex );
 
  if ( pos!=std::string::npos ) return name.substr( 0, pos );
 
  return name; 
} 

findrun()

std::string findrun

(

TFile *

f

)

go through sub directories

Definition at line 575 of file computils.cxx.

                              { 
 
  TDirectory* here = gDirectory;
  
  f->cd();
 
  std::cout << "gDirectory::GetName() " << gDirectory->GetName() << std::endl;
  
  //  gDirectory->pwd();
  
  //  gDirectory->ls();
  
  TList* tl  = gDirectory->GetListOfKeys();
  
 
  for ( int i=0 ; i<tl->GetSize() ; i++ ) { 
    
    TKey* tobj = (TKey*)tl->At(i);
    
    if ( tobj==0 ) continue;
    
    if ( std::string(tobj->GetClassName()).find("TDirectory")!=std::string::npos ) { 
      
      TDirectory* tnd = (TDirectory*)tobj->ReadObj();
      
      std::string name = tnd->GetName();
 
      if ( name.find( "run_" )==0 ) { 
    here->cd();
    return name;
      }
    }
  }  
 
  here->cd();
 
  return "";
}

findxrange()

std::vector<int> findxrange	(	TH1 *	h,
		bool	symmetric = false
	)

automatically set the xrange on a histogram

zoom on non-empty bins

Definition at line 448 of file computils.cxx.

                                                    { 
 
  int ilo = 1;
  int ihi = h->GetNbinsX();
 
  h->GetXaxis()->SetRange( ilo, ihi );
 
  std::vector<int> limits(2,0);
  limits[0] = ilo;
  limits[1] = ihi;
 
  if ( empty(h) ) return limits;
 
#if 1
 
  for ( ; ilo<=ihi ; ilo++ ) if ( h->GetBinContent(ilo)!=0 ) break; 
  for ( ; ihi>=ilo ; ihi-- ) if ( h->GetBinContent(ihi)!=0 ) break;
 
#else
 
 
 
  double icont = 1/content;
  
  double flo = 0;
  double fhi = 0;
  for ( ; ilo<=ihi ; ilo++ ) { 
    flo += h->GetBinContent(ilo);
    if ( (flo*icont)>0.0005 )  break;
  }
 
  for ( ; ihi>=ilo  ; ihi-- ) { 
    fhi += h->GetBinContent(ihi);
    if ( (fhi*icont)>0.0005 )  break;
  }
 
#endif 
 
  int delta_lo = ilo-1;
  int delta_hi = h->GetNbinsX()-ihi;
 
  if ( symmetric ) { 
    if ( delta_hi<delta_lo ) { 
      limits[0] = 1+delta_hi; 
      limits[1] = ihi; 
    }
    else { 
      limits[0] = 1+delta_lo; 
      limits[1] = h->GetNbinsX()-delta_lo;
    }
  }
  else { 
    
    if ( ilo>1 ) ilo--;
    if ( ihi<h->GetNbinsX() ) ihi++;
    
    limits[0] = ilo;
    limits[1] = ihi;
  }
 
  return limits; 
 
}

findxrangeuser()

std::vector<double> findxrangeuser	(	TH1 *	h,
		bool	symmetric = false
	)

Definition at line 525 of file computils.cxx.

                                                           {
  
  std::vector<int> limits = findxrange( h, symmetric );
 
  std::vector<double> dlimits(2,0);
 
  double dx = h->GetBinLowEdge(limits[1]+1)-h->GetBinLowEdge(limits[1]);  
 
  dlimits[0] = h->GetBinLowEdge(limits[0]);  
  dlimits[1] = h->GetBinLowEdge(limits[1]+1)-dx*1e-11;  
  
  return dlimits;
}

globbed()

std::string globbed

(

const std::string &

s

)

match a file name

glob for a file based on the pattern, then return the name of the first matching file

Definition at line 264 of file computils.cxx.

                                        { 
  
  glob_t glob_result;
  glob( s.c_str(), GLOB_TILDE, 0, &glob_result );
 
  std::vector<std::string> ret;
  for( unsigned i=0 ; i<glob_result.gl_pathc ; i++ ){
    ret.push_back( std::string(glob_result.gl_pathv[i]) );
  }
  globfree(&glob_result);
 
  
  if ( ret.empty() ) { 
    std::cerr << "no matching file: " << s << std::endl;
    return "";
  }  
 
  if ( ret.size()>1 ) { 
    for ( unsigned i=0 ; i<ret.size() ; i++ ) { 
      std::cout << "matched " << ret[i] << std::endl;
    }
  }    
 
  return ret[0];
}

head()

std::string head	(	std::string	s,
		const std::string &	pattern
	)

head of a string

Definition at line 312 of file computils.cxx.

                                                        {
  size_t pos = s.find_last_of(pattern);
  if ( pos != std::string::npos ) {
    s.erase( pos, s.size() );
  }
  return s;
}

hminus()

void hminus ( TH1 *  h )

inline

Definition at line 903 of file computils.h.

                           { 
  std::cout << __FUNCTION__ << std::endl; 
  for ( int i=1 ; i<=h->GetNbinsX() ; i++ ) { 
    double duff = h->GetBinContent(i);
    if (  duff<0 ) { 
      std::cout<< "\t\t" << __FUNCTION__ << " " << h->GetName() << "  " << i << " " << h->GetBinContent(i) << " " << (duff*1e6) << std::endl;  
    }
  }
  h->DrawCopy();
  gPad->Print( (std::string("duff-")+h->GetName()+".pdf").c_str() );
}

integral()

double integral

(

TH1 *

h

)

Definition at line 59 of file computils.cxx.

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

myText()

void myText	(	Double_t	x,
		Double_t	y,
		Color_t	color,
		const std::string &	text,
		Double_t	tsize
	)

Definition at line 205 of file computils.cxx.

                                                                                           {
 
  //Double_t tsize=0.05;
  TLatex lat; lat.SetTextAlign(12); lat.SetTextSize(tsize); 
  lat.SetNDC();
  lat.SetTextColor(color);
  lat.DrawLatex(x,y,text.c_str());
}

Norm()

void Norm	(	TH1 *	h,
		double	scale = 1
	)

Definition at line 67 of file computils.cxx.

                                  {
  double n = 0;
  for ( int i=h->GetNbinsX()+2 ; --i ; ) n += h->GetBinContent(i);
  if ( n!=0 ) {
    double in=scale/n;
    for ( int i=h->GetNbinsX()+2 ; --i ; ) {
      h->SetBinContent(i, h->GetBinContent(i)*in );
      h->SetBinError(i, h->GetBinError(i)*in );
    }
  }
  
}

operator<<() [1/4]

std::ostream& operator<< ( std::ostream &  s, const AxisInfo &  a  )

inline

Definition at line 322 of file computils.h.

                                                                  { 
  s << "[ " << a.tag() << ( a.log() ? " : log" : "" ) << " ";
  if      (  a.autoset() ) s << " : auto";
  else if ( a.rangeset() ) s << " : range " << a.lo() << " - " << a.hi();
  s << " ]";
  return s; 
}

operator<<() [2/4]

std::ostream& operator<< ( std::ostream &  s, const HistDetails &  h  )

inline

Definition at line 1313 of file computils.h.

                                                                     { 
  return s << "[ " << h.name() << "  \tx: \"" << h.xtitle() << "\"   " << h.xaxis() << "\t\ty: \"" << h.ytitle() << "\"   " << h.yaxis() << " ]"; 
}

operator<<() [3/4]

std::ostream& operator<< ( std::ostream &  s, const Panel &  p  )

inline

Definition at line 1372 of file computils.h.

                                                               { 
  s << "Panel: " << p.name();
  if ( p.size() == 1 ) s << "\t" << p[0];
  else for ( size_t i=0 ; i<p.size() ; i++ ) s << "\n\t" << p[i];
  return s;
}

operator<<() [4/4]

template<typename T >

std::ostream& operator<<	(	std::ostream &	s,
		std::vector< T > &	v
	)

Definition at line 109 of file computils.h.

                                                        { 
  for ( unsigned i=0 ; i<v.size() ; i++ ) s << "\t" << v[i];
  return s;
}

plotable()

double plotable

(

TH1 *

h

)

Definition at line 239 of file computils.cxx.

                          { // , double xlo, double xhi ) {
  double n = 0;
    
  double xlo = h->GetBinLowEdge(1);
  double xhi = h->GetBinLowEdge(h->GetNbinsX()+1);
 
  //  if ( xlo!=-999 ) _xlo = xlo;
  //  if ( xhi!=-999 ) _xhi = xhi;
 
  for ( int i=h->GetNbinsX()+1 ; --i ; ) {
    if ( h->GetBinCenter(i)>xlo && h->GetBinCenter(i)<xhi ) n += h->GetBinContent(i);
  } 
  return n;
}

realmax()

double realmax	(	TH1 *	h,
		bool	include_error = true,
		double	lo = 0,
		double	hi = 0
	)

Definition at line 381 of file computils.cxx.

                                                                   { 
  double rm = 0;
  if ( h->GetNbinsX()==0 )  return 0; 
 
  bool first = true;
  for ( int i=1 ; i<=h->GetNbinsX() ; i++ ) { 
 
    if ( lo!=hi ) { 
      double c = h->GetBinCenter(i);
      if ( lo>c || hi<c ) continue;
    }
 
    double re = h->GetBinContent(i);
    if ( include_error ) re += h->GetBinError(i);
    if ( re!=0 ) {
      if ( first || rm<re ) { 
    rm = re;
    first = false;
      }
    }
  }
 
  return rm;
}

realmin()

double realmin	(	TH1 *	h,
		bool	include_error = true,
		double	lo = 0,
		double	hi = 0
	)

Definition at line 408 of file computils.cxx.

                                                      { 
 
  if ( h->GetNbinsX()==0 )  return 0; 
 
  double rm = 0;
 
  bool first = true;
  for ( int i=1 ; i<=h->GetNbinsX() ; i++ ) { 
 
    if ( lo!=hi ) { 
      double c = h->GetBinCenter(i);
      if ( lo>c || hi<c ) continue;
    }
 
    double re = h->GetBinContent(i);
 
    if ( re!=0 ) {
      //      if ( include_error ) re -= h->GetBinError(i);
      if ( first || rm>re ) rm = re;
      first = false;
    } 
  }
 
  return rm;
}

setParameters()

template<typename T >

void setParameters	(	T *	h,
		TGraphAsymmErrors *	tg
	)

Definition at line 436 of file computils.h.

                                                  { 
  tg->SetLineColor( h->GetLineColor() );
  tg->SetMarkerColor( h->GetMarkerColor() );
  tg->SetMarkerStyle( h->GetMarkerStyle() );
  tg->SetMarkerSize( h->GetMarkerSize() );
  tg->SetLineWidth( h->GetLineWidth() );
  tg->SetLineStyle( h->GetLineStyle() );
}

stime()

std::string stime

(

)

return the current data and time

Definition at line 215 of file computils.cxx.

                  { 
  time_t t;
  time(&t);
  return label("%s", ctime(&t) );
}

tail()

std::string tail	(	std::string	s,
		const std::string &	pattern
	)

tail of a string

Definition at line 302 of file computils.cxx.

                                                        { 
  size_t pos = s.find(pattern);
  while ( pos != std::string::npos ) { 
    s.erase( 0, pos+1 ); 
    pos = s.find(pattern);
  }
  return s;
}

trim_tgraph()

void trim_tgraph	(	TH1 *	h,
		TGraphAsymmErrors *	t
	)

Definition at line 153 of file computils.cxx.

                                                 {
  
  double ylo = h->GetMinimum();
 
  int ih=1; 
 
  for ( int i=0 ; i<t->GetN() && ih<=h->GetNbinsX() ; i++, ih++ ) { 
   
    double yt = 0;
    double xt = 0;
    double ye = 0;
 
    t->GetPoint( i, xt, yt );
    ye = t->GetErrorYlow( i );
 
    double yh = h->GetBinContent(ih);
    double xh = h->GetBinCenter(ih);
 
    while( !almost_equal( xh, xt, 5 ) && ih<=h->GetNbinsX() ) { 
      ih++;
      yh = h->GetBinContent(ih);
      xh = h->GetBinCenter(ih);
    }
 
    if ( !almost_equal( yh, yt, 5 ) ) throw data_mismatch(std::string("for histogram ")+h->GetName());
 
    if ( (yt-ye) < ylo ) { 
      h->SetBinContent(ih, ylo-100 );
      t->SetPoint( i, xt, ylo-100 ); 
    }
 
  }
}

xrange()

void xrange	(	TH1 *	h,
		bool	symmetric = true
	)

Definition at line 517 of file computils.cxx.

                                     { 
  std::vector<int> limits = findxrange( h, symmetric );
  h->GetXaxis()->SetRange( limits[0], limits[1] );
}

zeroErrors()

template<typename T >

void zeroErrors

(

T *

h

)

Definition at line 447 of file computils.h.

                        {
  for ( int i=1 ; i<=h->GetNbinsX() ; i++ ) h->SetBinError( i, 1e-100 ); 
} 

Variable Documentation

colours

int colours[6]

Definition at line 46 of file computils.cxx.

LINEF

bool LINEF

Definition at line 40 of file computils.cxx.

LINES

bool LINES

Definition at line 41 of file computils.cxx.

markers

int markers[6]

Definition at line 47 of file computils.cxx.

msizes

double msizes[6]

Definition at line 48 of file computils.cxx.