]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGJE/macros/AddTaskJetCluster.C
fix bug from last commit - Marta V.
[u/mrichter/AliRoot.git] / PWGJE / macros / AddTaskJetCluster.C
index 87554e9184971b7b349b051e05ef3d8f71c97932..6f2c48ed44c6a84bc6e998bc728d86da07c16d69 100644 (file)
@@ -1,11 +1,12 @@
-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);
+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,Int_t dice=0,Int_t smear=0,Bool_t useOADB=kFALSE,Double_t changeEfficiencyFraction=0.);
+AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag);
 
-Int_t kBackgroundMode = 0;
-Float_t kPtTrackCut = 0.15;
-Float_t kTrackEtaWindow = 0.8;
-Float_t kVertexWindow = 10;
+Int_t kBackgroundModeCl = 0;
+Float_t kPtTrackCutCl = 0.15;
+Float_t kTrackEtaWindowCl = 0.8;
+Float_t kVertexWindowCl = 10;
 
-AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t kUseAODMC = kFALSE,UInt_t iPhysicsSelectionFlag = AliVEvent::kMB,Char_t *jf = "KT", UInt_t iFlag){
+AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask,Bool_t kUseAODMC,UInt_t iPhysicsSelectionFlag,Char_t *jf, UInt_t iFlag){
    AliAnalysisTaskJetCluster *js = 0;
    if(kUseAODMC&&false){// do not use the MC info yet
      if(iFlag&(1<<0))js = AddTaskJetCluster("AOD","AODMC",filterMask,iPhysicsSelectionFlag,jf,0.00001);
@@ -30,12 +31,12 @@ AliAnalysisTaskJetCluster *AddTaskJetClusterDelta(UInt_t filterMask = 16,Bool_t
  }
 
 
-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)
+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,Int_t dice,Int_t smear,Bool_t useOADB,Double_t changeEfficiencyFraction)
  {
  // Creates a jet fider task, configures it and adds it to the analysis manager.
-   kPtTrackCut = ptTrackCut;
-   kTrackEtaWindow = etaTrackWindow;
-   kVertexWindow = vertexWindow;
+   kPtTrackCutCl = ptTrackCut;
+   kTrackEtaWindowCl = etaTrackWindow;
+   kVertexWindowCl = vertexWindow;
 
    TString outputFile(deltaFile);
     // Get the pointer to the existing analysis manager via the static access method.
@@ -56,7 +57,7 @@ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filte
     TString type = mgr->GetInputEventHandler()->GetDataType();
     TString typeRec(bRec);
     TString typeGen(bGen);
-    if(!typeRec.Contains("AODextra")) {
+    if(!typeRec.Contains("AODextra") && !typeRec.Contains("AODMCextra")) {
       typeGen.ToUpper();
       typeRec.ToUpper();
     }
@@ -69,71 +70,89 @@ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filte
 
     TString cAdd = "";
     cAdd += Form("%02d_",(int)((radius+0.01)*10.));
-    cAdd += Form("B%d",(int)kBackgroundMode);
+    cAdd += Form("B%d",(int)kBackgroundModeCl);
     cAdd += Form("_Filter%05d",filterMask);
-    cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCut));
+    cAdd += Form("_Cut%05d",(int)(1000.*kPtTrackCutCl));
     cAdd += Form("_Skip%02d",nSkip);
-    Printf("%s %E",cAdd.Data(),kPtTrackCut);
-    AliAnalysisTaskJetCluster* pwg4clus = new  AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data()));
+    if(dice>0 || smear>0)
+      cAdd += Form("Detector%d%dFr%d",dice,smear,(int)(changeEfficiencyFraction*100.));
+    
+
+    Printf("%s %E %d %d",cAdd.Data(),kPtTrackCutCl,dice,smear);
+    AliAnalysisTaskJetCluster* clus = new  AliAnalysisTaskJetCluster(Form("JetCluster%s_%s%s",bRec,jf,cAdd.Data()));
       
    // or a config file
-   // pwg4clus->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);
-   // if(iAODanalysis)pwg4clus->SetAODInput(kTRUE);
-   // pwg4clus->SetDebugLevel(11); 
-   pwg4clus->SetFilterMask(filterMask); 
-   //   pwg4clus->SetUseGlobalSelection(kTRUE); 
-   pwg4clus->SetVtxCuts(kVertexWindow,1);
+   // clus->SetAnalysisType(AliAnalysisTaskJetCluster::kAnaMC);
+   // if(iAODanalysis)clus->SetAODInput(kTRUE);
+   // clus->SetDebugLevel(11); 
+   clus->SetFilterMask(filterMask); 
+   //   clus->SetUseGlobalSelection(kTRUE); 
+   clus->SetVtxCuts(kVertexWindowCl,1);
    if(type == "AOD"){
      // Assume all jet are produced already
-     pwg4clus->SetAODTrackInput(kTRUE);
-     pwg4clus->SetAODMCInput(kTRUE);
+     clus->SetAODTrackInput(kTRUE);
+     clus->SetAODMCInput(kTRUE);
    }
 
-   if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(kTrackEtaWindow);
+ if (typeRec.Contains("AODMCHF")){
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCHF);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
+   }
+   else if(typeRec.Contains("AODMC2b")){// work down from the top AODMC2b -> AODMC2 -> AODMC -> AOD
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCChargedAcceptance);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
    }
    else if (typeRec.Contains("AODMC2")){
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(5);
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCCharged);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(5);
+   }
+   else if (typeRec.Contains("AODMCextraonly")) {
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCextraonly);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
+   }
+   else if (typeRec.Contains("AODMCextra")) {
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCextra);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
    }
    else if (typeRec.Contains("AODMC")){
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(5);
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODMCAll);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(5);
    }
    else if (typeRec.Contains("AODextraonly")) {
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(kTrackEtaWindow);
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextraonly);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
    }
    else if (typeRec.Contains("AODextra")) {
-     cout << "AliAnalysisTaskJetCluster::kTrackAODextra: " << AliAnalysisTaskJetCluster::kTrackAODextra << endl;
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(kTrackEtaWindow);
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAODextra);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
    }
    else if (typeRec.Contains("AOD")) {
-     pwg4clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);
-     pwg4clus->SetTrackPtCut(kPtTrackCut);
-     pwg4clus->SetTrackEtaWindow(kTrackEtaWindow);
+     clus->SetTrackTypeRec(AliAnalysisTaskJetCluster::kTrackAOD);
+     clus->SetTrackPtCut(kPtTrackCutCl);
+     clus->SetTrackEtaWindow(kTrackEtaWindowCl);
    }
 
-   pwg4clus->SetRparam(radius);
-   pwg4clus->SetGhostArea(0.005);
-   pwg4clus->SetGhostEtamax(kTrackEtaWindow);
+   clus->SetRparam(radius);
+   clus->SetGhostArea(0.005);
+   clus->SetGhostEtamax(kTrackEtaWindowCl);
 
    switch (jf) {
    case "ANTIKT":
-     pwg4clus->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh
+     clus->SetAlgorithm(2); // antikt from fastjet/JetDefinition.hh
      break;
    case "CA":
-     pwg4clus->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh
+     clus->SetAlgorithm(1); // CA from fastjet/JetDefinition.hh
      break;
    case "KT":
-     pwg4clus->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh
+     clus->SetAlgorithm(0); // kt from fastjet/JetDefinition.hh
      break;
    default:
      ::Error("AddTaskJetCluster", "Wrong jet finder selected\n");
@@ -142,26 +161,32 @@ AliAnalysisTaskJetCluster *AddTaskJetCluster(char* bRec,char* bGen ,UInt_t filte
 
    
    if(kWriteAOD){
-     if(outputFile.Length())pwg4clus->SetJetOutputFile(outputFile);
-     pwg4clus->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));
-     pwg4clus->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold
+     if(outputFile.Length())clus->SetJetOutputFile(outputFile);
+     clus->SetJetOutputBranch(Form("clusters%s_%s%s",bRec,jf,cAdd.Data()));
+     clus->SetJetOutputMinPt(0); // store only jets / clusters above a certain threshold
    }
 
-   pwg4clus->SetNSkipLeadingRan(nSkip);
-   pwg4clus->SetNSkipLeadingCone(nSkipCone);
+   clus->SetNSkipLeadingRan(nSkip);
+   clus->SetNSkipLeadingCone(nSkipCone);
+
+   if(iPhysicsSelectionFlag)clus->SelectCollisionCandidates(iPhysicsSelectionFlag);
 
-   if(iPhysicsSelectionFlag)pwg4clus->SelectCollisionCandidates(iPhysicsSelectionFlag);
+   if(useOADB) {
+     clus->SetUseTrResolutionFromOADB();
+     clus->SetUseTrEfficiencyFromOADB();
+     clus->SetChangeEfficiencyFraction(changeEfficiencyFraction);
+   }
 
-   mgr->AddTask(pwg4clus);
+   mgr->AddTask(clus);
 
    // 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_Spec = mgr->CreateContainer(Form("pwg4cluster_%s_%s_%s%s",bRec,bGen,jf,cAdd.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_cluster_%s_%s_%s%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen,jf,cAdd.Data()));
+   AliAnalysisDataContainer *coutput1_clus = 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()));
 
-   mgr->ConnectInput  (pwg4clus, 0, mgr->GetCommonInputContainer());
-   mgr->ConnectOutput (pwg4clus, 0, mgr->GetCommonOutputContainer());
-   mgr->ConnectOutput (pwg4clus,  1, coutput1_Spec );
+   mgr->ConnectInput  (clus, 0, mgr->GetCommonInputContainer());
+   mgr->ConnectOutput (clus, 0, mgr->GetCommonOutputContainer());
+   mgr->ConnectOutput (clus,  1, coutput1_clus );
    
-   return pwg4clus;
+   return clus;
 }