Changes to run in central train

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / macros / AddTaskDStarSpectra.C

//if like define a different number of signal for TPC PID\r
//by default the task is anyway computing 1, 2 and 3 sigmas\r
const Int_t    numberOfSigmasPID  = 2;\r
// option to switch on and off the TPC PID.\r
const Bool_t usePID = kTRUE;\r
// analysis type... TO BE REMOVED!!!\r
const Bool_t anaType   = 0;//0 HD; 1 UU;\r
//----------------------------------------------------\r
\r
AliAnalysisTaskSEDStarSpectra *AddTaskDStarSpectra(Bool_t theMCon=kFALSE)\r
{\r
  \r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  if (!mgr) {\r
    ::Error("AddTaskDStarSpectra", "No analysis manager to connect to.");\r
    return NULL;\r
  }  \r
  \r
  // cuts are stored in a TFile generated by makeTFile4CutsDStartoKpipi.C in ./macros/\r
  // set there the cuts!!!!!\r
  Bool_t stdcuts=kFALSE;\r
  TFile* filecuts=new TFile("DStartoKpipiCuts.root");\r
  if(!filecuts->IsOpen()){\r
    cout<<"Input file not found: exit"<<endl;\r
    stdcuts=kTRUE;\r
  }\r
\r
  AliRDHFCutsDStartoKpipi* RDHFDStartoKpipi=new AliRDHFCutsDStartoKpipi();\r
  if(stdcuts) RDHFDStartoKpipi->SetStandardCutsPP2010();\r
  else RDHFDStartoKpipi = (AliRDHFCutsDStartoKpipi*)filecuts->Get("DStartoKpipiCuts");\r
  RDHFDStartoKpipi->SetName("DStartoKpipiCuts");\r
\r
  // mm let's see if everything is ok\r
  if(!RDHFDStartoKpipi){\r
    cout<<"Specific AliRDHFCuts not found"<<endl;\r
    return;\r
  }\r
 \r
  //CREATE THE TASK\r
  printf("CREATE TASK\n");\r
  // create the task\r
  AliAnalysisTaskSEDStarSpectra *task = new AliAnalysisTaskSEDStarSpectra("AliAnalysisTaskSEDStarSpectra",RDHFDStartoKpipi);\r
  task->SetAnalysisType(anaType);\r
  task->SetNSigmasPID(numberOfSigmasPID);\r
  task->SetMC(theMCon);\r
  task->SetPID(usePID);\r
  task->SetDebugLevel(0);\r
\r
  mgr->AddTask(task);\r
\r
  // Create and connect containers for input/output\r
  \r
  TString outputfile = AliAnalysisManager::GetCommonFileName();\r
  outputfile += ":PWG3_D2H_DStarSpectra";\r
  \r
  // ------ input data ------\r
\r
  //AliAnalysisDataContainer *cinput0  = mgr->GetCommonInputContainer();\r
  AliAnalysisDataContainer *cinput0  =  mgr->CreateContainer("indstar",TChain::Class(), \r
							     AliAnalysisManager::kInputContainer);\r
  // ----- output data -----\r
  \r
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("chist1",TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar1 = mgr->CreateContainer("DStarSpectrum",TList::Class(),AliAnalysisManager::kOutputContainer, outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar2 = mgr->CreateContainer("DStarAll",TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar3 = mgr->CreateContainer("DStarPID3",TList::Class(),AliAnalysisManager::kOutputContainer, outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar4 = mgr->CreateContainer("DStarPID2",TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar5 = mgr->CreateContainer("DStarPID1",TList::Class(),AliAnalysisManager::kOutputContainer, outputfile.Data());\r
  AliAnalysisDataContainer *coutputDStar6 = mgr->CreateContainer("cuts",AliRDHFCutsDStartoKpipi::Class(),AliAnalysisManager::kOutputContainer, outputfile.Data()); //cuts\r
\r
  mgr->ConnectInput(task,0,mgr->GetCommonInputContainer());\r
\r
  mgr->ConnectOutput(task,1,coutput1);\r
  mgr->ConnectOutput(task,2,coutputDStar1);\r
  mgr->ConnectOutput(task,3,coutputDStar2);\r
  mgr->ConnectOutput(task,4,coutputDStar3);\r
  mgr->ConnectOutput(task,5,coutputDStar4);\r
  mgr->ConnectOutput(task,6,coutputDStar5);\r
  mgr->ConnectOutput(task,7,coutputDStar6);\r
  \r
  return task;\r
}\r
\r