#ifndef ALILego_H
#define ALILego_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//                                                                           //
//    Utility class to compute and draw Radiation Length Map                 //
//                                                                           //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include <TH2.h>

#include "AliGenerator.h"

class AliLegoGenerator;

class AliLego : public TNamed  {

private:
  AliLegoGenerator *fGener;     //Lego generator
   Float_t    fTotRadl;         //Total Radiation length
   Float_t    fTotAbso;         //Total absorption length
   Float_t    fTotGcm2;         //Total G/CM2 traversed
   TH2F      *fHistRadl;        //Radiation length map 
   TH2F      *fHistAbso;        //Interaction length map
   TH2F      *fHistGcm2;        //g/cm2 length map
   TH2F      *fHistReta;        //Radiation length map as a function of eta
   
public:
  AliLego();
  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  StepManager();
  virtual void  BeginEvent();
  virtual void  FinishEvent();
  virtual void  FinishRun();
  
  ClassDef(AliLego,1) //Utility class to compute and draw Radiation Length Map

};

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