Overlaps corrected, new shape of sectors

[u/mrichter/AliRoot.git] / EVGEN / AliGenTHnSparse.h

diff --git a/EVGEN/AliGenTHnSparse.h b/EVGEN/AliGenTHnSparse.h
index e7e7c08a112d7bfd21995d57ac87c62d93b79acc..4782ac2ad4edaeb872e75aacb748acfd87eb4dd0 100644 (file)
--- a/EVGEN/AliGenTHnSparse.h
+++ b/EVGEN/AliGenTHnSparse.h
@@ -7,7 +7,6 @@
 // the root file (path and name to be set via the SetTHnSparse method).
 // This class is similar to AliGenFunction.
 
-#include "AliLog.h"
 #include "AliGenerator.h"
 #include "THnSparse.h"
 
@@ -21,7 +20,7 @@ public:
   virtual ~AliGenTHnSparse();
   virtual void Generate();
   virtual void Init();
-  virtual void SetPart(Int_t part) {fIpart=part;}
+  virtual void SetPart(Int_t part, Bool_t both) {fIpart=part; fBoth=both;}
   virtual void SetThnSparse(char *file_name, char *thn_name);
   
 private:
@@ -29,8 +28,9 @@ private:
   THnSparse *fHn; // Pointer to THnSparse object
   TFile *fFile;   // Pointer to input file
   Int_t fIpart;   // Particle type
-
-  ClassDef(AliGenTHnSparse,1)
+  Bool_t fBoth;   // Particle and anti-particle type
+      
+  ClassDef(AliGenTHnSparse,2)
 };
 
 #endif