[u/mrichter/AliRoot.git] / ESDCheck / AliTOFQATask.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 TOF data in simulated data
//
//*-- Silvia Arcelli
//////////////////////////////////////////////////////////////////////////////

#include <TChain.h>
#include <TCanvas.h>
#include <TFile.h> 

#include "AliTOFQATask.h" 
#include "AliESD.h" 
#inclued "AliESDtrack.h" 
#include "AliLog.h"

//______________________________________________________________________________
AliTOFQATask::AliTOFQATask(const char *name) : 
  AliAnalysisTask(name,""),  
  fChain(0),
  fESD(0), 
  fhTOF(0),
  fhTOFEnergy(0),
  fhTOFDigits(0),
  fhTOFRecParticles(0),
  fhTOFPhotons(0),
  fhTOFInvariantMass(0),
  fhTOFDigitsEvent(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()) ; 
}

//______________________________________________________________________________
void AliTOFQATask::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 
  fhTOFSector   = new TH1F("hSector",   " TOF TrackRefs, sector # ",  18, 0., 18.) ;
  fhTOFSectorM  = new TH1F("hSectorM",  " TOF Matched, sector # ",    18, 0., 18.) ;
  fhTOFSectorMF = new TH1F("hSectorMF", " TOF Matched G, sector # ",  18, 0., 18.) ;
  fhTOFSectorMG = new TH1F("hSectorMG", " TOF Matched F , sector # ", 18, 0., 18.) ;
  fhTOFprimP    = new TH1F("hprimP",    " TPC mom  tracks",           20, 0., 4.) ;
  fhTOFprimPpi  = new TH1F("hprimPpi",  " TPC mom  tracks",           20, 0., 4.) ;
  fhTOFprimPka  = new TH1F("hprimPka",  " TPC mom  tracks",           20, 0., 4.) ;
  fhTOFprimPpr  = new TH1F("hprimPpr",  " TPC mom  tracks",           20, 0., 4.) ;


   // Reaching TOF,prim
  fhTOFprimPTOF   = new TH1F("hprimPTOF",   " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpi = new TH1F("hprimPTOFpi", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFka = new TH1F("hprimPTOFka", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpr = new TH1F("hprimPTOFpr", " TPC mom  tracks", 20, 0., 4.) ;


   // Well matched,prim
  fhTOFprimPTOF3   = new TH1F("hprimPTOF3",   " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpi3 = new TH1F("hprimPTOFpi3", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFka3 = new TH1F("hprimPTOFka3", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpr3 = new TH1F("hprimPTOFpr3", " TPC mom  tracks", 20, 0., 4.) ;

   // bad matched,prim
  fhTOFprimPTOF4   = new TH1F("hprimPTOF4",   " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpi4 = new TH1F("hprimPTOFpi4", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFka4 = new TH1F("hprimPTOFka4", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpr4 = new TH1F("hprimPTOFpr4", " TPC mom  tracks", 20, 0., 4.) ;

   // matched,prim
  fhTOFprimPTOF34   = new TH1F("hprimPTOF34",   " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpi34 = new TH1F("hprimPTOFpi34", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFka34 = new TH1F("hprimPTOFka34", " TPC mom  tracks", 20, 0., 4.) ;
  fhTOFprimPTOFpr34 = new TH1F("hprimPTOFpr34", " TPC mom  tracks", 20,0 ., 4.) ;
  
  // create output container
  
  fOutputContainer = new TObjArray(24) ; 
  fOutputContainer->SetName(GetName()) ; 

  fOutputContainer->AddAt(fhTOFSector,            0) ; 
  fOutputContainer->AddAt(fhTOFSectorM,           1) ; 
  fOutputContainer->AddAt(fhTOFSectorMF,          2) ; 
  fOutputContainer->AddAt(fhTOFSectorMG,          3) ; 
  fOutputContainer->AddAt(fhTOFprimP,             4) ; 
  fOutputContainer->AddAt(fhTOFprimPpi,           5) ; 
  fOutputContainer->AddAt(fhTOFprimPka,           6) ; 
  fOutputContainer->AddAt(fhTOFprimPpr,           7) ; 
  fOutputContainer->AddAt(fhTOFprimPTOF,          8) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpi,        9) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFka,       10) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpr,       11) ; 
  fOutputContainer->AddAt(fhTOFprimPTOF3,        12) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpi3,      13) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFka3,      14) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpr3,      15) ; 
  fOutputContainer->AddAt(fhTOFprimPTOF4,        16) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpi4,      17) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFka4,      18) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpr4,      19) ; 
  fOutputContainer->AddAt(fhTOFprimPTOF34,       20) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpi34,     21) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFka34,     22) ; 
  fOutputContainer->AddAt(fhTOFprimPTOFpr34,     23) ; 
 
}

//______________________________________________________________________________
void AliTOFQATask::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)) ; 
  
  // ************************  TOF *************************************
  const Int_t knCalinSec = 8736 ;
  
  Int_t ntrk = fESD->GetNumberOfTracks() ;
  while ( ntrk-- ) {
    AliESDtrack * t = fESD->GetTrack(ntrk) ;
    if ( (t->GetStatus() & AliESDtrack::kTIME)==0 )
      continue;
    Int_t label               = TMath::Abs(t->GetLabel()) ;
    Double_t p                = t->GetP() ; 
    UInt_t assignedTOFcluster = t->GetTOFcluster() ;    //index of the assigned TOF cluster, >0 ?
    Int_t detid               = t->GetTOFCalChannel() ; //index of the assigned TOF cluster, >0 ?
    
    Int_t sector = detid / knCalinSec ;
    
    if(assignedTOFcluster){ //matched
      hSectorM->Fill(sector);
    }
  }
 
  PostData(0, fOutputContainer);

  
}

//______________________________________________________________________________
void AliTOFQATask::Terminate(Option_t *)
{
  // Processing when the event loop is ended
  
  // some plots

  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 TOF data in simulated data
	17	//
	18	//*-- Silvia Arcelli
	19	//////////////////////////////////////////////////////////////////////////////
	20
	21	#include <TChain.h>
	22	#include <TCanvas.h>
	23	#include <TFile.h>
	24
	25	#include "AliTOFQATask.h"
	26	#include "AliESD.h"
	27	#inclued "AliESDtrack.h"
	28	#include "AliLog.h"
	29
	30	//______________________________________________________________________________
	31	AliTOFQATask::AliTOFQATask(const char *name) :
	32	AliAnalysisTask(name,""),
	33	fChain(0),
	34	fESD(0),
	35	fhTOF(0),
	36	fhTOFEnergy(0),
	37	fhTOFDigits(0),
	38	fhTOFRecParticles(0),
	39	fhTOFPhotons(0),
	40	fhTOFInvariantMass(0),
	41	fhTOFDigitsEvent(0)
	42	{
	43	// Constructor.
	44	// Input slot #0 works with an Ntuple
	45	DefineInput(0, TChain::Class());
	46	// Output slot #0 writes into a TH1 container
	47	DefineOutput(0, TObjArray::Class()) ;
	48	}
	49
	50	//______________________________________________________________________________
	51	void AliTOFQATask::Init(const Option_t*)
	52	{
	53	// Initialisation of branch container and histograms
	54
	55	AliInfo(Form("*** Initialization of %s", GetName())) ;
	56
	57	// Get input data
	58	fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
	59	if (!fChain) {
	60	AliError(Form("Input 0 for %s not found\n", GetName()));
	61	return ;
	62	}
	63
	64	if (!fESD) {
65	// One should first check if the branch address was taken by some other task
66	char address = (char )GetBranchAddress(0, "ESD") ;
67	if (address)
68	fESD = (AliESD )(address) ;
69	if (!fESD)
70	fChain->SetBranchAddress("ESD", &fESD) ;
71	}
72	// The output objects will be written to
73	TDirectory * cdir = gDirectory ;
74	// Open a file for output #0
75	char outputName[1024] ;
76	sprintf(outputName, "%s.root", GetName() ) ;
77	OpenFile(0, outputName , "RECREATE") ;
78	if (cdir)
79	cdir->cd() ;
80
81	// create histograms
82	fhTOFSector = new TH1F("hSector", " TOF TrackRefs, sector # ", 18, 0., 18.) ;
83	fhTOFSectorM = new TH1F("hSectorM", " TOF Matched, sector # ", 18, 0., 18.) ;
84	fhTOFSectorMF = new TH1F("hSectorMF", " TOF Matched G, sector # ", 18, 0., 18.) ;
85	fhTOFSectorMG = new TH1F("hSectorMG", " TOF Matched F , sector # ", 18, 0., 18.) ;
86	fhTOFprimP = new TH1F("hprimP", " TPC mom tracks", 20, 0., 4.) ;
87	fhTOFprimPpi = new TH1F("hprimPpi", " TPC mom tracks", 20, 0., 4.) ;
88	fhTOFprimPka = new TH1F("hprimPka", " TPC mom tracks", 20, 0., 4.) ;
89	fhTOFprimPpr = new TH1F("hprimPpr", " TPC mom tracks", 20, 0., 4.) ;
90
91
92	// Reaching TOF,prim
93	fhTOFprimPTOF = new TH1F("hprimPTOF", " TPC mom tracks", 20, 0., 4.) ;
94	fhTOFprimPTOFpi = new TH1F("hprimPTOFpi", " TPC mom tracks", 20, 0., 4.) ;
95	fhTOFprimPTOFka = new TH1F("hprimPTOFka", " TPC mom tracks", 20, 0., 4.) ;
96	fhTOFprimPTOFpr = new TH1F("hprimPTOFpr", " TPC mom tracks", 20, 0., 4.) ;
97
98
99	// Well matched,prim
100	fhTOFprimPTOF3 = new TH1F("hprimPTOF3", " TPC mom tracks", 20, 0., 4.) ;
101	fhTOFprimPTOFpi3 = new TH1F("hprimPTOFpi3", " TPC mom tracks", 20, 0., 4.) ;
102	fhTOFprimPTOFka3 = new TH1F("hprimPTOFka3", " TPC mom tracks", 20, 0., 4.) ;
103	fhTOFprimPTOFpr3 = new TH1F("hprimPTOFpr3", " TPC mom tracks", 20, 0., 4.) ;
104
105	// bad matched,prim
106	fhTOFprimPTOF4 = new TH1F("hprimPTOF4", " TPC mom tracks", 20, 0., 4.) ;
107	fhTOFprimPTOFpi4 = new TH1F("hprimPTOFpi4", " TPC mom tracks", 20, 0., 4.) ;
108	fhTOFprimPTOFka4 = new TH1F("hprimPTOFka4", " TPC mom tracks", 20, 0., 4.) ;
109	fhTOFprimPTOFpr4 = new TH1F("hprimPTOFpr4", " TPC mom tracks", 20, 0., 4.) ;
110
111	// matched,prim
112	fhTOFprimPTOF34 = new TH1F("hprimPTOF34", " TPC mom tracks", 20, 0., 4.) ;
113	fhTOFprimPTOFpi34 = new TH1F("hprimPTOFpi34", " TPC mom tracks", 20, 0., 4.) ;
114	fhTOFprimPTOFka34 = new TH1F("hprimPTOFka34", " TPC mom tracks", 20, 0., 4.) ;
115	fhTOFprimPTOFpr34 = new TH1F("hprimPTOFpr34", " TPC mom tracks", 20,0 ., 4.) ;
116
117	// create output container
118
119	fOutputContainer = new TObjArray(24) ;
120	fOutputContainer->SetName(GetName()) ;
121
122	fOutputContainer->AddAt(fhTOFSector, 0) ;
123	fOutputContainer->AddAt(fhTOFSectorM, 1) ;
124	fOutputContainer->AddAt(fhTOFSectorMF, 2) ;
125	fOutputContainer->AddAt(fhTOFSectorMG, 3) ;
126	fOutputContainer->AddAt(fhTOFprimP, 4) ;
127	fOutputContainer->AddAt(fhTOFprimPpi, 5) ;
128	fOutputContainer->AddAt(fhTOFprimPka, 6) ;
129	fOutputContainer->AddAt(fhTOFprimPpr, 7) ;
130	fOutputContainer->AddAt(fhTOFprimPTOF, 8) ;
131	fOutputContainer->AddAt(fhTOFprimPTOFpi, 9) ;
132	fOutputContainer->AddAt(fhTOFprimPTOFka, 10) ;
133	fOutputContainer->AddAt(fhTOFprimPTOFpr, 11) ;
134	fOutputContainer->AddAt(fhTOFprimPTOF3, 12) ;
135	fOutputContainer->AddAt(fhTOFprimPTOFpi3, 13) ;
136	fOutputContainer->AddAt(fhTOFprimPTOFka3, 14) ;
137	fOutputContainer->AddAt(fhTOFprimPTOFpr3, 15) ;
138	fOutputContainer->AddAt(fhTOFprimPTOF4, 16) ;
139	fOutputContainer->AddAt(fhTOFprimPTOFpi4, 17) ;
140	fOutputContainer->AddAt(fhTOFprimPTOFka4, 18) ;
141	fOutputContainer->AddAt(fhTOFprimPTOFpr4, 19) ;
142	fOutputContainer->AddAt(fhTOFprimPTOF34, 20) ;
143	fOutputContainer->AddAt(fhTOFprimPTOFpi34, 21) ;
144	fOutputContainer->AddAt(fhTOFprimPTOFka34, 22) ;
145	fOutputContainer->AddAt(fhTOFprimPTOFpr34, 23) ;
146
147	}
148
149	//______________________________________________________________________________
150	void AliTOFQATask::Exec(Option_t *)
151	{
152	// Processing of one event
153
154	Long64_t entry = fChain->GetReadEntry() ;
155
156	if (!fESD) {
157	AliError("fESD is not connected to the input!") ;
158	return ;
159	}
160
161	if ( !((entry-1)%100) )
162	AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
163
164	// ********************** TOF ***********************************
165	const Int_t knCalinSec = 8736 ;
166
167	Int_t ntrk = fESD->GetNumberOfTracks() ;
168	while ( ntrk-- ) {
169	AliESDtrack * t = fESD->GetTrack(ntrk) ;
170	if ( (t->GetStatus() & AliESDtrack::kTIME)==0 )
171	continue;
172	Int_t label = TMath::Abs(t->GetLabel()) ;
173	Double_t p = t->GetP() ;
174	UInt_t assignedTOFcluster = t->GetTOFcluster() ; //index of the assigned TOF cluster, >0 ?
175	Int_t detid = t->GetTOFCalChannel() ; //index of the assigned TOF cluster, >0 ?
176
177	Int_t sector = detid / knCalinSec ;
178
179	if(assignedTOFcluster){ //matched
180	hSectorM->Fill(sector);
181	}
182	}
183
184	PostData(0, fOutputContainer);
185
186
187	}
188
189	//______________________________________________________________________________
190	void AliTOFQATask::Terminate(Option_t *)
191	{
192	// Processing when the event loop is ended
193
194	// some plots
195
196	char line[1024] ;
197	sprintf(line, ".!tar -zcvf %s.tar.gz *.eps", GetName()) ;
198	gROOT->ProcessLine(line);
199	sprintf(line, ".!rm -fR *.eps");
200	gROOT->ProcessLine(line);
201
202	AliInfo(Form("!!! All the eps files are in %s.tar.gz !!! \n", GetName())) ;
203	}