[u/mrichter/AliRoot.git] / test / QA / 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 "AliDAQ.h"
#include "AliLog.h"
#include "AliQA.h"
#include "AliQADataMakerSteer.h"
#include "AliRawReader.h"
#include "AliRawReaderRoot.h"
#include "AliTRDrawStreamBase.h"
#include "AliGeomManager.h"

TString ClassName() { return "rawqa" ; } 

//________________________________qa______________________________________
void rawqa(const Int_t runNumber, Int_t maxFiles = 10, const char* year = "08") 
{	
//	char * kDefaultOCDBStorage = Form("alien://folder=/alice/data/20%s/LHC%sd/OCDB/", year, year) ; 
	char * kDefaultOCDBStorage = Form("local://$ALICE_ROOT/OCDB") ; 
	//AliQA::SetQARefStorage(Form("%s%s/", AliQA::GetQARefDefaultStorage(), year)) ;  
	AliQA::SetQARefStorage("local://$ALICE_ROOT/QAref") ;
	
	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} ;
	TString 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("*.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 collectionFile(pattern) ; 
		collectionFile.ReplaceAll("*.root", ".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 { 
			TString baseDir; 
			baseDir.Form("/alice/data/20%s/",year);
			result = grid->Query(baseDir, pattern) ;  
		}
	} 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) ; 

	TString detectors  = ""; 
	TString detectorsW = ""; 
	UShort_t file = 0 ; 
	UShort_t filesProcessed = 0 ; 
	UShort_t eventsProcessed = 0 ; 
	AliCDBManager* man = AliCDBManager::Instance();
	man->SetDefaultStorage(kDefaultOCDBStorage) ;  
	man->SetRun(runNumber) ; 
	AliQAManager qam("rec") ; 
  qam->SetDefaultStorage(AliQA::GetQARefStorage()) ; 
  qam->SetRun(runNumber) ; 
    
	AliGeomManager::LoadGeometry();
	for ( file = 0 ; file < maxFiles ; file++) {
		if ( qas.GetCurrentEvent() >= maxEvents) 
			break ;

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