Lines getting the matched track moved to a method in AliCalorimeterUtils. Lines copie...

[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemto / AliFemtoModelGausLCMSFreezeOutGenerator.cxx

diff --git a/PWG2/FEMTOSCOPY/AliFemto/AliFemtoModelGausLCMSFreezeOutGenerator.cxx b/PWG2/FEMTOSCOPY/AliFemto/AliFemtoModelGausLCMSFreezeOutGenerator.cxx
index cf962be0ec652474d0e0d94d6660ecafbec732f3..2472a79e6cfc90ac63c0d11a7ee594024464baa5 100644 (file)
--- a/PWG2/FEMTOSCOPY/AliFemto/AliFemtoModelGausLCMSFreezeOutGenerator.cxx
+++ b/PWG2/FEMTOSCOPY/AliFemto/AliFemtoModelGausLCMSFreezeOutGenerator.cxx
@@ -13,6 +13,7 @@
 #include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
 #include "AliFemtoModelHiddenInfo.h"
 #include "AliFemtoLorentzVector.h"
+#include <TMath.h>
 
 //_______________________
 AliFemtoModelGausLCMSFreezeOutGenerator::AliFemtoModelGausLCMSFreezeOutGenerator() :
@@ -39,6 +40,18 @@ AliFemtoModelGausLCMSFreezeOutGenerator::~AliFemtoModelGausLCMSFreezeOutGenerato
   if (fRandom) delete fRandom;
 }
 //_______________________
+AliFemtoModelGausLCMSFreezeOutGenerator& AliFemtoModelGausLCMSFreezeOutGenerator::operator=(const AliFemtoModelGausLCMSFreezeOutGenerator &aModel)
+{
+  if (this != &aModel) {
+    fRandom = new TRandom2();
+    SetSizeOut(aModel.GetSizeOut());
+    SetSizeSide(aModel.GetSizeSide());
+    SetSizeLong(aModel.GetSizeLong());
+  }
+
+  return *this;
+}
+//_______________________
 void AliFemtoModelGausLCMSFreezeOutGenerator::GenerateFreezeOut(AliFemtoPair *aPair)
 {
   // Generate two particle emission points with respect
@@ -54,8 +67,8 @@ void AliFemtoModelGausLCMSFreezeOutGenerator::GenerateFreezeOut(AliFemtoPair *aP
   Double_t tPz = inf1->GetTrueMomentum()->z() + inf2->GetTrueMomentum()->z();
   Double_t tM1 = inf1->GetMass();
   Double_t tM2 = inf2->GetMass();
-  Double_t tE1 = sqrt(tM1*tM1 + inf1->GetTrueMomentum()->mag2());
-  Double_t tE2 = sqrt(tM2*tM2 + inf2->GetTrueMomentum()->mag2());
+  Double_t tE1 = sqrt(tM1*tM1 + inf1->GetTrueMomentum()->Mag2());
+  Double_t tE2 = sqrt(tM2*tM2 + inf2->GetTrueMomentum()->Mag2());
   Double_t tEs = tE1 + tE2;
 
   Double_t tPt = sqrt(tPx*tPx + tPy*tPy);
@@ -76,12 +89,14 @@ void AliFemtoModelGausLCMSFreezeOutGenerator::GenerateFreezeOut(AliFemtoPair *aP
   if (!(inf1->GetEmissionPoint())) {
     AliFemtoLorentzVector *tPos = new AliFemtoLorentzVector(0,0,0,0);
     inf1->SetEmissionPoint(tPos);
+    delete tPos;
   }
   else
     inf1->SetEmissionPoint(0,0,0,0);
   if (!(inf2->GetEmissionPoint())) {
     AliFemtoLorentzVector *tPos = new AliFemtoLorentzVector(tXout,tXside,tXlong,tXtime);
     inf2->SetEmissionPoint(tPos);
+    delete tPos;
   }
   else
     inf2->SetEmissionPoint(tXout, tXside, tXlong, tXtime);