//K//////////////////////////////////////////////////////////////////////////M//
//K                                                                          M//
//K AliFemtoModelAllHiddenInfo -                                             M//
//K derived class inherits  the base class AliFemtoModelHiddenInfo           M//
//K the hidden info for model calculations                                   M//
//K Stores information needed for the weight generation -                    M//
//K                                                                          M//
//K in addition to  the base class AliFemtoModelHiddenInfo - the true        M//
//K simulated momenta, freeze-out coordinates from model and particle PID    M//
//K New information was added                                                M//
//K 1. Mother ID                                                             M//
//K 2. Mother 4-Momentum                                                     M//
//K 3. Mother emission point 4-vector                                        M//
//K 4. Childs IDs                                                            M//
//K 5. Childs 4-Momentum                                                     M//
//K--------------------------------------------------------------------------M//                                                                            //
//K APR2008  Konstantin Mikhailov Konstantin.Mikhailov@itep.ru               M//
//K                                                                          M//
//K//////////////////////////////////////////////////////////////////////////M//
#include "AliFemtoModelAllHiddenInfo.h"

//_____________________________________________
AliFemtoModelAllHiddenInfo::AliFemtoModelAllHiddenInfo() :
  fTrueMomentumMother(0),
  fEmissionPointMother(0),
  fPDGPidMother(0),
  fTrueMomentumChild1(0),
  fTrueMomentumChild2(0),
  fPDGPidChild1(0),
  fPDGPidChild2(0)
{
  // Default constructor
}
//_____________________________________________
AliFemtoModelAllHiddenInfo::AliFemtoModelAllHiddenInfo(const AliFemtoModelAllHiddenInfo &aInfo) :
  AliFemtoModelHiddenInfo(aInfo),
  fTrueMomentumMother(new AliFemtoLorentzVector(*(aInfo.fTrueMomentumMother))),
  fEmissionPointMother(new AliFemtoLorentzVector(*(aInfo.fEmissionPointMother))),
  fPDGPidMother(aInfo.fPDGPidMother),
  fTrueMomentumChild1(new AliFemtoLorentzVector(*(aInfo.fTrueMomentumChild1))),
  fTrueMomentumChild2(new AliFemtoLorentzVector(*(aInfo.fTrueMomentumChild2))),
  fPDGPidChild1(aInfo.fPDGPidChild1),
  fPDGPidChild2(aInfo.fPDGPidChild2)
{
  //
  // Copy constructor
  //
}
//_____________________________________________
AliFemtoModelAllHiddenInfo::~AliFemtoModelAllHiddenInfo()
{
  // Destructor
  delete fTrueMomentumMother;
  delete fEmissionPointMother;
  delete fTrueMomentumChild1;
  delete fTrueMomentumChild2;
}
//_____________________________________________
AliFemtoModelAllHiddenInfo& AliFemtoModelAllHiddenInfo::operator=(const AliFemtoModelAllHiddenInfo& aInfo)
{
  //
  // assignment operator
  //
  if (this != &aInfo) {
    AliFemtoModelHiddenInfo::operator=(aInfo);
    delete fTrueMomentumMother;
    fTrueMomentumMother = new AliFemtoLorentzVector(*(aInfo.fTrueMomentumMother));
    delete fEmissionPointMother;
    fEmissionPointMother = new AliFemtoLorentzVector(*(aInfo.fEmissionPointMother));
    fPDGPidMother = aInfo.fPDGPidMother;
    delete fTrueMomentumChild1;
    fTrueMomentumChild1 = new AliFemtoLorentzVector(*(aInfo.fTrueMomentumChild1));
    delete fTrueMomentumChild2;
    fTrueMomentumChild2 = new AliFemtoLorentzVector(*(aInfo.fTrueMomentumChild2));
    fPDGPidChild1 = aInfo.fPDGPidChild1;      
    fPDGPidChild2 = aInfo.fPDGPidChild2;   
  }
  return *this;
}
//
//   GET
//
//_____________________________________________
AliFemtoLorentzVector *AliFemtoModelAllHiddenInfo::GetTrueMomentumMother() const
{
return fTrueMomentumMother;
}
//_____________________________________________
AliFemtoLorentzVector *AliFemtoModelAllHiddenInfo::GetEmissionPointMother() const
{
  return fEmissionPoint;
}
//_____________________________________________
  Int_t                AliFemtoModelAllHiddenInfo::GetPDGPidMother() const
{
  return fPDGPidMother;
}
//_____________________________________________
AliFemtoLorentzVector *AliFemtoModelAllHiddenInfo::GetTrueMomentumChild1() const
{
return fTrueMomentumChild1;
}
//_____________________________________________
AliFemtoLorentzVector *AliFemtoModelAllHiddenInfo::GetTrueMomentumChild2() const
{
return fTrueMomentumChild2;
}
//_____________________________________________
  Int_t                AliFemtoModelAllHiddenInfo::GetPDGPidChild1() const
{
  return fPDGPidChild1;
}
//_____________________________________________
  Int_t                AliFemtoModelAllHiddenInfo::GetPDGPidChild2() const
{
  return fPDGPidChild2;
}
//
//   SET
//
//_____________________________________________
//  Mother momentum
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumMother(AliFemtoLorentzVector *aMomMother)
{
  // Set momentum from vector
  if (fTrueMomentumMother) {
    fTrueMomentumMother->SetX(aMomMother->px());
    fTrueMomentumMother->SetY(aMomMother->py());
    fTrueMomentumMother->SetZ(aMomMother->pz());
    fTrueMomentumMother->SetT(aMomMother->e());
  }
  else {
    fTrueMomentumMother = new AliFemtoLorentzVector(*aMomMother);
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumMother(const AliFemtoLorentzVector& aMomMother)
{
  // Set momentum from vector and energy
  if (fTrueMomentumMother) {
    fTrueMomentumMother->SetX(aMomMother.px());
    fTrueMomentumMother->SetY(aMomMother.py());
    fTrueMomentumMother->SetZ(aMomMother.pz());
    fTrueMomentumMother->SetT(aMomMother.e());
  }
  else {
    fTrueMomentumMother = new AliFemtoLorentzVector();
    *fTrueMomentumMother = aMomMother;
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumMother(Double_t aMotherPx, Double_t aMotherPy, Double_t aMotherPz, Double_t aMotherE)
{
  // Set momentum from components and energy
  if (!fTrueMomentumMother) fTrueMomentumMother = new AliFemtoLorentzVector();
    fTrueMomentumMother->SetX(aMotherPx);
    fTrueMomentumMother->SetY(aMotherPy);
    fTrueMomentumMother->SetZ(aMotherPz);
    fTrueMomentumMother->SetT(aMotherE);
}
//_____________________________________________
//   Mother Emissin Point
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetEmissionPointMother(AliFemtoLorentzVector *aPos)
{
  // Set position from vector
  if (fEmissionPointMother) {
    fEmissionPointMother->SetX(aPos->px());
    fEmissionPointMother->SetY(aPos->py());
    fEmissionPointMother->SetZ(aPos->pz());
    fEmissionPointMother->SetT(aPos->e());
  }
  else {
    fEmissionPointMother = new AliFemtoLorentzVector(*aPos);
  }
}
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetEmissionPointMother(const AliFemtoLorentzVector& aPos)
{
  // Set position from vector
  if (fEmissionPointMother) {
    fEmissionPointMother->SetX(aPos.px());
    fEmissionPointMother->SetY(aPos.py());
    fEmissionPointMother->SetZ(aPos.pz());
    fEmissionPointMother->SetT(aPos.e());
  }
  else {
    fEmissionPointMother = new AliFemtoLorentzVector();
    *fEmissionPointMother = aPos;
  }
}
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetEmissionPointMother(Double_t aRx, Double_t aRy, Double_t aRz, Double_t aT)
{
  // Set position from components
  if (fEmissionPointMother) {
    fEmissionPointMother->SetX(aRx);
    fEmissionPointMother->SetY(aRy);
    fEmissionPointMother->SetZ(aRz);
    fEmissionPointMother->SetT(aT);
  }
  else {
    fEmissionPointMother = new AliFemtoLorentzVector(aRx, aRy, aRz, aT); 
  }
}
//_____________________________________________
//  Mother PID
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetPDGPidMother(Int_t aPidMother)
{
  fPDGPidMother = aPidMother;
}
//_____________________________________________
//  Child1 momentum
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild1(AliFemtoLorentzVector *aMomChild1)
{
  // Set momentum from vector
  if (fTrueMomentumChild1) {
    fTrueMomentumChild1->SetX(aMomChild1->px());
    fTrueMomentumChild1->SetY(aMomChild1->py());
    fTrueMomentumChild1->SetZ(aMomChild1->pz());
    fTrueMomentumChild1->SetT(aMomChild1->e());
  }
  else {
    fTrueMomentumChild1 = new AliFemtoLorentzVector(*aMomChild1);
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild1(const AliFemtoLorentzVector& aMomChild1)
{
  // Set momentum from vector and energy
  if (fTrueMomentumChild1) {
    fTrueMomentumChild1->SetX(aMomChild1.px());
    fTrueMomentumChild1->SetY(aMomChild1.py());
    fTrueMomentumChild1->SetZ(aMomChild1.pz());
    fTrueMomentumChild1->SetT(aMomChild1.e());
  }
  else {
    fTrueMomentumChild1 = new AliFemtoLorentzVector();
    *fTrueMomentumChild1 = aMomChild1;
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild1(Double_t aChild1Px, Double_t aChild1Py, Double_t aChild1Pz, Double_t aChild1E)
{
  // Set momentum from components and energy
  if (!fTrueMomentumChild1) fTrueMomentumChild1 = new AliFemtoLorentzVector();
    fTrueMomentumChild1->SetX(aChild1Px);
    fTrueMomentumChild1->SetY(aChild1Py);
    fTrueMomentumChild1->SetZ(aChild1Pz);
    fTrueMomentumChild1->SetT(aChild1E);
}
//_____________________________________________
//  Child2 momentum
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild2(AliFemtoLorentzVector *aMomChild2)
{
  // Set momentum from vector
  if (fTrueMomentumChild2) {
    fTrueMomentumChild2->SetX(aMomChild2->px());
    fTrueMomentumChild2->SetY(aMomChild2->py());
    fTrueMomentumChild2->SetZ(aMomChild2->pz());
    fTrueMomentumChild2->SetT(aMomChild2->e());
  }
  else {
    fTrueMomentumChild2 = new AliFemtoLorentzVector(*aMomChild2);
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild2(const AliFemtoLorentzVector& aMomChild2)
{
  // Set momentum from vector and energy
  if (fTrueMomentumChild2) {
    fTrueMomentumChild2->SetX(aMomChild2.px());
    fTrueMomentumChild2->SetY(aMomChild2.py());
    fTrueMomentumChild2->SetZ(aMomChild2.pz());
    fTrueMomentumChild2->SetT(aMomChild2.e());
  }
  else {
    fTrueMomentumChild2 = new AliFemtoLorentzVector();
    *fTrueMomentumChild2 = aMomChild2;
  }
}
//_____________________________________________
void AliFemtoModelAllHiddenInfo::SetTrueMomentumChild2(Double_t aChild2Px, Double_t aChild2Py, Double_t aChild2Pz, Double_t aChild2E)
{
  // Set momentum from components and energy
  if (!fTrueMomentumChild2) fTrueMomentumChild2 = new AliFemtoLorentzVector();
    fTrueMomentumChild2->SetX(aChild2Px);
    fTrueMomentumChild2->SetY(aChild2Py);
    fTrueMomentumChild2->SetZ(aChild2Pz);
    fTrueMomentumChild2->SetT(aChild2E);
}
//_____________________________________________
//  Child1 PID
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetPDGPidChild1(Int_t aPidChild1)
{
  fPDGPidMother = aPidChild1;
}
//_____________________________________________
//  Child2 PID
//_____________________________________________
void                   AliFemtoModelAllHiddenInfo::SetPDGPidChild2(Int_t aPidChild2)
{
  fPDGPidMother = aPidChild2;
}
//
//  RETURN COPY
//
//_____________________________________________
 AliFemtoModelHiddenInfo* AliFemtoModelAllHiddenInfo::GetParticleHiddenInfo() const
{
  // return copy of this hidden info
  AliFemtoModelAllHiddenInfo* tBuf = new AliFemtoModelAllHiddenInfo(*this);
  return tBuf;
}