[u/mrichter/AliRoot.git] / ESDCheck / AliPHOSQATask.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.                  *
 **************************************************************************/
//_________________________________________________________________________
// An analysis task to check the PHOS photon data in simulated data
//
//*-- Yves Schutz 
//////////////////////////////////////////////////////////////////////////////

#include <TChain.h>
#include <TH1.h>
#include <TH1F.h>
#include <TH1I.h>
#include <TNtuple.h>
#include <TCanvas.h>
#include <TLegend.h> 
#include <TVector3.h> 
#include <TFile.h> 

#include "AliPHOSQATask.h" 
#include "AliESD.h" 
#include "AliLog.h"

//______________________________________________________________________________
AliPHOSQATask::AliPHOSQATask(const char *name) : 
  AliAnalysisTask(name,""),  
  fChain(0),
  fESD(0), 
  fhPHOS(0),
  fhPHOSEnergy(0),
  fhPHOSDigits(0),
  fhPHOSRecParticles(0),
  fhPHOSPhotons(0),
  fhPHOSInvariantMass(0),
  fhPHOSDigitsEvent(0)
{
  // Constructor.
  // Input slot #0 works with an Ntuple
  DefineInput(0, TChain::Class());
  // Output slot #0 writes into a TH1 container
  DefineOutput(0,  TObjArray::Class()) ; 
}

//______________________________________________________________________________
AliPHOSQATask::~AliPHOSQATask()
{
  // dtor
  fOutputContainer->Clear() ; 
  delete fOutputContainer ;

  delete fhPHOSPos ;
  delete fhPHOS ;
  delete fhPHOSEnergy ;
  delete fhPHOSDigits ;
  delete fhPHOSRecParticles ;
  delete fhPHOSPhotons ;
  delete fhPHOSInvariantMass ;
  delete fhPHOSDigitsEvent ;
}


//______________________________________________________________________________
void AliPHOSQATask::Init(const Option_t*)
{
  // Initialisation of branch container and histograms 
    
  AliInfo(Form("*** Initialization of %s", GetName())) ; 
  
  // Get input data
  fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
  if (!fChain) {
    AliError(Form("Input 0 for %s not found\n", GetName()));
    return ;
  }
  
  if (!fESD) {
    // One should first check if the branch address was taken by some other task
    char ** address = (char **)GetBranchAddress(0, "ESD") ;
    if (address) 
      fESD = (AliESD *)(*address) ; 
    if (!fESD) 
      fChain->SetBranchAddress("ESD", &fESD) ;  
  }
  // The output objects will be written to 
  TDirectory * cdir = gDirectory ; 
  // Open a file for output #0
  char outputName[1024] ; 
  sprintf(outputName, "%s.root", GetName() ) ; 
  OpenFile(0, outputName , "RECREATE") ; 
  if (cdir) 
    cdir->cd() ; 
  
  // create histograms 
  
  fhPHOSPos            = new TNtuple("PHOSPos"         , "Position in PHOS"  , "x:y:z");
  fhPHOS               = new TNtuple("PHOS"            , "PHOS"  , "event:digits:clusters:photons");
  fhPHOSEnergy         = new TH1D("PHOSEnergy"         , "PHOSEnergy"        , 1000, 0., 10. ) ;
  fhPHOSDigits         = new TH1I("PHOSDigitsCluster"  , "PHOSDigits"        , 20 , 0 , 20  ) ;
  fhPHOSRecParticles   = new TH1D("PHOSRecParticles"   , "PHOSRecParticles" , 20 , 0., 20. ) ;
  fhPHOSPhotons        = new TH1I("PHOSPhotons"        , "PHOSPhotons"       , 20 , 0 , 20  ) ;
  fhPHOSInvariantMass  = new TH1D("PHOSInvariantMass"  , "PHOSInvariantMass" , 400, 0., 400.) ;
  fhPHOSDigitsEvent    = new TH1I("PHOSDigitsEvent"    , "PHOSDigitsEvent"   , 30 , 0 , 30  ) ;
  
  // create output container
  
  fOutputContainer = new TObjArray(8) ; 
  fOutputContainer->SetName(GetName()) ; 

  fOutputContainer->AddAt(fhPHOSPos,             0) ; 
  fOutputContainer->AddAt(fhPHOS,                1) ; 
  fOutputContainer->AddAt(fhPHOSEnergy,          2) ; 
  fOutputContainer->AddAt(fhPHOSDigits,          3) ; 
  fOutputContainer->AddAt(fhPHOSRecParticles,    4) ; 
  fOutputContainer->AddAt(fhPHOSPhotons,         5) ; 
  fOutputContainer->AddAt(fhPHOSInvariantMass,   6) ; 
  fOutputContainer->AddAt(fhPHOSDigitsEvent,     7) ; 
}

//______________________________________________________________________________
void AliPHOSQATask::Exec(Option_t *) 
{
  // Processing of one event
    
  Long64_t entry = fChain->GetReadEntry() ;
  
  if (!fESD) {
    AliError("fESD is not connected to the input!") ; 
    return ; 
  }
  
  if ( !((entry-1)%100) ) 
    AliInfo(Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ; 
  
  //************************  PHOS *************************************
      
  Int_t       firstPhosCluster       = fESD->GetFirstPHOSCluster() ;
  const Int_t numberOfPhosClusters   = fESD->GetNumberOfPHOSClusters() ;
  
  TVector3 ** phosVector       = new TVector3*[numberOfPhosClusters] ;
  Float_t  * phosPhotonsEnergy = new Float_t[numberOfPhosClusters] ;
  Int_t      phosCluster ; 
  Int_t      numberOfPhotonsInPhos  = 0 ;
  Int_t      numberOfDigitsInPhos   = 0 ;
  
  // loop over the PHOS Cluster
  for(phosCluster = firstPhosCluster ; phosCluster < firstPhosCluster + numberOfPhosClusters ; phosCluster++) {
    AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(phosCluster) ;
    if (caloCluster) {
      Float_t pos[3] ;
      caloCluster->GetGlobalPosition( pos ) ;
      fhPHOSEnergy->Fill( caloCluster->GetClusterEnergy() ) ;
      fhPHOSPos->Fill( pos[0], pos[1], pos[2] ) ;
      fhPHOSDigits->Fill(entry, caloCluster->GetNumberOfDigits() ) ;
      numberOfDigitsInPhos += caloCluster->GetNumberOfDigits() ;
      Float_t * pid = caloCluster->GetPid() ;
      if(pid[AliPID::kPhoton] > 0.9) {
	phosVector[numberOfPhotonsInPhos] = new TVector3(pos[0],pos[1],pos[2]) ;
	phosPhotonsEnergy[numberOfPhotonsInPhos]=caloCluster->GetClusterEnergy() ;
	numberOfPhotonsInPhos++;
      }
    }
  } //PHOS clusters
    
  fhPHOSRecParticles->Fill(numberOfPhosClusters);
  fhPHOSPhotons->Fill(numberOfPhotonsInPhos);
  fhPHOSDigitsEvent->Fill(numberOfDigitsInPhos);
  fhPHOS->Fill(entry, numberOfDigitsInPhos, numberOfPhosClusters, numberOfPhotonsInPhos) ; 

  // invariant Mass
  if (numberOfPhotonsInPhos > 1 ) {
    Int_t phosPhoton1, phosPhoton2 ; 
    for(phosPhoton1 = 0 ; phosPhoton1 < numberOfPhotonsInPhos ; phosPhoton1++) {
      for(phosPhoton2 = phosPhoton1 + 1 ; phosPhoton2 < numberOfPhotonsInPhos ; phosPhoton2++) {      
	Float_t tempMass = TMath::Sqrt( 2 * phosPhotonsEnergy[phosPhoton1] * phosPhotonsEnergy[phosPhoton2] *
					( 1 - TMath::Cos(phosVector[phosPhoton1]->Angle(*phosVector[phosPhoton2])) ) 
					);
	fhPHOSInvariantMass->Fill(tempMass*1000.);
      }
    }    
  }
  
  PostData(0, fOutputContainer);

  delete [] phosVector ; 
  delete [] phosPhotonsEnergy ; 
  
}

//______________________________________________________________________________
void AliPHOSQATask::Terminate(Option_t *)
{
  // Processing when the event loop is ended
  
  printf("PHOSEnergy Mean         : %5.3f , RMS : %5.3f \n", fhPHOSEnergy->GetMean(),         fhPHOSEnergy->GetRMS()         ) ;
  printf("PHOSDigits Mean         : %5.3f , RMS : %5.3f \n", fhPHOSDigits->GetMean(),         fhPHOSDigits->GetRMS()         ) ;
  printf("PHOSRecParticles Mean   : %5.3f , RMS : %5.3f \n", fhPHOSRecParticles->GetMean(),   fhPHOSRecParticles->GetRMS()   ) ;
  printf("PHOSPhotons Mean        : %5.3f , RMS : %5.3f \n", fhPHOSPhotons->GetMean(),        fhPHOSPhotons->GetRMS()        ) ;
  printf("PHOSInvariantMass Mean  : %5.3f , RMS : %5.3f \n", fhPHOSInvariantMass->GetMean(),  fhPHOSInvariantMass->GetRMS()  ) ;
  printf("PHOSDigitsEvent Mean    : %5.3f , RMS : %5.3f \n", fhPHOSDigitsEvent->GetMean(),    fhPHOSDigitsEvent->GetRMS()    ) ;

  TCanvas  * cPHOS = new TCanvas("cPHOS", "PHOS ESD Test", 400, 10, 600, 700) ;
  cPHOS->Divide(3, 2);

  cPHOS->cd(1) ; 
  gPad->SetLogy();
  fhPHOSEnergy->SetAxisRange(0, 25.);
  fhPHOSEnergy->SetLineColor(2);
  fhPHOSEnergy->Draw();

  cPHOS->cd(2) ; 
  fhPHOSDigits->SetAxisRange(0,25.);
  fhPHOSDigits->SetLineColor(2);
  fhPHOSDigits->Draw();

  cPHOS->cd(3) ; 
  gPad->SetLogy();
  fhPHOSRecParticles->SetAxisRange(0, 25.);
  fhPHOSRecParticles->SetLineColor(2);
  fhPHOSRecParticles->Draw();

  cPHOS->cd(4) ; 
  gPad->SetLogy();
  fhPHOSPhotons->SetAxisRange(0,25.);
  fhPHOSPhotons->SetLineColor(2);
  fhPHOSPhotons->Draw();

  cPHOS->cd(5) ; 
  fhPHOSInvariantMass->SetLineColor(2);
  fhPHOSInvariantMass->Draw();
 
  cPHOS->cd(6) ; 
  gPad->SetLogy();
  fhPHOSDigitsEvent->SetAxisRange(0,40.);
  fhPHOSDigitsEvent->SetLineColor(2);
  fhPHOSDigitsEvent->Draw();
 
  cPHOS->Print("PHOS.eps");
 
  char line[1024] ; 
  sprintf(line, ".!tar -zcvf %s.tar.gz *.eps", GetName()) ; 
  gROOT->ProcessLine(line);
  sprintf(line, ".!rm -fR *.eps"); 
  gROOT->ProcessLine(line);
 
  AliInfo(Form("!!! All the eps files are in %s.tar.gz !!! \n", GetName())) ;
}
Commit	Line	Data
1dfe075f	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	// An analysis task to check the PHOS photon data in simulated data
	17	//
	18	//*-- Yves Schutz
	19	//////////////////////////////////////////////////////////////////////////////
	20
	21	#include <TChain.h>
	22	#include <TH1.h>
	23	#include <TH1F.h>
	24	#include <TH1I.h>
	25	#include <TNtuple.h>
	26	#include <TCanvas.h>
	27	#include <TLegend.h>
	28	#include <TVector3.h>
	29	#include <TFile.h>
	30
	31	#include "AliPHOSQATask.h"
	32	#include "AliESD.h"
	33	#include "AliLog.h"
	34
	35	//______________________________________________________________________________
	36	AliPHOSQATask::AliPHOSQATask(const char *name) :
	37	AliAnalysisTask(name,""),
	38	fChain(0),
	39	fESD(0),
	40	fhPHOS(0),
	41	fhPHOSEnergy(0),
	42	fhPHOSDigits(0),
	43	fhPHOSRecParticles(0),
	44	fhPHOSPhotons(0),
	45	fhPHOSInvariantMass(0),
	46	fhPHOSDigitsEvent(0)
	47	{
	48	// Constructor.
	49	// Input slot #0 works with an Ntuple
	50	DefineInput(0, TChain::Class());
	51	// Output slot #0 writes into a TH1 container
	52	DefineOutput(0, TObjArray::Class()) ;
	53	}
	54
	55	//______________________________________________________________________________
	56	AliPHOSQATask::~AliPHOSQATask()
	57	{
	58	// dtor
	59	fOutputContainer->Clear() ;
	60	delete fOutputContainer ;
	61
	62	delete fhPHOSPos ;
	63	delete fhPHOS ;
	64	delete fhPHOSEnergy ;
65	delete fhPHOSDigits ;
66	delete fhPHOSRecParticles ;
67	delete fhPHOSPhotons ;
68	delete fhPHOSInvariantMass ;
69	delete fhPHOSDigitsEvent ;
70	}
71
72
73	//______________________________________________________________________________
74	void AliPHOSQATask::Init(const Option_t*)
75	{
76	// Initialisation of branch container and histograms
77
78	AliInfo(Form("*** Initialization of %s", GetName())) ;
79
80	// Get input data
81	fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
82	if (!fChain) {
83	AliError(Form("Input 0 for %s not found\n", GetName()));
84	return ;
85	}
86
87	if (!fESD) {
88	// One should first check if the branch address was taken by some other task
89	char address = (char )GetBranchAddress(0, "ESD") ;
90	if (address)
91	fESD = (AliESD )(address) ;
92	if (!fESD)
93	fChain->SetBranchAddress("ESD", &fESD) ;
94	}
95	// The output objects will be written to
96	TDirectory * cdir = gDirectory ;
97	// Open a file for output #0
98	char outputName[1024] ;
99	sprintf(outputName, "%s.root", GetName() ) ;
100	OpenFile(0, outputName , "RECREATE") ;
101	if (cdir)
102	cdir->cd() ;
103
104	// create histograms
105
106	fhPHOSPos = new TNtuple("PHOSPos" , "Position in PHOS" , "x:y:z");
107	fhPHOS = new TNtuple("PHOS" , "PHOS" , "event:digits:clusters:photons");
108	fhPHOSEnergy = new TH1D("PHOSEnergy" , "PHOSEnergy" , 1000, 0., 10. ) ;
109	fhPHOSDigits = new TH1I("PHOSDigitsCluster" , "PHOSDigits" , 20 , 0 , 20 ) ;
110	fhPHOSRecParticles = new TH1D("PHOSRecParticles" , "PHOSRecParticles" , 20 , 0., 20. ) ;
111	fhPHOSPhotons = new TH1I("PHOSPhotons" , "PHOSPhotons" , 20 , 0 , 20 ) ;
112	fhPHOSInvariantMass = new TH1D("PHOSInvariantMass" , "PHOSInvariantMass" , 400, 0., 400.) ;
113	fhPHOSDigitsEvent = new TH1I("PHOSDigitsEvent" , "PHOSDigitsEvent" , 30 , 0 , 30 ) ;
114
115	// create output container
116
117	fOutputContainer = new TObjArray(8) ;
118	fOutputContainer->SetName(GetName()) ;
119
120	fOutputContainer->AddAt(fhPHOSPos, 0) ;
121	fOutputContainer->AddAt(fhPHOS, 1) ;
122	fOutputContainer->AddAt(fhPHOSEnergy, 2) ;
123	fOutputContainer->AddAt(fhPHOSDigits, 3) ;
124	fOutputContainer->AddAt(fhPHOSRecParticles, 4) ;
125	fOutputContainer->AddAt(fhPHOSPhotons, 5) ;
126	fOutputContainer->AddAt(fhPHOSInvariantMass, 6) ;
127	fOutputContainer->AddAt(fhPHOSDigitsEvent, 7) ;
128	}
129
130	//______________________________________________________________________________
131	void AliPHOSQATask::Exec(Option_t *)
132	{
133	// Processing of one event
134
135	Long64_t entry = fChain->GetReadEntry() ;
136
137	if (!fESD) {
138	AliError("fESD is not connected to the input!") ;
139	return ;
140	}
141
142	if ( !((entry-1)%100) )
143	AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
144
145	//********************** PHOS ***********************************
146
147	Int_t firstPhosCluster = fESD->GetFirstPHOSCluster() ;
148	const Int_t numberOfPhosClusters = fESD->GetNumberOfPHOSClusters() ;
149
150	TVector3 ** phosVector = new TVector3*[numberOfPhosClusters] ;
151	Float_t * phosPhotonsEnergy = new Float_t[numberOfPhosClusters] ;
152	Int_t phosCluster ;
153	Int_t numberOfPhotonsInPhos = 0 ;
154	Int_t numberOfDigitsInPhos = 0 ;
155
156	// loop over the PHOS Cluster
157	for(phosCluster = firstPhosCluster ; phosCluster < firstPhosCluster + numberOfPhosClusters ; phosCluster++) {
158	AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(phosCluster) ;
159	if (caloCluster) {
160	Float_t pos[3] ;
161	caloCluster->GetGlobalPosition( pos ) ;
162	fhPHOSEnergy->Fill( caloCluster->GetClusterEnergy() ) ;
163	fhPHOSPos->Fill( pos[0], pos[1], pos[2] ) ;
164	fhPHOSDigits->Fill(entry, caloCluster->GetNumberOfDigits() ) ;
165	numberOfDigitsInPhos += caloCluster->GetNumberOfDigits() ;
166	Float_t * pid = caloCluster->GetPid() ;
167	if(pid[AliPID::kPhoton] > 0.9) {
168	phosVector[numberOfPhotonsInPhos] = new TVector3(pos[0],pos[1],pos[2]) ;
169	phosPhotonsEnergy[numberOfPhotonsInPhos]=caloCluster->GetClusterEnergy() ;
170	numberOfPhotonsInPhos++;
171	}
172	}
173	} //PHOS clusters
174
175	fhPHOSRecParticles->Fill(numberOfPhosClusters);
176	fhPHOSPhotons->Fill(numberOfPhotonsInPhos);
177	fhPHOSDigitsEvent->Fill(numberOfDigitsInPhos);
178	fhPHOS->Fill(entry, numberOfDigitsInPhos, numberOfPhosClusters, numberOfPhotonsInPhos) ;
179
180	// invariant Mass
181	if (numberOfPhotonsInPhos > 1 ) {
182	Int_t phosPhoton1, phosPhoton2 ;
183	for(phosPhoton1 = 0 ; phosPhoton1 < numberOfPhotonsInPhos ; phosPhoton1++) {
184	for(phosPhoton2 = phosPhoton1 + 1 ; phosPhoton2 < numberOfPhotonsInPhos ; phosPhoton2++) {
185	Float_t tempMass = TMath::Sqrt( 2 * phosPhotonsEnergy[phosPhoton1] * phosPhotonsEnergy[phosPhoton2] *
186	( 1 - TMath::Cos(phosVector[phosPhoton1]->Angle(*phosVector[phosPhoton2])) )
187	);
188	fhPHOSInvariantMass->Fill(tempMass*1000.);
189	}
190	}
191	}
192
193	PostData(0, fOutputContainer);
194
195	delete [] phosVector ;
196	delete [] phosPhotonsEnergy ;
197
198	}
199
200	//______________________________________________________________________________
201	void AliPHOSQATask::Terminate(Option_t *)
202	{
203	// Processing when the event loop is ended
204
205	printf("PHOSEnergy Mean : %5.3f , RMS : %5.3f \n", fhPHOSEnergy->GetMean(), fhPHOSEnergy->GetRMS() ) ;
206	printf("PHOSDigits Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigits->GetMean(), fhPHOSDigits->GetRMS() ) ;
207	printf("PHOSRecParticles Mean : %5.3f , RMS : %5.3f \n", fhPHOSRecParticles->GetMean(), fhPHOSRecParticles->GetRMS() ) ;
208	printf("PHOSPhotons Mean : %5.3f , RMS : %5.3f \n", fhPHOSPhotons->GetMean(), fhPHOSPhotons->GetRMS() ) ;
209	printf("PHOSInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhPHOSInvariantMass->GetMean(), fhPHOSInvariantMass->GetRMS() ) ;
210	printf("PHOSDigitsEvent Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigitsEvent->GetMean(), fhPHOSDigitsEvent->GetRMS() ) ;
211
212	TCanvas * cPHOS = new TCanvas("cPHOS", "PHOS ESD Test", 400, 10, 600, 700) ;
213	cPHOS->Divide(3, 2);
214
215	cPHOS->cd(1) ;
216	gPad->SetLogy();
217	fhPHOSEnergy->SetAxisRange(0, 25.);
218	fhPHOSEnergy->SetLineColor(2);
219	fhPHOSEnergy->Draw();
220
221	cPHOS->cd(2) ;
222	fhPHOSDigits->SetAxisRange(0,25.);
223	fhPHOSDigits->SetLineColor(2);
224	fhPHOSDigits->Draw();
225
226	cPHOS->cd(3) ;
227	gPad->SetLogy();
228	fhPHOSRecParticles->SetAxisRange(0, 25.);
229	fhPHOSRecParticles->SetLineColor(2);
230	fhPHOSRecParticles->Draw();
231
232	cPHOS->cd(4) ;
233	gPad->SetLogy();
234	fhPHOSPhotons->SetAxisRange(0,25.);
235	fhPHOSPhotons->SetLineColor(2);
236	fhPHOSPhotons->Draw();
237
238	cPHOS->cd(5) ;
239	fhPHOSInvariantMass->SetLineColor(2);
240	fhPHOSInvariantMass->Draw();
241
242	cPHOS->cd(6) ;
243	gPad->SetLogy();
244	fhPHOSDigitsEvent->SetAxisRange(0,40.);
245	fhPHOSDigitsEvent->SetLineColor(2);
246	fhPHOSDigitsEvent->Draw();
247
248	cPHOS->Print("PHOS.eps");
249
250	char line[1024] ;
251	sprintf(line, ".!tar -zcvf %s.tar.gz *.eps", GetName()) ;
252	gROOT->ProcessLine(line);
253	sprintf(line, ".!rm -fR *.eps");
254	gROOT->ProcessLine(line);
255
256	AliInfo(Form("!!! All the eps files are in %s.tar.gz !!! \n", GetName())) ;
257	}