PWGJE/macros/AddTaskJetCluster.C

   1 AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec = "AOD",char* bGen = "",UInt_t filterMask = 16, UInt_t iPhysicsSelectionFlag = AliVEvent::kAny,Char_t *jf = "KT", Float_t radius = 0.4,Int_t nSkip = 0,Int_t kWriteAOD = kFALSE,char* deltaFile = "",Float_t ptTrackCut = 0.15, Float_t etaTrackWindow = 0.9,Float_t vertexWindow = 10.,Int_t nSkipCone = 2);
   2
   3 Int_t kBackgroundMode = 0;
   4 Float_t kPtTrackCut = 0.15;
   5 Float_t kTrackEtaWindow = 0.8;
   6 Float_t kVertexWindow = 10;
   7
   8 AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag){
   9    AliAnalysisTaskJetCluster *js = 0;
  10    if(kUseAODMC&&false){// do not use the MC info yet
  11      if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001);
  12      if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.1);
  13      if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.2);
  14      if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.4);
  15      if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.6);
  16      if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.8);
  17      if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,1.0);
  18    }
  19    else{
  20      if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.00001);
  21      if(iFlag&(1<<1))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.1);
  22      if(iFlag&(1<<2))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.2);
  23      if(iFlag&(1<<4))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.4);
  24      if(iFlag&(1<<6))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.6);
  25      if(iFlag&(1<<8))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,0.8);
  26      if(iFlag&(1<<10))js = AddTaskJetCluster("AOD","",filterMask,iPhysicsSelectionFlag,jf,1.0);
  27    }
  28
  29    return js;
  30  }
  31
  32
  33 AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filterMask,UInt_t iPhysicsSelectionFlag,Char_t *jf,Float_t radius,Int_t nSkip,Int_t kWriteAOD,char *deltaFile,Float_t ptTrackCut,Float_t etaTrackWindow,Float_t vertexWindow,Int_t nSkipCone)
  34  {
  35  // Creates a jet fider task, configures it and adds it to the analysis manager.
  36    kPtTrackCut = ptTrackCut;
  37    kTrackEtaWindow = etaTrackWindow;
  38    kVertexWindow = vertexWindow;
  39
  40    TString outputFile(deltaFile);
  41     // Get the pointer to the existing analysis manager via the static access method.
  42     //==============================================================================
  43     AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  44     if (!mgr) {
  45        ::Error("AddTaskJetCluster", "No analysis manager to connect to.");
  46        return NULL;
  47     }
  48
  49     // Check the analysis type using the event handlers connected to the analysis manager.
  50     //==============================================================================
  51     if (!mgr->GetInputEventHandler()) {
  52       ::Error("AddTaskJetCluster", "This task requires an input event handler");
  53        return NULL;
  54     }
  55
  56     TString type = mgr->GetInputEventHandler()->GetDataType();
  57     TString typeRec(bRec);
  58     TString typeGen(bGen);
  59     if(!typeRec.Contains("AODextra")) {
  60       typeGen.ToUpper();
  61       typeRec.ToUpper();
  62     }
  63     cout << "typeRec: " << typeRec << endl;
  64     // Create the task and configure it.
  65     //===========================================================================
  66
  67
  68
  69
  70     TString cAdd = "";
  71     cAdd += Form("%02d_",(int)((radius+0.01)*10.));
  72     cAdd += Form("B%d",(int)kBackgroundMode);
  73     cAdd += Form("_Filter%05d",filterMask);
  74     cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCut));
  75     cAdd += Form("_Skip%02d",nSkip);
  76     Printf("%s %E",cAdd.Data(),kPtTrackCut);
  77     AliAnalysisTaskJetCluster* clus = new  AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data()));
  78
  79    // or a config file
  80    // clus->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);
  81    // if(iAODanalysis)clus->SetAODInput(kTRUE);
  82    // clus->SetDebugLevel(11);
  83    clus->SetFilterMask(filterMask);
  84    //   clus->SetUseGlobalSelection(kTRUE);
  85    clus->SetVtxCuts(kVertexWindow,1);
  86    if(type == "AOD"){
  87      // Assume all jet are produced already
  88      clus->SetAODTrackInput(kTRUE);
  89      clus->SetAODMCInput(kTRUE);
  90    }
  91
  92    if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD
  93      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);
  94      clus->SetTrackPtCut(kPtTrackCut);
  95      clus->SetTrackEtaWindow(kTrackEtaWindow);
  96    }
  97    else if (typeRec.Contains("AODMC2")){
  98      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);
  99      clus->SetTrackPtCut(kPtTrackCut);
 100      clus->SetTrackEtaWindow(5);
 101    }
 102    else if (typeRec.Contains("AODMC")){
 103      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);
 104      clus->SetTrackPtCut(kPtTrackCut);
 105      clus->SetTrackEtaWindow(5);
 106    }
 107    else if (typeRec.Contains("AODextraonly")) {
 108      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly);
 109      clus->SetTrackPtCut(kPtTrackCut);
 110      clus->SetTrackEtaWindow(kTrackEtaWindow);
 111    }
 112    else if (typeRec.Contains("AODextra")) {
 113      cout << "AliAnalysisTaskJetCluster::kTrackAODextra: " << AliAnalysisTaskJetCluster::kTrackAODextra << endl;
 114      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra);
 115      clus->SetTrackPtCut(kPtTrackCut);
 116      clus->SetTrackEtaWindow(kTrackEtaWindow);
 117    }
 118    else if (typeRec.Contains("AOD")) {
 119      clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);
 120      clus->SetTrackPtCut(kPtTrackCut);
 121      clus->SetTrackEtaWindow(kTrackEtaWindow);
 122    }
 123
 124    clus->SetRparam(radius);
 125    clus->SetGhostArea(0.005);
 126    clus->SetGhostEtamax(kTrackEtaWindow);
 127
 128    switch (jf) {
 129    case "ANTIKT":
 130      clus->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh
 131      break;
 132    case "CA":
 133      clus->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh
 134      break;
 135    case "KT":
 136      clus->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh
 137      break;
 138    default:
 139      ::Error("AddTaskJetCluster", "Wrong jet finder selected\n");
 140      return 0;
 141    }
 142
 143
 144    if(kWriteAOD){
 145      if(outputFile.Length())clus->SetJetOutputFile(outputFile);
 146      clus->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));
 147      clus->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold
 148    }
 149
 150    clus->SetNSkipLeadingRan(nSkip);
 151    clus->SetNSkipLeadingCone(nSkipCone);
 152
 153    if(iPhysicsSelectionFlag)clus->SelectCollisionCandidates(iPhysicsSelectionFlag);
 154
 155    mgr->AddTask(clus);
 156
 157    // Create ONLY the output containers for the data produced by the task.
 158    // Get and connect other common input/output containers via the manager as below
 159    //==============================================================================
 160    AliAnalysisDataContainer *coutput1_Spec = mgr->CreateContainer(Form("cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWGJE_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data()));
 161
 162    mgr->ConnectInput  (clus, 0, mgr->GetCommonInputContainer());
 163    mgr->ConnectOutput (clus, 0, mgr->GetCommonOutputContainer());
 164    mgr->ConnectOutput (clus,  1, coutput1_Spec );
 165
 166    return clus;
 167 }