Improved common vertex handling in cocktails.

[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONlib.h

diff --git a/EVGEN/AliGenMUONlib.h b/EVGEN/AliGenMUONlib.h
index e75406381374847937837fc8d1478f53760a7d3b..1fb44b00919f4770b41d270496ff8f6ed37df2d1 100644 (file)
--- a/EVGEN/AliGenMUONlib.h
+++ b/EVGEN/AliGenMUONlib.h
@@ -1,43 +1,90 @@
-#ifndef _AliGenMUONlib_H
-#define _AliGenMUONlib_H
-#include <TROOT.h>
+#ifndef ALIGENMUONLIB_H
+#define ALIGENMUONLIB_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */
+
+// Library class for particle pt and y distributions used for 
+// muon spectrometer simulations.
+// To be used with AliGenParam.
+//
+// andreas.morsch@cern.ch
+
+#include "AliGenLib.h"
+
 class AliGenMUONlib :
-public TObject
+  public AliGenLib
 {
  public:
-// pions and kaons
-    static Double_t PtPion(Double_t *px, Double_t *);
+    enum constants{kPhi, kOmega, kEta, kJpsi, kUpsilon,
+                  kCharm, kBeauty, kPion, kKaon};
+    
+    
+// pions
+    static Double_t PtPion(Double_t *px, Double_t *dummy);
     static Double_t PtScal(Double_t pt, Int_t np);
-    static Double_t EtaKaon( Double_t *py, Double_t *);
+    static Double_t YPion( Double_t *py, Double_t *dummy);
+    static Int_t    IpPion(TRandom *ran);
+// kaons
+    static Double_t PtKaon(Double_t *px, Double_t *dummy);
+    static Double_t YKaon( Double_t *py, Double_t *dummy);
+    static Int_t    IpKaon(TRandom *ran);
 // Phi
-    static Double_t PtPhi( Double_t *px, Double_t *);
-    static Double_t YPhi( Double_t *px, Double_t *);
-    static Int_t    IpPhi();
+    static Double_t PtPhi( Double_t *px, Double_t *dummy);
+    static Double_t YPhi( Double_t *px, Double_t *dummy);
+    static Int_t    IpPhi(TRandom *ran);
+// Omega
+    static Double_t PtOmega( Double_t *px, Double_t *dummy);
+    static Double_t YOmega( Double_t *px, Double_t *dummy);
+    static Int_t    IpOmega(TRandom *ran);
+// Eta
+    static Double_t PtEta( Double_t *px, Double_t *dummy);
+    static Double_t YEta( Double_t *px, Double_t *dummy);
+    static Int_t    IpEta(TRandom *ran);
 // J/Psi     
-    static Double_t PtJpsi( Double_t *px, Double_t *);
-    static Double_t YJpsi(Double_t *py, Double_t *);
-    static Int_t    IpJpsi();
+    static Double_t PtJpsi( Double_t *px, Double_t *dummy);
+    static Double_t YJpsi(Double_t *py, Double_t *dummy);
+    static Double_t PtJpsiPbPb( Double_t *px, Double_t *dummy);
+    static Double_t YJpsiPbPb(Double_t *py, Double_t *dummy);
+    static Double_t PtJpsiPP( Double_t *px, Double_t *dummy);
+    static Double_t YJpsiPP(Double_t *py, Double_t *dummy);
+    static Int_t    IpJpsi(TRandom *ran);
 // Upsilon    
-    static Double_t PtUpsilon( Double_t *px, Double_t * );
-    static Double_t YUpsilon(Double_t *py, Double_t *);
-    static Int_t    IpUpsilon();
+    static Double_t PtUpsilon( Double_t *px, Double_t *dummy );
+    static Double_t YUpsilon(Double_t *py, Double_t *dummy);
+    static Double_t PtUpsilonPbPb( Double_t *px, Double_t *dummy );
+    static Double_t YUpsilonPbPb(Double_t *py, Double_t *dummy);
+    static Double_t PtUpsilonPP( Double_t *px, Double_t *dummy );
+    static Double_t YUpsilonPP(Double_t *py, Double_t *dummy);
+    static Int_t    IpUpsilon(TRandom *ran);
 //
 // Charm    
-    static Double_t PtCharm( Double_t *px, Double_t * );
-    static Double_t YCharm(Double_t *py, Double_t *);
-    static Int_t    IpCharm();
+    static Double_t PtCharm( Double_t *px, Double_t *dummy );
+    static Double_t YCharm(Double_t *py, Double_t *dummy);
+    static Int_t    IpCharm(TRandom *ran);
 //
 // Beauty
-    static Double_t PtBeauty( Double_t *px, Double_t * );
-    static Double_t YBeauty(Double_t *py, Double_t *);
-    static Int_t    IpBeauty();
+    static Double_t PtBeauty( Double_t *px, Double_t *dummy );
+    static Double_t YBeauty(Double_t *py, Double_t *dummy);
+    static Int_t    IpBeauty(TRandom *ran);
 //
-    typedef Double_t (*GenFunc)  (Double_t *, Double_t *);
-    typedef Int_t    (*GenFuncIp)();    
-    static GenFunc   GetPt(Int_t ipart);
-    static GenFunc   GetY(Int_t ipart);
-    static GenFuncIp GetIp(Int_t ipart);    
-    ClassDef(AliGenMUONlib,1)
+    GenFunc   GetPt(Int_t param, const char* tname=0) const;
+    GenFunc   GetY (Int_t param, const char* tname=0) const;
+    GenFuncIp GetIp(Int_t param, const char* tname=0) const;
+ private:
+    
+    static Float_t Interpolate(Float_t x, Float_t* y, Float_t x0, 
+                       Float_t dx,
+                       Int_t n, Int_t no);
+    
+    ClassDef(AliGenMUONlib,0) // Library providing y and pT parameterisations
 };
 #endif
 
+
+
+
+
+
+