AliVMC and AliGeant3 classes introduced

[u/mrichter/AliRoot.git] / STEER / AliLego.h

diff --git a/STEER/AliLego.h b/STEER/AliLego.h
index 9bc6e6c9ae0383dfaf3187e044667960413c4c0e..b2409bd648f6375acda3b675919ec1c16300da86 100644 (file)
--- a/STEER/AliLego.h
+++ b/STEER/AliLego.h
@@ -15,22 +15,14 @@
 
 #include <TH2.h>
 
+#include "AliGenerator.h"
+
+class AliLegoGenerator;
+
 class AliLego : public TNamed  {
 
 private:
-   Float_t    fThetaMin;        //Minimum generation theta
-   Float_t    fThetaMax;        //Maximum generation theta
-   Float_t    fPhiMin;          //Minimum generation phi
-   Float_t    fPhiMax;          //Maximum generation phi
-   Float_t    fRadMin;          //Generation radius
-   Float_t    fRadMax;          //Maximum tracking radius
-   Float_t    fZMax;            //Maximum tracking Z
-   Int_t      fNtheta;          //Number of bins in Theta
-   Int_t      fNphi;            //Number of bins in Phi
-   Int_t      fThetaBin;        //Current theta bin
-   Int_t      fPhiBin;          //Current phi bin
-   Float_t    fCurTheta;        //Current theta of track
-   Float_t    fCurPhi;          //Current phi of track
+  AliLegoGenerator *fGener;     //Lego generator
    Float_t    fTotRadl;         //Total Radiation length
    Float_t    fTotAbso;         //Total absorption length
    Float_t    fTotGcm2;         //Total G/CM2 traversed
@@ -41,12 +33,10 @@ private:
    
 public:
   AliLego();
-  AliLego(const char *name, const char *title);
+  AliLego(const char *title, Int_t ntheta,Float_t themin,
+         Float_t themax, Int_t nphi, Float_t phimin,
+         Float_t phimax,Float_t rmin,Float_t rmax,Float_t zmax);
   virtual ~AliLego();
-  virtual void  GenerateKinematics();
-  virtual void  Init(Int_t ntheta,Float_t themin,Float_t themax, Int_t nphi, Float_t phimin,
-                     Float_t phimax,Float_t rmin,Float_t rmax,Float_t zmax);
-  Float_t       PropagateCylinder(Float_t *x, Float_t *v, Float_t r, Float_t z);
   virtual void  Run();
   virtual void  StepManager();
   
@@ -54,5 +44,36 @@ public:
 
 };
 
+class AliLegoGenerator : public AliGenerator
+{
+public:
+  AliLegoGenerator() {}
+  AliLegoGenerator(Int_t ntheta, Float_t themin, Float_t themax,
+                  Int_t nphi, Float_t phimin, Float_t phimax,
+                  Float_t rmin, Float_t rmax, Float_t zmax);
+  void    Generate();
+  Float_t CurTheta() const {return fCurTheta;}
+  Int_t   ThetaBin() const {return fThetaBin;}
+  Float_t CurPhi() const {return fCurPhi;}
+  Float_t ZMax() const {return fZMax;}
+  Float_t RadMax() const {return fRadMax;}
+  Int_t   PhiBin() const {return fPhiBin;}
+  Int_t   Nphi() const {return fNphi;}
+  Int_t   Ntheta() const {return fNtheta;}
+  Float_t       PropagateCylinder(Float_t *x, Float_t *v, Float_t r, Float_t z);
+private:
+   Float_t    fRadMin;          //Generation radius
+   Float_t    fRadMax;          //Maximum tracking radius
+   Float_t    fZMax;            //Maximum tracking Z
+   Int_t      fNtheta;          //Number of bins in Theta
+   Int_t      fNphi;            //Number of bins in Phi
+   Int_t      fThetaBin;        //Current theta bin
+   Int_t      fPhiBin;          //Current phi bin
+   Float_t    fCurTheta;        //Current theta of track
+   Float_t    fCurPhi;          //Current phi of track
+
+  ClassDef(AliLegoGenerator,1) //Lego generator
+};
+
 #endif