-
Wouter Verkerke authoredWouter Verkerke authored
stressRooFit.cxx 12.92 KiB
// @(#)root/roofitcore:$name: $:$id$
// Authors: Wouter Verkerke November 2007
#include "TWebFile.h"
#include "TSystem.h"
#include "TString.h"
#include "TStopwatch.h"
#include "TROOT.h"
#include "TLine.h"
#include "TFile.h"
#include "TClass.h"
#include "TCanvas.h"
#include "TH1.h"
#include "TBenchmark.h"
#include "RooGlobalFunc.h"
#include "RooMsgService.h"
#include "RooPlot.h"
#include "RooFitResult.h"
#include "RooDouble.h"
#include "RooWorkspace.h"
#include "Roo1DTable.h"
#include "RooCurve.h"
#include "RooHist.h"
#include "RooRandom.h"
#include "RooTrace.h"
#include "RooMath.h"
#include <string>
#include <list>
#include <iostream>
#include <math.h>
using namespace std ;
using namespace RooFit ;
//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*//
// //
// RooFit Examples, Wouter Verkerke //
// //
// //
//*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*_*//
Int_t stressRooFit(const char* refFile, Bool_t writeRef, Int_t doVerbose, Int_t oneTest, Bool_t dryRun) ;
//------------------------------------------------------------------------
void StatusPrint(Int_t id,const TString &title,Int_t status)
{
// Print test program number and its title
const Int_t kMAX = 65;
Char_t number[4];
sprintf(number,"%2d",id);
TString header = TString("Test ")+number+" : "+title;
const Int_t nch = header.Length();
for (Int_t i = nch; i < kMAX; i++) header += '.';
cout << header << (status>0 ? "OK" : (status<0 ? "SKIPPED" : "FAILED")) << endl;
}
//#include "stressRooFit_tests_direct.cxx"
#include "stressRooFit_tests.cxx"
//______________________________________________________________________________
Int_t stressRooFit(const char* refFile, Bool_t writeRef, Int_t doVerbose, Int_t oneTest, Bool_t dryRun, Bool_t doDump, Bool_t doTreeStore)
{
Int_t retVal = 0;
// Save memory directory location
RooUnitTest::setMemDir(gDirectory) ;
if (doTreeStore) {
RooAbsData::setDefaultStorageType(RooAbsData::Tree) ;
}
TFile* fref = 0 ;
if (!dryRun) {
if (TString(refFile).Contains("http:")) {
if (writeRef) {
cout << "stressRooFit ERROR: reference file must be local file in writing mode" << endl ;
return 1;
}
fref = new TWebFile(refFile) ;
} else {
fref = new TFile(refFile,writeRef?"RECREATE":"") ;
}
if (fref->IsZombie()) {
cout << "stressRooFit ERROR: cannot open reference file " << refFile << endl ;
return 1;
}
}
if (dryRun) {
// Preload singletons here so they don't show up in trace accounting
RooNumIntConfig::defaultConfig() ;
RooResolutionModel::identity() ;
RooTrace::active(1) ;
}
// Add dedicated logging stream for errors that will remain active in silent mode
RooMsgService::instance().addStream(RooFit::ERROR) ;
cout << "******************************************************************" <<endl;
cout << "* RooFit - S T R E S S suite *" <<endl;
cout << "******************************************************************" <<endl;
cout << "******************************************************************" <<endl;
TStopwatch timer;
timer.Start();
list<RooUnitTest*> testList ;
testList.push_back(new TestBasic101(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic102(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic103(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic105(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic108(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic109(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic110(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic111(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic201(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic202(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic203(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic204(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic205(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic208(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic209(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic301(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic302(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic303(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic304(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic305(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic306(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic307(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic308(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic310(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic311(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic312(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic313(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic315(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic314(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic316(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic402(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic403(fref,writeRef,doVerbose)) ; testList.push_back(new TestBasic404(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic405(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic406(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic501(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic599(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic601(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic602(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic604(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic605(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic606(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic607(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic609(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic701(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic702(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic703(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic704(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic705(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic706(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic707(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic708(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic801(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic802(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic803(fref,writeRef,doVerbose)) ;
testList.push_back(new TestBasic804(fref,writeRef,doVerbose)) ;
cout << "* Starting S T R E S S basic suite *" <<endl;
cout << "******************************************************************" <<endl;
if (doDump) {
TFile fdbg("stressRooFit_DEBUG.root","RECREATE") ;
}
gBenchmark->Start("StressRooFit");
Int_t i(1) ;
for (list<RooUnitTest*>::iterator iter = testList.begin () ; iter != testList.end() ; ++iter) {
if (oneTest<0 || oneTest==i) {
if (doDump) {
(*iter)->setDebug(kTRUE) ;
}
Int_t status = (*iter)->isTestAvailable()?(*iter)->runTest():-1;
StatusPrint( i,(*iter)->GetName(), status);
// increment retVal for every failed test
if (!status) ++retVal;
}
delete (*iter) ;
i++ ;
}
if (dryRun) {
RooTrace::dump() ;
}
gBenchmark->Stop("StressRooFit");
//Print table with results
Bool_t UNIX = strcmp(gSystem->GetName(), "Unix") == 0;
printf("******************************************************************\n");
if (UNIX) {
TString sp = gSystem->GetFromPipe("uname -a");
sp.Resize(60);
printf("* SYS: %s\n",sp.Data());
if (strstr(gSystem->GetBuildNode(),"Linux")) {
sp = gSystem->GetFromPipe("lsb_release -d -s");
printf("* SYS: %s\n",sp.Data());
}
if (strstr(gSystem->GetBuildNode(),"Darwin")) {
sp = gSystem->GetFromPipe("sw_vers -productVersion");
sp += " Mac OS X "; printf("* SYS: %s\n",sp.Data());
}
} else {
const Char_t *os = gSystem->Getenv("OS");
if (!os) printf("* SYS: Windows 95\n");
else printf("* SYS: %s %s \n",os,gSystem->Getenv("PROCESSOR_IDENTIFIER"));
}
printf("******************************************************************\n");
gBenchmark->Print("StressFit");
#ifdef __CINT__
Double_t reftime = 186.34; //pcbrun4 interpreted
#else
Double_t reftime = 93.59; //pcbrun4 compiled
#endif
const Double_t rootmarks = 860*reftime/gBenchmark->GetCpuTime("StressRooFit");
printf("******************************************************************\n");
printf("* ROOTMARKS =%6.1f * Root%-8s %d/%d\n",rootmarks,gROOT->GetVersion(),
gROOT->GetVersionDate(),gROOT->GetVersionTime());
printf("******************************************************************\n");
printf("Time at the end of job = %f seconds\n",timer.CpuTime());
if (fref) {
fref->Close() ;
delete fref ;
}
delete gBenchmark ;
gBenchmark = 0 ;
return retVal;
}
//_____________________________batch only_____________________
#ifndef __CINT__
int main(int argc,const char *argv[])
{
Bool_t doWrite = kFALSE ;
Int_t doVerbose = 0 ;
Int_t oneTest = -1 ;
Int_t dryRun = kFALSE ;
Bool_t doDump = kFALSE ;
Bool_t doTreeStore = kFALSE ;
string refFileName = "http://root.cern.ch/files/stressRooFit_v534_ref.root" ;
// Parse command line arguments
for (Int_t i=1 ; i<argc ; i++) {
string arg = argv[i] ;
if (arg=="-f") {
cout << "stressRooFit: using reference file " << argv[i+1] << endl ;
refFileName = argv[++i] ;
}
if (arg=="-w") {
cout << "stressRooFit: running in writing mode to updating reference file" << endl ;
doWrite = kTRUE ;
}
if (arg=="-mc") {
cout << "stressRooFit: running in memcheck mode, no regression tests are performed" << endl ;
dryRun=kTRUE ;
}
if (arg=="-ts") {
cout << "stressRooFit: setting tree-based storage for datasets" << endl ; doTreeStore=kTRUE ;
}
if (arg=="-v") {
cout << "stressRooFit: running in verbose mode" << endl ;
doVerbose = 1 ;
}
if (arg=="-vv") {
cout << "stressRooFit: running in very verbose mode" << endl ;
doVerbose = 2 ;
}
if (arg=="-n") {
cout << "stressRooFit: running single test " << argv[i+1] << endl ;
oneTest = atoi(argv[++i]) ;
}
if (arg=="-d") {
cout << "stressRooFit: setting gDebug to " << argv[i+1] << endl ;
gDebug = atoi(argv[++i]) ;
}
if (arg=="-c") {
cout << "stressRooFit: dumping comparison file for failed tests " << endl ;
doDump=kTRUE ;
}
if (arg=="-h") {
cout << "usage: stressRooFit [ options ] " << endl ;
cout << "" << endl ;
cout << " -f <file> : use given reference file instead of default (" << refFileName << ")" << endl ;
cout << " -w : write reference file, instead of reading file and running comparison tests" << endl ;
cout << " " << endl ;
cout << " -n N : Only run test with sequential number N instead of full suite of tests" << endl ;
cout << " -c : dump file stressRooFit_DEBUG.root to which results of both current result and reference for each failed test are written" << endl ;
cout << " -mc : memory check mode, no regression test are performed. Set this flag when running with valgrind" << endl ;
cout << " -vs : Use vector-based storage for all datasets (default is tree-based storage)" << endl ;
cout << " -v/-vv : set verbose mode (show result of each regression test) or very verbose mode (show all roofit output as well)" << endl ;
cout << " -d N : set ROOT gDebug flag to N" << endl ;
cout << " " << endl ;
return 0 ;
}
}
if (doWrite && refFileName.find("http:")==0) {
// Locate file name part in URL and update refFileName accordingly
char* buf = new char[refFileName.size()+1] ;
strcpy(buf,refFileName.c_str()) ;
char *ptr = strrchr(buf,'/') ;
if (!ptr) {
ptr = strrchr(buf,':') ;
}
refFileName = ptr+1 ;
delete[] buf ;
cout << "stressRooFit: WARNING running in write mode, but reference file is web file, writing local file instead: " << refFileName << endl ;
}
gBenchmark = new TBenchmark();
Int_t retVal = stressRooFit(refFileName.c_str(),doWrite,doVerbose,oneTest,dryRun,doDump,doTreeStore);
return retVal;
}
#endif