]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGHF/hfe/AliAnalysisTaskHFE.cxx
Update
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFE.cxx
index 5408818bddef8fef33086e123107ac033cb3af7f..fcb99e24d83650d0f5fd5ad5bba908824759ff8a 100644 (file)
@@ -45,6 +45,7 @@
 
 #include "AliESDtrackCuts.h"
 #include "AliAnalysisManager.h"
+#include "AliAnalysisUtils.h"
 #include "AliAODInputHandler.h"
 #include "AliAODMCParticle.h"
 #include "AliAODTrack.h"
@@ -65,6 +66,7 @@
 #include "AliOADBContainer.h"
 #include "AliStack.h"
 #include "AliTriggerAnalysis.h"
+#include "AliTRDTriggerAnalysis.h"
 #include "AliVVertex.h"
 
 #include "AliHFEcollection.h"
 #include "AliHFEpairs.h"
 #include "AliHFEpid.h"
 #include "AliHFEpidQAmanager.h"
-#include "AliHFEpostAnalysis.h"
 #include "AliHFEsecVtxs.h"
 #include "AliHFEsecVtx.h"
 #include "AliHFEsignalCuts.h"
 #include "AliHFEtaggedTrackAnalysis.h"
 #include "AliHFEtools.h"
+#include "AliHFEV0taginfo.h"
 #include "AliHFEvarManager.h"
 #include "AliAnalysisTaskHFE.h"
 #include "AliAODMCHeader.h"
@@ -96,11 +98,10 @@ AliAnalysisTaskSE("PID efficiency Analysis")
   , fAODArrayMCInfo(NULL)
   , fQAlevel(0)
   , fPlugins(0)
+  , fCollisionSystem(3)
   , fFillSignalOnly(kTRUE)
+  , fRejectMCFakeTracks(kFALSE)
   , fFillNoCuts(kFALSE)
-  , fUseFilterAOD(kTRUE)
-  , fApplyCutAOD(kFALSE)
-  , fFilter(1<<4)
   , fBackGroundFactorApply(kFALSE)
   , fRemovePileUp(kFALSE)
   , fIdentifiedAsPileUp(kFALSE)
@@ -109,14 +110,15 @@ AliAnalysisTaskSE("PID efficiency Analysis")
   , fRejectKinkMother(kTRUE)
   , fisppMultiBin(kFALSE)
   , fPbPbUserCentralityBinning(kFALSE)
+  , fRemoveFirstEvent(kFALSE)
   , fisNonHFEsystematics(kFALSE)
   , fSpecialTrigger(NULL)
   , fCentralityF(-1)
   , fCentralityPercent(-1)
+  , fCentralityEstimator("V0M")
   , fContributors(0.5)
   , fWeightBackGround(0.)
   , fVz(0.0)
-  , fHadronBackgroundOADB(NULL)
   , fContainer(NULL)
   , fVarManager(NULL)
   , fSignalCuts(NULL)
@@ -124,11 +126,14 @@ AliAnalysisTaskSE("PID efficiency Analysis")
   , fTriggerAnalysis(NULL)
   , fPID(NULL)
   , fPIDqa(NULL)
+  , fTRDTriggerAnalysismb(NULL)
+  , fTRDTriggerAnalysistrg(NULL)
   , fPIDpreselect(NULL)
   , fCuts(NULL)
   , fTaggedTrackCuts(NULL)
   , fCleanTaggedTrack(kFALSE)
   , fVariablesTRDTaggedTrack(kFALSE)
+  , fAnalysisUtils(NULL)
   , fCutspreselect(NULL)
   , fSecVtx(NULL)
   , fElecBackGround(NULL)
@@ -136,13 +141,15 @@ AliAnalysisTaskSE("PID efficiency Analysis")
   , fTaggedTrackAnalysis(NULL)
   , fExtraCuts(NULL)
   , fBackgroundSubtraction(NULL)
+  , fTRDTrigger(kFALSE)
+  , fWhichTRDTrigger(0)
+  , fV0Tagger(NULL)
   , fQA(NULL)
   , fOutput(NULL)
   , fHistMCQA(NULL)
   , fHistSECVTX(NULL)
   , fHistELECBACKGROUND(NULL)
   , fQACollection(NULL)
-  , fQAAODCollection(NULL)
 {
   //
   // Dummy constructor
@@ -153,7 +160,7 @@ AliAnalysisTaskSE("PID efficiency Analysis")
   memset(&fisppMultiBin, kFALSE, sizeof(fisppMultiBin));
   memset(fCentralityLimits, 0, sizeof(Float_t) * 12);
 
-
+  SetppAnalysis();
 }
 
 //____________________________________________________________
@@ -163,11 +170,10 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   , fAODArrayMCInfo(NULL)
   , fQAlevel(0)
   , fPlugins(0)
+  , fCollisionSystem(3)
   , fFillSignalOnly(kTRUE)
+  , fRejectMCFakeTracks(kFALSE)
   , fFillNoCuts(kFALSE)
-  , fUseFilterAOD(kTRUE)
-  , fApplyCutAOD(kFALSE)
-  , fFilter(1<<4)
   , fBackGroundFactorApply(kFALSE)
   , fRemovePileUp(kFALSE)
   , fIdentifiedAsPileUp(kFALSE)
@@ -176,14 +182,15 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   , fRejectKinkMother(kTRUE)
   , fisppMultiBin(kFALSE)
   , fPbPbUserCentralityBinning(kFALSE)
+  , fRemoveFirstEvent(kFALSE)
   , fisNonHFEsystematics(kFALSE)
   , fSpecialTrigger(NULL)
   , fCentralityF(-1)
   , fCentralityPercent(-1)
+  , fCentralityEstimator("V0M")
   , fContributors(0.5)
   , fWeightBackGround(0.)
   , fVz(0.0)
-  , fHadronBackgroundOADB(NULL)
   , fContainer(NULL)
   , fVarManager(NULL)
   , fSignalCuts(NULL)
@@ -191,11 +198,14 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   , fTriggerAnalysis(NULL)
   , fPID(NULL)
   , fPIDqa(NULL)
+  , fTRDTriggerAnalysismb(NULL)
+  , fTRDTriggerAnalysistrg(NULL)
   , fPIDpreselect(NULL)
   , fCuts(NULL)
   , fTaggedTrackCuts(NULL)
   , fCleanTaggedTrack(kFALSE)
   , fVariablesTRDTaggedTrack(kFALSE)
+  , fAnalysisUtils(NULL)
   , fCutspreselect(NULL)
   , fSecVtx(NULL)
   , fElecBackGround(NULL)
@@ -203,13 +213,15 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   , fTaggedTrackAnalysis(NULL)
   , fExtraCuts(NULL)
   , fBackgroundSubtraction(NULL)
+  , fTRDTrigger(kFALSE)
+  , fWhichTRDTrigger(0)
+  , fV0Tagger(NULL)
   , fQA(NULL)
   , fOutput(NULL)
   , fHistMCQA(NULL)
   , fHistSECVTX(NULL)
   , fHistELECBACKGROUND(NULL)
   , fQACollection(0x0)
-  , fQAAODCollection(NULL)
 {
   //
   // Default constructor
@@ -217,9 +229,14 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   DefineOutput(1, TList::Class());
   DefineOutput(2, TList::Class());
 
+  fV0Tagger = new AliHFEV0taginfo("Tagger");
   fPID = new AliHFEpid("hfePid");
   fPIDqa = new AliHFEpidQAmanager;
   fVarManager = new AliHFEvarManager("hfeVarManager");
+  fAnalysisUtils = new AliAnalysisUtils;
+  fTRDTriggerAnalysismb = new AliTRDTriggerAnalysis();
+  fTRDTriggerAnalysistrg = new AliTRDTriggerAnalysis();
+  fTRDTriggerAnalysistrg->SetRequireMatchElectron(kTRUE);
 
   memset(fElecBackgroundFactor, 0, sizeof(Double_t) * kElecBgSpecies * kBgPtBins * kCentBins * kBgLevels);
   memset(fkBackGroundFactorArray, 0, sizeof(TF1 *) * 12);
@@ -227,6 +244,7 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const char * name):
   memset(&fisppMultiBin, kFALSE, sizeof(fisppMultiBin));
   memset(fCentralityLimits, 0, sizeof(Float_t) * 12);
 
+  SetppAnalysis();
 }
 
 //____________________________________________________________
@@ -236,11 +254,10 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref):
   , fAODArrayMCInfo(NULL)
   , fQAlevel(0)
   , fPlugins(0)
+  , fCollisionSystem(ref.fCollisionSystem)
   , fFillSignalOnly(ref.fFillSignalOnly)
+  , fRejectMCFakeTracks(ref.fRejectMCFakeTracks)
   , fFillNoCuts(ref.fFillNoCuts)
-  , fUseFilterAOD(ref.fUseFilterAOD)
-  , fApplyCutAOD(ref.fApplyCutAOD)
-  , fFilter(ref.fFilter) 
   , fBackGroundFactorApply(ref.fBackGroundFactorApply)
   , fRemovePileUp(ref.fRemovePileUp)
   , fIdentifiedAsPileUp(ref.fIdentifiedAsPileUp)
@@ -249,14 +266,15 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref):
   , fRejectKinkMother(ref.fRejectKinkMother)
   , fisppMultiBin(ref.fisppMultiBin)
   , fPbPbUserCentralityBinning(ref.fPbPbUserCentralityBinning)
+  , fRemoveFirstEvent(ref.fRemoveFirstEvent)
   , fisNonHFEsystematics(ref.fisNonHFEsystematics)
   , fSpecialTrigger(ref.fSpecialTrigger)
   , fCentralityF(ref.fCentralityF)
   , fCentralityPercent(ref.fCentralityPercent)
+  , fCentralityEstimator(ref.fCentralityEstimator)
   , fContributors(ref.fContributors)
   , fWeightBackGround(ref.fWeightBackGround)
   , fVz(ref.fVz)
-  , fHadronBackgroundOADB(ref.fHadronBackgroundOADB)
   , fContainer(NULL)
   , fVarManager(NULL)
   , fSignalCuts(NULL)
@@ -264,11 +282,14 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref):
   , fTriggerAnalysis(NULL)
   , fPID(NULL)
   , fPIDqa(NULL)
+  , fTRDTriggerAnalysismb(NULL)
+  , fTRDTriggerAnalysistrg(NULL)
   , fPIDpreselect(NULL)
   , fCuts(NULL)
   , fTaggedTrackCuts(NULL)
   , fCleanTaggedTrack(ref.fCleanTaggedTrack)
   , fVariablesTRDTaggedTrack(ref.fVariablesTRDTaggedTrack)
+  , fAnalysisUtils(NULL)
   , fCutspreselect(NULL)
   , fSecVtx(NULL)
   , fElecBackGround(NULL)
@@ -276,13 +297,15 @@ AliAnalysisTaskHFE::AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref):
   , fTaggedTrackAnalysis(NULL)
   , fExtraCuts(NULL)
   , fBackgroundSubtraction(NULL)
+  , fTRDTrigger(ref.fTRDTrigger)
+  , fWhichTRDTrigger(ref.fWhichTRDTrigger)
+  , fV0Tagger(NULL)
   , fQA(NULL)
   , fOutput(NULL)
   , fHistMCQA(NULL)
   , fHistSECVTX(NULL)
   , fHistELECBACKGROUND(NULL)
   , fQACollection(NULL)
-  , fQAAODCollection(NULL)
 {
   //
   // Copy Constructor
@@ -310,11 +333,10 @@ void AliAnalysisTaskHFE::Copy(TObject &o) const {
   target.fAODArrayMCInfo = fAODArrayMCInfo;
   target.fQAlevel = fQAlevel;
   target.fPlugins = fPlugins;
+  target.fCollisionSystem = fCollisionSystem;
   target.fFillSignalOnly = fFillSignalOnly;
+  target.fRejectMCFakeTracks = fRejectMCFakeTracks;
   target.fFillNoCuts = fFillNoCuts;
-  target.fUseFilterAOD = fUseFilterAOD;
-  target.fApplyCutAOD = fApplyCutAOD;
-  target.fFilter = fFilter;
   target.fBackGroundFactorApply = fBackGroundFactorApply;
   target.fRemovePileUp = fRemovePileUp;
   target.fIdentifiedAsPileUp = fIdentifiedAsPileUp;
@@ -323,14 +345,15 @@ void AliAnalysisTaskHFE::Copy(TObject &o) const {
   target.fRejectKinkMother = fRejectKinkMother;
   target.fisppMultiBin =   fisppMultiBin;
   target.fPbPbUserCentralityBinning = fPbPbUserCentralityBinning;
+  target.fRemoveFirstEvent = fRemoveFirstEvent;
   target.fisNonHFEsystematics = fisNonHFEsystematics;
   target.fSpecialTrigger = fSpecialTrigger;
   target.fCentralityF = fCentralityF;
   target.fCentralityPercent = fCentralityPercent;
+  target.fCentralityEstimator = fCentralityEstimator;
   target.fContributors = fContributors;
   target.fWeightBackGround = fWeightBackGround;
   target.fVz = fVz;
-  target.fHadronBackgroundOADB = fHadronBackgroundOADB;
   target.fContainer = fContainer;
   target.fVarManager = fVarManager;
   target.fSignalCuts = fSignalCuts;
@@ -338,11 +361,14 @@ void AliAnalysisTaskHFE::Copy(TObject &o) const {
   target.fTriggerAnalysis = fTriggerAnalysis;
   target.fPID = fPID;
   target.fPIDqa = fPIDqa;
+  target.fTRDTriggerAnalysismb = fTRDTriggerAnalysismb;
+  target.fTRDTriggerAnalysistrg = fTRDTriggerAnalysistrg;
   target.fPIDpreselect = fPIDpreselect;
   target.fCuts = fCuts;
   target.fTaggedTrackCuts = fTaggedTrackCuts;
   target.fCleanTaggedTrack = fCleanTaggedTrack;
   target.fVariablesTRDTaggedTrack = fVariablesTRDTaggedTrack;
+  target.fAnalysisUtils = fAnalysisUtils;
   target.fCutspreselect = fCutspreselect;
   target.fSecVtx = fSecVtx;
   target.fElecBackGround = fElecBackGround;
@@ -350,13 +376,15 @@ void AliAnalysisTaskHFE::Copy(TObject &o) const {
   target.fTaggedTrackAnalysis = fTaggedTrackAnalysis;
   target.fExtraCuts = fExtraCuts;
   target.fBackgroundSubtraction = fBackgroundSubtraction;
+  target.fTRDTrigger = fTRDTrigger;
+  target.fWhichTRDTrigger = fWhichTRDTrigger;
+  target.fV0Tagger = fV0Tagger;
   target.fQA = fQA;
   target.fOutput = fOutput;
   target.fHistMCQA = fHistMCQA;
   target.fHistSECVTX = fHistSECVTX;
   target.fHistELECBACKGROUND = fHistELECBACKGROUND;
   target.fQACollection = fQACollection;
-  target.fQAAODCollection = fQAAODCollection;
 }
 
 //____________________________________________________________
@@ -367,6 +395,8 @@ AliAnalysisTaskHFE::~AliAnalysisTaskHFE(){
   if(fPID) delete fPID;
   if(fPIDpreselect) delete fPIDpreselect;
   if(fVarManager) delete fVarManager;
+  if(fTRDTriggerAnalysismb) delete fTRDTriggerAnalysismb;
+  if(fTRDTriggerAnalysistrg) delete fTRDTriggerAnalysistrg;
   if(fCFM) delete fCFM;
   if(fTriggerAnalysis) delete fTriggerAnalysis;
   if(fSignalCuts) delete fSignalCuts;
@@ -375,6 +405,8 @@ AliAnalysisTaskHFE::~AliAnalysisTaskHFE(){
   if(fElecBackGround) delete fElecBackGround;
   if(fBackgroundSubtraction) delete fBackgroundSubtraction;
   if(fSpecialTrigger) delete fSpecialTrigger;
+  if(fAnalysisUtils) delete fAnalysisUtils;
+  if(fV0Tagger) delete fV0Tagger;
   // Delete output objects only if we are not running in PROOF mode because otherwise this produces a crash during merging
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if(mgr && mgr->GetAnalysisType() != AliAnalysisManager::kProofAnalysis){
@@ -414,15 +446,6 @@ void AliAnalysisTaskHFE::UserCreateOutputObjects(){
       SetHasMCData();
   }
   printf("Analysis Mode: %s Analysis\n", IsAODanalysis() ? "AOD" : "ESD");
-  if(IsAODanalysis()) {
-    printf("AOD filter: %s \n", fUseFilterAOD ? "Yes" : "No");
-    if(fUseFilterAOD) printf("AOD filter used: %d \n", fFilter);
-    // First Part: Make QA histograms
-    fQAAODCollection = new AliHFEcollection("TaskQAAOD", "QA histos from the AOD Electron Task");
-    fQAAODCollection->CreateTH1F("Filterorigin", "AOD filter of tracks at the origin", 21, -1, 20);
-    fQAAODCollection->CreateTH1F("Filterend", "AOD filter of tracks after all cuts", 21, -1, 20);
-    fQA->Add(fQAAODCollection);
-  }
   printf("MC Data available %s\n", HasMCData() ? "Yes" : "No");
 
   // Enable Trigger Analysis
@@ -435,12 +458,16 @@ void AliAnalysisTaskHFE::UserCreateOutputObjects(){
   fQACollection->CreateTH1F("nElectronTracksEvent", "Number of Electron Candidates", 100, 0, 100);
   fQACollection->CreateTH1F("nElectron", "Number of electrons", 100, 0, 100);
   fQACollection->CreateTH2F("radius", "Production Vertex", 100, 0.0, 5.0, 100, 0.0, 5.0);
-  fQACollection->CreateTH2F("TPCdEdxBeforePID", "TPC dE/dx; p (GeV/c); TPC dE/dx (a.u.)", 1000, 0., 10., 200, 0., 200.); 
-  fQACollection->CreateTH2F("TPCnSigmaBeforePID", "TPC dE/dx; p (GeV/c); TPC dE/dx - <TPC dE/dx>|_{el} (#sigma)", 1000, 0., 10., 1000, -10., 10.);
+  fQACollection->CreateTH1F("nTriggerBit", "Histo Trigger Bit", 22, 0, 22);
+  fQACollection->CreateTH2F("TriggerAnalysis","TRD Trigger Analysis",10,0.,10.,10,0.,10.);
+  fQACollection->CreateTH1F("Filterbegin", "AOD filter of tracks after all cuts", 21, -1, 20);
+  fQACollection->CreateTH1F("Filterend", "AOD filter of tracks after all cuts", 21, -1, 20);
+  fQACollection->CreateTH2F("Kinkbefore", "Kink status before filter; p_{T} (GeV/c); kink status", 100, 0., 20., 3, -0.5, 2.5);
+  fQACollection->CreateTH2F("Kinkafter", "Kink status after filter; p_{T} (GeV/c); kink status", 100, 0., 20., 3, -0.5, 2.5);
+  fQACollection->CreateTH1F("HFPuzzle", "Source definition for electrons from HF", 11, -0.5, 10.5);
  
-  InitPIDperformanceQA();
-  InitContaminationQA();
   InitHistoITScluster();
+  InitContaminationQA();
   fQA->Add(fQACollection);
 
   // Initialize PID
@@ -454,8 +481,9 @@ void AliAnalysisTaskHFE::UserCreateOutputObjects(){
   fPID->SortDetectors();
 
   // Background subtraction-------------------------------------------------------------------
-  if (GetPlugin(kNonPhotonicElectron)) {
+  if (GetPlugin(kNonPhotonicElectron)||GetPlugin(kNonPhotonicElectronBeauty)) {
     if(!fBackgroundSubtraction) fBackgroundSubtraction = new AliHFENonPhotonicElectron();
+    if(IsAODanalysis()) fBackgroundSubtraction->SetAOD(kTRUE);
     fBackgroundSubtraction->Init();
     fOutput->Add(fBackgroundSubtraction->GetListOutput());
   }
@@ -536,6 +564,7 @@ void AliAnalysisTaskHFE::UserCreateOutputObjects(){
     fTaggedTrackAnalysis = new AliHFEtaggedTrackAnalysis(Form("taggedTrackAnalysis%s", GetName()));
     fTaggedTrackAnalysis->SetCuts(fTaggedTrackCuts);
     fTaggedTrackAnalysis->SetClean(fCleanTaggedTrack);
+    if(IsAODanalysis()) fTaggedTrackAnalysis->SetAOD();
     AliHFEvarManager *varManager = fTaggedTrackAnalysis->GetVarManager();
     TObjArray *array = fVarManager->GetVariables();
     Int_t nvars = array->GetEntriesFast();
@@ -565,6 +594,8 @@ void AliAnalysisTaskHFE::UserCreateOutputObjects(){
     fQA->Add(fTaggedTrackAnalysis->GetQAcollection());
   }
 
+  //fQA->Print();
+
   PrintStatus();
   // Done!!!
   PostData(1, fOutput);
@@ -603,18 +634,19 @@ void AliAnalysisTaskHFE::UserExec(Option_t *){
     fPID->InitializePID(fInputEvent->GetRunNumber());
   }
 
-  //printf("test0\n");
-
-  // Initialize hadronic background from OADB Container
-  AliDebug(2, Form("Apply background factors: %s, OADB Container %p", fBackGroundFactorApply ? "Yes" : "No", fHadronBackgroundOADB));
-  if(fBackGroundFactorApply && !TestBit(kBackgroundInitialized)){ 
-    AliDebug(2, "Initializing Background from OADB");
-    if(!InitializeHadronBackground(fInputEvent->GetRunNumber())) AliError("Failed initializing hadronic background parameterization from OADB");
-    else AliDebug(2, "Successfully loaded Background from OADB");
-    SetBit(kBackgroundInitialized); 
+  if(fRemoveFirstEvent){
+    if(fAnalysisUtils->IsFirstEventInChunk(fInputEvent)) return;
   }
 
-  //printf("test1\n");
+  AliESDEvent *ev = dynamic_cast<AliESDEvent *>(fInputEvent);
+  if(ev && fTRDTrigger && (fWhichTRDTrigger<6))
+  {
+      if(!CheckTRDTriggerESD(ev)) return;
+  }
+  if(fInputEvent && fTRDTrigger && (fWhichTRDTrigger>5))
+  {
+      if(!CheckTRDTrigger(fInputEvent)) return;
+  }
 
   if(IsESDanalysis() && HasMCData()){
     // Protect against missing MC trees
@@ -626,10 +658,9 @@ void AliAnalysisTaskHFE::UserExec(Option_t *){
     if(!mcH->InitOk()) return;
     if(!mcH->TreeK()) return;
     if(!mcH->TreeTR()) return;
+
     // Background subtraction-------------------------------------------------------------------
-    if(GetPlugin(kNonPhotonicElectron)){
-      fBackgroundSubtraction->SetMCEvent(fMCEvent);
-    }
+    if(GetPlugin(kNonPhotonicElectron)) fBackgroundSubtraction->SetMCEvent(fMCEvent);
     //------------------------------------------------------------------------------------------
   }
 
@@ -654,9 +685,7 @@ void AliAnalysisTaskHFE::UserExec(Option_t *){
     }
     fSignalCuts->SetMCAODInfo(fAODArrayMCInfo);
     // Background subtraction-------------------------------------------------------------------
-    if (GetPlugin(kNonPhotonicElectron)) {
-      fBackgroundSubtraction->SetAODArrayMCInfo(fAODArrayMCInfo);
-    }
+    if (GetPlugin(kNonPhotonicElectron)||GetPlugin(kNonPhotonicElectronBeauty)) fBackgroundSubtraction->SetAODArrayMCInfo(fAODArrayMCInfo);
     //------------------------------------------------------------------------------------------
   }
 
@@ -689,7 +718,7 @@ void AliAnalysisTaskHFE::UserExec(Option_t *){
   if(fPIDpreselect) fPIDpreselect->SetPIDResponse(pidResponse);
 
   // Background subtraction-------------------------------------------------------------------
-  if(GetPlugin(kNonPhotonicElectron)) fBackgroundSubtraction->InitRun(fInputEvent,pidResponse);
+  if(GetPlugin(kNonPhotonicElectron)||GetPlugin(kNonPhotonicElectronBeauty)) fBackgroundSubtraction->InitRun(fInputEvent,pidResponse);
   //------------------------------------------------------------------------------------------
 
   // Event loop
@@ -701,7 +730,6 @@ void AliAnalysisTaskHFE::UserExec(Option_t *){
     // Check Trigger selection
     if(specialTrigger){
       AliDebug(2, Form("Special Trigger requested: %s", specialTrigger));
-      AliESDEvent *ev = dynamic_cast<AliESDEvent *>(fInputEvent);
       if(!(ev && ev->IsTriggerClassFired(specialTrigger))){
         AliDebug(2, "Event not selected"); 
         return;
@@ -720,49 +748,8 @@ void AliAnalysisTaskHFE::Terminate(Option_t *){
   //
   // Terminate not implemented at the moment
   //
-  if(GetPlugin(kPostProcess)){
-    fOutput = dynamic_cast<TList *>(GetOutputData(1));
-    fQA = dynamic_cast<TList *>(GetOutputData(2));
-    if(!fOutput){
-      AliError("Results not available");
-      return;
-    }
-    if(!fQA){
-      AliError("QA output not available");
-      return;
-    }
-    fContainer = dynamic_cast<AliHFEcontainer *>(fOutput->FindObject("trackContainer")); 
-    if(!fContainer){
-      AliError("Track container not found");
-      return;
-    }
-    AliHFEpostAnalysis postanalysis;
-    postanalysis.SetTaskResults(fContainer);
-    TList *qalist = dynamic_cast<TList *>(fQA->FindObject("list_TaskQA"));
-    if(!qalist){
-      AliError("QA List not found");
-      return;
-    }
-    postanalysis.SetTaskQA(qalist);
-    //printf("Running post analysis\n");
-    //if(HasMCData())
-    postanalysis.DrawMCSignal2Background();
-    postanalysis.DrawEfficiency();
-    postanalysis.DrawPIDperformance();
-    postanalysis.DrawCutEfficiency();
-
-    if (GetPlugin(kIsElecBackGround)) {
-      AliHFEelecbackground elecBackGround;
-      TList *oe = 0x0;
-      if(!(oe = (TList*)dynamic_cast<TList *>(fOutput->FindObject("HFEelecbackground")))){
-        return;
-      }
-      elecBackGround.Load(oe);
-      elecBackGround.Plot();
-      elecBackGround.PostProcess();      
-    }
-  }
 }
+
 //_______________________________________________________________
 Bool_t AliAnalysisTaskHFE::IsEventInBinZero() {
   //
@@ -795,7 +782,7 @@ void AliAnalysisTaskHFE::ProcessMC(){
   // In case MC QA is on also MC QA loop is done
   //
   AliDebug(3, "Processing MC Information");
-  Double_t eventContainer [4];
+  Double_t eventContainer [4] = {0., 0., 0., 0.};
   if(IsESDanalysis()) eventContainer[0] = fMCEvent->GetPrimaryVertex()->GetZ();
   else eventContainer[0] = fAODMCHeader->GetVtxZ();
   eventContainer[2] = fCentralityF;
@@ -814,7 +801,7 @@ void AliAnalysisTaskHFE::ProcessMC(){
         fMCQA->SetMCEvent(fMCEvent);
         fMCQA->SetGenEventHeader(fMCEvent->GenEventHeader());
         fMCQA->SetCentrality(fCentralityF);
-        fMCQA->SetPercentrality(fCentralityPercent);
+        fMCQA->SetPercentrality(static_cast<Int_t>(fCentralityPercent));
         if(IsPbPb()) { fMCQA->SetPbPb();}
         else
         {
@@ -847,6 +834,38 @@ void AliAnalysisTaskHFE::ProcessMC(){
    fCFM->SetMCEventInfo(fMCEvent);
    // fCFM->CheckEventCuts(AliCFManager::kEvtRecCuts, fESD);
   } else {
+    fMCQA->SetMCArray(fAODArrayMCInfo);
+
+    if(!((fIdentifiedAsPileUp) || (TMath::Abs(fVz) > fCuts->GetVertexRange()) || (fCentralityF < 0))){ //kStepMCGeneratedZOutNoPileUpCentralityFine
+      if (HasMCData() && IsQAOn(kMCqa)) {
+        AliDebug(2, "Running MC QA");
+
+        fMCQA->SetCentrality(fCentralityF);
+        fMCQA->SetPercentrality(static_cast<Int_t>(fCentralityPercent));
+
+        if(IsPbPb()) { fMCQA->SetPbPb();}
+        else
+        {
+            if(fisppMultiBin) fMCQA->SetPPMultiBin();
+            else fMCQA->SetPP();
+        }
+        fMCQA->Init();
+
+        //fMCQA->GetMesonKine();
+
+        // loop over all tracks for decayed electrons
+        AliAODMCParticle * mcpart;
+        for (Int_t igen = 0; igen < fAODArrayMCInfo->GetEntriesFast(); igen++){
+          mcpart = dynamic_cast<AliAODMCParticle *>(fAODArrayMCInfo->At(igen));
+          if(!mcpart) continue;
+          fMCQA->GetDecayedKine(mcpart, AliHFEmcQA::kCharm,  AliHFEmcQA::kElectronPDG); // no accept cut
+          fMCQA->GetDecayedKine(mcpart, AliHFEmcQA::kBeauty, AliHFEmcQA::kElectronPDG); // no accept cut
+          fMCQA->GetDecayedKine(mcpart, AliHFEmcQA::kOthers, AliHFEmcQA::kElectronPDG); // no accept cut
+        }
+
+      } // end of MC QA loop
+    }
+
     fCFM->SetMCEventInfo(fInputEvent);
   }
   // Run MC loop
@@ -892,10 +911,18 @@ void AliAnalysisTaskHFE::ProcessESD(){
 
   // Set magnetic field if V0 task on
   if(fTaggedTrackAnalysis) {
+    // Tag all v0s in current event
+    if(fV0Tagger){
+      fV0Tagger->Reset();
+      fV0Tagger->TagV0Tracks(fESD);
+    }
     fTaggedTrackAnalysis->SetMagneticField(fESD->GetMagneticField());
     fTaggedTrackAnalysis->SetCentrality(fCentralityF);
     if(IsPbPb()) fTaggedTrackAnalysis->SetPbPb();
-    else fTaggedTrackAnalysis->SetPP();
+    else {
+           if(IspPb()) fTaggedTrackAnalysis->SetpPb();
+           else fTaggedTrackAnalysis->SetPP();
+    }
   }
 
   // Do event Normalization
@@ -903,7 +930,8 @@ void AliAnalysisTaskHFE::ProcessESD(){
   eventContainer[0] = 0.; 
   if(HasMCData()) eventContainer[0] = fVz;
   else {
-    if(fESD->GetPrimaryVertexSPD()) eventContainer[0] = fESD->GetPrimaryVertexSPD()->GetZ();
+    const AliESDVertex* vtxESD = fESD->GetPrimaryVertex();
+    if(vtxESD) eventContainer[0] = vtxESD->GetZ();
   }
   eventContainer[1] = 0.;
   eventContainer[2] = fCentralityF;
@@ -955,13 +983,11 @@ void AliAnalysisTaskHFE::ProcessESD(){
   Double_t container[10];
   memset(container, 0, sizeof(Double_t) * 10);
   // container for the output THnSparse
-  Double_t dataE[6]; // [pT, eta, Phi, type, 'C' or 'B']
   Double_t dataDca[6]; // [source, pT, dca, centrality]
   Int_t nElectronCandidates = 0;
   AliESDtrack *track = NULL, *htrack = NULL;
   AliMCParticle *mctrack = NULL;
   AliMCParticle *mctrackmother = NULL;
-  Int_t pid = 0;
 
   Bool_t signal = kTRUE;
 
@@ -977,41 +1003,41 @@ void AliAnalysisTaskHFE::ProcessESD(){
   // minjung for IP QA(temporary ~ 2weeks)
   if(!fExtraCuts){
     fExtraCuts = new AliHFEextraCuts("hfetmpCuts","HFE tmp Cuts");
-  }   
+  }
   fExtraCuts->SetRecEventInfo(fESD);
 
   // Electron background analysis 
   if (GetPlugin(kIsElecBackGround)) {
-    
+
     AliDebug(2, "Running BackGround Analysis");
-    
+
     fElecBackGround->Reset();
-    
+
   } // end of electron background analysis
-  
+
 
   // Background subtraction-------------------------------------------------------------------
-  if (GetPlugin(kNonPhotonicElectron)) {
-    fBackgroundSubtraction->FillPoolAssociatedTracks(fInputEvent,fCentralityF);
-  }
+  if (GetPlugin(kNonPhotonicElectron)) fBackgroundSubtraction->FillPoolAssociatedTracks(fInputEvent, fCentralityF);
   //------------------------------------------------------------------------------------------
 
   //
   // Loop ESD
   //
   AliDebug(3, Form("Number of Tracks: %d", fESD->GetNumberOfTracks()));
+  Bool_t kinkmother(kFALSE), kinkdaughter(kFALSE); Double_t kinkstatus(0);
   for(Int_t itrack = 0; itrack < fESD->GetNumberOfTracks(); itrack++){
     AliDebug(4, "New ESD track");
     track = fESD->GetTrack(itrack);
     track->SetESDEvent(fESD);
+    kinkmother = track->GetKinkIndex(0) < 0; kinkdaughter = track->GetKinkIndex(0) > 0;
+    kinkstatus = 0.;
+    if(kinkmother) kinkstatus = 1.;
+    else if(kinkdaughter) kinkstatus = 2.;
 
     // fill counts of v0-identified particles
-    Int_t v0pid = -1;
-    if(track->TestBit(BIT(14))) v0pid = AliPID::kElectron;
-    else if(track->TestBit(BIT(15))) v0pid = AliPID::kPion;
-    else if(track->TestBit(BIT(16))) v0pid = AliPID::kProton;
+    AliPID::EParticleType v0pid = fV0Tagger ? fV0Tagger->GetV0Info(track->GetID()) : AliPID::kUnknown;
     // here the tagged track analysis will run
-    if(fTaggedTrackAnalysis && v0pid > -1){ 
+    if(fTaggedTrackAnalysis && v0pid != AliPID::kUnknown){ 
       AliDebug(1, Form("Track identified as %s", AliPID::ParticleName(v0pid)));
       fTaggedTrackAnalysis->ProcessTrack(track, v0pid);
       AliDebug(1, "V0 PID done");
@@ -1038,6 +1064,7 @@ void AliAnalysisTaskHFE::ProcessESD(){
       if(!(mctrack = dynamic_cast<AliMCParticle *>(fMCEvent->GetTrack(TMath::Abs(track->GetLabel()))))) continue;
 
       if(fFillSignalOnly && !fCFM->CheckParticleCuts(AliHFEcuts::kStepMCGenerated, mctrack)) signal = kFALSE; 
+      if(fRejectMCFakeTracks && IsMCFakeTrack(track)) signal = kFALSE;
       else AliDebug(3, "Signal Electron");
 
       // Fill K pt for Ke3 contributions
@@ -1053,52 +1080,16 @@ void AliAnalysisTaskHFE::ProcessESD(){
         fVarManager->FillContainer(fContainer, "recTrackContMC", AliHFEcuts::kStepRecNoCut, kTRUE);
       }
     }
-        
-    if(fisNonHFEsystematics && IsPbPb())  {
-      //FillProductionVertex(track);     
-      if(fMCQA && signal){ 
-       fMCQA->SetCentrality(fCentralityF);
-       if(mctrack && (TMath::Abs(mctrack->Particle()->GetPdgCode()) == 11)){
-         Double_t weightElecBgV0[kBgLevels] = {0.,0.,0.};
-          
-         fMCQA->SetTrkKine(track->Pt(),track->Eta(), track->Phi());
-         fMCQA->SetContainerStep(4);
-          
-         weightElecBgV0[0] = fMCQA->GetWeightFactor(mctrack, 0); // positive:conversion e, negative: nonHFE 
-          
-         //Fill additional containers for electron source distinction
-         Int_t elecSource = 0;
-         elecSource = fMCQA->GetElecSource(mctrack->Particle());
-         const Char_t *sourceName[kElecBgSpecies]={"Pion","Eta","Omega","Phi","EtaPrime","Rho"};
-         const Char_t *levelName[kBgLevels]={"Best","Lower","Upper"};
-         Int_t iName = 0;
-         for(Int_t iSource = AliHFEmcQA::kPi0; iSource <=  AliHFEmcQA::kGammaRho0; iSource++){
-           if((iSource == AliHFEmcQA::kElse)||(iSource == AliHFEmcQA::kMisID)) continue;
-           if(elecSource == iSource){
-             
-             if(weightElecBgV0[0]>0){ 
-               fVarManager->FillContainer(fContainer, Form("conversionElecs%s%s",sourceName[iName], levelName[0]), 4, kTRUE, weightElecBgV0[0]);
-             } 
-             else if(weightElecBgV0[0]<0){ 
-               fVarManager->FillContainer(fContainer, Form("mesonElecs%s%s",sourceName[iName], levelName[0]), 4, kTRUE, -1*weightElecBgV0[0]);
-             }           
-             break;
-           }
-           iName++;
-           if(iName == kElecBgSpecies)iName = 0;
-         }
-       }
-      }
-    }
   
     // RecKine: ITSTPC cuts  
     if(!ProcessCutStep(AliHFEcuts::kStepRecKineITSTPC, track)) continue;
     
-    
+    fQACollection->Fill("Kinkbefore", track->Pt(), kinkstatus); 
     // RecPrim
     if(fRejectKinkMother) { 
       if(track->GetKinkIndex(0) != 0) continue; } // Quick and dirty fix to reject both kink mothers and daughters
     if(!ProcessCutStep(AliHFEcuts::kStepRecPrim, track)) continue;
+    fQACollection->Fill("Kinkafter", track->Pt(), kinkstatus); 
 
     // HFEcuts: ITS layers cuts
     if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsITS, track)) continue;
@@ -1226,12 +1217,6 @@ void AliAnalysisTaskHFE::ProcessESD(){
       }
     }
 
-    if(TMath::Abs(track->Eta()) < 0.5){
-      if(track->GetInnerParam())
-        fQACollection->Fill("TPCdEdxBeforePID", track->GetInnerParam()->P(), track->GetTPCsignal());
-      fQACollection->Fill("TPCnSigmaBeforePID", track->P(), fInputHandler->GetPIDResponse()->NumberOfSigmasTPC(track, AliPID::kElectron));
-    }
-
     AliHFEpidObject hfetrack;
     hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis);
     hfetrack.SetRecTrack(track);
@@ -1239,7 +1224,10 @@ void AliAnalysisTaskHFE::ProcessESD(){
     hfetrack.SetCentrality(fCentralityF);
     hfetrack.SetMulitplicity(ncontribVtx);
     if(IsPbPb()) hfetrack.SetPbPb();
-    else hfetrack.SetPP();
+    else {
+       if(IspPb()) hfetrack.SetpPb();
+       else hfetrack.SetPP();
+    }
     fPID->SetVarManager(fVarManager);
     if(!fPID->IsSelected(&hfetrack, fContainer, "recTrackCont", fPIDqa)) continue;
     nElectronCandidates++;
@@ -1248,26 +1236,34 @@ void AliAnalysisTaskHFE::ProcessESD(){
     if (GetPlugin(kNonPhotonicElectron)) {
       Int_t indexmother = -1;
       Int_t mcsource = -1;
+      Int_t mcQAsource = -1;
+      Double_t weightNonPhotonicFactor = 1.; 
       if(HasMCData()){
        mcsource = fBackgroundSubtraction->FindMother(mctrack->GetLabel(),indexmother);
+       if(fBackgroundSubtraction->GetLevelBack()>=0) {
+         if(fMCQA) {
+           mcQAsource = fMCQA->GetElecSource(mctrack);
+           weightNonPhotonicFactor = TMath::Abs(fMCQA->GetWeightFactor(mctrack, fBackgroundSubtraction->GetLevelBack())); // positive:conversion e, negative: nonHFE 
+         }
+       }
       }
-      fBackgroundSubtraction->LookAtNonHFE(itrack, track, fInputEvent, 1, fCentralityF, -1, mcsource, indexmother);
+      fBackgroundSubtraction->LookAtNonHFE(itrack, track, fInputEvent, weightNonPhotonicFactor, fCentralityF, -1, mcsource, indexmother,mcQAsource);
     }
     //-----------------------------------------------------------------------------------------------------------------
 
     // Temporary histogram for chi2/ITS cluster
     if(IsPbPb()) {
-            TBits shared = track->GetTPCSharedMap();
+      TBits shared = track->GetTPCSharedMap();
            Int_t sharebit=0;
-            if(shared.CountBits() >= 2) sharebit=1;
+      if(shared.CountBits() >= 2) sharebit=1;
 
            Double_t itschi2percluster = 0.0;
            Double_t itsnbcls = static_cast<Double_t>(track->GetNcls(0));
            if(itsnbcls > 0) itschi2percluster = track->GetITSchi2()/itsnbcls;
 
-            Double_t itsChi2[7] = {track->Pt(),track->Eta(), track->Phi(),
-                                  fCentralityF,track->GetTPCsignalN(), sharebit,itschi2percluster};
-            fQACollection->Fill("fChi2perITScluster", itsChi2);
+      Double_t itsChi2[7] = {track->Pt(),track->Eta(), track->Phi(),
+                       static_cast<Double_t>(fCentralityF),static_cast<Double_t>(track->GetTPCsignalN()), static_cast<Double_t>(sharebit),itschi2percluster};
+      fQACollection->Fill("fChi2perITScluster", itsChi2);
     }
     else{
 
@@ -1275,7 +1271,7 @@ void AliAnalysisTaskHFE::ProcessESD(){
       Double_t itsnbcls = static_cast<Double_t>(track->GetNcls(0));
       if(itsnbcls > 0) itschi2percluster = track->GetITSchi2()/itsnbcls;
 
-      Double_t itsChi2[3] = {track->Pt(), fCentralityF, itschi2percluster};
+      Double_t itsChi2[3] = {track->Pt(), static_cast<Double_t>(fCentralityF), itschi2percluster};
       fQACollection->Fill("fChi2perITScluster", itsChi2);
     }
 
@@ -1298,12 +1294,12 @@ void AliAnalysisTaskHFE::ProcessESD(){
     // Fill Containers
     if(signal) {
       // Apply weight for background contamination
-            if(fBackGroundFactorApply) {
-              if(IsPbPb() && fCentralityF >= 0) fWeightBackGround =  fkBackGroundFactorArray[fCentralityF >= 0 ? fCentralityF : 0]->Eval(TMath::Abs(track->P()));
-              else    fWeightBackGround =  fkBackGroundFactorArray[0]->Eval(TMath::Abs(track->P())); // pp case
+      if(fBackGroundFactorApply) {
+        if(IsPbPb() && fCentralityF >= 0) fWeightBackGround =  fkBackGroundFactorArray[fCentralityF >= 0 ? fCentralityF : 0]->Eval(TMath::Abs(track->P()));
+        else    fWeightBackGround =  fkBackGroundFactorArray[0]->Eval(TMath::Abs(track->P())); // for pp and pPb
 
-              if(fWeightBackGround < 0.0) fWeightBackGround = 0.0;
-              else if(fWeightBackGround > 1.0) fWeightBackGround = 1.0;
+        if(fWeightBackGround < 0.0) fWeightBackGround = 0.0;
+        else if(fWeightBackGround > 1.0) fWeightBackGround = 1.0;
         // weightBackGround as special weight
         fVarManager->FillContainer(fContainer, "hadronicBackground", 1, kFALSE, fWeightBackGround);
       }
@@ -1320,44 +1316,6 @@ void AliAnalysisTaskHFE::ProcessESD(){
       }
     }
 
-    if(HasMCData()){
-      dataE[0] = track->Pt();
-      dataE[1] = track->Eta();
-      dataE[2] = track->Phi();
-      dataE[3] = track->Charge();
-      dataE[4] = -1.;
-      dataE[5] = -1.;
-
-      // Track selected: distinguish between true and fake
-      AliDebug(1, Form("Candidate Selected, filling THnSparse, PID: %d\n", mctrack->Particle()->GetPdgCode()));
-      if((pid = TMath::Abs(mctrack->Particle()->GetPdgCode())) == 11){
-        Int_t type = 0;
-        if(fSignalCuts->IsCharmElectron(track))
-          type = 1;
-        else if(fSignalCuts->IsBeautyElectron(track))
-          type = 2;
-        AliDebug(1, Form("Type: %d\n", type));
-        if(type){
-                dataE[5] = type; // beauty[1] or charm[2]
-                dataE[4] = 2;  // signal electron
-        }
-        else{
-                dataE[4] = 1; // not a signal electron
-                dataE[5] = 0;
-        }
-      } 
-      else {
-        // Fill THnSparse with the information for Fake Electrons
-        dataE[4] = 0;
-        dataE[5] = 0;
-      }
-      // fill the performance THnSparse, if the mc origin could be defined
-      if(dataE[4] > -1){
-        AliDebug(1, Form("Entries: [%.3f|%.3f|%.3f|%f|%f|%f]\n", dataE[0],dataE[1],dataE[2],dataE[3],dataE[4],dataE[5]));
-        fQACollection->Fill("PIDperformance", dataE);
-      }
-    }
-
     // Electron background analysis 
     if (GetPlugin(kIsElecBackGround)) {
       
@@ -1498,6 +1456,11 @@ void AliAnalysisTaskHFE::ProcessESD(){
     }
 
   }
+
+  // Background subtraction-------------------------------------------------------------------
+  if (GetPlugin(kNonPhotonicElectron)) fBackgroundSubtraction->CountPoolAssociated(fInputEvent, fCentralityF);
+  //------------------------------------------------------------------------------------------
+
   fQACollection->Fill("nElectronTracksEvent", nElectronCandidates);
 }
 
@@ -1526,6 +1489,22 @@ void AliAnalysisTaskHFE::ProcessAOD(){
     AliError("AOD Event required for AOD Analysis");
       return;
   }
+
+  // Set magnetic field if V0 task on
+  if(fTaggedTrackAnalysis) {
+    // Tag all v0s in current event
+    if(fV0Tagger){
+      fV0Tagger->Reset();
+      fV0Tagger->TagV0Tracks(fAOD);
+    }
+    fTaggedTrackAnalysis->SetMagneticField(fAOD->GetMagneticField());
+    fTaggedTrackAnalysis->SetCentrality(fCentralityF);
+    if(IsPbPb()) fTaggedTrackAnalysis->SetPbPb();
+    else {
+      if(IspPb()) fTaggedTrackAnalysis->SetpPb();
+      else fTaggedTrackAnalysis->SetPP();
+    }
+  }
   
   //printf("Will fill\n");
   //
@@ -1579,48 +1558,53 @@ void AliAnalysisTaskHFE::ProcessAOD(){
   //printf("Number of kink mother in the events %d\n",numberofmotherkink);
 
   // Background subtraction-------------------------------------------------------------------
-  if (GetPlugin(kNonPhotonicElectron)) {
-    fBackgroundSubtraction->FillPoolAssociatedTracks(fInputEvent,fCentralityF);
-  }
+  if (GetPlugin(kNonPhotonicElectron)||GetPlugin(kNonPhotonicElectronBeauty)) fBackgroundSubtraction->FillPoolAssociatedTracks(fInputEvent, fCentralityF);
   //------------------------------------------------------------------------------------------
 
   // Loop over tracks
   AliAODTrack *track = NULL;
   AliAODMCParticle *mctrack = NULL;
-  Double_t dataE[6]; // [pT, eta, Phi, Charge, type, 'C' or 'B']
   Double_t dataDca[6]; // [source, pT, dca, centrality]
   Int_t nElectronCandidates = 0;
-  Int_t pid;
   Bool_t signal;
+
   //printf("Number of track %d\n",(Int_t) fAOD->GetNumberOfTracks());
+  Bool_t kinkmother(kFALSE), kinkdaughter(kFALSE); Double_t kinkstatus(0);
   for(Int_t itrack = 0; itrack < fAOD->GetNumberOfTracks(); itrack++){
+    kinkmother=kFALSE;
+    kinkdaughter=kFALSE;
+    kinkstatus = 0.;
     track = fAOD->GetTrack(itrack); mctrack = NULL;
     if(!track) continue;
-    // Begining
-    Bool_t passone = kFALSE;  
-    fQAAODCollection->Fill("Filterorigin", -1);
-    for(Int_t k=0; k<20; k++) {
-      Int_t u = 1<<k;     
-      if((track->TestFilterBit(u))) {
-       fQAAODCollection->Fill("Filterorigin", k);
-       passone = kTRUE;
+
+    for(int ivx = 0; ivx < numberofmotherkink; ivx++){
+      if(track->GetID() == listofmotherkink[ivx]){
+        kinkmother = kTRUE;
+        break;
       }
     }
-    //if(!passone) printf("what is the filter %d\n",track->GetFilterMap());
+    AliAODVertex *pvx = track->GetProdVertex();
+    if(pvx && (pvx->GetType() == AliAODVertex::kKink)) kinkdaughter = kTRUE;
+    kinkstatus = 0.;
+    if(kinkmother) kinkstatus = 1.;
+    else if(kinkdaughter) kinkstatus = 2.;
 
-    if(fUseFilterAOD){
-      //printf("Filter of the track  %d\n",track->GetFilterMap());
-      if(!(track->TestFilterBit(fFilter))) continue;  // Only process AOD tracks where the HFE is set
+    // fill counts of v0-identified particles
+    AliPID::EParticleType v0pid = fV0Tagger ? fV0Tagger->GetV0Info(track->GetID()) : AliPID::kUnknown;
+    // here the tagged track analysis will run
+    if(fTaggedTrackAnalysis && v0pid != AliPID::kUnknown){ 
+      AliDebug(1, Form("Track identified as %s", AliPID::ParticleName(v0pid)));
+      fTaggedTrackAnalysis->ProcessTrack(track, v0pid);
+      AliDebug(1, "V0 PID done");
     }
-    //printf("Pass the flag\n");
     
     signal = kTRUE;
     if(HasMCData()){
-
       Int_t label = TMath::Abs(track->GetLabel());
       if(label && label < fAODArrayMCInfo->GetEntriesFast())
         mctrack = dynamic_cast<AliAODMCParticle *>(fAODArrayMCInfo->At(label));
         if(fFillSignalOnly && !fCFM->CheckParticleCuts(AliHFEcuts::kStepMCGenerated, mctrack)) signal = kFALSE;
+        if(fRejectMCFakeTracks && IsMCFakeTrack(track)) signal = kFALSE;
     }
     
     fVarManager->NewTrack(track, mctrack, fCentralityF, -1, signal);
@@ -1632,38 +1616,46 @@ void AliAnalysisTaskHFE::ProcessAOD(){
       }
     }
 
-    if(fApplyCutAOD) {
-      //printf("Apply cuts\n");
-      // RecKine: ITSTPC cuts  
-      if(!ProcessCutStep(AliHFEcuts::kStepRecKineITSTPC, track)) continue;
+    // begin AOD QA
+    fQACollection->Fill("Filterbegin", -1);  
+    for(Int_t k=0; k<20; k++) {
+      Int_t u = 1<<k;
+      if((track->TestFilterBit(u))) {
+           fQACollection->Fill("Filterbegin", k);
+      }
+    }
 
-      // Reject kink mother
-      if(fRejectKinkMother) {
-       Bool_t kinkmotherpass = kTRUE;
-       for(Int_t kinkmother = 0; kinkmother < numberofmotherkink; kinkmother++) {
-         if(track->GetID() == listofmotherkink[kinkmother]) {
-           kinkmotherpass = kFALSE;
-           continue;
-         }
-       }
-       if(!kinkmotherpass) continue;
+    // RecKine: ITSTPC cuts  
+    if(!ProcessCutStep(AliHFEcuts::kStepRecKineITSTPC, track)) continue;
+
+    fQACollection->Fill("Kinkbefore", track->Pt(), kinkstatus); 
+    // Reject kink mother
+    if(fRejectKinkMother) {
+      Bool_t kinkmotherpass = kTRUE;
+      for(Int_t ikinkmother = 0; ikinkmother < numberofmotherkink; ikinkmother++) {
+        if(track->GetID() == listofmotherkink[ikinkmother]) {
+          kinkmotherpass = kFALSE;
+          continue;
+        }
       }
+      if(!kinkmotherpass) continue;
+    }       
+
+    // RecPrim
+    if(!ProcessCutStep(AliHFEcuts::kStepRecPrim, track)) continue;
+    fQACollection->Fill("Kinkafter", track->Pt(), kinkstatus); 
+
+    // HFEcuts: ITS layers cuts
+    if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsITS, track)) continue;
       
-      // RecPrim
-      if(!ProcessCutStep(AliHFEcuts::kStepRecPrim, track)) continue;
-      
-      // HFEcuts: ITS layers cuts
-      if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsITS, track)) continue;
-      
-      // HFE cuts: TOF PID and mismatch flag
-      if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTOF, track)) continue;
-      
-      // HFE cuts: TPC PID cleanup
-      if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTPC, track)) continue;
+    // HFE cuts: TOF PID and mismatch flag
+    if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTOF, track)) continue;
       
-      // HFEcuts: Nb of tracklets TRD0
-      if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTRD, track)) continue;
-    }
+    // HFE cuts: TPC PID cleanup
+    if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTPC, track)) continue;
+    
+    // HFEcuts: Nb of tracklets TRD0
+    if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTRD, track)) continue;
 
     // Fill correlation maps before PID
     if(signal && fContainer->GetCorrelationMatrix("correlationstepbeforePID")) {
@@ -1672,6 +1664,25 @@ void AliAnalysisTaskHFE::ProcessAOD(){
     }
 
     if(HasMCData()){
+
+      if(fMCQA && signal){
+        fMCQA->SetCentrality(fCentralityF);
+        if(mctrack && (TMath::Abs(mctrack->GetPdgCode()) == 11)){
+         Double_t weightElecBgV0[kBgLevels] = {0.,0.,0.};
+         for(Int_t iLevel = 0; iLevel < kBgLevels; iLevel++){
+           weightElecBgV0[iLevel] = fMCQA->GetWeightFactor(mctrack, iLevel); // positive:conversion e, negative: nonHFE 
+         }
+         if(weightElecBgV0[0]>0) {
+             fVarManager->FillContainer(fContainer, "conversionElecs", 3, kFALSE, weightElecBgV0[0]);
+             fVarManager->FillContainer(fContainer, "conversionElecs", 4, kTRUE, weightElecBgV0[0]);
+         }
+         else if(weightElecBgV0[0]<0) {
+             fVarManager->FillContainer(fContainer, "mesonElecs", 3, kFALSE, -1*weightElecBgV0[0]);
+             fVarManager->FillContainer(fContainer, "mesonElecs", 4, kTRUE, -1*weightElecBgV0[0]);
+         }
+        }
+      }
+
       Double_t hfeimpactR4all=0., hfeimpactnsigmaR4all=0.;
       Int_t sourceDca =-1;
       if(mctrack && (TMath::Abs(mctrack->GetPdgCode()) == 211)){
@@ -1728,26 +1739,40 @@ void AliAnalysisTaskHFE::ProcessAOD(){
     hfetrack.SetCentrality(fCentralityF);
     hfetrack.SetMulitplicity(ncontribVtx); // for correction
     if(IsPbPb()) hfetrack.SetPbPb();
-    else hfetrack.SetPP();
+    else{
+      if(IspPb()) hfetrack.SetpPb();
+      else hfetrack.SetPP();
+    }
     fPID->SetVarManager(fVarManager);
     if(!fPID->IsSelected(&hfetrack, fContainer, "recTrackCont", fPIDqa)) continue;   
     // we will do PID here as soon as possible
 
     // Background subtraction----------------------------------------------------------------------------------------------
-    if (GetPlugin(kNonPhotonicElectron)) {
+    if (GetPlugin(kNonPhotonicElectron)&&!GetPlugin(kNonPhotonicElectronBeauty)) {
       Int_t indexmother = -1;
-      Int_t mcsource = -1;
-      if(HasMCData())  mcsource = fBackgroundSubtraction->FindMother(mctrack->GetLabel(),indexmother);
-      fBackgroundSubtraction->LookAtNonHFE(itrack, track, fInputEvent, 1, fCentralityF, -1,mcsource, indexmother);
+      Int_t mcsource = -1;  
+      Int_t mcQAsource = -1;
+      Double_t weightNonPhotonicFactor = 1.; 
+      //printf("weight %f \n",weightNonPhotonicFactor);
+      if(HasMCData() && mctrack){  
+        mcsource = fBackgroundSubtraction->FindMother(TMath::Abs(track->GetLabel()),indexmother);
+       if(fBackgroundSubtraction->GetLevelBack()>=0) {
+         if(fMCQA) {
+           mcQAsource = fMCQA->GetElecSource(mctrack);
+           weightNonPhotonicFactor = TMath::Abs(fMCQA->GetWeightFactor(mctrack, fBackgroundSubtraction->GetLevelBack())); // positive:conversion e, negative: nonHFE 
+         }
+       }
+      }
+       fBackgroundSubtraction->LookAtNonHFE(itrack, track, fInputEvent, weightNonPhotonicFactor, fCentralityF, -1,mcsource, indexmother,mcQAsource);
     }
     //---------------------------------------------------------------------------------------------------------------------
 
     // end AOD QA
-    fQAAODCollection->Fill("Filterend", -1);  
+    fQACollection->Fill("Filterend", -1);  
     for(Int_t k=0; k<20; k++) {
       Int_t u = 1<<k;
       if((track->TestFilterBit(u))) {
-       fQAAODCollection->Fill("Filterend", k);
+             fQACollection->Fill("Filterend", k);
       }
     }
        
@@ -1756,11 +1781,11 @@ void AliAnalysisTaskHFE::ProcessAOD(){
     if(signal) {
       // Apply weight for background contamination
       if(fBackGroundFactorApply) {
-       if(IsPbPb() && fCentralityF >= 0) fWeightBackGround =  fkBackGroundFactorArray[fCentralityF >= 0 ? fCentralityF : 0]->Eval(TMath::Abs(track->P()));
-       else    fWeightBackGround =  fkBackGroundFactorArray[0]->Eval(TMath::Abs(track->P())); // pp case
+             if(IsPbPb() && fCentralityF >= 0) fWeightBackGround =  fkBackGroundFactorArray[fCentralityF >= 0 ? fCentralityF : 0]->Eval(TMath::Abs(track->P()));
+             else    fWeightBackGround =  fkBackGroundFactorArray[0]->Eval(TMath::Abs(track->P())); // pp case
        
-       if(fWeightBackGround < 0.0) fWeightBackGround = 0.0;
-       else if(fWeightBackGround > 1.0) fWeightBackGround = 1.0;
+             if(fWeightBackGround < 0.0) fWeightBackGround = 0.0;
+             else if(fWeightBackGround > 1.0) fWeightBackGround = 1.0;
         // weightBackGround as special weight
         fVarManager->FillContainer(fContainer, "hadronicBackground", 1, kFALSE, fWeightBackGround);
       }
@@ -1768,43 +1793,6 @@ void AliAnalysisTaskHFE::ProcessAOD(){
     }
     
     nElectronCandidates++;    
-    if(HasMCData()){
-      dataE[0] = track->Pt();
-      dataE[1] = track->Eta();
-      dataE[2] = track->Phi();
-      dataE[3] = track->Charge();
-      dataE[4] = -1;
-      dataE[5] = -1;
-      // Track selected: distinguish between true and fake
-      AliDebug(1, Form("Candidate Selected, filling THnSparse, PID: %d\n", mctrack ? mctrack->GetPdgCode(): -1));
-      if(mctrack && ((pid = TMath::Abs(mctrack->GetPdgCode())) == 11)){
-      
-        Int_t type = 0;
-        if(fSignalCuts->IsCharmElectron(track))
-          type = 1;
-        else if(fSignalCuts->IsBeautyElectron(track))
-          type = 2;
-        AliDebug(1, Form("Type: %d\n", type));
-        if(type){
-                dataE[5] = type; // beauty[1] or charm[2]
-                dataE[4] = 2;  // signal electron
-        }
-        else{
-                dataE[4] = 1; // not a signal electron
-                dataE[5] = 0;
-        }
-      } 
-      else {
-        // Fill THnSparse with the information for Fake Electrons
-        dataE[4] = 0;
-        dataE[5] = 0;
-      }
-      // fill the performance THnSparse, if the mc origin could be defined
-      if(dataE[4] > -1){
-        AliDebug(1, Form("Entries: [%.3f|%.3f|%.3f|%f|%f|%f]\n", dataE[0],dataE[1],dataE[2],dataE[3],dataE[4],dataE[5]));
-        fQACollection->Fill("PIDperformance", dataE);
-      }
-    }
 
     if (GetPlugin(kDEstep)) {
       if (!HasMCData()){
@@ -1830,9 +1818,39 @@ void AliAnalysisTaskHFE::ProcessAOD(){
               // weightBackGround as special weight
               fVarManager->FillContainer(fContainer, "hadronicBackground", 2, kFALSE, fWeightBackGround);
         }
+
+        fVarManager->FillContainer(fContainer, "recTrackContDEReco", AliHFEcuts::kStepHFEcutsDca, kFALSE);
+        fVarManager->FillContainer(fContainer, "recTrackContDEMC", AliHFEcuts::kStepHFEcutsDca, kTRUE);
+        fVarManager->FillCorrelationMatrix(fContainer->GetCorrelationMatrix("correlationstepafterDE"));
+      }
+    }
+
+    // Background subtraction----------------------------------------------------------------------------------------------
+    if (!GetPlugin(kNonPhotonicElectron)&&GetPlugin(kNonPhotonicElectronBeauty)) {
+      Int_t indexmother = -1;
+      Int_t mcsource = -1;  
+      Int_t mcQAsource = -1;
+      Double_t weightNonPhotonicFactor = 1.; 
+      //printf("weight %f \n",weightNonPhotonicFactor);
+      if(HasMCData() && mctrack){  
+        mcsource = fBackgroundSubtraction->FindMother(TMath::Abs(track->GetLabel()),indexmother);
+       if(fBackgroundSubtraction->GetLevelBack()>=0) {
+         if(fMCQA) {
+           mcQAsource = fMCQA->GetElecSource(mctrack);
+           weightNonPhotonicFactor = TMath::Abs(fMCQA->GetWeightFactor(mctrack, fBackgroundSubtraction->GetLevelBack())); // positive:conversion e, negative: nonHFE 
+         }
+       }
       }
+      fBackgroundSubtraction->LookAtNonHFE(itrack, track, fInputEvent, weightNonPhotonicFactor, fCentralityF, -1,mcsource, indexmother,mcQAsource);
     }
+    //---------------------------------------------------------------------------------------------------------------------
+
   }
+
+  // Background subtraction-------------------------------------------------------------------
+  if (GetPlugin(kNonPhotonicElectron)||GetPlugin(kNonPhotonicElectronBeauty)) fBackgroundSubtraction->CountPoolAssociated(fInputEvent, fCentralityF);
+  //------------------------------------------------------------------------------------------
+
   fQACollection->Fill("nElectronTracksEvent", nElectronCandidates);
 }
 
@@ -1844,12 +1862,7 @@ Bool_t AliAnalysisTaskHFE::ProcessMCtrack(AliVParticle *track){
   // Works for AOD and MC analysis Type
   //
   fVarManager->NewTrack(track, NULL, fCentralityF, -1, kTRUE);
-  Double_t signalContainer[6];
 
-  signalContainer[0] = track->Pt();
-  signalContainer[1] = track->Eta();
-  signalContainer[2] = track->Phi();
-  signalContainer[3] = track->Charge()/3;
 
   Double_t vertex[3] = {0.,0.,0.}; // Production vertex cut to mask gammas which are NOT supposed to have hits in the first ITS layer(s)
   if(IsESDanalysis()){
@@ -1867,29 +1880,6 @@ Bool_t AliAnalysisTaskHFE::ProcessMCtrack(AliVParticle *track){
   if(!fCFM->CheckParticleCuts(AliHFEcuts::kStepMCGenerated, track)) return kFALSE;
   //printf("MC Generated pass\n");
   fVarManager->FillContainer(fContainer, "MCTrackCont", AliHFEcuts::kStepMCGenerated, kFALSE);
-  signalContainer[4] = 0;
-  if(fSignalCuts->IsSelected(track)){
-    //fVarManager->FillContainer(fContainer, "MCTrackCont", AliHFEcuts::kStepMCsignal, kFALSE);
-    // Filling of the Signal/Background histogram using the 
-    // definition of the codes for charm and beauty as below in
-    // th crearion of the histogram
-    if(fSignalCuts->IsCharmElectron(track))
-      signalContainer[4] = 1;
-    else 
-      signalContainer[4] = 2;
-  } else {
-    signalContainer[4] = 0; // (and other background)
-  }
-  signalContainer[5] = 0;
-  // apply cut on the sqrt of the production vertex
-  Double_t radVertex = TMath::Sqrt(vertex[0]*vertex[0] + vertex[1] * vertex[1]);
-  if(radVertex < 3.5){
-    // Within first ITS layer(2) -> Background we cannot reject by ITS cut, let it pass
-    signalContainer[5] = 1;
-  } else if (radVertex < 7.5){
-    signalContainer[5] = 2;
-  }
-  fQACollection->Fill("SignalToBackgroundMC", signalContainer);
 
   // Step GeneratedZOutNoPileUp
   if((fIdentifiedAsPileUp) || (TMath::Abs(fVz) > fCuts->GetVertexRange()) || (fCentralityF < 0)) return kFALSE;
@@ -1900,8 +1890,10 @@ Bool_t AliAnalysisTaskHFE::ProcessMCtrack(AliVParticle *track){
   if(!fPassTheEventCut) return kFALSE;
   fVarManager->FillContainer(fContainer, "MCTrackCont", AliHFEcuts::kStepMCGeneratedEventCut, kFALSE);
 
-  if(!fCFM->CheckParticleCuts(AliHFEcuts::kStepMCInAcceptance, track)) return kFALSE;
-  fVarManager->FillContainer(fContainer, "MCTrackCont", AliHFEcuts::kStepMCInAcceptance, kFALSE);
+  if(IsESDanalysis()){
+    if(!fCFM->CheckParticleCuts(AliHFEcuts::kStepMCInAcceptance, track)) return kFALSE;
+    fVarManager->FillContainer(fContainer, "MCTrackCont", AliHFEcuts::kStepMCInAcceptance, kFALSE);
+  }
   return kTRUE;
 }
 
@@ -2028,31 +2020,6 @@ void AliAnalysisTaskHFE::MakeParticleContainer(){
     fContainer->SetStepTitle("recTrackContMC", fPID->SortedDetectorName(ipid), AliHFEcuts::kNcutStepsRecTrack + ipid);
   }
 }
-
-//____________________________________________________________
-void AliAnalysisTaskHFE::InitPIDperformanceQA(){
-  // Add a histogram for Fake electrons
-  const Int_t nDim=6;
-  Int_t nBin[nDim] = {40, 8, 18, 2, 3, 3};
-  //number of variables on the grid:pt,eta,phi,charge,
-  const Double_t kPtbound[2] = {0.1, 20.};
-  const Double_t kEtabound[2] = {-0.8, 0.8};
-  const Double_t kPhibound[2] = {0., 2. * TMath::Pi()}; 
-  const Double_t kChargebound[2] = {-1.1, 1.1};
-  const Double_t kAddInf1bound[2] = {0., 3.};
-  const Double_t kAddInf2bound[2] = {0., 3.};
-  Double_t minima[nDim] = {kPtbound[0], kEtabound[0], kPhibound[0], kChargebound[0], kAddInf1bound[0], kAddInf2bound[0]}; 
-  Double_t maxima[nDim] = {kPtbound[1], kEtabound[1], kPhibound[1], kChargebound[1], kAddInf1bound[1], kAddInf2bound[1]}; 
-  
-  fQACollection->CreateTHnSparse("PIDperformance", "PID performance; pT [GeV/c]; theta [rad]; phi [rad]; charge; type (0 - not el, 1 - other el, 2 - HF el; flavor (0 - no, 1 - charm, 2 - bottom)", nDim, nBin, minima, maxima);
-  fQACollection->CreateTHnSparse("SignalToBackgroundMC", "PID performance; pT [GeV/c]; theta [rad]; phi [rad]; charge; flavor (0 - no, 1 - charm, 2 - bottom); ITS Cluster (0 - no, 1 - first (and maybe second), 2 - second)", nDim, nBin, minima, maxima);
-
-  fQACollection->BinLogAxis("PIDperformance", 0);
-  fQACollection->BinLogAxis("SignalToBackgroundMC", 0);
-  fQACollection->Sumw2("PIDperformance");
-  fQACollection->Sumw2("SignalToBackgroundMC");
-}
-
 //____________________________________________________________
 void AliAnalysisTaskHFE::InitContaminationQA(){
   // 
@@ -2228,6 +2195,7 @@ void AliAnalysisTaskHFE::SwitchOnPlugin(Int_t plug){
     case kDEstep: SETBIT(fPlugins, plug); break;
     case kTaggedTrackAnalysis: SETBIT(fPlugins, plug); break;
     case kNonPhotonicElectron: SETBIT(fPlugins, plug); break; 
+    case kNonPhotonicElectronBeauty: SETBIT(fPlugins, plug); break; 
     default: AliError("Unknown Plugin");
   };
 }
@@ -2252,16 +2220,16 @@ Bool_t AliAnalysisTaskHFE::ReadCentrality() {
   //
   
   Float_t fCentralityLimitstemp[12];
-  Float_t fCentralityLimitsdefault[12]= {0.,5.,10., 20., 30., 40., 50., 60.,70.,80., 90., 100.};
+  Float_t fCentralityLimitsdefault[12]= {0.,5.,10., 20., 30., 40., 50., 60.,70.,80., 90., 100.00001};
   if(!fPbPbUserCentralityBinning) memcpy(fCentralityLimitstemp,fCentralityLimitsdefault,sizeof(fCentralityLimitsdefault));
   else memcpy(fCentralityLimitstemp,fCentralityLimits,sizeof(fCentralityLimitsdefault));
   
 
   Int_t bin = -1;
-  if(IsPbPb()) {
+  if(IsPbPb()||IspPb()) {
     // Centrality
     AliCentrality *centrality = fInputEvent->GetCentrality();
-    fCentralityPercent = centrality->GetCentralityPercentile("V0M");
+    fCentralityPercent = centrality->GetCentralityPercentile(fCentralityEstimator.Data());
     //printf("centrality %f\n",fCentralityPercent);
 
     for(Int_t ibin = 0; ibin < 11; ibin++){
@@ -2304,7 +2272,7 @@ Bool_t AliAnalysisTaskHFE::ReadCentrality() {
       AliError("ESD Event required for ESD Analysis");
       return kFALSE;
     }
-    vtx = fESD->GetPrimaryVertexSPD();
+    vtx = fESD->GetPrimaryVertex() ;
   }
   if(!vtx){ 
     fContributors = 0.5;
@@ -2350,29 +2318,6 @@ Int_t AliAnalysisTaskHFE::GetITSMultiplicity(AliVEvent *ev){
   return nAcc;
 }
 
-//___________________________________________________
-Bool_t AliAnalysisTaskHFE::InitializeHadronBackground(Int_t run){
-  //
-  // Initialize background factors array from the AliOADBContainer
-  // The container is expected to provide a TObjArray with the name
-  // "hadronBackground" and the size 12 for the given run number
-  //
-  AliDebug(1, "Deriving hadronic background parameterization from OADB container");
-  TObjArray *functions = dynamic_cast<TObjArray *>(fHadronBackgroundOADB->GetObject(run, "hadronBackground"));
-  if(!functions){
-    AliDebug(1, "Content in the OADB Container is not a TObjArray");
-    fBackGroundFactorApply = kFALSE;
-    return kFALSE;
-  } 
-  if(functions->GetSize() < 12){
-    AliDebug(1, Form("Size not matching: 12 expected, %d provided", functions->GetSize()));
-    fBackGroundFactorApply = kFALSE;
-    return kFALSE;
-  }
-  for(Int_t icent = 0; icent < 12; icent++) fkBackGroundFactorArray[icent] = dynamic_cast<const TF1 *>(functions->UncheckedAt(icent));
-  return kTRUE;
-}
-
 //___________________________________________________
 void AliAnalysisTaskHFE::RejectionPileUpVertexRangeEventCut() {
   //
@@ -2408,26 +2353,340 @@ void AliAnalysisTaskHFE::RejectionPileUpVertexRangeEventCut() {
    // PileUp
    fIdentifiedAsPileUp = kFALSE;
    if(fRemovePileUp && fESD->IsPileupFromSPD()) fIdentifiedAsPileUp = kTRUE; 
+   
+
+
    // Z vertex
    fIdentifiedAsOutInz = kFALSE;
-   if (IsPbPb()) {
-     //printf("PbPb\n");
-     if(fESD->GetPrimaryVertex()){
-       if(TMath::Abs(fESD->GetPrimaryVertex()->GetZ()) > fCuts->GetVertexRange()) fIdentifiedAsOutInz = kTRUE;
-     }
-   }
-   else {
-     //printf("pp\n");
-     if(fESD->GetPrimaryVertexTracks()){
-       if(TMath::Abs(fESD->GetPrimaryVertexTracks()->GetZ()) > fCuts->GetVertexRange()) fIdentifiedAsOutInz = kTRUE;
-     }
+   Bool_t findvertex = kTRUE;
+   const AliESDVertex* vtxESD = fESD->GetPrimaryVertex();
+   if((!vtxESD) || (vtxESD->GetNContributors() <= 0)) findvertex = kFALSE;
+   if(findvertex) {
+     if(TMath::Abs(vtxESD->GetZ()) > fCuts->GetVertexRange()) fIdentifiedAsOutInz = kTRUE;
    }
+   
    //Event Cut
    fPassTheEventCut = kTRUE;
    if(!fCFM->CheckEventCuts(AliHFEcuts::kEventStepReconstructed, fESD)) fPassTheEventCut = kFALSE;   
   
-
  }
 
 }
 
+//___________________________________________________
+Bool_t AliAnalysisTaskHFE::CheckTRDTriggerESD(AliESDEvent *ev) {
+//
+// Check TRD trigger; pPb settings
+//
+    Bool_t cint8=kFALSE;
+    Bool_t cint7=kFALSE;
+    Bool_t cint5=kFALSE;
+    Bool_t cint8s=kFALSE;
+    Bool_t cint7s=kFALSE;
+    Bool_t cint7ppb=kFALSE;
+    Bool_t trdtrgevent=kFALSE;
+
+  //  printf("TRIGGERS %s \n",ev->GetFiredTriggerClasses().Data());
+
+    fTRDTriggerAnalysismb->CalcTriggers(ev);
+
+    // mb selection of WU events
+    if(fWhichTRDTrigger==1)
+    {
+//     if(ev->IsTriggerClassFired("CINT7WU-B-NOPF-ALL"))
+       if((ev->IsTriggerClassFired("CINT7WU-B-NOPF-ALL"))||(ev->IsTriggerClassFired("CINT7WU-S-NOPF-ALL"))||(ev->IsTriggerClassFired("CINT8WU-S-NOPF-ALL")))
+       {
+           DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+        else return kFALSE;
+    }
+
+
+    // HSE no cleanup
+    if(fWhichTRDTrigger==2)
+    {
+       cint8= ev->IsTriggerClassFired("CINT8WUHSE-B-NOPF-CENT");
+       cint7= ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT");  // pPb rare
+       cint7ppb= ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-ALL"); // pPb mb
+       cint8s= ev->IsTriggerClassFired("CINT7WUHSE-S-NOPF-CENT");
+       cint7s= ev->IsTriggerClassFired("CINT8WUHSE-S-NOPF-CENT");
+       cint5= (ev->IsTriggerClassFired("CINT5WU-B-NOPF-ALL")) &&
+           (ev->GetHeader()->GetL1TriggerInputs() & (1 << 10));
+//        printf("hse trigger %i %i %i %i %i \n",cint7,cint7ppb,cint8,cint7s,cint8s);
+       if((cint7==kFALSE)&&(cint7ppb==kFALSE)&&(cint8==kFALSE)&&(cint7s==kFALSE)&&(cint8s==kFALSE)&&(cint5==kFALSE)) return kFALSE;
+       else
+       {
+           DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+    }
+
+   
+
+    //HQU no cleanup
+    if(fWhichTRDTrigger==3)
+    {
+       cint8= ev->IsTriggerClassFired("CINT8WUHQU-B-NOPF-CENT");
+       cint7= ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT");    // pPb rare
+       cint7ppb= ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-ALL"); // pPb mb
+       cint8s= ev->IsTriggerClassFired("CINT7WUHQU-S-NOPF-CENT");
+       cint7s= ev->IsTriggerClassFired("CINT8WUHQU-S-NOPF-CENT");
+       cint5= (ev->IsTriggerClassFired("CINT5WU-B-NOPF-ALL")) &&
+           (ev->GetHeader()->GetL1TriggerInputs() & (1 << 12));
+ //       printf("hqu trigger %i %i %i %i %i \n",cint7,cint7ppb,cint8,cint7s,cint8s);
+       if((cint7==kFALSE)&&(cint7ppb==kFALSE)&&(cint8==kFALSE)&&(cint7s==kFALSE)&&(cint8s==kFALSE)&&(cint5==kFALSE)) return kFALSE;
+       else
+       {
+           DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+    }
+
+    if(fWhichTRDTrigger==4)
+    {
+//     printf("trigger %i %i \n", ev->GetHeader()->IsTriggerInputFired("1HSE"),(ev->GetHeader()->GetL1TriggerInputs() & (1 << 10))); // bug in IsTriggerInputFired; reported in savannah
+
+//     if(ev->IsTriggerClassFired("CINT7WU-B-NOPF-ALL"))
+//     {
+//         Int_t trginput=0;
+//         trginput=ev->GetHeader()->GetL1TriggerInputs() & (1 << 10);  // HSE
+           //      if(trginput==1024)
+//         if(fTRDTriggerAnalysismb->CheckCondition(AliTRDTriggerAnalysis::kHSE))
+//         if(fTRDTriggerAnalysismb->HasFired(AliTRDTriggerAnalysis::kHSE))
+           if(fTRDTriggerAnalysismb->HasTriggered(AliTRDTriggerAnalysis::kHSE))
+           {
+               DrawTRDTrigger(ev);
+               DrawTRDTriggerAnalysis(ev);
+               return kTRUE;
+           } else return kFALSE;
+    }// else return kFALSE;
+//    }
+    if(fWhichTRDTrigger==5)
+    {
+//     printf("trigger %i %i \n", ev->GetHeader()->IsTriggerInputFired("1HQU"),(ev->GetHeader()->GetL1TriggerInputs() & (1 << 12))); // bug in IsTriggerInputFired; reported in savannah
+//     if(ev->IsTriggerClassFired("CINT7WU-B-NOPF-ALL"))
+//     {
+
+//         Int_t trginput=0;
+//         trginput=ev->GetHeader()->GetL1TriggerInputs() & (1 << 12);  //HQU
+           //        printf("triggerinput %i \n",trginput);
+           //      if(trginput==4096)
+//         if(fTRDTriggerAnalysismb->CheckCondition(AliTRDTriggerAnalysis::kHQU))
+//          if(fTRDTriggerAnalysismb->HasFired(AliTRDTriggerAnalysis::kHQU))
+          if(fTRDTriggerAnalysismb->HasTriggered(AliTRDTriggerAnalysis::kHQU))
+           {
+               DrawTRDTrigger(ev);
+               DrawTRDTriggerAnalysis(ev);
+               return kTRUE;
+           } else return kFALSE;
+    } //else return kFALSE;
+//    }
+   
+
+    return trdtrgevent;
+
+}
+
+
+//___________________________________________________
+Bool_t AliAnalysisTaskHFE::CheckTRDTrigger(AliVEvent *ev) {
+//
+// Check TRD trigger; pPb settings
+//
+
+    fTRDTriggerAnalysistrg->CalcTriggers(ev);
+
+    // HSE cleanup
+    if(fWhichTRDTrigger==6)
+    {
+       if(!fTRDTriggerAnalysistrg->HasTriggeredConfirmed(AliTRDTriggerAnalysis::kHSE))
+       {
+           return kFALSE;
+       }
+       else
+       {
+           //   DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+    }
+
+
+    // HQU cleanup
+    if(fWhichTRDTrigger==7)
+    {
+
+       if(!fTRDTriggerAnalysistrg->HasTriggeredConfirmed(AliTRDTriggerAnalysis::kHQU))
+       {
+           return kFALSE;
+       }
+       else
+       {
+           //     DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+    }
+
+    // HSE or HQU cleanup
+    if(fWhichTRDTrigger==8)
+    {
+       if((fTRDTriggerAnalysistrg->HasTriggeredConfirmed(AliTRDTriggerAnalysis::kHSE))||(fTRDTriggerAnalysistrg->HasTriggeredConfirmed(AliTRDTriggerAnalysis::kHQU)))
+       {
+           //    DrawTRDTrigger(ev);
+           DrawTRDTriggerAnalysis(ev);
+           return kTRUE;
+       }
+       else
+       {
+           return kFALSE; 
+       }
+    }
+
+    return kFALSE;
+
+}
+
+//___________________________________________________
+void AliAnalysisTaskHFE::DrawTRDTrigger(AliESDEvent *ev) {
+
+    Int_t ntriggerbit=0;
+    fQACollection->Fill("nTriggerBit",ntriggerbit);
+    if(ev->IsTriggerClassFired("CINT7-B-NOPF-ALLNOTRD"))
+    {
+       ntriggerbit=2;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+    }
+    if(ev->IsTriggerClassFired("CINT7WU-B-NOPF-ALL"))
+    {
+       ntriggerbit=3;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+       if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")) {
+           ntriggerbit=18;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+       if(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT")) {
+           ntriggerbit=19;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHJT-B-NOPF-CENT"))
+    {
+       ntriggerbit=4;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+
+       if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")) {
+           ntriggerbit=13;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+       if(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT")) {
+           ntriggerbit=14;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+           if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")) {
+               ntriggerbit=17;
+               fQACollection->Fill("nTriggerBit",ntriggerbit);
+           }
+       }
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT"))
+    {
+       ntriggerbit=5;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+       if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")) {
+           ntriggerbit=11;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+       if((!(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")))&&(!(ev->IsTriggerClassFired("CINT7WUHJT-B-NOPF-CENT")))) {
+           ntriggerbit=21;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+
+            /*
+           Int_t nTrdTracks = ev->GetNumberOfTrdTracks();
+           for (Int_t iTrack = 0; iTrack < nTrdTracks; ++iTrack) {
+               AliESDTrdTrack* trdTrack = ev->GetTrdTrack(iTrack);
+               printf("GTU track %3i: pt = %5.1f, PID = %3i\n", iTrack, trdTrack->Pt(), trdTrack->GetPID());
+           }*/
+
+
+       }
+
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT"))
+    {
+       ntriggerbit=6;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+       if(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT")) {
+           ntriggerbit=12;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+       if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-FAST")){
+           ntriggerbit=15;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+
+           if((!(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-CENT")))&&(!(ev->IsTriggerClassFired("CINT7WUHJT-B-NOPF-CENT")))) {
+               ntriggerbit=20;
+               fQACollection->Fill("nTriggerBit",ntriggerbit);
+               /*
+                Int_t nTrdTracks = ev->GetNumberOfTrdTracks();
+                for (Int_t iTrack = 0; iTrack < nTrdTracks; ++iTrack) {
+                AliESDTrdTrack* trdTrack = ev->GetTrdTrack(iTrack);
+                printf("HSE GTU track %3i: pt = %5.1f, PID = %3i\n", iTrack, trdTrack->Pt(), trdTrack->GetPID());
+                }                          */
+
+           }
+
+       }
+
+    }
+    if(ev->IsTriggerClassFired("CEMC7WUHEE-B-NOPF-CENT")) {
+       ntriggerbit=7;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHJT-B-NOPF-FAST")){
+       ntriggerbit=8;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHQU-B-NOPF-FAST")){
+       ntriggerbit=9;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+    }
+    if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-FAST")){
+       ntriggerbit=10;
+       fQACollection->Fill("nTriggerBit",ntriggerbit);
+       if(ev->IsTriggerClassFired("CINT7WUHSE-B-NOPF-CENT")) {
+           ntriggerbit=16;
+           fQACollection->Fill("nTriggerBit",ntriggerbit);
+       }
+    }
+    if(ntriggerbit==0) fQACollection->Fill("nTriggerBit",1);
+
+}
+
+
+//___________________________________________________
+void AliAnalysisTaskHFE::DrawTRDTriggerAnalysis(AliVEvent *ev) {
+
+    fTRDTriggerAnalysistrg->CalcTriggers(ev);
+    for(Int_t itrg=0;itrg<AliTRDTriggerAnalysis::kHlast;itrg++)
+    {
+        Int_t trdtrgstatus=0;
+       if(fTRDTriggerAnalysistrg->CheckCondition((AliTRDTriggerAnalysis::TRDTrigger_t) itrg))trdtrgstatus=1;
+       if(fTRDTriggerAnalysistrg->HasFired((AliTRDTriggerAnalysis::TRDTrigger_t) itrg))trdtrgstatus=2;
+       if(fTRDTriggerAnalysistrg->HasTriggered((AliTRDTriggerAnalysis::TRDTrigger_t) itrg))trdtrgstatus=3;
+       if(fTRDTriggerAnalysistrg->HasTriggeredConfirmed((AliTRDTriggerAnalysis::TRDTrigger_t) itrg))trdtrgstatus=4;
+        fQACollection->Fill("TriggerAnalysis",(Float_t)itrg,(Float_t)trdtrgstatus);
+    }
+}
+
+//___________________________________________________
+Bool_t AliAnalysisTaskHFE::IsMCFakeTrack(const AliVTrack *const trk) const {
+  //
+  // Check whether track is MC Fake track using the sign of the track label
+  //
+  return trk->GetLabel() < 0;
+}