computils.cxx File Reference

#include <stdlib.h>
#include <sys/time.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <glob.h>
#include <stdint.h>
#include <iostream>
#include <string>
#include <vector>
#include "label.h"
#include "DrawLabel.h"
#include "TFile.h"
#include "TKey.h"
#include "TTree.h"
#include "TList.h"
#include "TObject.h"
#include "TDirectory.h"
#include "TH1D.h"
#include "TLegend.h"
#include "TColor.h"
#include "computils.h"

Go to the source code of this file.

Classes
union	floaty_t

Functions
double	Entries (TH1 *h)
double	integral (TH1 *h)
void	Norm (TH1 *h, double scale)
bool	almost_equal (floaty_t a, floaty_t b, int max_diff)
bool	almost_equal (float a, float b, int max_diff)
bool	operator== (floaty_t a, floaty_t b)
void	trim_tgraph (TH1 *h, TGraphAsymmErrors *t)
void	ATLASFORAPP_LABEL (double x, double y, int color, double size)
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...
bool	contains (const std::string &s, const std::string &p)
	contains a string More...
bool	contains (const std::string &s, char p) noexcept
bool	fcontains (const std::string &s, const std::string &p)
	contains a string at the beginning of the string More...
double	plotable (TH1 *h)
bool	exists (const std::string &filename)
	does a file exist More...
std::string	globbed (const std::string &s)
	match a file name More...
bool	empty (TH1 *h)
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...
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, const std::string &path)
double	realmax (TH1 *h, bool include_error, double lo, double hi)
double	realmin (TH1 *h, bool, double lo, double hi)
double	hmean (TH1 *h)
std::vector< int >	findxrange (TH1 *h, bool symmetric)
	automatically set the xrange on a histogram More...
void	xrange (TH1 *h, bool symmetric)
std::vector< double >	findxrangeuser (TH1 *h, bool symmetric)
void	xrangeuser (TH1 *h, bool symmetric)
std::string	findcell (std::string name, const std::string &regex, const std::string &splitex)
std::string	findrun (TFile *f)
void	copyReleaseInfo (TFile *finput, TFile *foutdir)
	copy the release info TTree More...

Variables
bool	LINEF = true
bool	LINES = false
int	colours [6] = { 1, 2, kBlue-4, 6, kCyan-2, kMagenta+2 }
int	markers [6] = { 20, 24, 25, 26, 25, 22 }
double	msizes [6] = { 0.85, 1, 1, 1, 1, 1 }

Detailed Description

Author

mark sutton

Date

Sat Aug 30 2014 14:38:03 CEST

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

Definition in file computils.cxx.

Function Documentation

almost_equal() [1/2]

bool almost_equal	(	float	a,
		float	b,
		int	max_diff
	)

Definition at line 140 of file computils.cxx.

                                                    {
    return almost_equal( floaty_t(a), floaty_t(b), max_diff );
}

almost_equal() [2/2]

bool almost_equal	(	floaty_t	a,
		floaty_t	b,
		int	max_diff
	)

Definition at line 124 of file computils.cxx.

                                                          {
 
  // Check for trivial equality to make sure +0==-0
  if ( a.f == b.f ) return true;
  
  // Different signs means they do not match.
  if ( a.negative() != b.negative() ) return false;
  
  // Find the difference in last place units
  int ulps_diff = std::abs( a.i - b.i );
  if (ulps_diff <= max_diff) return true;
  
  return false;
}

ATLASFORAPP_LABEL()

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

Definition at line 186 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 226 of file computils.cxx.

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

contains() [2/2]

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

contains a string

does a string contain the substring

Definition at line 221 of file computils.cxx.

                                                       { 
  return (s.find(p)!=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 319 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,
		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 329 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 release info TTree

copy the TTree of release info from one directory to another

Definition at line 618 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 294 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 50 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 254 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
	)

contains a string at the beginning of the string

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

Definition at line 232 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 545 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 573 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
	)

automatically set the xrange on a histogram

zoom on non-empty bins

Definition at line 446 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
	)

Definition at line 523 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 262 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 310 of file computils.cxx.

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

hmean()

double hmean

(

TH1 *

h

)

Definition at line 433 of file computils.cxx.

                       {
  double N = integral(h);
  double n=0;
  for ( int i=h->GetNbinsX() ; i>0 ; i-- ) { 
    n += h->GetBinContent(i);
    if ( 2*n>N ) return h->GetBinCenter(i);
  }
  return 0;
}

integral()

double integral

(

TH1 *

h

)

Definition at line 57 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 203 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
	)

Definition at line 65 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==()

bool operator==	(	floaty_t	a,
		floaty_t	b
	)

use a maximum 5 float separation between the two - could be more precise

Definition at line 145 of file computils.cxx.

                                          { 
  return almost_equal( a, b, 5 ); 
}

plotable()

double plotable

(

TH1 *

h

)

Definition at line 237 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,
		double	lo,
		double	hi
	)

Definition at line 379 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	,
		double	lo,
		double	hi
	)

Definition at line 406 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;
}

stime()

std::string stime

(

)

return the current data and time

Definition at line 213 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 300 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 151 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
	)

Definition at line 515 of file computils.cxx.

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

xrangeuser()

void xrangeuser	(	TH1 *	h,
		bool	symmetric
	)

Definition at line 538 of file computils.cxx.

                                         { 
  std::vector<double> limits = findxrangeuser( h, symmetric );
  h->GetXaxis()->SetRangeUser( limits[0], limits[1] );
}

Variable Documentation

colours

int colours[6] = { 1, 2, kBlue-4, 6, kCyan-2, kMagenta+2 }

Definition at line 44 of file computils.cxx.

LINEF

bool LINEF = true

Definition at line 38 of file computils.cxx.

LINES

bool LINES = false

Definition at line 39 of file computils.cxx.

markers

int markers[6] = { 20, 24, 25, 26, 25, 22 }

Definition at line 45 of file computils.cxx.

msizes

double msizes[6] = { 0.85, 1, 1, 1, 1, 1 }

Definition at line 46 of file computils.cxx.