PWGHF/vertexingHF/macros/AddTaskHFQA.C

   1 AliAnalysisTaskSEHFQA* AddTaskHFQA(AliAnalysisTaskSEHFQA::DecChannel ch,TString filecutsname="",Bool_t readMC=kFALSE, Bool_t simplemode=kFALSE, Int_t system=1 /*0=pp, 1=PbPb*/, TString finDirname="",Bool_t trackon=kTRUE,Bool_t pidon=kTRUE,Bool_t centralityon=kTRUE, Bool_t eventselon=kTRUE, Bool_t flowobson=kFALSE){
   2   //
   3   // Test macro for the AliAnalysisTaskSE for HF mesons quality assurance
   4   //Author: C.Bianchin chiara.bianchin@pd.infn.it
   5
   6   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   7   if (!mgr) {
   8     ::Error("AddTaskHFQA", "No analysis manager to connect to.");
   9     return NULL;
  10   }
  11
  12   Bool_t stdcuts=kFALSE;
  13   TFile* filecuts;
  14   if( filecutsname.EqualTo("") ) {
  15     stdcuts=kTRUE;
  16   } else {
  17       filecuts=TFile::Open(filecutsname.Data());
  18       if(!filecuts ||(filecuts&& !filecuts->IsOpen())){
  19         AliFatal("Input file not found : check your cut object");
  20       }
  21   }
  22
  23   if(system==0) centralityon=kFALSE;
  24
  25   AliRDHFCuts *analysiscuts=0x0;
  26
  27   TString filename="",out1name="nEntriesQA",out2name="outputPid",out3name="outputTrack",out4name="cuts",out5name="countersCentrality",out6name="outputCentrCheck",out7name="outputEvSel",out8name="outputFlowObs",inname="input",suffix="",cutsobjname="",centr="";
  28   filename = AliAnalysisManager::GetCommonFileName();
  29   filename += ":PWG3_D2H_QA";
  30   filename += finDirname.Data();
  31
  32   switch (ch){
  33   case 0:
  34     cutsobjname="AnalysisCuts";
  35     if(stdcuts) {
  36       analysiscuts = new AliRDHFCutsDplustoKpipi();
  37       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  38       else analysiscuts->SetStandardCutsPbPb2011();
  39     }
  40     else analysiscuts = (AliRDHFCutsDplustoKpipi*)filecuts->Get(cutsobjname);
  41     suffix="Dplus";
  42     break;
  43   case 1:
  44     cutsobjname="D0toKpiCuts";
  45     if(stdcuts) {
  46       analysiscuts = new AliRDHFCutsD0toKpi();
  47       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  48       else analysiscuts->SetStandardCutsPbPb2011();
  49     }
  50     else analysiscuts = (AliRDHFCutsD0toKpi*)filecuts->Get(cutsobjname);
  51     suffix="D0";
  52     break;
  53   case 2:
  54     cutsobjname="DStartoKpipiCuts";
  55     if(stdcuts) {
  56       analysiscuts = new AliRDHFCutsDStartoKpipi();
  57       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  58       else analysiscuts->SetStandardCutsPbPb2011();
  59     }
  60     else analysiscuts = (AliRDHFCutsDstartoKpipi*)filecuts->Get(cutsobjname);
  61     suffix="Dstar";
  62     break;
  63   case 3:
  64     cutsobjname="DstoKKpiCuts";
  65     if(stdcuts) {
  66       analysiscuts = new AliRDHFCutsDstoKKpi();
  67       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  68       else analysiscuts->SetStandardCutsPbPb2010();
  69     }
  70     else analysiscuts = (AliRDHFCutsDstoKKpi*)filecuts->Get(cutsobjname);
  71     suffix="Ds";
  72     break;
  73   case 4:
  74     cutsobjname="D0toKpipipiCuts";
  75     if(stdcuts) {
  76       analysiscuts = new AliRDHFCutsD0toKpipipi();
  77       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  78       else analysiscuts->SetStandardCutsPbPb2010();
  79     }
  80     else analysiscuts = (AliRDHFCutsD0toKpipipi*)filecuts->Get(cutsobjname);
  81     suffix="D04";
  82     break;
  83   case 5:
  84     cutsobjname="LctopKpiAnalysisCuts";
  85     if(stdcuts) {
  86       analysiscuts = new AliRDHFCutsLctopKpi();
  87       if (system == 0) analysiscuts->SetStandardCutsPP2010();
  88       else analysiscuts->SetStandardCutsPbPb2010();
  89     }
  90     else analysiscuts = (AliRDHFCutsLctopKpi*)filecuts->Get(cutsobjname);
  91     suffix="Lc";
  92     break;
  93   }
  94
  95   inname+=suffix;
  96   out1name+=suffix;
  97   out2name+=suffix;
  98   out3name+=suffix;
  99   out4name=cutsobjname;
 100   out5name+=suffix;
 101   out6name+=suffix;
 102   out7name+=suffix;
 103   out8name+=suffix;
 104
 105   if(!analysiscuts && filecutsname!="none"){
 106     cout<<"Specific AliRDHFCuts not found"<<endl;
 107     return;
 108   }
 109
 110   centr=Form("%.0f%.0f",analysiscuts->GetMinCentrality(),analysiscuts->GetMaxCentrality());
 111   inname+=centr;
 112   out1name+=centr;
 113   out2name+=centr;
 114   out3name+=centr;
 115   out4name+=centr;
 116   out5name+=centr;
 117   out6name+=centr;
 118   out7name+=centr;
 119   out8name+=centr;
 120   inname+= finDirname.Data();
 121   out1name+= finDirname.Data();
 122   out2name+= finDirname.Data();
 123   out3name+= finDirname.Data();
 124   out4name+= finDirname.Data();
 125   out5name+= finDirname.Data();
 126   out6name+= finDirname.Data();
 127   out7name+= finDirname.Data();
 128   out8name+= finDirname.Data();
 129
 130
 131   AliAnalysisTaskSEHFQA* taskQA=new AliAnalysisTaskSEHFQA(Form("QA%s",suffix.Data()),ch,analysiscuts);
 132
 133   taskQA->SetReadMC(readMC);
 134   taskQA->SetSimpleMode(simplemode); // set to kTRUE to go faster in PbPb
 135   taskQA->SetTrackOn(trackon);
 136   taskQA->SetPIDOn(pidon);
 137   taskQA->SetCentralityOn(centralityon);
 138   taskQA->SetEvSelectionOn(eventselon);
 139   taskQA->SetFlowObsOn(flowobson);
 140   mgr->AddTask(taskQA);
 141
 142   //
 143   // Create containers for input/output
 144   AliAnalysisDataContainer *cinput = mgr->CreateContainer(inname,TChain::Class(), AliAnalysisManager::kInputContainer);
 145   mgr->ConnectInput(taskQA,0,mgr->GetCommonInputContainer());
 146
 147   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(out1name,TH1F::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //events analysed
 148   mgr->ConnectOutput(taskQA,1,coutput1);
 149
 150   AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(out2name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //PID
 151   if(pidon) mgr->ConnectOutput(taskQA,2,coutput2);
 152
 153   AliAnalysisDataContainer *coutput3 = mgr->CreateContainer(out3name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of tracks
 154   if(trackon) mgr->ConnectOutput(taskQA,3,coutput3);
 155
 156   AliAnalysisDataContainer *coutput4 = mgr->CreateContainer(out4name,AliRDHFCuts::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //cuts
 157   mgr->ConnectOutput(taskQA,4,coutput4);
 158
 159   AliAnalysisDataContainer *coutput5 = mgr->CreateContainer(out5name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of centrality
 160   if(centralityon) mgr->ConnectOutput(taskQA,5,coutput5);
 161
 162   AliAnalysisDataContainer *coutput6 = mgr->CreateContainer(out6name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of centrality
 163   if(centralityon) mgr->ConnectOutput(taskQA,6,coutput6);
 164
 165   AliAnalysisDataContainer *coutput7 = mgr->CreateContainer(out7name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //event selection
 166   if(eventselon) mgr->ConnectOutput(taskQA,7,coutput7);
 167
 168   AliAnalysisDataContainer *coutput8 = mgr->CreateContainer(out8name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //flow observables
 169   if(flowobson) mgr->ConnectOutput(taskQA,8,coutput8);
 170
 171  return taskQA;
 172 }
 173