Merge branch 'master' into TRDdev

[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / macros / AddTaskEMCALIsoPhoton.C

AliAnalysisTaskEMCALIsoPhoton *AddTaskEMCALIsoPhoton(
						     TString period = "LHC11d",
						     TString trigbitname = "kEMC7",
						     TString geoname="EMCAL_COMPLETEV1",
						     TString pathstrsel = "/",
						     TString trackSelType = "standard"
						     )
{
  // Get the pointer to the existing analysis manager via the static access method.
  //==============================================================================
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    ::Error("AddTaskEMCALIsoPhoton", "No analysis manager to connect to.");
    return NULL;
  }  
  
  // Create the task and configure it.
  //===========================================================================
  AliAnalysisTaskEMCALIsoPhoton* ana = new  AliAnalysisTaskEMCALIsoPhoton("");
  
  Bool_t isMC = (mgr->GetMCtruthEventHandler() != NULL);

  if(!isMC)
    ana->SelectCollisionCandidates( AliVEvent::kEMC1 | AliVEvent::kMB | AliVEvent::kEMC7 | AliVEvent::kINT7);
  
  //ana->SetClusThreshold(clusTh);
  
  ana->SetTrainMode(kTRUE);
  ana->SetTriggerBit(trigbitname);
  ana->SetMcMode(isMC);
  ana->SetPathStringSelect(pathstrsel.Data());
  gROOT->LoadMacro("$ALICE_ROOT/PWGJE/macros/CreateTrackCutsPWGJE.C");
  AliESDtrackCuts *cutsp = new AliESDtrackCuts;
  if(trackSelType == "standard"){
    cutsp->SetMinNClustersTPC(70);
    cutsp->SetMinRatioCrossedRowsOverFindableClustersTPC(0.8);
    cutsp->SetMaxChi2PerClusterTPC(4);
    cutsp->SetRequireTPCRefit(kTRUE);
    cutsp->SetAcceptKinkDaughters(kFALSE);
    cutsp->SetMaxDCAToVertexZ(3.2);
    cutsp->SetMaxDCAToVertexXY(2.4);
    cutsp->SetDCAToVertex2D(kTRUE);
    cutsp->SetPtRange(0.2);
    cutsp->SetEtaRange(-1.0,1.0);
  }
  if(trackSelType == "hybrid"){
    cutsp = CreateTrackCutsPWGJE(10001008);
    ana->SetHybridOn();
  }
  if(trackSelType == "hybridp"){
    cutsp = CreateTrackCutsPWGJE(10011008);
    ana->SetHybridOn();
  }
  ana->SetPrimTrackCuts(cutsp);
  ana->SetPeriod(period.Data());
  ana->SetGeoName(geoname.Data());  
  //ana->SetTrackFilterBit(128);
  mgr->AddTask(ana);
  TString containername = "histEMCIsoPhoton."+trigbitname;
  TString containernameQA = "histosQA."+trigbitname;
  if(pathstrsel != "/"){
    TString dirpth = (TSubString)pathstrsel.operator()(1,1);
    containername += dirpth;
    containernameQA  += dirpth;
  }
  
  // Create ONLY the output containers for the data produced by the task.
  // Get and connect other common input/output containers via the manager as below
  //==============================================================================
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(containername.Data(), 
							    TList::Class(),AliAnalysisManager::kOutputContainer,
							    Form("%s", AliAnalysisManager::GetCommonFileName()));

  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(containernameQA.Data(), 
							    TList::Class(),AliAnalysisManager::kOutputContainer,
							    Form("%s", AliAnalysisManager::GetCommonFileName()));
  
  mgr->ConnectInput  (ana, 0, mgr->GetCommonInputContainer());
  mgr->ConnectOutput (ana, 1, coutput1 );
  mgr->ConnectOutput (ana, 2, coutput2 );
   
  return ana;
}