2 // Calculate the multiplicity in the forward regions event-by-event
14 #include "AliForwardMultiplicityTask.h"
15 #include "AliTriggerAnalysis.h"
16 #include "AliPhysicsSelection.h"
18 #include "AliESDEvent.h"
19 #include "AliAODHandler.h"
20 #include "AliMultiplicity.h"
21 #include "AliInputEventHandler.h"
22 #include "AliForwardCorrectionManager.h"
23 #include "AliAnalysisManager.h"
25 #include <TDirectory.h>
30 //====================================================================
31 AliForwardMultiplicityTask::AliForwardMultiplicityTask()
32 : AliForwardMultiplicityBase(),
50 //____________________________________________________________________
51 AliForwardMultiplicityTask::AliForwardMultiplicityTask(const char* name)
52 : AliForwardMultiplicityBase(name),
57 fEventInspector("event"),
58 fEnergyFitter("energy"),
59 fSharingFilter("sharing"),
60 fDensityCalculator("density"),
61 fCorrections("corrections"),
62 fHistCollector("collector"),
71 DefineOutput(1, TList::Class());
74 //____________________________________________________________________
75 AliForwardMultiplicityTask::AliForwardMultiplicityTask(const AliForwardMultiplicityTask& o)
76 : AliForwardMultiplicityBase(o),
81 fEventInspector(o.fEventInspector),
82 fEnergyFitter(o.fEnergyFitter),
83 fSharingFilter(o.fSharingFilter),
84 fDensityCalculator(o.fDensityCalculator),
85 fCorrections(o.fCorrections),
86 fHistCollector(o.fHistCollector),
93 // o Object to copy from
95 DefineOutput(1, TList::Class());
98 //____________________________________________________________________
99 AliForwardMultiplicityTask&
100 AliForwardMultiplicityTask::operator=(const AliForwardMultiplicityTask& o)
103 // Assignment operator
106 // o Object to assign from
109 // Reference to this object
111 AliForwardMultiplicityBase::operator=(o);
114 fEventInspector = o.fEventInspector;
115 fEnergyFitter = o.fEnergyFitter;
116 fSharingFilter = o.fSharingFilter;
117 fDensityCalculator = o.fDensityCalculator;
118 fCorrections = o.fCorrections;
119 fHistCollector = o.fHistCollector;
127 //____________________________________________________________________
129 AliForwardMultiplicityTask::SetDebug(Int_t dbg)
137 fEventInspector.SetDebug(dbg);
138 fEnergyFitter.SetDebug(dbg);
139 fSharingFilter.SetDebug(dbg);
140 fDensityCalculator.SetDebug(dbg);
141 fCorrections.SetDebug(dbg);
142 fHistCollector.SetDebug(dbg);
145 //____________________________________________________________________
147 AliForwardMultiplicityTask::InitializeSubs()
150 // Initialise the sub objects and stuff. Called on first event
156 if (!ReadCorrections(pe,pv)) return;
161 fHData = static_cast<TH2D*>(fAODFMD.GetHistogram().Clone("d2Ndetadphi"));
163 fHData->SetDirectory(0);
166 fEnergyFitter.Init(*pe);
167 fEventInspector.Init(*pv);
168 fDensityCalculator.Init(*pe);
169 fCorrections.Init(*pe);
170 fHistCollector.Init(*pv);
175 //____________________________________________________________________
177 AliForwardMultiplicityTask::UserCreateOutputObjects()
180 // Create output objects
185 AliAnalysisManager* am = AliAnalysisManager::GetAnalysisManager();
187 dynamic_cast<AliAODHandler*>(am->GetOutputEventHandler());
188 if (!ah) AliFatal("No AOD output handler set in analysis manager");
191 TObject* obj = &fAODFMD;
192 ah->AddBranch("AliAODForwardMult", &obj);
194 fEventInspector.DefineOutput(fList);
195 fEnergyFitter.DefineOutput(fList);
196 fSharingFilter.DefineOutput(fList);
197 fDensityCalculator.DefineOutput(fList);
198 fCorrections.DefineOutput(fList);
202 //____________________________________________________________________
204 AliForwardMultiplicityTask::UserExec(Option_t*)
207 // Process each event
213 // static Int_t cnt = 0;
215 // Get the input data
216 AliESDEvent* esd = GetESDEvent();
223 Bool_t lowFlux = kFALSE;
227 UInt_t found = fEventInspector.Process(esd, triggers, lowFlux, ivz, vz);
229 if (found & AliFMDEventInspector::kNoEvent) return;
230 if (found & AliFMDEventInspector::kNoTriggers) return;
232 // Set trigger bits, and mark this event for storage
233 fAODFMD.SetTriggerBits(triggers);
236 if (found & AliFMDEventInspector::kNoSPD) return;
237 if (found & AliFMDEventInspector::kNoFMD) return;
238 if (found & AliFMDEventInspector::kNoVertex) return;
241 if (found & AliFMDEventInspector::kBadVertex) return;
243 // We we do not want to use low flux specific code, we disable it here.
244 if (!fEnableLowFlux) lowFlux = false;
247 AliESDFMD* esdFMD = esd->GetFMDData();
248 // Apply the sharing filter (or hit merging or clustering if you like)
249 if (!fSharingFilter.Filter(*esdFMD, lowFlux, fESDFMD)) {
250 AliWarning("Sharing filter failed!");
254 // Do the energy stuff
255 if (!fEnergyFitter.Accumulate(*esdFMD, triggers & AliAODForwardMult::kEmpty)){
256 AliWarning("Energy fitter failed");
260 // Calculate the inclusive charged particle density
261 if (!fDensityCalculator.Calculate(fESDFMD, fHistos, ivz, lowFlux)) {
262 AliWarning("Density calculator failed!");
266 // Do the secondary and other corrections.
267 if (!fCorrections.Correct(fHistos, ivz)) {
268 AliWarning("Corrections failed");
272 if (!fHistCollector.Collect(fHistos, ivz, fAODFMD.GetHistogram())) {
273 AliWarning("Histogram collector failed");
277 if (fAODFMD.IsTriggerBits(AliAODForwardMult::kInel))
278 fHData->Add(&(fAODFMD.GetHistogram()));
283 //____________________________________________________________________
285 AliForwardMultiplicityTask::Terminate(Option_t*)
294 TList* list = dynamic_cast<TList*>(GetOutputData(1));
296 AliError(Form("No output list defined (%p)", GetOutputData(1)));
297 if (GetOutputData(1)) GetOutputData(1)->Print();
301 // Get our histograms from the container
302 TH1I* hEventsTr = 0;//static_cast<TH1I*>(list->FindObject("nEventsTr"));
303 TH1I* hEventsTrVtx = 0;//static_cast<TH1I*>(list->FindObject("nEventsTrVtx"));
305 if (!fEventInspector.FetchHistograms(list, hEventsTr,
306 hEventsTrVtx, hTriggers)) {
307 AliError(Form("Didn't get histograms from event selector "
308 "(hEventsTr=%p,hEventsTrVtx=%p)",
309 hEventsTr, hEventsTrVtx));
314 TH2D* hData = static_cast<TH2D*>(list->FindObject("d2Ndetadphi"));
316 AliError(Form("Couldn't get our summed histogram from output "
317 "list %s (d2Ndetadphi=%p)", list->GetName(), hData));
322 // TH1D* dNdeta = fHData->ProjectionX("dNdeta", 0, -1, "e");
323 TH1D* dNdeta = hData->ProjectionX("dNdeta", 1, -1, "e");
324 TH1D* norm = hData->ProjectionX("norm", 0, 1, "");
325 dNdeta->SetTitle("dN_{ch}/d#eta in the forward regions");
326 dNdeta->SetYTitle("#frac{1}{N}#frac{dN_{ch}}{d#eta}");
327 dNdeta->Divide(norm);
329 dNdeta->Scale(Double_t(hEventsTrVtx->GetEntries())/hEventsTr->GetEntries(),
334 fEnergyFitter.Fit(list);
335 fSharingFilter.ScaleHistograms(list,Int_t(hEventsTr->Integral()));
336 fDensityCalculator.ScaleHistograms(list,Int_t(hEventsTrVtx->Integral()));
337 fCorrections.ScaleHistograms(list,Int_t(hEventsTrVtx->Integral()));