Fixes for building of DA (Anshul)
[u/mrichter/AliRoot.git] / TPC / AliTPCAnalysisTaskcalib.cxx
CommitLineData
73d51118 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///////////////////////////////////////////////////////////////////////////////
18// //
fc492d0d 19// ANALYSIS task to perrorm TPC calibration //
20
73d51118 21// //
22///////////////////////////////////////////////////////////////////////////////
23#include "AliTPCAnalysisTaskcalib.h"
24#include "TChain.h"
25#include "AliTPCcalibBase.h"
26#include "AliESDEvent.h"
27#include "AliESDfriend.h"
f7f33dec 28#include "AliESDtrack.h"
73d51118 29#include "AliESDfriendTrack.h"
30#include "AliTPCseed.h"
31#include "AliESDInputHandler.h"
32#include "AliAnalysisManager.h"
679480d0 33#include "TFile.h"
34#include "TSystem.h"
35#include "TTimeStamp.h"
73d51118 36
37ClassImp(AliTPCAnalysisTaskcalib)
38
f7f33dec 39
40AliTPCAnalysisTaskcalib::AliTPCAnalysisTaskcalib()
41 :AliAnalysisTask(),
42 fCalibJobs(0),
43 fESD(0),
44 fESDfriend(0),
5647625c 45 fDebugOutputPath("")
f7f33dec 46{
47 //
48 // default constructor
49 //
50
51}
52
53
73d51118 54AliTPCAnalysisTaskcalib::AliTPCAnalysisTaskcalib(const char *name)
55 :AliAnalysisTask(name,""),
56 fCalibJobs(0),
0db06ea6 57 fESD(0),
f7f33dec 58 fESDfriend(0),
5647625c 59 fDebugOutputPath("")
73d51118 60{
61 //
62 // Constructor
63 //
64 DefineInput(0, TChain::Class());
65 DefineOutput(0, TObjArray::Class());
f7f33dec 66 fCalibJobs = new TObjArray(0);
42a48f93 67 fCalibJobs->SetOwner(kTRUE);
73d51118 68}
69
70AliTPCAnalysisTaskcalib::~AliTPCAnalysisTaskcalib() {
71 //
72 // destructor
73 //
f7f33dec 74 printf("AliTPCAnalysisTaskcalib::~AliTPCAnalysisTaskcalib");
edb92719 75 fCalibJobs->Delete();
73d51118 76}
77
78void AliTPCAnalysisTaskcalib::Exec(Option_t *) {
79 //
80 // Exec function
81 // Loop over tracks and call Process function
82 if (!fESD) {
2acad464 83 //Printf("ERROR: fESD not available");
73d51118 84 return;
85 }
86 fESDfriend=static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));
87 if (!fESDfriend) {
2acad464 88 //Printf("ERROR: fESDfriend not available");
73d51118 89 return;
90 }
012c4694 91 if (fESDfriend->TestSkipBit()) return;
73d51118 92 Int_t n=fESD->GetNumberOfTracks();
f7f33dec 93 Process(fESD);
5822e507 94 Int_t run = fESD->GetRunNumber();
73d51118 95 for (Int_t i=0;i<n;++i) {
96 AliESDfriendTrack *friendTrack=fESDfriend->GetTrack(i);
f7f33dec 97 AliESDtrack *track=fESD->GetTrack(i);
0db06ea6 98 TObject *calibObject=0;
73d51118 99 AliTPCseed *seed=0;
1f3b7de0 100 if (!friendTrack) continue;
0db06ea6 101 for (Int_t j=0;(calibObject=friendTrack->GetCalibObject(j));++j)
102 if ((seed=dynamic_cast<AliTPCseed*>(calibObject)))
73d51118 103 break;
5822e507 104 if (track) Process(track, run);
73d51118 105 if (seed)
106 Process(seed);
107 }
c51653e8 108 PostData(0,fCalibJobs);
73d51118 109}
110
111void AliTPCAnalysisTaskcalib::ConnectInputData(Option_t *) {
112 //
113 //
114 //
115 TTree* tree=dynamic_cast<TTree*>(GetInputData(0));
116 if (!tree) {
2acad464 117 //Printf("ERROR: Could not read chain from input slot 0");
73d51118 118 }
119 else {
120 AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
121 if (!esdH) {
2acad464 122 //Printf("ERROR: Could not get ESDInputHandler");
73d51118 123 }
124 else {
125 fESD = esdH->GetEvent();
684602c8 126 //Printf("*** CONNECTED NEW EVENT ****");
73d51118 127 }
128 }
129}
130
131void AliTPCAnalysisTaskcalib::CreateOutputObjects() {
132 //
133 //
134 //
54b8f8e1 135 //OpenFile(0, "RECREATE");
77d31cb8 136
137 PostData(0,fCalibJobs);
73d51118 138}
77d31cb8 139
0db06ea6 140void AliTPCAnalysisTaskcalib::Terminate(Option_t */*option*/) {
7eaa723e 141 //
08825fb4 142 // Terminate
7eaa723e 143 //
08825fb4 144 AliTPCcalibBase *job=0;
f7f33dec 145 Int_t njobs = fCalibJobs->GetEntriesFast();
08825fb4 146 for (Int_t i=0;i<njobs;i++){
f7f33dec 147 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
08825fb4 148 if (job) job->Terminate();
7eaa723e 149 }
42a48f93 150
73d51118 151}
152
f7f33dec 153void AliTPCAnalysisTaskcalib::FinishTaskOutput()
154{
155 //
156 // According description in AliAnalisysTask this method is call
157 // on the slaves before sending data
158 //
159 Terminate("slave");
5647625c 160 if(!fDebugOutputPath.IsNull()) {
161 RegisterDebugOutput();
162 }
679480d0 163
f7f33dec 164}
165
166
73d51118 167void AliTPCAnalysisTaskcalib::Process(AliESDEvent *event) {
08825fb4 168 //
169 // Process ESD event
170 //
171 AliTPCcalibBase *job=0;
f7f33dec 172 Int_t njobs = fCalibJobs->GetEntriesFast();
08825fb4 173 for (Int_t i=0;i<njobs;i++){
f7f33dec 174 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
42a48f93 175 if (job) {
176 job->UpdateEventInfo(event);
c51653e8 177 if (job->AcceptTrigger())
178 job->Process(event);
42a48f93 179 }
08825fb4 180 }
73d51118 181}
182
183void AliTPCAnalysisTaskcalib::Process(AliTPCseed *track) {
08825fb4 184 //
185 // Process TPC track
186 //
187 AliTPCcalibBase *job=0;
f7f33dec 188 Int_t njobs = fCalibJobs->GetEntriesFast();
08825fb4 189 for (Int_t i=0;i<njobs;i++){
f7f33dec 190 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
c51653e8 191 if (job)
192 if (job->AcceptTrigger())
193 job->Process(track);
f7f33dec 194 }
195}
196
5822e507 197void AliTPCAnalysisTaskcalib::Process(AliESDtrack *track, Int_t run) {
f7f33dec 198 //
199 // Process ESD track
200 //
201 AliTPCcalibBase *job=0;
202 Int_t njobs = fCalibJobs->GetEntriesFast();
203 for (Int_t i=0;i<njobs;i++){
204 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
c51653e8 205 if (job)
206 if (job->AcceptTrigger())
207 job->Process(track,run);
08825fb4 208 }
73d51118 209}
210
211Long64_t AliTPCAnalysisTaskcalib::Merge(TCollection *li) {
f7f33dec 212 TIterator *i=fCalibJobs->MakeIterator();
73d51118 213 AliTPCcalibBase *job;
214 Long64_t n=0;
0db06ea6 215 while ((job=dynamic_cast<AliTPCcalibBase*>(i->Next())))
73d51118 216 n+=job->Merge(li);
217 return n;
218}
219
220void AliTPCAnalysisTaskcalib::Analyze() {
08825fb4 221 //
222 // Analyze the content of the task
223 //
224 AliTPCcalibBase *job=0;
f7f33dec 225 Int_t njobs = fCalibJobs->GetEntriesFast();
08825fb4 226 for (Int_t i=0;i<njobs;i++){
f7f33dec 227 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
08825fb4 228 if (job) job->Analyze();
229 }
73d51118 230}
f7f33dec 231
232
233void AliTPCAnalysisTaskcalib::RegisterDebugOutput(){
234 //
235 //
236 //
237 AliTPCcalibBase *job=0;
238 Int_t njobs = fCalibJobs->GetEntriesFast();
239 for (Int_t i=0;i<njobs;i++){
240 job = (AliTPCcalibBase*)fCalibJobs->UncheckedAt(i);
241 if (job) job->RegisterDebugOutput(fDebugOutputPath.Data());
242 }
86cd06f0 243 TString dsName=GetName();
244 dsName+=".root";
245 TFile fff(dsName.Data(),"recreate");
c51653e8 246 fCalibJobs->Write("TPCCalib",TObject::kSingleKey);
679480d0 247 fff.Close();
248 //
249 // store - copy debug output to the destination position
250 // currently ONLY for local copy
679480d0 251 TString dsName2=fDebugOutputPath.Data();
252 gSystem->MakeDirectory(dsName2.Data());
253 dsName2+=gSystem->HostName();
254 gSystem->MakeDirectory(dsName2.Data());
255 dsName2+="/";
256 TTimeStamp s;
257 dsName2+=Int_t(s.GetNanoSec());
258 dsName2+="/";
259 gSystem->MakeDirectory(dsName2.Data());
260 dsName2+=dsName;
261 AliInfo(Form("copy %s\t%s\n",dsName.Data(),dsName2.Data()));
262 printf("copy %s\t%s\n",dsName.Data(),dsName2.Data());
263 TFile::Cp(dsName.Data(),dsName2.Data());
264
f7f33dec 265}