Mods for K0 and D0 flow

[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowTasks / AliFlowCandidateTrack.h

diff --git a/PWG2/FLOW/AliFlowTasks/AliFlowCandidateTrack.h b/PWG2/FLOW/AliFlowTasks/AliFlowCandidateTrack.h
index 5603c9f71fd59d77ee31b20c18b5246559018b30..9ac042803e85fbd15a8002f81675917c40ab7763 100644 (file)
--- a/PWG2/FLOW/AliFlowTasks/AliFlowCandidateTrack.h
+++ b/PWG2/FLOW/AliFlowTasks/AliFlowCandidateTrack.h
@@ -15,9 +15,10 @@
 
 class AliFlowCandidateTrack : public AliFlowTrack {
   protected:
-    Double_t fMass;      // mass
-    Int_t fNDaughters;   // number of daughters (5 max)
-    Int_t fDaughter[5]; // index of daughters
+    Double_t fMass;           // mass
+    Int_t fNDaughters;        // number of daughters (5 max)
+    Int_t fDaughter[5];       // fID of daughter, points back to ESD track
+    AliFlowTrack *fTrack[5];  // pointer to daughter in FlowEvent
     
   public:
     AliFlowCandidateTrack();
@@ -28,8 +29,13 @@ class AliFlowCandidateTrack : public AliFlowTrack {
     Double_t Mass(void)            { return fMass; }
     void SetMass(Double_t value)   { fMass=value; }
 
-    Int_t GetNDaughters(void)      { return fNDaughters; }
-    void AddDaughter(Int_t value)  { if(fNDaughters<3) fDaughter[fNDaughters++]=value; }
+    Int_t GetNDaughters(void)           { return fNDaughters; }
+    void  AddDaughter(Int_t value)      { if(fNDaughters<3) fDaughter[fNDaughters++]=value; }
+    Int_t GetIDDaughter(Int_t value) { return fDaughter[value]; }
+
+    void SetDaughter(Int_t value, AliFlowTrack *track) { fTrack[value]=track; }
+    AliFlowTrack *GetDaughter(Int_t value) { return fTrack[value]; }
+
 
     ClassDef(AliFlowCandidateTrack, 1);
 };