Added Gautes changes from Bergen.

[u/mrichter/AliRoot.git] / RALICE / AliCollider.h

diff --git a/RALICE/AliCollider.h b/RALICE/AliCollider.h
index 968e4e1f9301d32123ddb21c5dfce35cecab2af9..58b4024f7dc50c80bc0db57cdce456e18e5d9e7b 100644 (file)
--- a/RALICE/AliCollider.h
+++ b/RALICE/AliCollider.h
@@ -3,12 +3,13 @@
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
-// $Id: AliCollider.h,v 1.5 2003/10/26 14:53:44 nick Exp $
+// $Id: AliCollider.h,v 1.8 2004/02/13 11:08:16 nick Exp $
 
 #include "TPythia6.h"
 #include "TString.h"
 #include "TFile.h"
-#include "TTree.h" 
+#include "TTree.h"
+#include "TArrayI.h" 
 
 #include "AliEvent.h"
 #include "AliRandom.h"
@@ -20,19 +21,23 @@ class AliCollider : public TPythia6
   virtual ~AliCollider();                               // Default destructor
   void SetOutputFile(TString name);                     // Initialise the ROOT output data file
   void SetVertexMode(Int_t mode);                       // Select mode for (sec.) vertex structure creation
-  Int_t GetVertexMode();                                // Provide vertex structure creation mode
+  Int_t GetVertexMode() const;                          // Provide vertex structure creation mode
   void SetResolution(Double_t res);                     // Set resolution (in cm) for resolving (sec.) vertices
-  Double_t GetResolution();                             // Provide (sec.) vertex resolving resolution (in cm)
+  Double_t GetResolution() const;                       // Provide (sec.) vertex resolving resolution (in cm)
   void SetRunNumber(Int_t run);                         // Set user defined run number
-  Int_t GetRunNumber();                                 // Provide the user defined run number
+  Int_t GetRunNumber() const;                           // Provide the user defined run number
   void SetPrintFreq(Int_t n);                           // Set print frequency for every 'n' events
-  Int_t GetPrintFreq();                                 // Provide the print frequency
+  Int_t GetPrintFreq() const;                           // Provide the print frequency
   void Init(char* frame,char* beam,char* target,Float_t win); // Standard Pythia initialisation
   void Init(char* frame,Int_t zp,Int_t ap,Int_t zt,Int_t at,Float_t win); // Nucl-Nucl initialisation
   void SetStable(Int_t id,Int_t mode=1);                // Declare a certain particle as stable or not
+  void SelectEvent(Int_t id);                           // Select only events containing specified particles
+  Int_t GetSelectionFlag() const;                       // Return the selection flag for this event
   void MakeEvent(Int_t npt=0,Int_t mlist=-1,Int_t medit=1);// Generate a single event with npt participant nucleons
   void EndRun();                                        // Properly close all buffers and output file
-  AliEvent* GetEvent();                                 // Provide pointer to the generated event structure
+  AliEvent* GetEvent(Int_t select=0) const;             // Provide pointer to the generated event structure
+  void SetSpectatorPmin(Float_t pmin);                  // Set minimal momentum for spectator track to be stored
+  Float_t GetSpectatorPmin() const;                     // Provide the minimal momentum for spectator tracks
 
  protected:
   Int_t fVertexmode;    // The vertex structure creation mode
@@ -53,12 +58,18 @@ class AliCollider : public TPythia6
   Float_t fFracnn;      // Fraction of n+n collisions
   AliRandom fRan;       // Random number generator
   AliEvent* fEvent;     // The produced event structure
+  Float_t fSpecpmin;    // The minimal momentum for spectator tracks to be stored
 
   TFile* fOutFile;      // The user defined output data file 
   TTree* fOutTree;      // The standard ROOT output tree
 
+  TArrayI* fSelections; // The particle KC codes for event selection
+  Int_t fSelect;        // Flag to indicate whether the total event is selected (1) or not (0)
+
+  Int_t IsSelected();   // Check whether (sub)event passed the selection criteria
   void GetFractions(Float_t zp,Float_t ap,Float_t zt,Float_t at); // Determine various N-N collision fractions
+  TString GetPyname(Int_t kf); // Provide the correctly truncated Pythia particle name for PDG code kf  
 
- ClassDef(AliCollider,5) // Pythia based universal physics event generator
+ ClassDef(AliCollider,8) // Pythia based universal physics event generator
 };
 #endif