PWGJE/macros/AddTaskJetsReader.C

   1 Float_t kPtTrackMin = 0.15; // pt cut on tracks
   2 Int_t kDetectorType = 1;    // 0 = MC, 1 = TPC
   3
   4 AliAnalysisDataContainer*  cout_jet_cont = 0x0;
   5
   6 void SetJetReaderExchangeContainer(AliAnalysisDataContainer* acont) {
   7   cout_jet_cont = acont;
   8
   9 }
  10
  11 AliAnalysisDataContainer* GetJetReaderExchangeContainer() { return cout_jet_cont;}
  12 AliAnalysisDataContainer* AddJetExchangeContainer(const char* contname) { return AliAnalysisManager::GetAnalysisManager()->CreateContainer(contname, TTree::Class(),AliAnalysisManager::kExchangeContainer);}
  13
  14 AliJetReader *CreateJetReader(AliAnalysisDataContainer* contname,Char_t *jr,UInt_t filterMask); // Common config
  15
  16 AliAnalysisTaskJetsReader *AddTaskJetsReader(AliAnalysisDataContainer* contname,Char_t *jr, UInt_t filterMask = 0,Float_t ptTrackMin = 0.15 , Int_t dettype = 1); // for the new AF
  17 Int_t AddTaskJetsReaderDelta(AliAnalysisDataContainer* contname,char *nonStdFile = "",UInt_t filterMask = 0,Bool_t kUseAODMC = kTRUE,UInt_t runFlag = 1|4|32|128|256);
  18 AliAnalysisTaskJetsReader *AddTaskJetsReader(AliAnalysisDataContainer* contname,UInt_t filterMask = 0);
  19
  20 AliAnalysisTaskJetsReader *AddTaskJetsReader(const char* contname, UInt_t filterMask = 0) { return AddTaskJetsReader(AddJetExchangeContainer(contname),filterMask); } // LEGO trains
  21 AliAnalysisTaskJetsReader *AddTaskJetsReader(const char* contname,Char_t *jr, UInt_t filterMask = 0,Float_t ptTrackMin = 0.15 , Int_t dettype = 1) {
  22  return AddTaskJetsReader(AddJetExchangeContainer(contname),jr,filterMask,ptTrackMin,dettype);
  23
  24 }
  25
  26 AliAnalysisTaskJetsReader *AddTaskJetsReader(AliAnalysisDataContainer* contname,UInt_t filterMask){
  27   // fills the standard input "jets" branch in the AOD
  28   // need the ESDFilter to run before, to access the AODtracks
  29   // Tracks selected by the first Filter (1<<0)
  30   // needs to be adapted for different cuts
  31
  32
  33   return AddTaskJetsReader(contname,"AOD",filterMask);
  34
  35 }
  36
  37
  38
  39 Int_t AddTaskJetsReaderDelta(AliAnalysisDataContainer* contname,char *nonStdFile,UInt_t filterMask,Bool_t kUseAODMC,UInt_t runFlag){
  40
  41   // Adds a whole set of jet finders  all to be written
  42   // to a delta AOD
  43
  44   // this can in principle be done also with on the fly
  45   // if we have an ESD input jet fidner task does automatically fetch the ouput aod
  46
  47
  48   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  49   if (!mgr) {
  50     ::Error("AddTaskJetsReaderDelta", "No analysis manager to connect to.");
  51     return -1;
  52   }
  53
  54
  55   // Check the analysis type using the event handlers connected to the analysis manager.
  56   //==============================================================================
  57   if (!mgr->GetInputEventHandler()) {
  58     ::Error("AddTaskJetsReaderDelta", "This task requires an input event handler");
  59     return -1;
  60   }
  61
  62
  63   AliAODHandler *aodh = (AliAODHandler*)mgr->GetOutputEventHandler();
  64   if (!aodh) {
  65     ::Error("AddTaskJetsReaderDelta", "This task needs an output event handler");
  66     return -1;
  67   }
  68
  69   TString outFile(nonStdFile);
  70
  71
  72   AliAnalysisTaskJetsReader *taskjetsReader = 0;
  73   Int_t iCount = 0;
  74
  75   const char *cReader[4] = {"AOD","AODMC","AODMC2","AODMC2b"};
  76
  77
  78
  79   // Ligne a revoir
  80   //  if(!kUseAODMC)ib=1; // switch OFF MC if we do not have it
  81   for(int ib = 0;ib<4;ib++){
  82     if(1<<ib){
  83       cout << "cReader[" << ib << "]: " << cReader[ib] << ", filterMask: " << filterMask << endl;
  84       taskjetsReader = AddTaskJetsReader(contname,cReader[ib],filterMask);
  85       if(taskjetsReader){
  86         if(outFile.Length()>0)taskjetsReader->SetNonStdOutputFile(outFile.Data());
  87       }
  88     }
  89   }
  90   return 0;
  91 }
  92
  93 AliAnalysisTaskJetsReader *AddTaskJetsReader(AliAnalysisDataContainer* contname,Char_t *jr, UInt_t filterMask,Float_t ptTrackMin,Int_t dettype)
  94 {
  95   // Creates a jet finder task, configures it and adds it to the analysis manager.
  96
  97   kPtTrackMin = ptTrackMin;
  98   kDetectorType = dettype;
  99
 100   // Get the pointer to the existing analysis manager via the static access method.
 101   //==============================================================================
 102   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 103   if (!mgr) {
 104     ::Error("AddTaskJetsReader", "No analysis manager to connect to.");
 105     return NULL;
 106   }
 107
 108   // Check the analysis type using the event handlers connected to the analysis manager.
 109   //==============================================================================
 110   if (!mgr->GetInputEventHandler()) {
 111     ::Error("AddTaskJetsReader", "This task requires an input event handler");
 112     return NULL;
 113   }
 114
 115   AliAODHandler *aodh = (AliAODHandler*)mgr->GetOutputEventHandler();
 116   if (!aodh) {
 117     ::Error("AddTaskJetsReader", "This task needs an output event handler");
 118     return NULL;
 119   }
 120
 121
 122   // Create the task and configure it.
 123   //===========================================================================
 124   AliAnalysisTaskJetsReader *taskjetsReader;
 125   AliJetReader *er = CreateJetReader(contname,jr,filterMask);
 126
 127   taskjetsReader = new AliAnalysisTaskJetsReader(Form("JetAnalysis%s",jr));
 128
 129   // Connect jet finder to task.
 130   taskjetsReader->SetJetReader(er);
 131   taskjetsReader->SetConfigFile("");
 132   taskjetsReader->SetDebugLevel(-1);
 133   mgr->SetGlobalStr(cout_jet_cont->GetName(),Form("%s#_Filter%05d_Cut%05d",jr,filterMask,(int)((1000.*ptTrackMin))));
 134   mgr->AddTask(taskjetsReader);
 135
 136   cout_jet_cont = GetJetReaderExchangeContainer();
 137   if(! cout_jet_cont) { ::Error("AddTaskJetsReader", "This task needs an Exchange container");  return NULL;}
 138   // Create ONLY the output containers for the data produced by the task.
 139   // Get and connect other common input/output containers via the manager as below
 140   //==============================================================================
 141   mgr->ConnectInput  (taskjetsReader, 0, mgr->GetCommonInputContainer());
 142   mgr->ConnectOutput (taskjetsReader, 0, mgr->GetCommonOutputContainer());
 143   mgr->ConnectOutput (taskjetsReader, 1, cout_jet_cont);
 144
 145
 146   return taskjetsReader;
 147 }
 148
 149
 150 AliJetReader *CreateJetReader(AliAnalysisDataContainer* contname,Char_t *jr,UInt_t filterMask){
 151   AliJetReader *er = 0;
 152   SetJetReaderExchangeContainer(contname);
 153   Float_t ptCut  = kPtTrackMin ; // cut on track p_T
 154
 155   switch (jr) {
 156   case "ESD":
 157     AliJetReaderHeader *jrh = new AliJetReaderHeader(kDetectorType);
 158     jrh->SetDataType("ESD");
 159     jrh->SetComment("ESD Reader");
 160     jrh->SetDebug(-1);
 161     jrh->SetPtCut(ptCut);
 162     jrh->SetFiducialEta(-0.9,0.9);
 163     jrh->SetFiducialPhi(0,2*TMath::Pi());
 164     // Define reader and set its header
 165     er = new AliJetReader();
 166     er->SetReaderHeader(jrh);
 167     break;
 168   case "AOD":
 169     AliJetReaderHeader *jrh = new AliJetReaderHeader(kDetectorType);
 170     jrh->SetDataType("AOD");
 171     jrh->SetComment("AOD Reader");
 172     jrh->SetDebug(-1);
 173     jrh->SetPtCut(ptCut);
 174     jrh->SetTestFilterMask(32); // Change this one for a different set of cuts
 175     if(filterMask>0)jrh->SetTestFilterMask(filterMask);
 176     jrh->SetFilterType(0); // Filter type
 177     jrh->SetFiducialEta(-0.9,0.9);
 178     jrh->SetFiducialPhi(0,2*TMath::Pi());
 179     // Define reader and set its header
 180     er = new AliJetReader();
 181     er->SetReaderHeader(jrh);
 182     break;
 183   case "AODMC":
 184     AliJetReaderHeader *jrh = new AliJetReaderHeader(/*kDetectorType =*/0);
 185     jrh->SetComment("AOD MC Reader");
 186     jrh->SetDataType("AODMC");
 187     jrh->SetDebug(-1);
 188     jrh->SetPtCut(ptCut);
 189     jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0.9
 190     // Define reader and set its header
 191     er = new AliJetReader();
 192     er->SetReaderHeader(jrh);
 193     break;
 194   case "AODMC2":
 195     AliJetReaderHeader *jrh = new AliJetReaderHeader(/*kDetectorType =*/0);
 196     jrh->SetDataType("AODMC2");
 197     jrh->SetComment("AOD MC Reader");
 198     jrh->SetDebug(-1);
 199     jrh->SetPtCut(ptCut);
 200     jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0.9
 201     // Define reader and set its header
 202     er = new AliJetReader();
 203     er->SetReaderHeader(jrh);
 204     break;
 205   case "AODMC2b":
 206     AliJetReaderHeader *jrh = new AliJetReaderHeader(/*kDetectorType =*/0);
 207     jrh->SetDataType("AODMC2b");
 208     jrh->SetComment("AOD MC Reader");
 209     jrh->SetDebug(-1);
 210     jrh->SetPtCut(ptCut);
 211     jrh->SetFiducialEta(-0.9,0.9);
 212     // Define reader and set its header
 213     er = new AliJetReader();
 214     er->SetReaderHeader(jrh);
 215     break;
 216   case "AODextra":
 217     AliJetReaderHeader *jrh = new AliJetReaderHeader(kDetectorType);
 218     jrh->SetDataType("AODextra");
 219     jrh->SetComment("AOD Reader with extra branch");
 220     jrh->SetDebug(-1);
 221     jrh->SetPtCut(ptCut);
 222     jrh->SetTestFilterMask(32); // Change this one for a different set of cuts
 223     if(filterMask>0)jrh->SetTestFilterMask(filterMask);
 224     // Define reader and set its header
 225     er = new AliJetReader();
 226     er->SetReaderHeader(jrh);
 227     break;
 228   case "AODextraonly":
 229     AliJetReaderHeader *jrh = new AliJetReaderHeader(kDetectorType);
 230     jrh->SetDataType("AODextraonly");
 231     jrh->SetComment("AOD Reader with extra branch");
 232     jrh->SetDebug(-1);
 233     jrh->SetPtCut(ptCut);
 234     jrh->SetTestFilterMask(32); // Change this one for a different set of cuts
 235     if(filterMask>0)jrh->SetTestFilterMask(filterMask);
 236     // Define reader and set its header
 237     er = new AliJetReader();
 238     er->SetReaderHeader(jrh);
 239     break;
 240
 241   case "MC":
 242     AliJetKineReaderHeader *jrh = new AliJetKineReaderHeader();
 243     jrh->SetDataType("MC");
 244     jrh->SetComment("MC full Kinematics");
 245     jrh->SetDebug(-1);
 246     jrh->SetFastSimTPC(kFALSE);
 247     jrh->SetFastSimEMCAL(kFALSE);
 248     jrh->SetPtCut(ptCut);
 249     jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0.9
 250     // Define reader and set its header
 251     er = new AliJetReader();
 252     er->SetReaderHeader(jrh);
 253     break;
 254   case "MC2":
 255     AliJetKineReaderHeader *jrh = new AliJetKineReaderHeader();
 256     jrh->SetDataType("MC2");
 257     jrh->SetComment("MC full Kinematics spearate config charged only");
 258     jrh->SetDebug(-1);
 259     jrh->SetFastSimTPC(kFALSE);
 260     jrh->SetFastSimEMCAL(kFALSE);
 261     jrh->SetChargedOnly(kTRUE);
 262     jrh->SetPtCut(ptCut);
 263     jrh->SetFiducialEta(-2.1,2.1); // to take all MC particles default is 0 .9
 264     // Define reader and set its header
 265     er = new AliJetReader();
 266     er->SetReaderHeader(jrh);
 267     break;
 268
 269   default:
 270     ::Error("AddTaskJetsReader", Form("Wrong jet reader selected: %s\n",jr));
 271     return 0x0;
 272     break;
 273   }
 274
 275   return er;
 276
 277 }