Page MenuHomeHEPForge
Files F19244254

doRefCheck.cc
No OneTemporary
Actions

Size
3 KB
Referenced Files
None
Subscribers
None

doRefCheck.cc
View Options

// Lange
// September 29, 2006
//
// Script to automate reference comparisons
//
//
#include "TFile.h"
#include "TKey.h"
#include "TCollection.h"
#include "TH1F.h"
#include "TROOT.h"
#include "TStyle.h"
#include "TCanvas.h"
#include <vector>
#include <string>
#include <iostream>
#include <map>
#include <fstream>
void doRefCheck(TString file, bool makepsfiles=false, bool writeRef=false) {
// read in root file
TFile *f1=new TFile(file,"READ");
// f1->ls();
TKey *key;
TIter nextkey(f1->GetListOfKeys());
std::vector<std::string> names;
while (key= (TKey*)nextkey()) {
if ( strcmp(key->GetClassName(),"TH1F")!=0 ) continue;
TH1F *h1=(TH1F*)key->ReadObj();
names.push_back((h1->GetName()));
}
if ( names.size()==0 ) {
std::cout << "NO histograms found?? Missing file?\n";
return;
}
// read in references if they are found.
std::map<std::string, TH1F*> references;
TString refFile=file;
refFile+=".ref";
ifstream *infile=new ifstream(refFile.Data(),std::ios::in);
char tChar[100];
std::vector<double> datums;
std::string refTitle;
double refLow=0.;
double refHigh=0.;
while ( infile->good()) {
infile->getline(tChar,100);
if ( infile->good() ) {
if ( strcmp(tChar,"start histogram")==0) {
if ( datums.size() > 0 ) {
TString tTitle="ref";
tTitle+=references.size();
TH1F *hTemp=new TH1F(tTitle,tTitle,datums.size()-2,refLow,refHigh);
for ( unsigned int i=0; i<datums.size(); i++) {
hTemp->SetBinContent(i,datums[i]);
//std::cout << datums[i] << std::endl;
}
references[refTitle]=hTemp;
datums.clear();
}
infile->getline(tChar,100);
refTitle=std::string(tChar);
infile->getline(tChar,100);
refLow=atof(tChar);
infile->getline(tChar,100);
refHigh=atof(tChar);
}
else{
datums.push_back(atof(tChar));
}
}
}
if ( datums.size() > 0 ) {
TString tTitle="ref";
tTitle+=references.size();
TH1F *hTemp=new TH1F(tTitle,tTitle,datums.size()-2,refLow,refHigh);
for ( unsigned int i=0; i<datums.size(); i++) hTemp->SetBinContent(i,datums[i]);
references[refTitle]=hTemp;
}
gROOT->SetStyle("Plain"); // make plot background truly white
gStyle->SetPalette(1); // palette set from violet to red, root chooses scale
fstream *outf=0;
if (writeRef) {
TString refOut=file;
refOut+=".ref.new";
outf=new fstream(refOut.Data(),std::ios::out);
}
for ( unsigned int i=0; i<names.size(); i++) {
TString canName="can"; canName+=i;
TCanvas *can=new TCanvas(canName,canName,600,400);
TH1F *h1=(TH1F*)f1->Get(names[i].c_str());
h1->SetMinimum(0.);
h1->Draw();
TH1F *h2=references[std::string(h1->GetTitle())];
std::cout << "Histogram: " << h1->GetTitle();
if ( h2 != 0 ) {
h2->SetLineColor(kRed);
h2->Draw("same");
if ( h1->GetNbinsX() != h2->GetNbinsX() ) {
std::cout << " reference has different binning! " << h1->GetNbinsX() << " " << h2->GetNbinsX() << std::endl;
}
else{
int nDiffs=0;
for ( int i=0; i<h1->GetNbinsX(); i++ )
if ( h1->GetBinContent(i) != h2->GetBinContent(i) ) nDiffs++;
if ( nDiffs > 0 )
std::cout << " has " << nDiffs << " bins that were different from reference" << std::endl;
else
std::cout << " agrees with reference\n";
}
}
else{
std::cout << " no reference found\n";
}
TString title=h1->GetTitle();
if ( outf != 0 ) {
(*outf) << "start histogram\n";
(*outf) << title.Data() << std::endl;
(*outf) << h1->GetBinLowEdge(1) << std::endl
<< h1->GetBinLowEdge(1+h1->GetNbinsX()) << std::endl;
for ( int j=0; j<= h1->GetNbinsX()+1; j++)
(*outf) << h1->GetBinContent(j) << std::endl;
}
if ( makepsfiles ) {
TString psName=file;
psName+="_"; psName+=i; psName+=".ps";
can->Print(psName);
}
}
if ( outf != 0 ) outf->close();
}

File Metadata

Mime Type
text/x-c
Expires
Tue, Sep 30, 4:40 AM (22 h, 38 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
6523255
Default Alt Text
doRefCheck.cc (3 KB)

Event Timeline

HEPForge.org · .