2+1 analysis from Raghava included in PWGCFCorrelationsDPhi libraries

[u/mrichter/AliRoot.git] / PWGCF / Correlations / macros / twoplusone / AddTaskDiJetCorrelations.C

// $Id: AddTaskDiJetCorrelations.C

AliAnalysisTaskDiJetCorrelations *AddTaskDiJetCorrelations(TString suffixName="",
							   Bool_t SEorME = kTRUE,
							   Bool_t ppOrPbPb = kTRUE,
							   Double_t pTrg1min = 12.0,
							   Double_t pTrg1max = 16.0,
							   Double_t pTrg2min = 5.0,
							   Double_t pTrg2max = 8.0)
{
  
  Bool_t UseFbits = kTRUE;
  
  //____________________________________| Correlation class setting..
  AliAnalysisTaskDiJetCorrelations *dijetcorrelations = new AliAnalysisTaskDiJetCorrelations("");
  dijetcorrelations->
    (AliVEvent::kMB);
  dijetcorrelations->SetSystem(ppOrPbPb); //PbPb = kTRUE
  dijetcorrelations->SetSEorME(SEorME); //kTRUE for mixed events
  if(SEorME)dijetcorrelations->SetMESettings(500, 25000, 8); //evt,track,minMixEvents
  dijetcorrelations->SetTrigger1PTValue(pTrg1min, pTrg1max); //GeV/c
  dijetcorrelations->SetTrigger2PTValue(pTrg2min, pTrg2max); //GeV/c
  dijetcorrelations->SetFilterBit(UseFbits);
  if(UseFbits)dijetcorrelations->SetFilterType(272);
  if(ppOrPbPb)dijetcorrelations->SetCentralityRange(0., 100.1); // 0-100%
  dijetcorrelations->SetDataType(kTRUE); //track Data/MC tracks=1 or MC Part=0?
  dijetcorrelations->SetVarCentBin(kTRUE); // kTRUE have some trouble ! FIX ME
  dijetcorrelations->SetVarPtBin(kTRUE); // kTRUE have some trouble ! FIX ME
   
  // Create containers for input/output        
  TString finDirname         = "_DiJetMayCERN";
  TString inname             = "cinputDiJetCorrelations";
  TString outBasicname       = "coutputDiJetBasicPlots";
  TString outCorrname        = "coutputDiJetCorrHistos";

  finDirname += suffixName.Data();
  inname            +=   finDirname.Data();
  outBasicname      +=   finDirname.Data();
  outCorrname       +=   finDirname.Data();
  
  
  // Get the pointer to the existing analysis manager via the static access method.                                                     
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    cout<<"AddTaskDiJetCorrelations", "No analysis manager to connect to."<<endl;
  }
  
  mgr->AddTask(dijetcorrelations);
  
  //Input and Output Slots:
  AliAnalysisDataContainer *cinputDiJetCorrelations = mgr->CreateContainer(inname,TChain::Class(), AliAnalysisManager::kInputContainer);
  //TString outputfile = AliAnalysisManager::GetCommonFileName();
  //outputfile += ":PWGCF_Di_Jet_Corr";
  TString outputfile = "resultsDiJetCorrelationsT112to16T25to8Dec8.root";

  
  AliAnalysisDataContainer *coutputDiJetCorrelations1 = mgr->CreateContainer(outBasicname,TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
  AliAnalysisDataContainer *coutputDiJetCorrelations2 = mgr->CreateContainer(outCorrname,TList::Class(),AliAnalysisManager::kOutputContainer,outputfile.Data());
  
  
  mgr->ConnectInput(dijetcorrelations,0,mgr->GetCommonInputContainer());
  mgr->ConnectOutput(dijetcorrelations,1,coutputDiJetCorrelations1);
  mgr->ConnectOutput(dijetcorrelations,2,coutputDiJetCorrelations2);
  
  return dijetcorrelations;
  
}