[u/mrichter/AliRoot.git] / test / cosmic / rawqa.C

#include <iostream>
#include <fstream>

#include <TAlienCollection.h>
#include <TFile.h>
#include <TGrid.h>
#include <TGridResult.h>
#include <TMath.h>
#include <TROOT.h>
#include <TString.h>
#include <TSystem.h>

#include "AliCDBManager.h"
#include "AliLog.h"
#include "AliQA.h"
#include "AliQADataMakerSteer.h"
#include "AliRawReader.h"
#include "AliRawReaderRoot.h"
#include "AliDAQ.h"

TString ClassName() { return "rawqa" ; } 

//________________________________qa______________________________________
void rawqa(const Int_t runNumber, Int_t maxFiles = 10, const char* year = "08") 
{	
	const char * kDefaultOCDBStorage = Form("alien://folder=/alice/data/20%s/LHC%sa/OCDB/", year, year) ; 
	
	UInt_t maxEvents = 99999 ;
	if ( maxFiles < 0 ) {
		maxEvents = TMath::Abs(maxFiles) ; 
		maxFiles = 99 ; 
	}
	AliLog::SetGlobalDebugLevel(0) ; 
	// connect to the grid 
	TGrid * grid = 0x0 ; 
       	grid = TGrid::Connect("alien://") ; 
		
	Bool_t detIn[AliDAQ::kNDetectors] = {kFALSE} ;
	char * detNameOff[AliDAQ::kNDetectors] = {"ITS", "ITS", "ITS", "TPC", "TRD", "TOF", "HMPID", "PHOS", "PHOS", "PMD", "MUON", "MUON", "FMD", "T0", "VZERO", "ZDC", "ACORDE", "TRG", "EMCAL", "DAQ_TEST", "HLT"} ; 
	// make the file name pattern year and run number
	TString pattern;
	pattern.Form("%9d",runNumber);
	pattern.ReplaceAll(" ", "0") ; 
	pattern.Prepend(year);
	pattern.Append("*0.root");

	// find the files associated to this run
	TGridResult * result = 0x0 ; 
	Bool_t local = kFALSE ; 
	if (grid) { // get the list of files from AliEn directly 
	  TString baseDir; 
	  baseDir.Form("/alice/data/20%s/",year);
	  result = grid->Query(baseDir, pattern) ;  
	} else {
	  TString collectionFile(pattern) ; 
	  collectionFile.Append(".xml") ; 
	  if ( gSystem->AccessPathName(collectionFile) == 0 ) { // get the list of files from an a-priori created collection file
	    TAlienCollection collection(collectionFile.Data(), maxFiles) ; 
	    result = collection.GetGridResult("", 0, 0); 
	  } else { // get the list of files from the local current directory 
	    local = kTRUE ; 
	    char line[100] ; 
	    sprintf(line, ".! ls %s*.root > tempo.txt", pattern.Data()) ; 
	    gROOT->ProcessLine(line) ;
	  }
	}
	AliLog::Flush();
	ifstream in ; 
	if (local) 
		in.open("tempo.txt", ifstream::in) ; 

	AliCDBManager* man = AliCDBManager::Instance();
	man->SetDefaultStorage(kDefaultOCDBStorage) ;  
	AliQA::SetQARefStorage("alien://folder=/alice/QA/2008/") ; 
	man->SetSpecificStorage("*", AliQA::GetQARefStorage());
	AliQADataMakerSteer qas ; 
	TString detectors  = ""; 
	TString detectorsW = ""; 
	UShort_t file = 0 ; 
	UShort_t filesProcessed = 0 ; 
	for ( file = 0 ; file < maxFiles ; file++) {
		TString fileName ; 
		if ( local) {
			in >> fileName ;
		} 
		else 
			fileName = result->GetKey(file, "turl");
	       	if ( fileName == "" )  
		        break ;
		if ( fileName.Contains("tag") )
			continue; 
                filesProcessed++ ;
		char input[200] ; 
		if (local) 
			sprintf(input, "%s", fileName.Data()) ; 
		else 
			sprintf(input, "%s", result->GetKey(file, "turl")); 
		AliInfo(Form("Proccessing file # %d --> %s", file, input)) ;
		AliLog::Flush();
		// check which detectors are present 
		AliRawReader * rawReader = new AliRawReaderRoot(input);
		while ( rawReader->NextEvent() ) {
			man->SetRun(rawReader->GetRunNumber());
			AliLog::Flush();
			UChar_t * data ; 
			while (rawReader->ReadNextData(data)) {
				Int_t detID = rawReader->GetDetectorID();
				if (detID < 0 || detID >= AliDAQ::kNDetectors) {
					AliError("Wrong detector ID! Skipping payload...");
					continue;
				}
				detIn[detID] = kTRUE ; 
			}
			for (Int_t detID = 0; detID < AliDAQ::kNDetectors ; detID++) {
				if (detIn[detID]) {
					if ( ! detectors.Contains(detNameOff[detID]) ) {
						detectors.Append(detNameOff[detID]) ;
						detectors.Append(" ") ;
					}
				}
			}
			if ( !detectors.IsNull() )
				break ; 
		}
		AliInfo(Form("Current = %d Max = %d", qas.GetCurrentEvent(), maxEvents)) ; 
		AliLog::Flush();
		if ( qas.GetCurrentEvent() > maxEvents) 
			break ;
		// TEMPORARY REMOVAL OF TRD!!!
		detectors.ReplaceAll("TRD", "") ;
		// TEMPORARY REMOVAL OF TRD!!!
		if ( !detectors.IsNull() ) {
			qas.SetMaxEvents(maxEvents) ; 	
			detectorsW = qas.Run(detectors, rawReader) ;
			qas.Reset() ;
		} else {
			AliError("No valid detectors found") ; 
		} 
		delete rawReader ;
	}
	AliLog::Flush();
	qas.Merge(runNumber) ; 
	
	AliLog::Flush();
	// The summary 
	AliInfo(Form("\n\n********** Summary for run %d **********", runNumber)) ; 
	printf("     detectors present in the run        : %s\n", detectors.Data()) ; 
	printf("     detectors present in the run with QA: %s\n", detectorsW.Data()) ; 
	printf("     number of files processed           : %d\n", filesProcessed) ; 
	TFile * qaResult = TFile::Open(AliQA::GetQAResultFileName()) ; 
	AliQA * qa = dynamic_cast<AliQA *>(qaResult->Get("QA")) ; 
	for (Int_t index = 0 ; index < AliQA::kNDET ; index++)
		if (detectorsW.Contains(AliQA::GetDetName(AliQA::DETECTORINDEX(index)))) 
			qa->ShowStatus(AliQA::DETECTORINDEX(index)) ;
}
Commit	Line	Data
40d0f38f	1	#include <iostream>
	2	#include <fstream>
	3
	4	#include <TAlienCollection.h>
	5	#include <TFile.h>
	6	#include <TGrid.h>
	7	#include <TGridResult.h>
aa22f9e3	8	#include <TMath.h>
40d0f38f	9	#include <TROOT.h>
	10	#include <TString.h>
	11	#include <TSystem.h>
	12
	13	#include "AliCDBManager.h"
	14	#include "AliLog.h"
	15	#include "AliQA.h"
	16	#include "AliQADataMakerSteer.h"
	17	#include "AliRawReader.h"
	18	#include "AliRawReaderRoot.h"
	19	#include "AliDAQ.h"
	20
	21	TString ClassName() { return "rawqa" ; }
	22
	23	//________________________________qa______________________________________
aa22f9e3	24	void rawqa(const Int_t runNumber, Int_t maxFiles = 10, const char* year = "08")
40d0f38f	25	{
aa22f9e3	26	const char * kDefaultOCDBStorage = Form("alien://folder=/alice/data/20%s/LHC%sa/OCDB/", year, year) ;
	27
	28	UInt_t maxEvents = 99999 ;
	29	if ( maxFiles < 0 ) {
aa22f9e3	30	maxEvents = TMath::Abs(maxFiles) ;
f0c6fb4b	31	maxFiles = 99 ;
aa22f9e3	32	}
40d0f38f	33	AliLog::SetGlobalDebugLevel(0) ;
	34	// connect to the grid
	35	TGrid * grid = 0x0 ;
	36	grid = TGrid::Connect("alien://") ;
	37
	38	Bool_t detIn[AliDAQ::kNDetectors] = {kFALSE} ;
	39	char * detNameOff[AliDAQ::kNDetectors] = {"ITS", "ITS", "ITS", "TPC", "TRD", "TOF", "HMPID", "PHOS", "PHOS", "PMD", "MUON", "MUON", "FMD", "T0", "VZERO", "ZDC", "ACORDE", "TRG", "EMCAL", "DAQ_TEST", "HLT"} ;
	40	// make the file name pattern year and run number
	41	TString pattern;
	42	pattern.Form("%9d",runNumber);
	43	pattern.ReplaceAll(" ", "0") ;
	44	pattern.Prepend(year);
	45	pattern.Append("*0.root");
	46
	47	// find the files associated to this run
	48	TGridResult * result = 0x0 ;
40d0f38f	49	Bool_t local = kFALSE ;
	50	if (grid) { // get the list of files from AliEn directly
	51	TString baseDir;
	52	baseDir.Form("/alice/data/20%s/",year);
	53	result = grid->Query(baseDir, pattern) ;
	54	} else {
	55	TString collectionFile(pattern) ;
	56	collectionFile.Append(".xml") ;
	57	if ( gSystem->AccessPathName(collectionFile) == 0 ) { // get the list of files from an a-priori created collection file
aa22f9e3	58	TAlienCollection collection(collectionFile.Data(), maxFiles) ;
40d0f38f	59	result = collection.GetGridResult("", 0, 0);
	60	} else { // get the list of files from the local current directory
	61	local = kTRUE ;
	62	char line[100] ;
	63	sprintf(line, ".! ls %s*.root > tempo.txt", pattern.Data()) ;
	64	gROOT->ProcessLine(line) ;
	65	}
	66	}
	67	AliLog::Flush();
	68	ifstream in ;
	69	if (local)
	70	in.open("tempo.txt", ifstream::in) ;
	71
	72	AliCDBManager* man = AliCDBManager::Instance();
018c2b09	73	man->SetDefaultStorage(kDefaultOCDBStorage) ;
9af5c2c8	74	AliQA::SetQARefStorage("alien://folder=/alice/QA/2008/") ;
01f3ddbc	75	man->SetSpecificStorage("*", AliQA::GetQARefStorage());
40d0f38f	76	AliQADataMakerSteer qas ;
	77	TString detectors = "";
	78	TString detectorsW = "";
	79	UShort_t file = 0 ;
aa22f9e3	80	UShort_t filesProcessed = 0 ;
aa22f9e3	81	for ( file = 0 ; file < maxFiles ; file++) {
40d0f38f	82	TString fileName ;
	83	if ( local) {
	84	in >> fileName ;
	85	}
	86	else
	87	fileName = result->GetKey(file, "turl");
	88	if ( fileName == "" )
	89	break ;
	90	if ( fileName.Contains("tag") )
	91	continue;
87929a37	92	filesProcessed++ ;
40d0f38f	93	char input[200] ;
	94	if (local)
	95	sprintf(input, "%s", fileName.Data()) ;
	96	else
	97	sprintf(input, "%s", result->GetKey(file, "turl"));
	98	AliInfo(Form("Proccessing file # %d --> %s", file, input)) ;
	99	AliLog::Flush();
	100	// check which detectors are present
	101	AliRawReader * rawReader = new AliRawReaderRoot(input);
aa22f9e3	102	while ( rawReader->NextEvent() ) {
aa22f9e3	103	man->SetRun(rawReader->GetRunNumber());
f0c6fb4b	104	AliLog::Flush();
aa22f9e3	105	UChar_t * data ;
	106	while (rawReader->ReadNextData(data)) {
	107	Int_t detID = rawReader->GetDetectorID();
	108	if (detID < 0 \|\| detID >= AliDAQ::kNDetectors) {
	109	AliError("Wrong detector ID! Skipping payload...");
	110	continue;
	111	}
	112	detIn[detID] = kTRUE ;
40d0f38f	113	}
aa22f9e3	114	for (Int_t detID = 0; detID < AliDAQ::kNDetectors ; detID++) {
	115	if (detIn[detID]) {
	116	if ( ! detectors.Contains(detNameOff[detID]) ) {
	117	detectors.Append(detNameOff[detID]) ;
	118	detectors.Append(" ") ;
	119	}
40d0f38f	120	}
40d0f38f	121	}
aa22f9e3	122	if ( !detectors.IsNull() )
aa22f9e3	123	break ;
40d0f38f	124	}
f0c6fb4b	125	AliInfo(Form("Current = %d Max = %d", qas.GetCurrentEvent(), maxEvents)) ;
f0c6fb4b	126	AliLog::Flush();
aa22f9e3	127	if ( qas.GetCurrentEvent() > maxEvents)
aa22f9e3	128	break ;
40d0f38f	129	// TEMPORARY REMOVAL OF TRD!!!
aa22f9e3	130	detectors.ReplaceAll("TRD", "") ;
40d0f38f	131	// TEMPORARY REMOVAL OF TRD!!!
40d0f38f	132	if ( !detectors.IsNull() ) {
f0c6fb4b	133	qas.SetMaxEvents(maxEvents) ;
	134	detectorsW = qas.Run(detectors, rawReader) ;
	135	qas.Reset() ;
40d0f38f	136	} else {
f0c6fb4b	137	AliError("No valid detectors found") ;
40d0f38f	138	}
f0c6fb4b	139	delete rawReader ;
40d0f38f	140	}
	141	AliLog::Flush();
	142	qas.Merge(runNumber) ;
	143
	144	AliLog::Flush();
	145	// The summary
	146	AliInfo(Form("\n\n******** Summary for run %d ********", runNumber)) ;
	147	printf(" detectors present in the run : %s\n", detectors.Data()) ;
	148	printf(" detectors present in the run with QA: %s\n", detectorsW.Data()) ;
87929a37	149	printf(" number of files processed : %d\n", filesProcessed) ;
40d0f38f	150	TFile * qaResult = TFile::Open(AliQA::GetQAResultFileName()) ;
	151	AliQA * qa = dynamic_cast<AliQA *>(qaResult->Get("QA")) ;
	152	for (Int_t index = 0 ; index < AliQA::kNDET ; index++)
	153	if (detectorsW.Contains(AliQA::GetDetName(AliQA::DETECTORINDEX(index))))
	154	qa->ShowStatus(AliQA::DETECTORINDEX(index)) ;
	155	}