#ifndef ALIGENDEUTERON_H
#define ALIGENDEUTERON_H
#include "AliGenerator.h"
class AliGenDeuteron: public AliGenerator
{
public:
AliGenDeuteron(Int_t sign=1, Double_t pmax=0.2, Double_t rmax=2.1, Int_t cluster=0 );
virtual ~AliGenDeuteron();
virtual void Init();
virtual void Generate();
Int_t GetSign() const { return fSign;}
Double_t GetCoalescenceMomentum() const { return fPmax; }
Double_t GetCoalescenceDistance() const { return fRmax; }
Double_t GetSpinProbability() const { return fSpinProb; }
Double_t GetCoalescenceProbability(const TParticle* nucleon1, const TParticle* nucleon2) const;
Int_t GetClusterType() const { return fClusterType; }
Int_t GetFreezeOutModel() const { return fModel; }
Double_t GetFreezeOutTime() const { return fTimeLength; }
void SetSign(Int_t sign) { fSign = sign > 0 ? 1 : -1;}
void SetCoalescenceMomentum(Double_t p) { fPmax = p; }
void SetCoalescenceDistance(Double_t r) { fRmax = r; }
void SetSpinProbability(Double_t s) { fSpinProb = s; }
void SetClusterType(Int_t cluster) { fClusterType = cluster; }
void SetFreezeOutModel(Int_t model, Double_t timeLength=2.5) { fModel = model; fTimeLength=timeLength;}
public:
enum { kFirstPartner=0, kLowestMomentum, kLowestDistance, kBoth };
enum { kNone=0, kThermal, kExpansion };
enum { kCluster=77 };
private:
AliGenDeuteron(const AliGenDeuteron &other);
AliGenDeuteron& operator=(const AliGenDeuteron &other);
void FixProductionVertex(class TParticle* i);
void FirstPartner(const TList* protons, TList* neutrons);
void WeightMatrix(const TList* protons, const TList* neutrons);
void PushDeuteron(TParticle* parent1, TParticle* parent2);
Double_t GetS(Double_t p1x, Double_t p1y, Double_t p1z, Double_t m1, Double_t p2x, Double_t p2y, Double_t p2z, Double_t m2) const;
Double_t GetPcm(Double_t p1x, Double_t p1y, Double_t p1z, Double_t m1, Double_t p2x, Double_t p2y, Double_t p2z, Double_t m2) const;
Double_t GetPcm(const TVector3& p1, Double_t m1, const TVector3& p2, Double_t m2) const;
private:
Int_t fSign;
Double_t fPmax;
Double_t fRmax;
Double_t fSpinProb;
Int_t fClusterType;
Int_t fModel;
Double_t fTimeLength;
Double_t fB;
Double_t fR;
Double_t fPsiR;
AliStack* fCurStack;
ClassDef(AliGenDeuteron,2)
};
#endif // ALIGENDEUTERON_H