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 reconstruction of heavy flavor
21 // decays, using the class AliAnalysisVertexingHF.
23 // Author: A.Dainese, andrea.dainese@lnl.infn.it
24 /////////////////////////////////////////////////////////////
28 #include <TClonesArray.h>
32 #include "AliVEvent.h"
33 #include "AliAODEvent.h"
34 #include "AliAODHandler.h"
35 #include "AliESDEvent.h"
36 #include "AliAnalysisVertexingHF.h"
37 #include "AliAnalysisTaskSE.h"
38 #include "AliAnalysisManager.h"
39 #include "AliAnalysisTaskSEVertexingHF.h"
41 #include "AliESDUtils.h"
42 #include "AliAODHFUtil.h"
45 ClassImp(AliAnalysisTaskSEVertexingHF)
48 //________________________________________________________________________
49 AliAnalysisTaskSEVertexingHF::AliAnalysisTaskSEVertexingHF():
53 fDeltaAODFileName("AliAOD.VertexingHF.root"),
57 fCharm3ProngTClArr(0),
58 fCharm4ProngTClArr(0),
61 fLikeSign2ProngTClArr(0),
62 fLikeSign3ProngTClArr(0),
65 // Default constructor
68 //________________________________________________________________________
69 AliAnalysisTaskSEVertexingHF::AliAnalysisTaskSEVertexingHF(const char *name):
70 AliAnalysisTaskSE(name),
73 fDeltaAODFileName("AliAOD.VertexingHF.root"),
77 fCharm3ProngTClArr(0),
78 fCharm4ProngTClArr(0),
81 fLikeSign2ProngTClArr(0),
82 fLikeSign3ProngTClArr(0),
85 // Standard constructor
87 DefineOutput(1,TList::Class()); // analysis cuts
90 //________________________________________________________________________
91 AliAnalysisTaskSEVertexingHF::~AliAnalysisTaskSEVertexingHF()
102 //________________________________________________________________________
103 void AliAnalysisTaskSEVertexingHF::Init()
106 // Instanciates vHF and loads its parameters
108 if(fDebug > 1) printf("AnalysisTaskSEVertexingHF::Init() \n");
110 if(gROOT->LoadMacro("ConfigVertexingHF.C")) {
111 printf("AnalysisTaskSEVertexingHF::Init() \n Using $ALICE_ROOT/PWGHF/vertexingHF/ConfigVertexingHF.C\n");
112 gROOT->LoadMacro("$ALICE_ROOT/PWGHF/vertexingHF/ConfigVertexingHF.C");
115 fVHF = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
119 // write the objects AliRDHFCuts to a list to store in the output
121 fListOfCuts = fVHF->FillListOfCuts();
123 PostData(1,fListOfCuts);
125 AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile(fDeltaAODFileName.Data());
130 //________________________________________________________________________
131 void AliAnalysisTaskSEVertexingHF::UserCreateOutputObjects()
133 // Create the output container
135 if(fDebug > 1) printf("AnalysisTaskSEVertexingHF::UserCreateOutPutData() \n");
136 // Support both the case when the AOD + deltaAOD are produced in an ESD
137 // analysis or if the deltaAOD is produced on an analysis on AOD's. (A.G. 27/04/09)
138 if(!AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()) {
139 Fatal("UserCreateOutputObjects", "This task needs an AOD handler");
142 TString filename = fDeltaAODFileName;
143 // When running on standard AOD to produce deltas, IsStandardAOD is never set,
144 // If AODEvent is NULL, new branches have to be added to the new file(s) (A.G. 15/01/10)
145 if(!IsStandardAOD() && AODEvent()) filename = "";
147 printf("AnalysisTaskSEVertexingHF::UserCreateOutPutData() \n ERROR! no fvHF!\n");
151 fVerticesHFTClArr = new TClonesArray("AliAODVertex", 0);
152 fVerticesHFTClArr->SetName("VerticesHF");
153 AddAODBranch("TClonesArray", &fVerticesHFTClArr, filename);
155 if(fVHF->GetD0toKpi()) {
156 fD0toKpiTClArr = new TClonesArray("AliAODRecoDecayHF2Prong", 0);
157 fD0toKpiTClArr->SetName("D0toKpi");
158 AddAODBranch("TClonesArray", &fD0toKpiTClArr, filename);
161 if(fVHF->GetJPSItoEle()) {
162 fJPSItoEleTClArr = new TClonesArray("AliAODRecoDecayHF2Prong", 0);
163 fJPSItoEleTClArr->SetName("JPSItoEle");
164 AddAODBranch("TClonesArray", &fJPSItoEleTClArr, filename);
167 if(fVHF->Get3Prong()) {
168 fCharm3ProngTClArr = new TClonesArray("AliAODRecoDecayHF3Prong", 0);
169 fCharm3ProngTClArr->SetName("Charm3Prong");
170 AddAODBranch("TClonesArray", &fCharm3ProngTClArr, filename);
173 if(fVHF->Get4Prong()) {
174 fCharm4ProngTClArr = new TClonesArray("AliAODRecoDecayHF4Prong", 0);
175 fCharm4ProngTClArr->SetName("Charm4Prong");
176 AddAODBranch("TClonesArray", &fCharm4ProngTClArr, filename);
179 if(fVHF->GetDstar()) {
180 fDstarTClArr = new TClonesArray("AliAODRecoCascadeHF", 0);
181 fDstarTClArr->SetName("Dstar");
182 AddAODBranch("TClonesArray", &fDstarTClArr, filename);
185 if(fVHF->GetCascades()){
186 fCascadesTClArr = new TClonesArray("AliAODRecoCascadeHF", 0);
187 fCascadesTClArr->SetName("CascadesHF");
188 AddAODBranch("TClonesArray", &fCascadesTClArr, filename);
191 if(fVHF->GetLikeSign()) {
192 fLikeSign2ProngTClArr = new TClonesArray("AliAODRecoDecayHF2Prong", 0);
193 fLikeSign2ProngTClArr->SetName("LikeSign2Prong");
194 AddAODBranch("TClonesArray", &fLikeSign2ProngTClArr, filename);
197 if(fVHF->GetLikeSign() && fVHF->Get3Prong()) {
198 fLikeSign3ProngTClArr = new TClonesArray("AliAODRecoDecayHF3Prong", 0);
199 fLikeSign3ProngTClArr->SetName("LikeSign3Prong");
200 AddAODBranch("TClonesArray", &fLikeSign3ProngTClArr, filename);
203 //---Way to pass information temporarily not available in AOD---
205 fHFUtilInfo = new AliAODHFUtil("fHFUtilInfoC");
206 fHFUtilInfo->SetName("fHFUtilInfo");
207 AddAODBranch( "AliAODHFUtil", &fHFUtilInfo, filename);
209 //--------------------------------------------------------------
214 //________________________________________________________________________
215 void AliAnalysisTaskSEVertexingHF::UserExec(Option_t */*option*/)
217 // Execute analysis for current event:
218 // heavy flavor vertexing
220 AliVEvent *event = dynamic_cast<AliVEvent*> (InputEvent());
221 // In case there is an AOD handler writing a standard AOD, use the AOD
222 // event in memory rather than the input (ESD) event. (A.G. 27/04/09)
223 if (AODEvent() && IsStandardAOD()) event = dynamic_cast<AliVEvent*> (AODEvent());
225 if (dynamic_cast<AliAODEvent*>(event)) {
226 AliAODHandler *aodhandler = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
227 if (aodhandler) aodhandler->SetFillExtension(kTRUE);
230 // heavy flavor vertexing
231 fVHF->FindCandidates(event,
239 fLikeSign2ProngTClArr,
240 fLikeSign3ProngTClArr);
242 //---Way to pass information temporarily not available in AOD---
243 AliESDEvent *eventE = dynamic_cast<AliESDEvent*> (InputEvent());
245 Float_t *vChCorr = new Float_t[64];
247 AliESDUtils::GetCorrV0(eventE,dummy,NULL,vChCorr);
248 fHFUtilInfo->SetVZERO( vChCorr );
251 //--------------------------------------------------------------
256 //________________________________________________________________________
257 void AliAnalysisTaskSEVertexingHF::Terminate(Option_t */*option*/)
259 // Terminate analysis
261 if(fDebug > 1) printf("AnalysisTaskSEVertexingHF: Terminate() \n");