1 /**************************************************************************
2 * Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
18 /////////////////////////////////////////////////////////////
20 // AliAnalysisTaskSE for the single muon and dimuon from HF analysis,
21 // using the classes AliMuonsHFHeader,
22 // AliMuonInfoStoreRD,
23 // AliDimuInfoStoreRD,
24 // AliMuonInfoStoreMC,
25 // AliDimuInfoStoreMC.
27 // Author: X-M. Zhang, zhang@clermont.in2p3.fr
28 // zhangxm@iopp.ccnu.edu.cn
29 /////////////////////////////////////////////////////////////
32 #include <TClonesArray.h>
34 #include "AliESDEvent.h"
35 #include "AliAODEvent.h"
36 #include "AliESDHeader.h"
37 #include "AliAODHeader.h"
38 #include "AliESDMuonTrack.h"
39 #include "AliAODTrack.h"
40 #include "AliCentrality.h"
41 #include "AliMuonsHFHeader.h"
42 #include "AliMuonInfoStoreRD.h"
43 #include "AliMuonInfoStoreMC.h"
44 #include "AliDimuInfoStoreRD.h"
45 #include "AliDimuInfoStoreMC.h"
46 #include "AliAnalysisManager.h"
47 #include "AliInputEventHandler.h"
48 #include "AliAnalysisTaskSEMuonsHF.h"
50 ClassImp(AliAnalysisTaskSEMuonsHF)
52 //_____________________________________________________________________________
53 AliAnalysisTaskSEMuonsHF::AliAnalysisTaskSEMuonsHF() :
56 fIsOutputTree(kFALSE),
64 // Default constructor
68 //_____________________________________________________________________________
69 AliAnalysisTaskSEMuonsHF::AliAnalysisTaskSEMuonsHF(const char *name) :
70 AliAnalysisTaskSE(name),
72 fIsOutputTree(kFALSE),
82 DefineOutput(1, TList::Class());
85 //_____________________________________________________________________________
86 AliAnalysisTaskSEMuonsHF::~AliAnalysisTaskSEMuonsHF()
91 if (fHeader) { delete fHeader; fHeader =NULL; }
92 if (fMuonClArr) { delete fMuonClArr; fMuonClArr =NULL; }
93 if (fDimuClArr) { delete fDimuClArr; fDimuClArr =NULL; }
94 if (fListOutput) { delete fListOutput; fListOutput=NULL; }
97 //_____________________________________________________________________________
98 void AliAnalysisTaskSEMuonsHF::Init()
101 // Setting and initializing the running mode and status
103 AliMuonsHFHeader::SetAnaMode(fAnaMode);
104 AliMuonsHFHeader::SetIsMC(fIsMC);
106 fHeader = new AliMuonsHFHeader();
107 fHeader->SetName(AliMuonsHFHeader::StdBranchName());
112 fMuonClArr = new TClonesArray("AliMuonInfoStoreMC", 0);
113 fMuonClArr->SetName(AliMuonInfoStoreMC::StdBranchName());
115 fMuonClArr = new TClonesArray("AliMuonInfoStoreRD", 0);
116 fMuonClArr->SetName(AliMuonInfoStoreRD::StdBranchName());
120 if (fAnaMode!=1 && !fDimuClArr) {
122 fDimuClArr = new TClonesArray("AliDimuInfoStoreMC", 0);
123 fDimuClArr->SetName(AliDimuInfoStoreMC::StdBranchName());
125 fDimuClArr = new TClonesArray("AliDimuInfoStoreRD", 0);
126 fDimuClArr->SetName(AliDimuInfoStoreRD::StdBranchName());
133 //_____________________________________________________________________________
134 void AliAnalysisTaskSEMuonsHF::UserCreateOutputObjects()
136 // Create the output container
138 if (!fListOutput) fListOutput = new TList();
139 fHeader->CreateHistograms(fListOutput);
142 AddAODBranch("AliMuonsHFHeader", &fHeader);
143 AddAODBranch("TClonesArray", &fMuonClArr);
144 if (fAnaMode!=1) AddAODBranch("TClonesArray", &fDimuClArr);
150 //_____________________________________________________________________________
151 void AliAnalysisTaskSEMuonsHF::UserExec(Option_t *)
153 // Execute analysis for current event:
154 // muon event header & (di)muon info store
158 if (MCEvent()->GetNumberOfTracks()<=0)
159 { AliError("MC event not found. Nothing done!"); return; }
160 } else { AliError("MC event not found. Nothing done!"); return; }
164 AliAODEvent *aod = 0;
165 AliESDEvent *esd = 0;
166 if (((TString)InputEvent()->IsA()->GetName())=="AliAODEvent") {
167 aod = dynamic_cast<AliAODEvent*>(InputEvent());
168 if (!aod) { AliError("AOD event not found. Nothing done!"); return; }
169 if (!fIsMC && (aod->GetHeader()->GetEventType()!=7)) return;
170 ntrks = aod->GetNTracks();
171 fHeader->SetFiredTriggerClass(aod->GetFiredTriggerClasses());
173 esd = dynamic_cast<AliESDEvent*>(InputEvent());
174 if (!esd) { AliError("ESD event not found. Nothing done!"); return; }
175 if (!fIsMC && (esd->GetHeader()->GetEventType()!=7)) return;
176 ntrks = esd->GetNumberOfMuonTracks();
177 fHeader->SetFiredTriggerClass(esd->GetFiredTriggerClasses());
178 AliCentrality *cent = esd->GetCentrality();
179 fHeader->SetCentrality(cent->GetCentralityPercentile("V0M")); // Just for ESD
182 if (fIsOutputTree) AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()->SetFillAOD(kTRUE);
183 if (fInputHandler && fInputHandler->GetEventSelection()) fHeader->SetSelectionMask(fInputHandler->IsEventSelected());
184 fHeader->SetVertex(((AliVVertex*)InputEvent()->GetPrimaryVertex()));
185 fHeader->FillHistosEvnH(fListOutput);
187 fMuonClArr->Delete();
188 TClonesArray &muonRef = *fMuonClArr;
189 Int_t countN = fMuonClArr->GetEntriesFast();
191 AliAODTrack *trkAOD = 0;
192 AliESDMuonTrack *trkESD = 0;
193 AliMuonInfoStoreRD *muonRD = 0;
194 AliMuonInfoStoreMC *muonMC = 0;
195 for (Int_t itrk=0; itrk<ntrks; itrk++) { // loop over all tracks
197 trkAOD = (AliAODTrack*)aod->GetTrack(itrk);
198 if (!trkAOD->IsMuonTrack()) { trkAOD=0; continue; }
199 if (fIsMC) muonMC = new AliMuonInfoStoreMC(trkAOD, MCEvent());
200 else muonRD = new AliMuonInfoStoreRD(trkAOD);
203 trkESD = (AliESDMuonTrack*)esd->GetMuonTrack(itrk);
204 if (!trkESD->ContainTrackerData()) { trkESD=0; continue; }
205 if (fIsMC) muonMC = new AliMuonInfoStoreMC(trkESD, MCEvent());
206 else muonRD = new AliMuonInfoStoreRD(trkESD);
211 new(muonRef[countN++]) AliMuonInfoStoreRD(*muonRD);
212 if (fAnaMode!=2) fHeader->FillHistosMuon(fListOutput, muonRD);
215 new(muonRef[countN++]) AliMuonInfoStoreMC(*muonMC);
216 if (fAnaMode!=2) fHeader->FillHistosMuon(fListOutput, muonMC, muonMC->Source());
219 if (muonRD) { delete muonRD; muonRD=0; }
220 if (muonMC) { delete muonMC; muonMC=0; }
221 } // end loop of all tracks
224 if (fAnaMode==1) { PostData(1,fListOutput); return; }
226 fDimuClArr->Delete();
227 countN = fDimuClArr->GetEntriesFast();
228 TClonesArray &dimuRef = *fDimuClArr;
230 AliDimuInfoStoreRD *dimuRD = 0;
231 AliDimuInfoStoreMC *dimuMC = 0;
232 ntrks = fMuonClArr->GetEntriesFast();
233 for (Int_t itrk=0; itrk<ntrks-1; itrk++) { // 1st loop over muon tracks
234 for (Int_t jtrk=itrk+1; jtrk<ntrks; jtrk++) { // 2nd loop ofver muon tracks
236 dimuMC = new AliDimuInfoStoreMC((AliMuonInfoStoreMC*)fMuonClArr->At(itrk), (AliMuonInfoStoreMC*)fMuonClArr->At(jtrk));
238 dimuRD = new AliDimuInfoStoreRD((AliMuonInfoStoreRD*)fMuonClArr->At(itrk), (AliMuonInfoStoreRD*)fMuonClArr->At(jtrk));
242 fHeader->FillHistosDimu(fListOutput, dimuRD);
243 if (fIsOutputTree) new(dimuRef[countN++]) AliDimuInfoStoreRD(*dimuRD);
246 fHeader->FillHistosDimu(fListOutput, dimuMC, dimuMC->Source());
247 if (fIsOutputTree) new(dimuRef[countN++]) AliDimuInfoStoreMC(*dimuMC);
250 if (dimuRD) { delete dimuRD; dimuRD=0; }
251 if (dimuMC) { delete dimuMC; dimuMC=0; }
252 } // end 2nd loop of muon tracks
253 } // end 1st loop of muon tracks
255 PostData(1, fListOutput);
259 //_____________________________________________________________________________
260 void AliAnalysisTaskSEMuonsHF::Terminate(Option_t *)
262 // Terminate analysis
264 // add the correction matrix