PWG2/FEMTOSCOPY/AliFemto/AliFemtoModelHiddenInfo.cxx

   1 ////////////////////////////////////////////////////////////////////////////////
   2 ///                                                                          ///
   3 /// AliFemtoModelHiddenInfo - the hidden info for model calculations         ///
   4 /// Stores information needed for the weight generation - the true           ///
   5 /// simulated momenta, freeze-out coordinates from model and particle PID    ///
   6 ///                                                                          ///
   7 ////////////////////////////////////////////////////////////////////////////////
   8 #include "AliFemtoModelHiddenInfo.h"
   9
  10 //_____________________________________________
  11 AliFemtoModelHiddenInfo::AliFemtoModelHiddenInfo() :
  12   fTrueMomentum(0),
  13   fEmissionPoint(0),
  14   fPDGPid(0),
  15   fMass(0)
  16 {
  17   // Default constructor
  18 }
  19 //_____________________________________________
  20 AliFemtoModelHiddenInfo::AliFemtoModelHiddenInfo(const AliFemtoModelHiddenInfo &aInfo) :
  21   AliFemtoHiddenInfo(aInfo),
  22   fTrueMomentum(0),
  23   fEmissionPoint(0),
  24   fPDGPid(0),
  25   fMass(0)
  26 {
  27   // Copy constructor
  28   if (aInfo.GetTrueMomentum())
  29     SetTrueMomentum(aInfo.GetTrueMomentum());
  30   if (aInfo.GetEmissionPoint())
  31     SetEmissionPoint(aInfo.GetEmissionPoint());
  32   fPDGPid = aInfo.GetPDGPid();
  33   fMass = aInfo.GetMass();
  34 }
  35 //_____________________________________________
  36 AliFemtoModelHiddenInfo::~AliFemtoModelHiddenInfo()
  37 {
  38   // Destructor
  39   if (fTrueMomentum) delete fTrueMomentum;
  40   if (fEmissionPoint) delete fEmissionPoint;
  41 }
  42 //_____________________________________________
  43 AliFemtoModelHiddenInfo& AliFemtoModelHiddenInfo::operator=(const AliFemtoModelHiddenInfo& aInfo)
  44 {
  45   // assignment operator
  46   if (this == &aInfo)
  47     return *this;
  48
  49   if (aInfo.GetTrueMomentum())
  50     SetTrueMomentum(aInfo.GetTrueMomentum());
  51   else SetTrueMomentum(0);
  52   if (aInfo.GetEmissionPoint())
  53     SetEmissionPoint(aInfo.GetEmissionPoint());
  54   else SetEmissionPoint(0);
  55   fPDGPid = aInfo.GetPDGPid();
  56   fMass = aInfo.GetMass();
  57
  58   return *this;
  59 }
  60 //_____________________________________________
  61 AliFemtoThreeVector   *AliFemtoModelHiddenInfo::GetTrueMomentum() const
  62 {
  63   return fTrueMomentum;
  64 }
  65 //_____________________________________________
  66 AliFemtoLorentzVector *AliFemtoModelHiddenInfo::GetEmissionPoint() const
  67 {
  68   return fEmissionPoint;
  69 }
  70 //_____________________________________________
  71 Int_t                  AliFemtoModelHiddenInfo::GetPDGPid() const
  72 {
  73   return fPDGPid;
  74 }
  75 //_____________________________________________
  76 Double_t                  AliFemtoModelHiddenInfo::GetMass() const
  77 {
  78   return fMass;
  79 }
  80 //_____________________________________________
  81 void                   AliFemtoModelHiddenInfo::SetTrueMomentum(AliFemtoThreeVector *aMom)
  82 {
  83   // Set momentum from vector
  84   if (fTrueMomentum) {
  85     fTrueMomentum->SetX(aMom->x());
  86     fTrueMomentum->SetY(aMom->y());
  87     fTrueMomentum->SetZ(aMom->z());
  88   }
  89   else {
  90     fTrueMomentum = new AliFemtoThreeVector(*aMom);
  91   }
  92 }
  93 //_____________________________________________
  94 void                   AliFemtoModelHiddenInfo::SetTrueMomentum(const AliFemtoThreeVector& aMom)
  95 {
  96   // Set momentum from vector
  97   if (fTrueMomentum) {
  98     fTrueMomentum->SetX(aMom.x());
  99     fTrueMomentum->SetY(aMom.y());
 100     fTrueMomentum->SetZ(aMom.z());
 101   }
 102   else {
 103     fTrueMomentum = new AliFemtoThreeVector();
 104     *fTrueMomentum = aMom;
 105   }
 106 }
 107 //_____________________________________________
 108 void                   AliFemtoModelHiddenInfo::SetTrueMomentum(Double_t aPx, Double_t aPy, Double_t aPz)
 109 {
 110   // Set momentum from components
 111   if (!fTrueMomentum) fTrueMomentum = new AliFemtoThreeVector();
 112     fTrueMomentum->SetX(aPx);
 113     fTrueMomentum->SetY(aPy);
 114     fTrueMomentum->SetZ(aPz);
 115 }
 116 //_____________________________________________
 117 void                   AliFemtoModelHiddenInfo::SetEmissionPoint(AliFemtoLorentzVector *aPos)
 118 {
 119   // Set position from vector
 120   if (fEmissionPoint) {
 121     fEmissionPoint->SetX(aPos->px());
 122     fEmissionPoint->SetY(aPos->py());
 123     fEmissionPoint->SetZ(aPos->pz());
 124     fEmissionPoint->SetT(aPos->e());
 125   }
 126   else {
 127     fEmissionPoint = new AliFemtoLorentzVector(*aPos);
 128   }
 129 }
 130 //_____________________________________________
 131 void                   AliFemtoModelHiddenInfo::SetEmissionPoint(const AliFemtoLorentzVector& aPos)
 132 {
 133   // Set position from vector
 134   if (fEmissionPoint) {
 135     fEmissionPoint->SetX(aPos.px());
 136     fEmissionPoint->SetY(aPos.py());
 137     fEmissionPoint->SetZ(aPos.pz());
 138     fEmissionPoint->SetT(aPos.e());
 139   }
 140   else {
 141     fEmissionPoint = new AliFemtoLorentzVector();
 142     *fEmissionPoint = aPos;
 143   }
 144 }
 145 //_____________________________________________
 146 void                   AliFemtoModelHiddenInfo::SetPDGPid(Int_t aPid)
 147 {
 148   fPDGPid = aPid;
 149 }
 150 //_____________________________________________
 151 void                   AliFemtoModelHiddenInfo::SetMass(Double_t aMass)
 152 {
 153   fMass = aMass;
 154 }
 155 //_____________________________________________
 156 void                   AliFemtoModelHiddenInfo::SetEmissionPoint(Double_t aRx, Double_t aRy, Double_t aRz, Double_t aT)
 157 {
 158   // Set position from components
 159   if (fEmissionPoint) {
 160     fEmissionPoint->SetX(aRx);
 161     fEmissionPoint->SetY(aRy);
 162     fEmissionPoint->SetZ(aRz);
 163     fEmissionPoint->SetT(aT);
 164   }
 165   else {
 166     fEmissionPoint = new AliFemtoLorentzVector(aRx, aRy, aRz, aT);
 167   }
 168 }
 169 //_____________________________________________
 170  AliFemtoHiddenInfo* AliFemtoModelHiddenInfo::GetParticleHiddenInfo() const
 171 {
 172   // return copy of this hidden info
 173   AliFemtoModelHiddenInfo* tBuf = new AliFemtoModelHiddenInfo(*this);
 174   return tBuf;
 175 }