Clean-up
[u/mrichter/AliRoot.git] / EVGEN / AliGenTHnSparse.h
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   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   ClassDef(AliGenTHnSparse,2)
34 };
35
36 #endif