Line data    Source code

       1             : #ifndef ALIGENFUNCTION_H
       2             : #define ALIGENFUNCTION_H
       3             : /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
       4             :  * See cxx source for full Copyright notice                               */
       5             : // Generator for particles according generic functions
       6             : //  
       7             : //  TF1 *   fFMomentum;           // momentum distribution function inGeV
       8             : //  TF1 *   fFPhi;                // phi distribution function in rad
       9             : //  TF1 *   fFTheta;              // theta distribution function in rad
      10             : //  TF3 *   fFPosition;           // position distribution function in cm
      11             : //  TF1 *   fFPdg;                // pdg distribution function  
      12             : //  We assume that the moment, postion and PDG code of particles are independent  
      13             : //  Only tracks/particle crossing the reference radius at given z range
      14             : //
      15             : // Origin: marian.ivanov@cern.ch
      16             : 
      17             : 
      18             : #include "AliGenerator.h"
      19             : class TF3;
      20             : 
      21             : class AliGenFunction : public AliGenerator
      22             : {
      23             : public:
      24             : 
      25             :   AliGenFunction();
      26             :   AliGenFunction(const AliGenFunction& func);
      27             :   AliGenFunction &operator=(const AliGenFunction& func);
      28           0 :   virtual ~AliGenFunction() {}
      29             :   virtual void Generate();
      30             :   virtual void Init();
      31             :   void SetFunctions(TF1 * momentum, TF1 *fphi, TF1 *ftheta, TF3 * position, TF1* pdg);
      32             :   void SetCylinder(Double_t refR, Double_t zmin, Double_t zmax);
      33           0 :   void SetBkG(Float_t b) { fBkG=b; return; }
      34             : private:
      35             :   Bool_t IntersectCylinder(Float_t r,Float_t zmin, Float_t zmax,Int_t pdg,
      36             :                            Float_t o[3],Float_t p[3]) const;  
      37             : 
      38             :   Float_t fBkG;                 // field in kGauss
      39             :   TF1 *   fFMomentum;           // momentum distribution function
      40             :   TF1 *   fFPhi;                // phi distribution function
      41             :   TF1 *   fFTheta;              // theta distribution function
      42             :   TF3 *   fFPosition;           // position distribution function 
      43             :   TF1 *   fFPdg;                // pdg distribution function  
      44             :   //
      45             :   Double_t fRefRadius;          // reference radius to be crossed
      46             :   Double_t fZmin;               // minimal z at reference radius
      47             :   Double_t fZmax;               // maximal z at reference radius
      48             :   Int_t    fMaxTrial;           // maximal number of attempts
      49             :   //
      50           6 :   ClassDef(AliGenFunction,1) // parametrized cosmics generator
      51             : };
      52             : 
      53             : #endif