introducing SDD, SSD layer misal (Andrea Dainese)
[u/mrichter/AliRoot.git] / EVGEN / AliGenSlowNucleons.h
index bec9e8c..399ba17 100644 (file)
 #include "AliGenerator.h"
 class AliSlowNucleonModel;
 class TH2F;
-
+class TH1F;
+class TF1;
 
 class AliGenSlowNucleons : public AliGenerator
 {
 public:
     AliGenSlowNucleons();
     AliGenSlowNucleons(Int_t npart);
-    AliGenSlowNucleons(const AliGenSlowNucleons &sn);
     virtual ~AliGenSlowNucleons();
     virtual void Init();
     virtual void FinishRun();
@@ -43,13 +43,14 @@ public:
     virtual void   SetDebug(Int_t flag = 0) {fDebug = flag;}
     virtual void   SetNumbersOfSlowNucleons(Int_t ngp, Int_t ngn, Int_t nbp, Int_t nbn)
        {fNgp = ngp; fNgn = ngn; fNbp = nbp; fNbn = nbn;}
+    //
+    // Added by Chiara to take into account angular distribution 4 gray tracks
+    virtual void   SetThetaDist(Int_t flag=0) {fThetaDistribution = flag;}
     
  protected:
-    void     GenerateSlow(Int_t charge, Double_t T, Double_t beta, Float_t* q);
+    void     GenerateSlow(Int_t charge, Double_t T, Double_t beta, Float_t* q, Float_t &theta);
     Double_t Maxwell(Double_t m, Double_t p, Double_t t);
     void     Lorentz(Double_t m, Double_t beta, Float_t* q);
-    void Copy(TObject&) const;
-    AliGenSlowNucleons & operator=(const AliGenSlowNucleons & rhs);
  protected:
     Float_t  fCMS;             // Center of mass energy
     Float_t  fMomentum;        // Target nucleus momentum
@@ -70,10 +71,19 @@ public:
     Int_t    fDebug;           // Debug flag
     TH2F*    fDebugHist1;      // Histogram for debugging
     TH2F*    fDebugHist2;      // Histogram for debugging
+    // Added by Chiara to take into account angular distribution 4 gray tracks
+    Int_t    fThetaDistribution;// 0 -> flat dist., 1 -> fwd. peaked distribution
+    TH1F*    fCosThetaGrayHist; // Histogram for debugging
+    TF1*     fCosTheta;         // Function for non-uniform cos(theta) distribution
     
     //
     AliSlowNucleonModel* fSlowNucleonModel; // The slow nucleon model
-    ClassDef(AliGenSlowNucleons,1) // Slow Nucleon Generator
+
+ private:
+    AliGenSlowNucleons(const AliGenSlowNucleons &sn);
+    AliGenSlowNucleons & operator=(const AliGenSlowNucleons & rhs);
+
+    ClassDef(AliGenSlowNucleons,2) // Slow Nucleon Generator
 };
 #endif