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