1 ///////////////////////////////////////////////////////
3 // Basic class for Performance/Calibration TRD tasks
5 // It performs generic tasks like :
6 // - data file manegment
7 // - reference container management
8 // - debug container management
9 // - interaction with AliAnalysisManager
10 // - Plot functor loop
12 // Author: Alexandru Bercuci <A.Bercuci@gsi.de>, 10/09/2008
14 /////////////////////////////////////////////////////////
18 #include "TMethodCall.h"
19 #include "TMethodArg.h"
26 #include "TObjArray.h"
27 #include "TDirectory.h"
28 #include "TTreeStream.h"
31 #include "AliAnalysisTask.h"
33 #include "AliTRDrecoTask.h"
35 ClassImp(AliTRDrecoTask)
37 TList* AliTRDrecoTask::fgTrendPoint(NULL);
38 TTreeSRedirector* AliTRDrecoTask::fgDebugStream(NULL);
39 //_______________________________________________________
40 AliTRDrecoTask::AliTRDrecoTask()
50 // Default constructor
53 //_______________________________________________________
54 AliTRDrecoTask::AliTRDrecoTask(const char *name, const char *title)
55 : AliAnalysisTaskSE(name)
64 // Constructor for all derived performance tasks
67 DefineInput (1, TObjArray::Class()); // track list
68 DefineOutput(1, TObjArray::Class()); // histogram list
71 //_______________________________________________________
72 AliTRDrecoTask::~AliTRDrecoTask()
75 // Generic task destructor
77 AliDebug(2, Form(" Ending task %s[%s]", GetName(), GetTitle()));
84 fPlotFuncList->Delete();
90 if(fContainer->IsOwner()) fContainer->Delete();
96 TFile::Open("TRD.PerformanceTrend.root", "UPDATE");
97 fgTrendPoint->Write();
104 //_______________________________________________________
105 void AliTRDrecoTask::ConnectInputData(Option_t *)
108 // Connect input data
110 AliAnalysisTaskSE::ConnectInputData();
111 fTracks = dynamic_cast<TObjArray *>(GetInputData(1));
114 //_______________________________________________________
115 void AliTRDrecoTask::UserExec(Option_t *)
117 // Loop over Plot functors published by particular tasks
121 AliWarning("No functor list defined for the reference plots");
125 if(!fTracks->GetEntriesFast()) return;
126 else AliDebug(2, Form("Tracks[%d] for %s", fTracks->GetEntriesFast(), GetName()));
128 AliTRDtrackInfo *trackInfo = NULL;
129 TIter plotIter(fPlotFuncList);
130 TObjArrayIter trackIter(fTracks);
131 while((trackInfo = dynamic_cast<AliTRDtrackInfo*>(trackIter()))){
132 fkTrack = trackInfo->GetTrack();
133 fkMC = trackInfo->GetMCinfo();
134 fkESD = trackInfo->GetESDinfo();
136 TMethodCall *plot = NULL;
138 while((plot=dynamic_cast<TMethodCall*>(plotIter()))){
142 PostData(1, fContainer);
145 //_______________________________________________________
146 Bool_t AliTRDrecoTask::GetRefFigure(Int_t /*ifig*/)
148 AliWarning("Retrieving reference figures not implemented.");
152 //_______________________________________________________
153 Bool_t AliTRDrecoTask::PutTrendValue(const Char_t *name, Double_t val)
155 // Generic publisher for trend values
158 fgTrendPoint = new TList();
159 fgTrendPoint->SetOwner();
161 fgTrendPoint->AddLast(new TNamed(Form("%s_%s", GetName(), name), Form("%f", val)));
165 //_______________________________________________________
166 void AliTRDrecoTask::InitFunctorList()
168 // Initialize list of functors
170 TClass *c = this->IsA();
173 TIter methIter(c->GetListOfMethods());
174 while((m=dynamic_cast<TMethod*>(methIter()))){
175 TString name(m->GetName());
176 if(!name.BeginsWith("Plot")) continue;
177 if(!fPlotFuncList) fPlotFuncList = new TList();
178 fPlotFuncList->AddLast(new TMethodCall(c, (const char*)name, ""));
182 //_______________________________________________________
183 Bool_t AliTRDrecoTask::Load(const Char_t *filename)
185 // Generic container loader
187 if(!TFile::Open(filename)){
188 AliWarning(Form("Couldn't open file %s.", filename));
192 if(!(o = (TObjArray*)gFile->Get(GetName()))){
193 AliWarning("Missing histogram container.");
196 fContainer = (TObjArray*)o->Clone(GetName());
201 //________________________________________________________
202 Bool_t AliTRDrecoTask::Save(TObjArray * const results){
204 // Store the output graphs in a ROOT file
205 // Input TObject array will not be written as Key to the file,
206 // only content itself
209 TDirectory *cwd = gDirectory;
210 if(!TFile::Open(Form("TRD.Result%s.root", GetName()), "RECREATE")) return kFALSE;
212 TIterator *iter = results->MakeIterator();
213 TObject *inObject = NULL, *outObject = NULL;
214 while((inObject = iter->Next())){
215 outObject = inObject->Clone();
216 outObject->Write(NULL, TObject::kSingleKey);
219 gFile->Close(); delete gFile;
224 //_______________________________________________________
225 Bool_t AliTRDrecoTask::PostProcess()
227 // To be implemented by particular tasks
229 AliWarning("Post processing of reference histograms not implemented.");
233 //_______________________________________________________
234 void AliTRDrecoTask::SetDebugLevel(Int_t level)
236 // Generic debug handler
238 AliAnalysisTaskSE::SetDebugLevel(level);
240 TDirectory *savedir = gDirectory;
241 fgDebugStream = new TTreeSRedirector("TRD.DebugPerformance.root");
246 //____________________________________________________________________
247 void AliTRDrecoTask::Terminate(Option_t *)
254 delete fgDebugStream;
255 fgDebugStream = NULL;
257 if(HasPostProcess()) PostProcess();
260 //________________________________________________________
261 void AliTRDrecoTask::Adjust(TF1 *f, TH1 * const h)
263 // Helper function to avoid duplication of code
264 // Make first guesses on the fit parameters
266 // find the intial parameters of the fit !! (thanks George)
267 Int_t nbinsy = Int_t(.5*h->GetNbinsX());
269 for(Int_t jbin=nbinsy-4; jbin<=nbinsy+4; jbin++) sum+=h->GetBinContent(jbin); sum/=9.;
270 f->SetParLimits(0, 0., 3.*sum);
271 f->SetParameter(0, .9*sum);
273 f->SetParLimits(1, -.2, .2);
274 f->SetParameter(1, -0.1);
276 f->SetParLimits(2, 0., 4.e-1);
277 f->SetParameter(2, 2.e-2);
278 if(f->GetNpar() <= 4) return;
280 f->SetParLimits(3, 0., sum);
281 f->SetParameter(3, .1*sum);
283 f->SetParLimits(4, -.3, .3);
284 f->SetParameter(4, 0.);
286 f->SetParLimits(5, 0., 1.e2);
287 f->SetParameter(5, 2.e-1);