Line data    Source code

       1             : #ifndef ALIGENTHNSPARSE_H
       2             : #define ALIGENTHNSPARSE_H
       3             : /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
       4             :  * See cxx source for full Copyright notice                               */
       5             : // Particle generator according to 4 correlated variables : here
       6             : // z, ptot, r, theta. The input is a THnSparse object included in
       7             : // the root file (path and name to be set via the SetTHnSparse method).
       8             : // This class is similar to AliGenFunction.
       9             : 
      10             : #include "AliGenerator.h"
      11             : #include "THnSparse.h"
      12             : 
      13             : class AliGenTHnSparse : public AliGenerator
      14             : {
      15             : public:
      16             : 
      17             :   AliGenTHnSparse();
      18             :   AliGenTHnSparse(const AliGenTHnSparse& func);
      19             :   AliGenTHnSparse &operator=(const AliGenTHnSparse& func);
      20             :   virtual ~AliGenTHnSparse();
      21             :   virtual void Generate();
      22             :   virtual void Init();
      23           0 :   virtual void SetPart(Int_t part, Bool_t both) {fIpart=part; fBoth=both;}
      24             :   virtual void SetThnSparse(char *file_name, char *thn_name);
      25             :   
      26             : private:
      27             : 
      28             :   THnSparse *fHn; // Pointer to THnSparse object
      29             :   TFile *fFile;   // Pointer to input file
      30             :   Int_t fIpart;   // Particle type
      31             :   Bool_t fBoth;   // Particle and anti-particle type
      32             :       
      33           6 :   ClassDef(AliGenTHnSparse,2)
      34             : };
      35             : 
      36             : #endif