Use TFile::Open in all AddTask macros

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / macros / AddTaskHFQA.C

AliAnalysisTaskSEHFQA* AddTaskHFQA(AliAnalysisTaskSEHFQA::DecChannel ch,TString filecutsname="D0toKpiCuts.root",Bool_t readMC=kFALSE, Bool_t simplemode=kFALSE, Int_t system=1 /*0=pp, 1=PbPb*/, TString finDirname=""){\r
  //\r
  // Test macro for the AliAnalysisTaskSE for HF mesons quality assurance\r
  //Author: C.Bianchin chiara.bianchin@pd.infn.it\r
\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  if (!mgr) {\r
    ::Error("AddTaskHFQA", "No analysis manager to connect to.");\r
    return NULL;\r
  }\r
\r
  Bool_t stdcuts=kFALSE;\r
  TFile* filecuts=TFile::Open(filecutsname.Data());\r
  if(!filecuts->IsOpen()){\r
    cout<<"Input file not found: using std cut object"<<endl;\r
    stdcuts=kTRUE;\r
  }\r
\r
  Bool_t onoff[4]={kTRUE,kTRUE,kTRUE,kTRUE}; //tracks (0), PID (1), centrality (2), event selection(3) \r
  if(system==0) onoff[2]=kFALSE;\r
\r
  AliRDHFCuts *analysiscuts=0x0;\r
\r
  TString filename="",out1name="nEntriesQA",out2name="outputPid",out3name="outputTrack",out4name="cuts",out5name="countersCentrality",out6name="outputCentrCheck",out7name="outputEvSel",inname="input",suffix="",cutsobjname="",centr="";\r
  filename = AliAnalysisManager::GetCommonFileName();\r
  filename += ":PWG3_D2H_QA";\r
  filename += finDirname.Data();\r
\r
  switch (ch){\r
  case 0:\r
    cutsobjname="AnalysisCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsDplustoKpipi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsDplustoKpipi*)filecuts->Get(cutsobjname);\r
    suffix="Dplus";\r
    break;\r
  case 1:\r
    cutsobjname="D0toKpiCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsD0toKpi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsD0toKpi*)filecuts->Get(cutsobjname);\r
    suffix="D0";\r
    break;\r
  case 2:\r
    cutsobjname="DStartoKpipiCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsDStartoKpipi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsDstartoKpipi*)filecuts->Get(cutsobjname);\r
    suffix="Dstar";\r
    break;\r
  case 3:\r
    cutsobjname="DstoKKpiCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsDstoKKpi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsDstoKKpi*)filecuts->Get(cutsobjname);\r
    suffix="Ds";\r
    break;\r
  case 4:\r
    cutsobjname="D0toKpipipiCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsD0toKpipipi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsD0toKpipipi*)filecuts->Get(cutsobjname);\r
    suffix="D04";\r
    break;\r
  case 5:\r
    cutsobjname="LctopKpiAnalysisCuts";\r
    if(stdcuts) {\r
      analysiscuts = new AliRDHFCutsLctopKpi();\r
      if (system == 0) analysiscuts->SetStandardCutsPP2010();\r
      else analysiscuts->SetStandardCutsPbPb2010();\r
    }\r
    else analysiscuts = (AliRDHFCutsLctopKpi*)filecuts->Get(cutsobjname);\r
    suffix="Lc";\r
    break;\r
  }\r
\r
  inname+=suffix;\r
  out1name+=suffix;\r
  out2name+=suffix;\r
  out3name+=suffix;\r
  out4name=cutsobjname;\r
  out5name+=suffix;\r
  out6name+=suffix;\r
  out7name+=suffix;\r
\r
  if(!analysiscuts && filecutsname!="none"){\r
    cout<<"Specific AliRDHFCuts not found"<<endl;\r
    return;\r
  }\r
\r
  centr=Form("%.0f%.0f",analysiscuts->GetMinCentrality(),analysiscuts->GetMaxCentrality());\r
  inname+=centr;\r
  out1name+=centr;\r
  out2name+=centr;\r
  out3name+=centr;\r
  out4name+=centr;\r
  out5name+=centr;\r
  out6name+=centr;\r
  out7name+=centr;\r
  inname+= finDirname.Data();\r
  out1name+= finDirname.Data();\r
  out2name+= finDirname.Data();\r
  out3name+= finDirname.Data();\r
  out4name+= finDirname.Data();\r
  out5name+= finDirname.Data();\r
  out6name+= finDirname.Data();\r
  out7name+= finDirname.Data();\r
\r
 \r
  AliAnalysisTaskSEHFQA* taskQA=new AliAnalysisTaskSEHFQA(Form("QA%s",suffix.Data()),ch,analysiscuts);\r
\r
  taskQA->SetReadMC(readMC);\r
  taskQA->SetSimpleMode(simplemode); // set to kTRUE to go faster in PbPb\r
  taskQA->SetTrackOn(onoff[0]);\r
  taskQA->SetPIDOn(onoff[1]);\r
  taskQA->SetCentralityOn(onoff[2]);\r
  taskQA->SetEvSelectionOn(onoff[3]);\r
  mgr->AddTask(taskQA);\r
\r
  //\r
  // Create containers for input/output\r
  AliAnalysisDataContainer *cinput = mgr->CreateContainer(inname,TChain::Class(), AliAnalysisManager::kInputContainer);\r
  mgr->ConnectInput(taskQA,0,mgr->GetCommonInputContainer());\r
\r
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(out1name,TH1F::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //events analysed\r
  mgr->ConnectOutput(taskQA,1,coutput1);\r
\r
  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(out2name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //PID\r
  if(onoff[1]) mgr->ConnectOutput(taskQA,2,coutput2);\r
\r
  AliAnalysisDataContainer *coutput3 = mgr->CreateContainer(out3name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of tracks\r
  if(onoff[0]) mgr->ConnectOutput(taskQA,3,coutput3);\r
\r
  AliAnalysisDataContainer *coutput4 = mgr->CreateContainer(out4name,AliRDHFCuts::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //cuts\r
  mgr->ConnectOutput(taskQA,4,coutput4);\r
\r
  AliAnalysisDataContainer *coutput5 = mgr->CreateContainer(out5name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of centrality\r
  if(onoff[2]) mgr->ConnectOutput(taskQA,5,coutput5);\r
\r
  AliAnalysisDataContainer *coutput6 = mgr->CreateContainer(out6name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //quality of centrality\r
  if(onoff[2]) mgr->ConnectOutput(taskQA,6,coutput6);\r
\r
  AliAnalysisDataContainer *coutput7 = mgr->CreateContainer(out7name,TList::Class(),AliAnalysisManager::kOutputContainer, filename.Data()); //event selection\r
  if(onoff[3]) mgr->ConnectOutput(taskQA,7,coutput7);\r
\r
 return taskQA;\r
}\r
\r