added new class for analysis
[u/mrichter/AliRoot.git] / ESDCheck / AliT0QATask.cxx
CommitLineData
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 **************************************************************************/
0b28fd57 15
16/* $Id$ */
17
1dfe075f 18//______________________________________________________________________________
19// An analysis task to check the T0 data in simulated data
20//
21//*-- Alla Maevskaya
22//////////////////////////////////////////////////////////////////////////////
23
0b28fd57 24#include <TCanvas.h>
1dfe075f 25#include <TChain.h>
0b28fd57 26#include <TFile.h>
1dfe075f 27#include <TH1F.h>
1dfe075f 28#include <TLegend.h>
0b28fd57 29#include <TROOT.h>
1dfe075f 30
31#include "AliT0QATask.h"
32#include "AliESD.h"
33#include "AliLog.h"
34#include "AliESDVertex.h"
35
36//______________________________________________________________________________
37AliT0QATask::AliT0QATask(const char *name) :
38 AliAnalysisTask(name,""),
39 fChain(0),
40 fESD(0),
41 fhT01(0),
42 fhT02(0),
43 fhT03(0)
44{
45 // Constructor.
46 // Input slot #0 works with an Ntuple
47 DefineInput(0, TChain::Class());
48 // Output slot #0 writes into a TH1 container
49 DefineOutput(0, TObjArray::Class()) ;
50}
51
52//______________________________________________________________________________
53AliT0QATask::~AliT0QATask()
54{
55 // dtor
56 fOutputContainer->Clear() ;
57 delete fOutputContainer ;
58
59 delete fhT01 ;
60 delete fhT02 ;
61 delete fhT03 ;
62}
63
64//______________________________________________________________________________
c52c2132 65void AliT0QATask::ConnectInputData(const Option_t*)
1dfe075f 66{
67 // Initialisation of branch container and histograms
68
69 AliInfo(Form("*** Initialization of %s", GetName())) ;
70
71 // Get input data
72 fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
73 if (!fChain) {
74 AliError(Form("Input 0 for %s not found\n", GetName()));
75 return ;
76 }
77
c52c2132 78 // One should first check if the branch address was taken by some other task
79 char ** address = (char **)GetBranchAddress(0, "ESD");
80 if (address) {
81 fESD = (AliESD*)(*address);
82 } else {
83 fESD = new AliESD();
84 SetBranchAddress(0, "ESD", &fESD);
1dfe075f 85 }
c52c2132 86}
1dfe075f 87
c52c2132 88//________________________________________________________________________
89void AliT0QATask::CreateOutputObjects()
90{
91 // create histograms
1e20f195 92
93 OpenFile(0) ;
94
1dfe075f 95 fhT01 = new TH1F("hRealVertex", "Primary vertex", 100, -20, 20);
96 fhT02 = new TH1F("hT0start", "T0 start time", 100, 12400, 12600);
97 fhT03 = new TH1F("hT0vertex", "T0vertex", 100, -20, 20);
98
99
100 // create output container
101
102 fOutputContainer = new TObjArray(3) ;
103 fOutputContainer->SetName(GetName()) ;
104
105 fOutputContainer->AddAt(fhT01, 0) ;
106 fOutputContainer->AddAt(fhT02, 1) ;
107 fOutputContainer->AddAt(fhT03, 2) ;
108}
109
110//______________________________________________________________________________
111void AliT0QATask::Exec(Option_t *)
112{
113 // Processing of one event
114
115 Long64_t entry = fChain->GetReadEntry() ;
116
117 if (!fESD) {
118 AliError("fESD is not connected to the input!") ;
119 return ;
120 }
121
122 if ( !((entry-1)%100) )
123 AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
124
125 // ************************ T0 *************************************
126
127 const AliESDVertex * vertex = fESD->GetPrimaryVertex() ;
128 Double_t posZ = vertex->GetZv() ;
129 fhT01->Fill( posZ ) ;
130
131 fhT02->Fill( fESD->GetT0() ) ;
132
133 fhT03->Fill( fESD->GetT0zVertex() / 2. ) ;
134
135 PostData(0, fOutputContainer);
136
137}
138
139//______________________________________________________________________________
140void AliT0QATask::Terminate(Option_t *)
141{
142 // Processing when the event loop is ended
c52c2132 143 fOutputContainer = (TObjArray*)GetOutputData(0);
144 fhT01 = (TH1F*)fOutputContainer->At(0);
145 fhT02 = (TH1F*)fOutputContainer->At(1);
146 fhT03 = (TH1F*)fOutputContainer->At(2);
147
2704006a 148 Bool_t problem = kFALSE ;
84eb42a1 149 AliInfo(Form(" *** %s Report:", GetName())) ;
150
1dfe075f 151 Float_t mean = fhT02->GetMean();
152
153 printf ("mean time T0 ps %f\n", mean) ;
154
2704006a 155 if ( mean > 12600 || mean < 12400 ) {
1dfe075f 156 AliWarning (" !!!!!!!!!!-----events sample is WRONG - T0 unreal -------");
2704006a 157 problem = kTRUE ;
158 }
1dfe075f 159 TCanvas * cTO1 = new TCanvas("cT01", "T0 ESD Test", 400, 10, 600, 700) ;
160 cTO1->Divide(2, 2) ;
161
162 cTO1->cd(1) ;
163 fhT01->Draw() ;
164
165 cTO1->cd(2) ;
166 fhT02->Draw() ;
167
168 cTO1->cd(3) ;
169 fhT03->Draw() ;
170
171
172 cTO1->Print("T0.eps");
173
174 char line[1024] ;
84eb42a1 175 sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
1dfe075f 176 gROOT->ProcessLine(line);
177 sprintf(line, ".!rm -fR *.eps");
178 gROOT->ProcessLine(line);
179
2704006a 180 AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
181
182 char * report ;
183 if(problem)
184 report="Problems found, please check!!!";
185 else
186 report="OK";
187
188 AliInfo(Form("*** %s Summary Report: %s\n",GetName(), report)) ;
1dfe075f 189}