RALICE/AliCollider.h

   1 #ifndef ALICOLLIDER_H
   2 #define ALICOLLIDER_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 // $Id: AliCollider.h,v 1.8 2004/02/13 11:08:16 nick Exp $
   7
   8 #include "TPythia6.h"
   9 #include "TString.h"
  10 #include "TFile.h"
  11 #include "TTree.h"
  12 #include "TArrayI.h"
  13
  14 #include "AliEvent.h"
  15 #include "AliRandom.h"
  16
  17 class AliCollider : public TPythia6
  18 {
  19  public:
  20   AliCollider();                                        // Default constructor
  21   virtual ~AliCollider();                               // Default destructor
  22   void SetOutputFile(TString name);                     // Initialise the ROOT output data file
  23   void SetVertexMode(Int_t mode);                       // Select mode for (sec.) vertex structure creation
  24   Int_t GetVertexMode() const;                          // Provide vertex structure creation mode
  25   void SetResolution(Double_t res);                     // Set resolution (in cm) for resolving (sec.) vertices
  26   Double_t GetResolution() const;                       // Provide (sec.) vertex resolving resolution (in cm)
  27   void SetRunNumber(Int_t run);                         // Set user defined run number
  28   Int_t GetRunNumber() const;                           // Provide the user defined run number
  29   void SetPrintFreq(Int_t n);                           // Set print frequency for every 'n' events
  30   Int_t GetPrintFreq() const;                           // Provide the print frequency
  31   void Init(char* frame,char* beam,char* target,Float_t win); // Standard Pythia initialisation
  32   void Init(char* frame,Int_t zp,Int_t ap,Int_t zt,Int_t at,Float_t win); // Nucl-Nucl initialisation
  33   void SetStable(Int_t id,Int_t mode=1);                // Declare a certain particle as stable or not
  34   void SelectEvent(Int_t id);                           // Select only events containing specified particles
  35   Int_t GetSelectionFlag() const;                       // Return the selection flag for this event
  36   void MakeEvent(Int_t npt=0,Int_t mlist=-1,Int_t medit=1);// Generate a single event with npt participant nucleons
  37   void EndRun();                                        // Properly close all buffers and output file
  38   AliEvent* GetEvent(Int_t select=0) const;             // Provide pointer to the generated event structure
  39   void SetSpectatorPmin(Float_t pmin);                  // Set minimal momentum for spectator track to be stored
  40   Float_t GetSpectatorPmin() const;                     // Provide the minimal momentum for spectator tracks
  41
  42  protected:
  43   Int_t fVertexmode;    // The vertex structure creation mode
  44   Double_t fResolution; // The resolution (in cm) for resolving (sec.) vertices
  45   Int_t fRunnum;        // The user defined run number
  46   Int_t fEventnum;      // The automatically updated event number
  47   Int_t fPrintfreq;     // The user selected print frequency
  48   char* fFrame;         // The Pythia frame indicator
  49   Float_t fWin;         // The Pythia energy indicator
  50   Int_t fNucl;          // Flag to denote nucleus-nucleus (1) or standard Pythia (0) running
  51   Int_t fZproj;         // Z of the projectile particle
  52   Int_t fAproj;         // A of the projectile particle
  53   Int_t fZtarg;         // Z of the target particle
  54   Int_t fAtarg;         // A of the target particle
  55   Float_t fFracpp;      // Fraction of p+p collisions
  56   Float_t fFracnp;      // Fraction of n+p collisions
  57   Float_t fFracpn;      // Fraction of p+n collisions
  58   Float_t fFracnn;      // Fraction of n+n collisions
  59   AliRandom fRan;       // Random number generator
  60   AliEvent* fEvent;     // The produced event structure
  61   Float_t fSpecpmin;    // The minimal momentum for spectator tracks to be stored
  62
  63   TFile* fOutFile;      // The user defined output data file
  64   TTree* fOutTree;      // The standard ROOT output tree
  65
  66   TArrayI* fSelections; // The particle KC codes for event selection
  67   Int_t fSelect;        // Flag to indicate whether the total event is selected (1) or not (0)
  68
  69   Int_t IsSelected();   // Check whether (sub)event passed the selection criteria
  70   void GetFractions(Float_t zp,Float_t ap,Float_t zt,Float_t at); // Determine various N-N collision fractions
  71   TString GetPyname(Int_t kf); // Provide the correctly truncated Pythia particle name for PDG code kf
  72
  73  ClassDef(AliCollider,8) // Pythia based universal physics event generator
  74 };
  75 #endif