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);
228 if (found & AliFMDEventInspector::kNoEvent) return;
229 if (found & AliFMDEventInspector::kNoTriggers) return;
231 // Set trigger bits, and mark this event for storage
232 fAODFMD.SetTriggerBits(triggers);
235 if (found & AliFMDEventInspector::kNoSPD) return;
236 if (found & AliFMDEventInspector::kNoFMD) return;
237 if (found & AliFMDEventInspector::kNoVertex) return;
240 if (found & AliFMDEventInspector::kBadVertex) return;
242 // We we do not want to use low flux specific code, we disable it here.
243 if (!fEnableLowFlux) lowFlux = false;
246 AliESDFMD* esdFMD = esd->GetFMDData();
247 // Apply the sharing filter (or hit merging or clustering if you like)
248 if (!fSharingFilter.Filter(*esdFMD, lowFlux, fESDFMD)) {
249 AliWarning("Sharing filter failed!");
253 // Do the energy stuff
254 if (!fEnergyFitter.Accumulate(*esdFMD, triggers & AliAODForwardMult::kEmpty)){
255 AliWarning("Energy fitter failed");
259 // Calculate the inclusive charged particle density
260 if (!fDensityCalculator.Calculate(fESDFMD, fHistos, ivz, lowFlux)) {
261 AliWarning("Density calculator failed!");
265 // Do the secondary and other corrections.
266 if (!fCorrections.Correct(fHistos, ivz)) {
267 AliWarning("Corrections failed");
271 if (!fHistCollector.Collect(fHistos, ivz, fAODFMD.GetHistogram())) {
272 AliWarning("Histogram collector failed");
276 if (fAODFMD.IsTriggerBits(AliAODForwardMult::kInel))
277 fHData->Add(&(fAODFMD.GetHistogram()));
282 //____________________________________________________________________
284 AliForwardMultiplicityTask::Terminate(Option_t*)
293 TList* list = dynamic_cast<TList*>(GetOutputData(1));
295 AliError(Form("No output list defined (%p)", GetOutputData(1)));
296 if (GetOutputData(1)) GetOutputData(1)->Print();
300 // Get our histograms from the container
301 TH1I* hEventsTr = 0;//static_cast<TH1I*>(list->FindObject("nEventsTr"));
302 TH1I* hEventsTrVtx = 0;//static_cast<TH1I*>(list->FindObject("nEventsTrVtx"));
304 if (!fEventInspector.FetchHistograms(list, hEventsTr,
305 hEventsTrVtx, hTriggers)) {
306 AliError(Form("Didn't get histograms from event selector "
307 "(hEventsTr=%p,hEventsTrVtx=%p)",
308 hEventsTr, hEventsTrVtx));
313 TH2D* hData = static_cast<TH2D*>(list->FindObject("d2Ndetadphi"));
315 AliError(Form("Couldn't get our summed histogram from output "
316 "list %s (d2Ndetadphi=%p)", list->GetName(), hData));
321 // TH1D* dNdeta = fHData->ProjectionX("dNdeta", 0, -1, "e");
322 TH1D* dNdeta = hData->ProjectionX("dNdeta", 1, -1, "e");
323 TH1D* norm = hData->ProjectionX("norm", 0, 1, "");
324 dNdeta->SetTitle("dN_{ch}/d#eta in the forward regions");
325 dNdeta->SetYTitle("#frac{1}{N}#frac{dN_{ch}}{d#eta}");
326 dNdeta->Divide(norm);
328 dNdeta->Scale(Double_t(hEventsTrVtx->GetEntries())/hEventsTr->GetEntries(),
333 fEnergyFitter.Fit(list);
334 fSharingFilter.ScaleHistograms(list,hEventsTr->Integral());
335 fDensityCalculator.ScaleHistograms(list,hEventsTrVtx->Integral());
336 fCorrections.ScaleHistograms(list,hEventsTrVtx->Integral());