Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliFemtoModelHiddenInfo.h

diff --git a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoModelHiddenInfo.h b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoModelHiddenInfo.h
index 2fb03614aa602112539a7ebe5e66557ce9859ba3..487fa376a79ac197719ab4fe3e3946738547aff3 100644 (file)
--- a/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoModelHiddenInfo.h
+++ b/PWGCF/FEMTOSCOPY/AliFemto/AliFemtoModelHiddenInfo.h
@@ -26,8 +26,19 @@ public:
   AliFemtoThreeVector   *GetTrueMomentum() const;
   AliFemtoLorentzVector *GetEmissionPoint() const;
   Int_t                  GetPDGPid() const;
+  Int_t                  GetMotherPdgCode() const;
   Double_t               GetMass() const;
 
+  AliFemtoThreeVector   *GetTrueMomentumPos() const;
+  AliFemtoLorentzVector *GetEmissionPointPos() const;
+  Int_t                  GetPDGPidPos() const;
+  Double_t               GetMassPos() const;
+
+  AliFemtoThreeVector   *GetTrueMomentumNeg() const;
+  AliFemtoLorentzVector *GetEmissionPointNeg() const;
+  Int_t                  GetPDGPidNeg() const;
+  Double_t               GetMassNeg() const;
+
   void                   SetTrueMomentum(AliFemtoThreeVector *aMom);
   void                   SetTrueMomentum(const AliFemtoThreeVector& aMom);
   void                   SetTrueMomentum(Double_t aPx, Double_t aPy, Double_t aPz);
@@ -35,19 +46,50 @@ public:
   void                   SetEmissionPoint(const AliFemtoLorentzVector& aPos);
   void                   SetEmissionPoint(Double_t aRx, Double_t aRy, Double_t aRz, Double_t aT);
   void                   SetPDGPid(Int_t aPid);
+  void                   SetMotherPdgCode(Int_t motherPdg);
   void                   SetMass(Double_t aMass);
 
+  void                   SetTrueMomentumPos(AliFemtoThreeVector *aMom);
+  void                   SetTrueMomentumPos(const AliFemtoThreeVector& aMom);
+  void                   SetTrueMomentumPos(Double_t aPx, Double_t aPy, Double_t aPz);
+  void                   SetEmissionPointPos(AliFemtoLorentzVector *aPos);
+  void                   SetEmissionPointPos(const AliFemtoLorentzVector& aPos);
+  void                   SetEmissionPointPos(Double_t aRx, Double_t aRy, Double_t aRz, Double_t aT);
+  void                   SetPDGPidPos(Int_t aPid);
+  void                   SetMassPos(Double_t aMass);
+
+  void                   SetTrueMomentumNeg(AliFemtoThreeVector *aMom);
+  void                   SetTrueMomentumNeg(const AliFemtoThreeVector& aMom);
+  void                   SetTrueMomentumNeg(Double_t aPx, Double_t aPy, Double_t aPz);
+  void                   SetEmissionPointNeg(AliFemtoLorentzVector *aPos);
+  void                   SetEmissionPointNeg(const AliFemtoLorentzVector& aPos);
+  void                   SetEmissionPointNeg(Double_t aRx, Double_t aRy, Double_t aRz, Double_t aT);
+  void                   SetPDGPidNeg(Int_t aPid);
+  void                   SetMassNeg(Double_t aMass);
+
 // !!! MANDATORY !!!
 // --- Copy the hidden info from AliFemtoTrack to AliFemtoParticle
   virtual AliFemtoHiddenInfo* Clone() const;
-  
+
  protected:
   virtual AliFemtoHiddenInfo* GetParticleHiddenInfo() const;
 
   AliFemtoThreeVector   *fTrueMomentum;  // True (simulated) momentum
   AliFemtoLorentzVector *fEmissionPoint; // Emission point coordinates
   Int_t                  fPDGPid;        // True PID of the particle
+  Int_t                  fMotherPdg;     // PDG code of particle's mother
   Double_t               fMass;          // True particle mass
+
+  //daughter particles
+  AliFemtoThreeVector   *fTrueMomentumPos;  // True (simulated) momentum of positive daughter
+  AliFemtoLorentzVector *fEmissionPointPos; // Emission point coordinates of positive daughter
+  Int_t                  fPDGPidPos;        // True PID of positive daughter
+  Double_t               fMassPos;          // True particle mass of positive daughter
+
+  AliFemtoThreeVector   *fTrueMomentumNeg;  // True (simulated) momentum of negative daughter
+  AliFemtoLorentzVector *fEmissionPointNeg; // Emission point coordinates of negative daughter
+  Int_t                  fPDGPidNeg;        // True PID of negative daughter
+  Double_t               fMassNeg;          // True particle mass of negative daughter
 };
 //_______________________________________
 inline AliFemtoHiddenInfo* AliFemtoModelHiddenInfo::Clone() const{