PWG4/macros/AddTaskFastEmbedding.C

   1
   2 AliAnalysisTaskFastEmbedding* AddTaskFastEmbedding(){
   3
   4    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   5    if(!mgr){
   6       ::Error("AddTaskCentralitySelection", "No analysis manager to connect ot.");
   7       return NULL;
   8    }
   9    if(!mgr->GetInputEventHandler()){
  10       ::Error("AddTaskCentralitySelection", "This task requires an input event handler.");
  11       return NULL;
  12    }
  13
  14
  15    AliAnalysisTaskFastEmbedding *task = new AliAnalysisTaskFastEmbedding("FastEmbedding");
  16    // ## set embedding mode ##
  17    // kAODFull=0, kAODJetTracks, kAODJet4Mom, kToySingle4Mom
  18    task->SetEmbedMode(AliAnalysisTaskFastEmbedding::kToyTracks);
  19
  20    // ## set ranges for toy ##
  21    //SetToyTrackRanges(
  22    Double_t minPt = 50.;   Double_t maxPt = 100.;
  23    Double_t minEta = -0.5; Double_t maxEta = 0.5;
  24    //Double_t minEta = -0.4; Double_t maxEta = 0.4;  // for LHC10h pass1
  25    Double_t minPhi = 0.;   Double_t maxPhi = 2*TMath::Pi();
  26    //fToyDistributionTrackPt: 0 = uniform distribution
  27    //                         else = exponential / power law (not implemented yet)
  28    //task->SetToyNumberOfTrackRange(4,4);
  29    //task->SetToyTrackRanges(0.15, 300., 5,-.9, .9, 0., 2*TMath::Pi());
  30    task->SetToyTrackRanges(minPt,maxPt,0.,minEta,maxEta,minPhi,maxPhi);
  31    task->SetToyFilterMap((1<<32)-1);
  32
  33    // ## set event selection for events of the addition AOD ##
  34    // kEventsAll=0; kEventsJetPt
  35    task->SetEvtSelecMode(AliAnalysisTaskFastEmbedding::kEventsJetPt);
  36
  37    // ## set jet pT range for event selection ##
  38    // SetEvtSelJetPtRange(Float_t minPt, Float_t maxPt)
  39    task->SetEvtSelJetPtRange(50.,110.);
  40    //task->SetEvtSelJetEtaRange(-0.4, 0.4); // smaller eta window for LHC10h pass1
  41    task->SetEvtSelJetEtaRange(-0.5, 0.5);
  42
  43    task->SetTrackFilterMap(272);
  44
  45    // event selection
  46    task->SetOfflineTrgMask(AliVEvent::kMB);
  47    task->SetCentMin(0.);
  48    task->SetCentMax(80.);
  49
  50    //task->SetVtxMin(-10.);
  51    //task->SetVtxMax(10.);
  52
  53    mgr->AddTask(task);
  54
  55    // ## create the output containers ##
  56    AliAnalysisDataContainer *coutputFastEmbedding = mgr->CreateContainer(
  57    "fastembedding", TList::Class(), AliAnalysisManager::kOutputContainer,
  58    Form("%s:PWG4_FastEmbedding", AliAnalysisManager::GetCommonFileName()));
  59
  60    mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer());
  61    mgr->ConnectOutput(task, 0, mgr->GetCommonOutputContainer());
  62    mgr->ConnectOutput(task, 1, coutputFastEmbedding);
  63
  64
  65    return task;
  66
  67 }
  68
  69 AliAnalysisTaskFastEmbedding* AddTaskFastEmbedding(TObjArray* aodarray){
  70
  71    AliAnalysisTaskFastEmbedding *task = AddTaskFastEmbedding();
  72    if(aodarray){
  73       task->SetArrayOfAODPaths(aodarray);
  74       task->SetEmbedMode(AliAnalysisTaskFastEmbedding::kAODFull);
  75    }
  76
  77    return task;
  78 }
  79
  80
  81 AliAnalysisTaskFastEmbedding* AddTaskFastEmbedding(const char* filepath, Int_t mode = 0){
  82
  83    AliAnalysisTaskFastEmbedding *task = AddTaskFastEmbedding();
  84    if(strlen(filepath)){
  85
  86       if(mode==0){ // path to single AOD
  87          task->SetAODPath(filepath);
  88       }
  89       if(mode==1){ // path to text file with list of paths of multiple AODs
  90          Printf("Read aod paths from file %s", filepath);
  91          TObjArray* array = new TObjArray();
  92          TObjString* ostr = 0;
  93          TString line;
  94          ifstream in;
  95          in.open(filepath);
  96          while(in.good()){
  97             in >> line;
  98             if(line.Length() == 0) continue;
  99             Printf("found aod path %s", line.Data());
 100             ostr = new TObjString(line.Data());
 101             array->Add(ostr);
 102          }
 103          Printf("-> %d aod paths found", array->GetEntries());
 104
 105          task->SetArrayOfAODPaths(array);
 106       }
 107
 108       task->SetEmbedMode(AliAnalysisTaskFastEmbedding::kAODFull);
 109    }
 110
 111    return task;
 112 }
 113
 114 AliAnalysisTaskFastEmbedding* AddTaskFastEmbedding(const char* aodpath, const char* entriespath){
 115
 116    AliAnalysisTaskFastEmbedding *task = AddTaskFastEmbedding(aodpath, 1);
 117
 118    Printf("Read entries of aod files from %s", entriespath);
 119    TArrayI* array = new TArrayI();
 120    Int_t count = 0;
 121    Int_t iEntry = -1;
 122    Int_t iEntrySum = 0;
 123    Int_t iEntryMax = 0;
 124    TString line;
 125    ifstream in;
 126    in.open(entriespath);
 127    while(in.good()){
 128       in >> line;
 129       iEntry = line.Atoi();
 130
 131       array->Set(count+1);
 132       array->AddAt(iEntry,count);
 133       count++;
 134       iEntrySum += iEntry;
 135       if(iEntry>iEntryMax) iEntryMax = iEntry;
 136    }
 137
 138    task->SetArrayOfAODEntries(array);
 139    task->SetAODEntriesSum(iEntrySum);
 140    task->SetAODEntriesMax(iEntryMax);
 141
 142    return task;
 143 }
 144