select MC particles for TT when embedding

diff --git a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.cxx b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.cxx
index c92384ad248a8babdb1bf30b6af527b2d581ca8f..3a9e3fe1167ab9119323c7597e0800cd16504de4 100644 (file)
--- a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.cxx
+++ b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.cxx
@@ -50,6 +50,7 @@ namespace EmcalHJetMassAnalysis {
     fPtTTMin(0),
     fPtTTMax(0),
     fRandom(0),
+    fEmbConstSel(0),
     fh1PtHadron(0),
     fh3PtHPtJDPhi(0),
     fh3PtJet1VsMassVsHPtAllSel(0),
@@ -106,6 +107,7 @@ namespace EmcalHJetMassAnalysis {
     fPtTTMin(0),
     fPtTTMax(0),
     fRandom(0),
+    fEmbConstSel(0),
     fh1PtHadron(0),
     fh3PtHPtJDPhi(0),
     fh3PtJet1VsMassVsHPtAllSel(0),
@@ -273,7 +275,7 @@ namespace EmcalHJetMassAnalysis {
             Double_t dphi = GetDeltaPhi(vp,jet)-TMath::Pi();
             fh3PtHPtJDPhi[fCentBin]->Fill(vp->Pt(),jet->Pt() - GetRhoVal(fContainerBase)*jet->Area(),dphi);
             if(dphi>fDPhiHJetMax) continue;
-            FillHJetHistograms(vp->Pt(),jet);
+            FillHJetHistograms(vp,jet);
           }
         }
       }
@@ -289,7 +291,7 @@ namespace EmcalHJetMassAnalysis {
          Double_t dphi = GetDeltaPhi(vp,jet)-TMath::Pi();
          fh3PtHPtJDPhi[fCentBin]->Fill(vp->Pt(),jet->Pt() - GetRhoVal(fContainerBase)*jet->Area(),dphi);
          if(dphi>fDPhiHJetMax) continue;
-         FillHJetHistograms(vp->Pt(),jet);
+         FillHJetHistograms(vp,jet);
        }
      }//trigger track types
     }
@@ -319,9 +321,10 @@ namespace EmcalHJetMassAnalysis {
   }
 
   //________________________________________________________________________
-  Bool_t AliAnalysisTaskEmcalHJetMass::FillHJetHistograms(Double_t pt, const AliEmcalJet *jet)
+  Bool_t AliAnalysisTaskEmcalHJetMass::FillHJetHistograms(const AliVParticle *vp, const AliEmcalJet *jet)
   {
     // Fill hadron-jet histograms.
+    Double_t pt = vp->Pt();
     Double_t ptJet = jet->Pt() - GetRhoVal(fContainerBase)*jet->Area();
     Double_t mJet = GetJetMass(jet);
     Double_t rat = -1.;
@@ -353,8 +356,10 @@ namespace EmcalHJetMassAnalysis {
     }
   
     if(fMinFractionShared>0. && fraction>fMinFractionShared) {
-      fh3PtJet1VsMassVsHPtAllSelMatch[fCentBin]->Fill(ptJet,mJet,pt);
-      fh3PtJet1VsRatVsHPtAllSelMatch[fCentBin]->Fill(ptJet,rat,pt);
+      if(vp->TestBits(fEmbConstSel) == (Int_t)fEmbConstSel) {
+        fh3PtJet1VsMassVsHPtAllSelMatch[fCentBin]->Fill(ptJet,mJet,pt);
+        fh3PtJet1VsRatVsHPtAllSelMatch[fCentBin]->Fill(ptJet,rat,pt);
+      }
     }
 
     if(jet->GetTagStatus()<1 || !jet->GetTaggedJet())
@@ -364,8 +369,10 @@ namespace EmcalHJetMassAnalysis {
     fh3PtJet1VsRatVsHPtTagged[fCentBin]->Fill(ptJet,rat,pt);
 
     if(fMinFractionShared>0. && fraction>fMinFractionShared) {
-      fh3PtJet1VsMassVsHPtTaggedMatch[fCentBin]->Fill(ptJet,mJet,pt);
-      fh3PtJet1VsRatVsHPtTaggedMatch[fCentBin]->Fill(ptJet,rat,pt);
+      if(vp->TestBits(fEmbConstSel) == (Int_t)fEmbConstSel) {
+        fh3PtJet1VsMassVsHPtTaggedMatch[fCentBin]->Fill(ptJet,mJet,pt);
+        fh3PtJet1VsRatVsHPtTaggedMatch[fCentBin]->Fill(ptJet,rat,pt);
+      }
     }
     return kTRUE;
   }

diff --git a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.h b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.h
index da4793e4f5e449b662ce479d1d34e8a265dfb612..e00260699424a459cd051f5c2b9f498deec954bd 100644 (file)
--- a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.h
+++ b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskEmcalHJetMass.h
@@ -46,11 +46,12 @@ namespace EmcalHJetMassAnalysis {
     void SetMaxDeltaPhi(Double_t dphi)                                 { fDPhiHJetMax       = dphi; }
     void SetTriggerTrackType(TriggerTrackType t)                       { fTriggerTrackType  = t   ; }
     void AddTriggerTrackPtCuts(Float_t min, Float_t max);
+    void SelectConstituents(UInt_t constSel)                           { fEmbConstSel = constSel  ; }
 
   protected:
     Bool_t                              RetrieveEventObjects();
     Bool_t                              Run();
-    Bool_t                              FillHJetHistograms(Double_t pt, const AliEmcalJet *jet);
+    Bool_t                              FillHJetHistograms(const AliVParticle *vp, const AliEmcalJet *jet);
 
     Double_t                            GetJetMass(const AliEmcalJet *jet) const;
     Double_t                            GetDeltaPhi(const AliVParticle *vp, const AliEmcalJet* jet) const;
@@ -67,6 +68,7 @@ namespace EmcalHJetMassAnalysis {
     TArrayF                            *fPtTTMin;                    // minimum pt of trigger tracks
     TArrayF                            *fPtTTMax;                    // maximum pt of trigger tracks
     TRandom3                           *fRandom;                     //! Random number generator
+    UInt_t                              fEmbConstSel;                // select embedded constituents
 
     TH1F            **fh1PtHadron;                        //!pt of hadrons
     TH3F            **fh3PtHPtJDPhi;                      //!pt hadron vs pt jet vs delta phi
@@ -84,7 +86,7 @@ namespace EmcalHJetMassAnalysis {
     AliAnalysisTaskEmcalHJetMass(const AliAnalysisTaskEmcalHJetMass&);            // not implemented
     AliAnalysisTaskEmcalHJetMass &operator=(const AliAnalysisTaskEmcalHJetMass&); // not implemented
 
-    ClassDef(AliAnalysisTaskEmcalHJetMass, 4)
+    ClassDef(AliAnalysisTaskEmcalHJetMass, 5)
       };
 }
 #endif