check the leading particle in the cone, apply the isolation cone condition just on...

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEreducedMCParticle.h

diff --git a/PWGHF/hfe/AliHFEreducedMCParticle.h b/PWGHF/hfe/AliHFEreducedMCParticle.h
index ea510e0212646a39646b9f12e196189248c2416a..8da7af13063a94510c0af8cca33ad41820e5162d 100644 (file)
--- a/PWGHF/hfe/AliHFEreducedMCParticle.h
+++ b/PWGHF/hfe/AliHFEreducedMCParticle.h
@@ -40,11 +40,15 @@ class AliHFEreducedMCParticle : public TObject{
   Int_t Pdg() const { return fPdg; }
   Int_t MotherPdg() const { return fMotherPdg; }
   Int_t Source() const { return static_cast<Int_t>(fSource); }
+  Int_t GetElectronSource() const { return static_cast<Int_t>(fEleSource); }
   Bool_t IsSignal() const { return fSignal; }
   Double_t RadialProductionVertex() const { return TMath::Abs(fProductionVertex[0]*fProductionVertex[0]+fProductionVertex[1]*fProductionVertex[1]); }
   Double_t VX() const { return fProductionVertex[0]; }
   Double_t VY() const { return fProductionVertex[1]; }
   Double_t VZ() const { return fProductionVertex[2]; }
+  Double_t GetMotherVX() const { return fMotherProductionVertex[0]; }
+  Double_t GetMotherVY() const { return fMotherProductionVertex[1]; }
+  Double_t GetMotherVZ() const { return fMotherProductionVertex[2]; }
   
   void SetSignedPt(Double_t pt, Bool_t positiveCharge){
     double chargesign = positiveCharge ? 1. : -1.;
@@ -54,14 +58,20 @@ class AliHFEreducedMCParticle : public TObject{
   void SetEta(Double_t eta) { fEta = eta; }
   void SetPhi(Double_t phi) { fPhi = phi; }
   void SetPdg(Int_t pdg) { fPdg = pdg; }
-  void SetMotherPdg(Int_t pdg) { fPdg = pdg; }
+  void SetMotherPdg(Int_t pdg) { fMotherPdg = pdg; }
   void SetSource(Int_t source) { fSource = static_cast<Char_t>(source); }
+  void SetElectronSource(Int_t source) { fEleSource = static_cast<UChar_t>(source); }
   void SetSignal() { fSignal = kTRUE; }
   void SetProductionVertex(Double_t vx, Double_t vy, Double_t vz) {
     fProductionVertex[0] = vx;
     fProductionVertex[1] = vy;
     fProductionVertex[2] = vz;
   }
+  void SetMotherProductionVertex(Double_t vx, Double_t vy, Double_t vz) {
+    fMotherProductionVertex[0] = vx;
+    fMotherProductionVertex[1] = vy;
+    fMotherProductionVertex[2] = vz;
+  }
   
  private:
   Double_t  fSignedPt;              // signed pt
@@ -73,8 +83,10 @@ class AliHFEreducedMCParticle : public TObject{
   Char_t    fSource;                // source
   Bool_t    fSignal;                // signal
   Double_t  fProductionVertex[3];   // production vertex
+  Double_t  fMotherProductionVertex[3];   // production vertex
+  UChar_t   fEleSource;             // Electron source (AliHFEmcQA)
   
-  ClassDef(AliHFEreducedMCParticle, 1)
+  ClassDef(AliHFEreducedMCParticle, 2)
   
 };
 #endif