#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:

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
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
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
bool	exists (const std::string &filename)
	does a file exist
std::string	tail (std::string s, const std::string &pattern)
	tail of a string
std::string	head (std::string s, const std::string &pattern)
	head of a string
std::string	globbed (const std::string &s)
	match a file name
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
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
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
static std::string	release
int	colours [6]
int	markers [6]
double	msizes [6]
template<typename T>
bool	tPlotter< T >::s_plotref = true
	use non c++17 format for improved external compatability
template<typename T>
bool	tPlotter< T >::s_meanplotref = true

Detailed Description

Author: mark sutton

Date: Sat Aug 30 2014 14:38:03 CEST

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

Definition at line 116 of file hcg.cxx.

116{ 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 ) { 
      //coverity[DEADCODE]
      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 release info TTree

copy the TTree of release info from one directory to another

Definition at line 621 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

contains a string 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 548 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 576 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 449 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 526 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 1316 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 1375 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 382 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 409 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 518 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]

extern

Definition at line 46 of file computils.cxx.

46{ 1, 2, kBlue-4, 6, kCyan-2, kMagenta+2 };

◆ LINEF

bool LINEF

extern

Definition at line 40 of file computils.cxx.

◆ LINES

bool LINES

extern

Definition at line 41 of file computils.cxx.

◆ markers

int markers[6]

extern

Definition at line 47 of file computils.cxx.

47{ 20, 24, 25, 26, 25, 22 };

◆ msizes

double msizes[6]

extern

Definition at line 48 of file computils.cxx.

48{ 0.85, 1, 1, 1, 1, 1 };

◆ release

std::string release

static

Definition at line 50 of file computils.h.

◆ tPlotter< T >::s_meanplotref

template<typename T>

bool tPlotter< T >::s_meanplotref = true

Definition at line 895 of file computils.h.

◆ tPlotter< T >::s_plotref

template<typename T>

bool tPlotter< T >::s_plotref = true

use non c++17 format for improved external compatability

Definition at line 891 of file computils.h.

Classes

Typedefs

Functions

Variables

Detailed Description

Typedef Documentation

◆ Plotter

Function Documentation

◆ ATLASFORAPP_LABEL()

◆ contains() [1/2]

◆ contains() [2/2]

◆ contents() [1/2]

◆ contents() [2/2]

◆ copyReleaseInfo()

◆ empty()

◆ Entries()

◆ exists()

◆ fcontains()

◆ findcell()

◆ findrun()

◆ findxrange()

◆ findxrangeuser()

◆ globbed()

◆ head()

◆ hminus()

◆ integral()

◆ myText()

◆ Norm()

◆ operator<<() [1/4]

◆ operator<<() [2/4]

◆ operator<<() [3/4]

◆ operator<<() [4/4]

◆ plotable()

◆ realmax()

◆ realmin()

◆ setParameters()

◆ stime()

◆ tail()

◆ trim_tgraph()

◆ xrange()

◆ zeroErrors()

Variable Documentation

◆ colours

◆ LINEF

◆ LINES

◆ markers

◆ msizes

◆ release

◆ tPlotter< T >::s_meanplotref

◆ tPlotter< T >::s_plotref