[u/mrichter/AliRoot.git] / STEER / AliQADataMaker.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/


/* $Id$ */

//
//  Base Class
//  Produces the data needed to calculate the quality assurance. 
//  All data must be mergeable objects.
//  Y. Schutz CERN July 2007
//

// --- ROOT system ---
#include <TROOT.h> 
#include <TSystem.h> 
#include <TFile.h>
#include <TList.h> 
#include <TTree.h>
#include <TClonesArray.h>
#include <TParameter.h>
#include <TH1K.h>
#include <TH2C.h>
#include <TH2D.h>
#include <TH2F.h>
#include <TH2I.h>
#include <TH3C.h>
#include <TH3D.h>
#include <TH3F.h>
#include <TH3I.h>
#include <TH3S.h>

// --- Standard library ---

// --- AliRoot header files ---
#include "AliLog.h"
#include "AliQADataMaker.h"
#include "AliQAChecker.h"
#include "AliESDEvent.h"
#include "AliRawReader.h"
#include "AliDetectorRecoParam.h"


ClassImp(AliQADataMaker)
             
//____________________________________________________________________________ 
AliQADataMaker::AliQADataMaker(const Char_t * name, const Char_t * title) : 
  TNamed(name, title), 
  fOutput(0x0),
  fDetectorDir(0x0),
  fDetectorDirName(""), 
  fCurrentCycle(0), 
  fCycle(9999999), 
  fCycleCounter(0), 
  fWriteExpert(kFALSE),
  fParameterList(new TList*[AliRecoParam::kNSpecies]), 
  fRun(0), 
  fEventSpecie(AliRecoParam::kDefault), 
  fDigitsArray(NULL) 
{
  // ctor
  fDetectorDirName = GetName() ; 
    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
      fParameterList[specie] = NULL ; 
    }
}

//____________________________________________________________________________ 
AliQADataMaker::AliQADataMaker(const AliQADataMaker& qadm) :
  TNamed(qadm.GetName(), qadm.GetTitle()),
  fOutput(qadm.fOutput),
  fDetectorDir(qadm.fDetectorDir),
  fDetectorDirName(qadm.fDetectorDirName),
  fCurrentCycle(qadm.fCurrentCycle), 
  fCycle(qadm.fCycle), 
  fCycleCounter(qadm.fCycleCounter), 
  fWriteExpert(qadm.fWriteExpert),
  fParameterList(qadm.fParameterList),  
  fRun(qadm.fRun), 
  fEventSpecie(qadm.fEventSpecie),
  fDigitsArray(NULL) 
{
  //copy ctor
  fDetectorDirName = GetName() ; 
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
    fParameterList[specie] = qadm.fParameterList[specie] ; 
  //  fImage[specie] = qadm.fImage[specie] ; 
  }
}

//____________________________________________________________________________ 
AliQADataMaker::~AliQADataMaker()
{
  for (Int_t esIndex = 0 ; esIndex < AliRecoParam::kNSpecies ; esIndex++) {
    if (fParameterList[esIndex] )
      delete fParameterList[esIndex] ; 
  }
  delete[] fParameterList ; 

  if (fDigitsArray) {
    fDigitsArray->Clear() ; 
    delete fDigitsArray ;
  }
}

//____________________________________________________________________________
Int_t AliQADataMaker::Add2List(TH1 * hist, const Int_t index, TObjArray ** list, const Bool_t expert, const Bool_t image, const Bool_t saveForCorr) 
{ 
	// Set histograms memory resident and add to the list
	// Maximm allowed is 10000
  
  Int_t rv = -1 ; 
  TClass * classType = hist->Class() ;
  TString className(classType->GetName()) ; 
  if( ! className.BeginsWith("T") && ! classType->InheritsFrom("TH1") ) {
    AliError(Form("QA data Object must be a generic ROOT object and derive fom TH1 and not %s", className.Data())) ; 
	} else if ( index > AliQAv1::GetMaxQAObj() ) {
		AliError(Form("Max number of authorized QA objects is %d", AliQAv1::GetMaxQAObj())) ; 
  } else {
    hist->SetDirectory(0) ; 
    if (expert) 
      hist->SetBit(AliQAv1::GetExpertBit()) ;
    if (image) 
      hist->SetBit(AliQAv1::GetImageBit()) ;  
    const Char_t * name = Form("%s_%s", AliRecoParam::GetEventSpecieName(fEventSpecie), hist->GetName()) ;
    hist->SetName(name) ; 
    if(saveForCorr) {  
      const Char_t * cname = Form("%s_%s", list[AliRecoParam::AConvert(AliRecoParam::kDefault)]->GetName(), hist->GetName()) ;  
      TParameter<double> * p = new TParameter<double>(cname, 9999.9999) ;
      if ( fParameterList[AliRecoParam::AConvert(fEventSpecie)] == NULL )
        fParameterList[AliRecoParam::AConvert(fEventSpecie)] = new TList() ; 
      fParameterList[AliRecoParam::AConvert(fEventSpecie)]->Add(p) ;
    }
    TObject* old = list[AliRecoParam::AConvert(fEventSpecie)]->At(index);
    if (old) {
      AliError(Form("%s - OUPS ! Already got an object (%p,%s) for index=%d => will most probably get a memory leak by replacing it with (%p,%s) !",
                    GetName(),old,old->GetName(),index,hist,hist->GetName()));
    }
    list[AliRecoParam::AConvert(fEventSpecie)]->AddAtAndExpand(hist, index) ; 
    rv = list[AliRecoParam::AConvert(fEventSpecie)]->GetLast() ;
  }
  return rv ; 
}

//____________________________________________________________________________
TH1 *  AliQADataMaker::CloneMe(TH1 * hist, Int_t specie) const  
{
  // clones a histogram 
  const Char_t * name = Form("%s_%s", AliRecoParam::GetEventSpecieName(specie), hist->GetName()) ;
  TH1 * hClone = static_cast<TH1 *>(hist->Clone(name)) ; 
  if ( hist->TestBit(AliQAv1::GetExpertBit()) )
    hClone->SetBit(AliQAv1::GetExpertBit()) ; 
  if ( hist->TestBit(AliQAv1::GetImageBit()) )
    hClone->SetBit(AliQAv1::GetImageBit()) ; 
  return hClone ; 
}

//____________________________________________________________________________
void AliQADataMaker::DefaultEndOfDetectorCycle(AliQAv1::TASKINDEX_t task) 
{
	// this method must be oveloaded by detectors
	// sets the QA result to Fatal
	AliQAv1::Instance(AliQAv1::GetDetIndex(GetName())) ;
	AliQAv1 * qa = AliQAv1::Instance(task) ;
  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) 
    qa->Set(AliQAv1::kFATAL, specie) ; 
	AliQAv1::GetQAResultFile()->cd() ; 
	qa->Write(AliQAv1::GetQAName(), kWriteDelete) ;   
	AliQAv1::GetQAResultFile()->Close() ; 
}

//____________________________________________________________________________ 
void AliQADataMaker::Finish() const 
{ 
	// write to the output File
	if (fOutput) 
		fOutput->Close() ; 
} 

//____________________________________________________________________________ 
TObject * AliQADataMaker::GetData(TObjArray ** list, const Int_t index)  
{ 
	// Returns the QA object at index. Limit is AliQAv1::GetMaxQAObj() 
  if ( ! list ) {
		AliError("Data list is NULL !!") ; 
		return NULL ; 		
	}

  SetEventSpecie(fEventSpecie) ;  
  Int_t esindex = AliRecoParam::AConvert(fEventSpecie) ; 
  TH1 * histClone = NULL ; 
  TObjArray * arr = list[esindex] ; 
	if (arr) {
    if ( ! arr->GetEntriesFast() ) {
      // Initializes the histograms 
      TString arrName(arr->GetName()) ; 
      if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kRAWS)))
        InitRaws() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kHITS)))
        InitHits() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kSDIGITS)))
        InitSDigits() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kDIGITS)))
        InitDigits() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kDIGITSR)))
        InitDigits() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kRECPOINTS)))
        InitRecPoints() ; 
      else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kESDS)))
        InitESDs() ; 
    }
		if ( index > AliQAv1::GetMaxQAObj() ) {
			AliError(Form("Max number of authorized QA objects is %d", AliQAv1::GetMaxQAObj())) ; 
		} else {
      if ( arr->At(index) )  {
        histClone = static_cast<TH1*>(arr->At(index)) ; 
      } 	
    }
  }
  return histClone ; 		
}

//____________________________________________________________________________ 
TObjArray*  AliQADataMaker::Init(AliQAv1::TASKINDEX_t task, AliRecoParam::EventSpecie_t es, Int_t cycles)
{
  // Initialializes and  returns the QAData list for a given event specie
  TObjArray ** ar = Init(task, cycles) ; 
  return ar[AliRecoParam::AConvert(es)] ;  
}

//____________________________________________________________________________ 
Bool_t AliQADataMaker::IsValidEventSpecie(Int_t eventSpecieIndex, TObjArray ** list)
{
  // check if event specie was present in current run or 
  // if histograms of this event specie have been created
  if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(eventSpecieIndex)) || ! list[eventSpecieIndex]->GetEntriesFast() )
    return kFALSE ;
  else
    return kTRUE ;
}
Commit	Line	Data
	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16
	17	/* $Id$ */
	18
	19	//
	20	// Base Class
	21	// Produces the data needed to calculate the quality assurance.
	22	// All data must be mergeable objects.
	23	// Y. Schutz CERN July 2007
	24	//
	25
	26	// --- ROOT system ---
	27	#include <TROOT.h>
	28	#include <TSystem.h>
	29	#include <TFile.h>
	30	#include <TList.h>
	31	#include <TTree.h>
	32	#include <TClonesArray.h>
	33	#include <TParameter.h>
	34	#include <TH1K.h>
	35	#include <TH2C.h>
	36	#include <TH2D.h>
	37	#include <TH2F.h>
	38	#include <TH2I.h>
	39	#include <TH3C.h>
	40	#include <TH3D.h>
	41	#include <TH3F.h>
	42	#include <TH3I.h>
	43	#include <TH3S.h>
	44
	45	// --- Standard library ---
	46
	47	// --- AliRoot header files ---
	48	#include "AliLog.h"
	49	#include "AliQADataMaker.h"
	50	#include "AliQAChecker.h"
	51	#include "AliESDEvent.h"
	52	#include "AliRawReader.h"
	53	#include "AliDetectorRecoParam.h"
	54
	55
	56	ClassImp(AliQADataMaker)
	57
	58	//____________________________________________________________________________
	59	AliQADataMaker::AliQADataMaker(const Char_t * name, const Char_t * title) :
	60	TNamed(name, title),
	61	fOutput(0x0),
	62	fDetectorDir(0x0),
	63	fDetectorDirName(""),
	64	fCurrentCycle(0),
	65	fCycle(9999999),
	66	fCycleCounter(0),
	67	fWriteExpert(kFALSE),
	68	fParameterList(new TList*[AliRecoParam::kNSpecies]),
	69	fRun(0),
	70	fEventSpecie(AliRecoParam::kDefault),
	71	fDigitsArray(NULL)
	72	{
	73	// ctor
	74	fDetectorDirName = GetName() ;
	75	for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
	76	fParameterList[specie] = NULL ;
	77	}
	78	}
	79
	80	//____________________________________________________________________________
	81	AliQADataMaker::AliQADataMaker(const AliQADataMaker& qadm) :
	82	TNamed(qadm.GetName(), qadm.GetTitle()),
	83	fOutput(qadm.fOutput),
	84	fDetectorDir(qadm.fDetectorDir),
	85	fDetectorDirName(qadm.fDetectorDirName),
	86	fCurrentCycle(qadm.fCurrentCycle),
	87	fCycle(qadm.fCycle),
	88	fCycleCounter(qadm.fCycleCounter),
	89	fWriteExpert(qadm.fWriteExpert),
	90	fParameterList(qadm.fParameterList),
	91	fRun(qadm.fRun),
	92	fEventSpecie(qadm.fEventSpecie),
	93	fDigitsArray(NULL)
	94	{
	95	//copy ctor
	96	fDetectorDirName = GetName() ;
	97	for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
	98	fParameterList[specie] = qadm.fParameterList[specie] ;
	99	// fImage[specie] = qadm.fImage[specie] ;
	100	}
	101	}
	102
	103	//____________________________________________________________________________
	104	AliQADataMaker::~AliQADataMaker()
	105	{
	106	for (Int_t esIndex = 0 ; esIndex < AliRecoParam::kNSpecies ; esIndex++) {
	107	if (fParameterList[esIndex] )
	108	delete fParameterList[esIndex] ;
	109	}
	110	delete[] fParameterList ;
	111
	112	if (fDigitsArray) {
	113	fDigitsArray->Clear() ;
	114	delete fDigitsArray ;
	115	}
	116	}
	117
	118	//____________________________________________________________________________
	119	Int_t AliQADataMaker::Add2List(TH1 * hist, const Int_t index, TObjArray ** list, const Bool_t expert, const Bool_t image, const Bool_t saveForCorr)
	120	{
	121	// Set histograms memory resident and add to the list
	122	// Maximm allowed is 10000
	123
	124	Int_t rv = -1 ;
	125	TClass * classType = hist->Class() ;
	126	TString className(classType->GetName()) ;
	127	if( ! className.BeginsWith("T") && ! classType->InheritsFrom("TH1") ) {
	128	AliError(Form("QA data Object must be a generic ROOT object and derive fom TH1 and not %s", className.Data())) ;
	129	} else if ( index > AliQAv1::GetMaxQAObj() ) {
	130	AliError(Form("Max number of authorized QA objects is %d", AliQAv1::GetMaxQAObj())) ;
	131	} else {
	132	hist->SetDirectory(0) ;
	133	if (expert)
	134	hist->SetBit(AliQAv1::GetExpertBit()) ;
	135	if (image)
	136	hist->SetBit(AliQAv1::GetImageBit()) ;
	137	const Char_t * name = Form("%s_%s", AliRecoParam::GetEventSpecieName(fEventSpecie), hist->GetName()) ;
	138	hist->SetName(name) ;
	139	if(saveForCorr) {
	140	const Char_t * cname = Form("%s_%s", list[AliRecoParam::AConvert(AliRecoParam::kDefault)]->GetName(), hist->GetName()) ;
	141	TParameter<double> * p = new TParameter<double>(cname, 9999.9999) ;
	142	if ( fParameterList[AliRecoParam::AConvert(fEventSpecie)] == NULL )
	143	fParameterList[AliRecoParam::AConvert(fEventSpecie)] = new TList() ;
	144	fParameterList[AliRecoParam::AConvert(fEventSpecie)]->Add(p) ;
	145	}
	146	TObject* old = list[AliRecoParam::AConvert(fEventSpecie)]->At(index);
	147	if (old) {
	148	AliError(Form("%s - OUPS ! Already got an object (%p,%s) for index=%d => will most probably get a memory leak by replacing it with (%p,%s) !",
	149	GetName(),old,old->GetName(),index,hist,hist->GetName()));
	150	}
	151	list[AliRecoParam::AConvert(fEventSpecie)]->AddAtAndExpand(hist, index) ;
	152	rv = list[AliRecoParam::AConvert(fEventSpecie)]->GetLast() ;
	153	}
	154	return rv ;
	155	}
	156
	157	//____________________________________________________________________________
	158	TH1 * AliQADataMaker::CloneMe(TH1 * hist, Int_t specie) const
	159	{
	160	// clones a histogram
	161	const Char_t * name = Form("%s_%s", AliRecoParam::GetEventSpecieName(specie), hist->GetName()) ;
	162	TH1 * hClone = static_cast<TH1 *>(hist->Clone(name)) ;
	163	if ( hist->TestBit(AliQAv1::GetExpertBit()) )
	164	hClone->SetBit(AliQAv1::GetExpertBit()) ;
	165	if ( hist->TestBit(AliQAv1::GetImageBit()) )
	166	hClone->SetBit(AliQAv1::GetImageBit()) ;
	167	return hClone ;
	168	}
	169
	170	//____________________________________________________________________________
	171	void AliQADataMaker::DefaultEndOfDetectorCycle(AliQAv1::TASKINDEX_t task)
	172	{
	173	// this method must be oveloaded by detectors
	174	// sets the QA result to Fatal
	175	AliQAv1::Instance(AliQAv1::GetDetIndex(GetName())) ;
	176	AliQAv1 * qa = AliQAv1::Instance(task) ;
	177	for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++)
	178	qa->Set(AliQAv1::kFATAL, specie) ;
	179	AliQAv1::GetQAResultFile()->cd() ;
	180	qa->Write(AliQAv1::GetQAName(), kWriteDelete) ;
	181	AliQAv1::GetQAResultFile()->Close() ;
	182	}
	183
	184	//____________________________________________________________________________
	185	void AliQADataMaker::Finish() const
	186	{
	187	// write to the output File
	188	if (fOutput)
	189	fOutput->Close() ;
	190	}
	191
	192	//____________________________________________________________________________
	193	TObject * AliQADataMaker::GetData(TObjArray ** list, const Int_t index)
	194	{
	195	// Returns the QA object at index. Limit is AliQAv1::GetMaxQAObj()
	196	if ( ! list ) {
	197	AliError("Data list is NULL !!") ;
	198	return NULL ;
	199	}
	200
	201	SetEventSpecie(fEventSpecie) ;
	202	Int_t esindex = AliRecoParam::AConvert(fEventSpecie) ;
	203	TH1 * histClone = NULL ;
	204	TObjArray * arr = list[esindex] ;
	205	if (arr) {
	206	if ( ! arr->GetEntriesFast() ) {
	207	// Initializes the histograms
	208	TString arrName(arr->GetName()) ;
	209	if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kRAWS)))
	210	InitRaws() ;
	211	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kHITS)))
	212	InitHits() ;
	213	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kSDIGITS)))
	214	InitSDigits() ;
	215	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kDIGITS)))
	216	InitDigits() ;
	217	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kDIGITSR)))
	218	InitDigits() ;
	219	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kRECPOINTS)))
	220	InitRecPoints() ;
	221	else if (arrName.Contains(AliQAv1::GetTaskName(AliQAv1::kESDS)))
	222	InitESDs() ;
	223	}
	224	if ( index > AliQAv1::GetMaxQAObj() ) {
	225	AliError(Form("Max number of authorized QA objects is %d", AliQAv1::GetMaxQAObj())) ;
	226	} else {
	227	if ( arr->At(index) ) {
	228	histClone = static_cast<TH1*>(arr->At(index)) ;
	229	}
	230	}
	231	}
	232	return histClone ;
	233	}
	234
	235	//____________________________________________________________________________
	236	TObjArray* AliQADataMaker::Init(AliQAv1::TASKINDEX_t task, AliRecoParam::EventSpecie_t es, Int_t cycles)
	237	{
	238	// Initialializes and returns the QAData list for a given event specie
	239	TObjArray ** ar = Init(task, cycles) ;
	240	return ar[AliRecoParam::AConvert(es)] ;
	241	}
	242
	243	//____________________________________________________________________________
	244	Bool_t AliQADataMaker::IsValidEventSpecie(Int_t eventSpecieIndex, TObjArray ** list)
	245	{
	246	// check if event specie was present in current run or
	247	// if histograms of this event specie have been created
	248	if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(eventSpecieIndex)) \|\| ! list[eventSpecieIndex]->GetEntriesFast() )
	249	return kFALSE ;
	250	else
	251	return kTRUE ;
	252	}
	253