Macro to create std AOD + MC

diff --git a/PWG3/vertexingHF/CreateStandardAODfromESD.C b/PWG3/vertexingHF/CreateStandardAODfromESD.C
new file mode 100644 (file)
index 0000000..7bf31cc
--- /dev/null
+++ b/PWG3/vertexingHF/CreateStandardAODfromESD.C
@@ -0,0 +1,98 @@
+#if !defined(__CINT__) || defined(__MAKECINT__)
+#include <TChain.h>
+#include <TSystem.h>
+#include "AliAnalysisManager.h"
+#include "AliESDInputHandler.h"
+#include "AliAODHandler.h"
+#include "AliAnalysisTaskESDfilter.h"
+#include "AliAnalysisDataContainer.h"
+#endif
+
+void CreateStandardAODfromESD(const char *inFileName = "AliESDs.root",
+                             const char *outFileName = "AliAOD.root",
+                             Bool_t writeKineToAOD = kTRUE) 
+{
+  
+  gSystem->Load("libTree");
+  gSystem->Load("libGeom");
+  gSystem->Load("libPhysics");
+  gSystem->Load("libVMC");
+  gSystem->Load("libSTEERBase");
+  gSystem->Load("libESD");
+  gSystem->Load("libAOD");
+  
+  gSystem->Load("libANALYSIS");
+  gSystem->Load("libANALYSISalice");
+  gSystem->Load("libPWG3muon");
+  
+  TChain *chain = new TChain("esdTree");
+  // Steering input chain
+  chain->Add(inFileName);
+  AliAnalysisManager *mgr  = new AliAnalysisManager("ESD to AOD", "Analysis Manager");
+  
+  // Input
+  AliESDInputHandler* inpHandler = new AliESDInputHandler();
+  mgr->SetInputEventHandler (inpHandler);
+  
+  // Output
+  AliAODHandler* aodHandler   = new AliAODHandler();
+  aodHandler->SetOutputFileName(outFileName);
+  mgr->SetOutputEventHandler(aodHandler);
+ 
+  // MC Truth
+  AliMCEventHandler* mcHandler = new AliMCEventHandler();
+  if(writeKineToAOD) mgr->SetMCtruthEventHandler(mcHandler);
+
+  // Tasks
+  
+  // Filtering of MC particles (decays conversions etc)
+  // this task is also needed to set the MCEventHandler
+  // to the AODHandler, this will not be needed when
+  // AODHandler goes to ANALYSISalice
+  AliAnalysisTaskMCParticleFilter *kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Filter");
+  if(writeKineToAOD) mgr->AddTask(kinefilter);
+  
+  // Barrel Tracks
+  AliAnalysisTaskESDfilter *filter = new AliAnalysisTaskESDfilter("Filter");
+  mgr->AddTask(filter);
+
+  // Muons
+  AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
+  mgr->AddTask(esdmuonfilter);
+
+  AliESDtrackCuts* esdTrackCutsHF = new AliESDtrackCuts("AliESDtrackCuts", "Heavy flavour");
+  esdTrackCutsHF->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
+
+  AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
+  trackFilter->AddCuts(esdTrackCutsHF);
+
+  filter->SetTrackFilter(trackFilter);
+
+  // Pipelining
+  AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("cchain", TChain::Class(),
+                                                          AliAnalysisManager::kInputContainer);
+  
+  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("tree", TTree::Class(),
+                                                           AliAnalysisManager::kOutputContainer,
+                                                           "default");
+  
+  
+  mgr->ConnectInput(filter,0,cinput1);
+  mgr->ConnectOutput(filter,0,coutput1);
+
+  mgr->ConnectInput(esdmuonfilter,0,cinput1);
+  mgr->ConnectOutput(esdmuonfilter,0,coutput1);
+
+  if(writeKineToAOD) {
+    mgr->ConnectInput(kinefilter,0,cinput1);
+    mgr->ConnectOutput(kinefilter,0,coutput1);
+  }
+  //
+  // Run the analysis
+  //
+  mgr->InitAnalysis();
+  mgr->PrintStatus();
+  mgr->StartAnalysis("local",chain);
+
+  return;
+}