PWGJE/EMCALJetTasks/macros/AddTaskEmcalDiJetResponse.C

   1 AliAnalysisTaskEmcalDiJetResponse* AddTaskEmcalDiJetResponse(TString     kTracksName         = "PicoTracks",
   2                                                              TString     kClusName           = "caloClusterCorr",
   3                                                              TString     kMCTracksName       = "MCParticles",
   4                                                              Double_t    R                   = 0.4,
   5                                                              Double_t    ptminTrack          = 0.15,
   6                                                              Double_t    etminClus           = 0.3,
   7                                                              Int_t       rhoType             = 0,
   8                                                              TString     trigClass           = "",
   9                                                              const char *CentEst             = "V0A",
  10                                                              Int_t       pSel                = AliVEvent::kINT7,
  11                                                              Int_t       matchFullCh         = AliAnalysisTaskEmcalDiJetBase::kNoMatching,
  12                                                              Double_t    ptTrackBias         = 0.,
  13                                                              Int_t       responseVar         = 0,
  14                                                              Int_t       corrType            = AliAnalysisTaskEmcalDiJetBase::kCorrelateTwo,
  15                                                              Float_t     nefCut              = 0.95
  16                                                              ) {
  17
  18   enum AlgoType {kKT, kANTIKT};
  19   enum JetType  {kFULLJETS, kCHARGEDJETS, kNEUTRALJETS};
  20
  21   // #### Define manager and data container names
  22   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  23   if (!mgr) {
  24     ::Error("AddTaskEmcalDiJet", "No analysis manager to connect to.");
  25     return NULL;
  26   }
  27
  28   // Check the analysis type using the event handlers connected to the analysis manager.
  29   //==============================================================================
  30   if (!mgr->GetInputEventHandler())
  31     {
  32       ::Error("AddTaskEmcalDiJet", "This task requires an input event handler");
  33       return NULL;
  34     }
  35
  36   // #### Add necessary jet finder tasks
  37   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJet.C");
  38
  39   AliEmcalJetTask* jetFinderTaskFull    = AddTaskEmcalJet(kTracksName, kClusName, kANTIKT, R, kFULLJETS, ptminTrack, etminClus);
  40   AliEmcalJetTask* jetFinderTaskCharged = AddTaskEmcalJet(kTracksName, kClusName, kANTIKT, R, kCHARGEDJETS, ptminTrack, etminClus);
  41   jetFinderTaskFull->SelectCollisionCandidates(AliVEvent::kAny);
  42   jetFinderTaskCharged->SelectCollisionCandidates(AliVEvent::kAny);
  43
  44   AliEmcalJetTask* jetFinderTaskFullMC = AddTaskEmcalJet(kMCTracksName ,"", kANTIKT, R, kFULLJETS, kPartLevPtCut, kPartLevPtCut);
  45   AliEmcalJetTask* jetFinderTaskChargedMC = AddTaskEmcalJet(kMCTracksName ,"", kANTIKT, R, kCHARGEDJETS, kPartLevPtCut, kPartLevPtCut);
  46   jetFinderTaskFullMC->SelectCollisionCandidates(AliVEvent::kAny);
  47   jetFinderTaskChargedMC->SelectCollisionCandidates(AliVEvent::kAny);
  48
  49   TString strJetsFull = jetFinderTaskFull->GetName();
  50   TString strJetsCh   = jetFinderTaskCharged->GetName();
  51
  52   TString strJetsFullMC = jetFinderTaskFullMC->GetName();
  53   TString strJetsChMC   = jetFinderTaskChargedMC->GetName();
  54
  55
  56   TString wagonName = Form("DiJetResponse_%s_%s_Rho%dTC%sMatch%dHadTrig%dRV%d",strJetsFull.Data(),strJetsFullMC.Data(),rhoType,trigClass.Data(),matchFullCh,(Int_t)(ptTrackBias),responseVar);
  57
  58   //Configure DiJet task
  59   AliAnalysisTaskEmcalDiJetResponse *taskDiJet = new AliAnalysisTaskEmcalDiJetResponse(wagonName.Data());
  60
  61   Printf("strJetsFull: %s",strJetsFull.Data());
  62   Printf("strJetsCh: %s",strJetsCh.Data());
  63
  64   taskDiJet->SetUseAliAnaUtils(kTRUE);
  65   taskDiJet->SetIsPythia(kTRUE);
  66
  67   taskDiJet->SetJetCorrelationType(corrType);
  68
  69   taskDiJet->SetContainerFull(0);
  70   taskDiJet->SetContainerCharged(1);
  71   taskDiJet->SetContainerFullMC(2);
  72   taskDiJet->SetContainerChargedMC(3);
  73
  74   taskDiJet->AddParticleContainer(kTracksName.Data());
  75   taskDiJet->AddClusterContainer(kClusName.Data());
  76
  77   taskDiJet->AddJetContainer(strJetsFull.Data(),"EMCAL",R);
  78   taskDiJet->AddJetContainer(strJetsCh.Data(),"TPC",R);
  79   taskDiJet->AddJetContainer(strJetsFullMC.Data(),"EMCAL",R);
  80   taskDiJet->AddJetContainer(strJetsChMC.Data(),"TPC",R);
  81
  82   taskDiJet->SetZLeadingCut(0.98,0.98,0);
  83   taskDiJet->SetZLeadingCut(0.98,0.98,2);
  84
  85   taskDiJet->SetNEFCut(0.,nefCut,0);
  86   taskDiJet->SetNEFCut(0.,nefCut,2);
  87
  88   for(Int_t i=0; i<4; i++) {
  89     taskDiJet->SetPercAreaCut(0.6, i);
  90     taskDiJet->SetPtBiasJetTrack(ptTrackBias,i);
  91   }
  92
  93   taskDiJet->SetRhoType(rhoType);
  94
  95   taskDiJet->SetCentralityEstimator(CentEst);
  96
  97   taskDiJet->SelectCollisionCandidates(pSel);
  98
  99   taskDiJet->SetFullChargedMatchingType(matchFullCh);
 100
 101   taskDiJet->SetDoChargedCharged(kTRUE);
 102   taskDiJet->SetDoFullCharged(kTRUE);
 103   taskDiJet->SetMatchFullCharged(kTRUE);
 104
 105   taskDiJet->SetResponseVar(responseVar);
 106
 107   taskDiJet->SetPtMinTriggerJet(0.);
 108
 109   mgr->AddTask(taskDiJet);
 110
 111   //Connnect input
 112   mgr->ConnectInput (taskDiJet, 0, mgr->GetCommonInputContainer() );
 113
 114   //Connect output
 115   AliAnalysisDataContainer *coutput1 = 0x0;
 116
 117   TString contName(wagonName);
 118   contName += "_histos";
 119
 120   //  TString outputfile = Form("%s:%s",AliAnalysisManager::GetCommonFileName(),wagonName.Data());
 121   TString outputfile = Form("%s",AliAnalysisManager::GetCommonFileName());
 122
 123   coutput1 = mgr->CreateContainer(contName.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,outputfile);
 124
 125   mgr->ConnectOutput(taskDiJet,1,coutput1);
 126
 127   return taskDiJet;
 128 }