STEER/CreateAODfromESD.C

   1 #if !defined(__CINT__) || defined(__MAKECINT__)
   2 #include <TChain.h>
   3 #include <TSystem.h>
   4 #include "AliAnalysisManager.h"
   5 #include "AliESDInputHandler.h"
   6 #include "AliAODHandler.h"
   7 #include "AliAnalysisTaskESDfilter.h"
   8 #include "AliAnalysisDataContainer.h"
   9 #endif
  10
  11 void CreateAODfromESD(const char *inFileName = "AliESDs.root",
  12                       const char *outFileName = "AliAOD.root") {
  13
  14     gSystem->Load("libTree");
  15     gSystem->Load("libGeom");
  16     gSystem->Load("libPhysics");
  17     gSystem->Load("libVMC");
  18     gSystem->Load("libSTEERBase");
  19     gSystem->Load("libESD");
  20     gSystem->Load("libAOD");
  21
  22     gSystem->Load("libANALYSIS");
  23     gSystem->Load("libANALYSISalice");
  24     gSystem->Load("libPWG3muon");
  25
  26     TChain *chain = new TChain("esdTree");
  27     // Steering input chain
  28     chain->Add(inFileName);
  29     AliAnalysisManager *mgr  = new AliAnalysisManager("ESD to AOD", "Analysis Manager");
  30
  31     // Input
  32     AliESDInputHandler* inpHandler = new AliESDInputHandler();
  33     inpHandler->SetReadTags();
  34     mgr->SetInputEventHandler  (inpHandler);
  35
  36     // Output
  37     AliAODHandler* aodHandler   = new AliAODHandler();
  38     aodHandler->SetOutputFileName(outFileName);
  39     mgr->SetOutputEventHandler(aodHandler);
  40
  41     // Task
  42     // Barrel Tracks
  43     AliAnalysisTaskESDfilter *filter = new AliAnalysisTaskESDfilter("Filter");
  44     mgr->AddTask(filter);
  45     // Muons
  46     AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
  47     mgr->AddTask(esdmuonfilter);
  48
  49     // Cuts on primary tracks
  50     AliESDtrackCuts* esdTrackCutsL = new AliESDtrackCuts("AliESDtrackCuts", "Standard");
  51     esdTrackCutsL->SetMinNClustersTPC(50);
  52     esdTrackCutsL->SetMaxChi2PerClusterTPC(3.5);
  53     esdTrackCutsL->SetMaxCovDiagonalElements(2, 2, 0.5, 0.5, 2);
  54     esdTrackCutsL->SetRequireTPCRefit(kTRUE);
  55     esdTrackCutsL->SetDCAToVertex(3.0);
  56     esdTrackCutsL->SetRequireSigmaToVertex(kFALSE);
  57     esdTrackCutsL->SetAcceptKingDaughters(kFALSE);
  58
  59     AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
  60     trackFilter->AddCuts(esdTrackCutsL);
  61
  62     // Cuts on V0s
  63     AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("AliESDv0Cuts", "Standard pp");
  64     esdV0Cuts->SetMinRadius(0.2);
  65     esdV0Cuts->SetMaxRadius(100);
  66     esdV0Cuts->SetMinDcaPosToVertex(0.05);
  67     esdV0Cuts->SetMinDcaNegToVertex(0.05);
  68     esdV0Cuts->SetMaxDcaV0Daughters(0.5);
  69     esdV0Cuts->SetMinCosinePointingAngle(0.99);
  70     AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
  71     v0Filter->AddCuts(esdV0Cuts);
  72
  73
  74 //
  75     filter->SetTrackFilter(trackFilter);
  76     filter->SetV0Filter(v0Filter);
  77
  78
  79 //  Create AOD Tags
  80     AliAnalysisTaskTagCreator* tagTask = new AliAnalysisTaskTagCreator("AOD Tag Creator");
  81     mgr->AddTask(tagTask);
  82
  83     // Pipelining
  84     AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("cchain", TChain::Class(),
  85                                                              AliAnalysisManager::kInputContainer);
  86
  87     AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("tree", TTree::Class(),
  88                                                               AliAnalysisManager::kOutputContainer,
  89                                                               "default");
  90     AliAnalysisDataContainer *coutputT
  91         = mgr->CreateContainer("cTag",  TTree::Class(), AliAnalysisManager::kOutputContainer, "AOD.tag.root");
  92
  93     mgr->ConnectInput (filter, 0, cinput1 );
  94     mgr->ConnectOutput(filter, 0, coutput1);
  95
  96     mgr->ConnectInput (esdmuonfilter, 0, cinput1 );
  97     mgr->ConnectOutput(esdmuonfilter, 0, coutput1);
  98
  99     mgr->ConnectInput (tagTask, 0, cinput1);
 100     mgr->ConnectOutput(tagTask, 1, coutputT);
 101
 102     //
 103     // Run the analysis
 104     //
 105     mgr->InitAnalysis();
 106     mgr->PrintStatus();
 107     mgr->StartAnalysis("local", chain);
 108 }