New generator: TUHKMgen

[u/mrichter/AliRoot.git] / TUHKMgen / UHKM / DatabasePDG.h

diff --git a/TUHKMgen/UHKM/DatabasePDG.h b/TUHKMgen/UHKM/DatabasePDG.h
new file mode 100644 (file)
index 0000000..9486751
--- /dev/null
+++ b/TUHKMgen/UHKM/DatabasePDG.h
@@ -0,0 +1,92 @@
+/*
+  Copyright   : The FASTMC and SPHMC Collaboration
+  Author      : Ionut Cristian Arsene 
+  Affiliation : Oslo University, Norway & Institute for Space Sciences, Bucharest, Romania
+  e-mail      : i.c.arsene@fys.uio.no
+  Date        : 2007/05/30
+
+  This class is using the particle and decay lists provided by the 
+  THERMINATOR (Computer Physics Communications 174 669 (2006)) and
+  SHARE (Computer Physics Communications 167 229 (2005)) collaborations.
+*/
+
+#ifndef DATABASE_PDG
+#define DATABASE_PDG
+
+#include "Rtypes.h"
+#ifndef PARTICLE_PDG
+#include "ParticlePDG.h"
+#endif
+
+const Int_t kMaxParticles = 500;
+
+class DatabasePDG {
+ private:
+  Int_t fNParticles;                        // no. of particles in database
+  ParticlePDG *fParticles[kMaxParticles];   // array of particle pointers
+  Bool_t fStatus[kMaxParticles];            // status of each particle
+  Char_t fParticleFilename[256];            // particle list filename
+  Char_t fDecayFilename[256];               // decay channels filename
+  Bool_t fUseCharmParticles;                // flag for using (or not) charm particles
+  Double_t fMinimumWidth;                   // minimum allowed width for resonances
+  Double_t fMaximumWidth;                   // maximum allowed width for resonances
+  Double_t fMinimumMass;                    // minimum allowed mass for resonances
+  Double_t fMaximumMass;                    // maximum allowed mass for resonances
+
+  Bool_t LoadParticles();
+  Bool_t LoadDecays();
+  void SortParticles();                     // put the good status particles at the beggining of the list
+ public:
+  DatabasePDG();
+  ~DatabasePDG();
+
+  // Load the particle PDG information from the particle and decay files
+  Bool_t LoadData();                        
+  
+  // Set particle and decay filenames
+  void SetParticleFilename(Char_t *filename);
+  void SetDecayFilename(Char_t *filename);
+  // Set criteria for using particles. Those particle which do not match
+  // these criteria will be flagged with FALSE in the fStatus array.
+  void SetUseCharmParticles(Bool_t flag);
+  void SetMinimumWidth(Double_t value);
+  void SetMaximumWidth(Double_t value);
+  void SetMinimumMass(Double_t value);
+  void SetMaximumMass(Double_t value);
+  void SetWidthRange(Double_t min, Double_t max);
+  void SetMassRange(Double_t min, Double_t max);
+  
+  // Read a list of pdg codes from a specified file. The corresponding particles
+  // will be flagged as good particles. If the exclusive flag is TRUE than
+  // only this criteria will be used in selecting particles and, in consequence,
+  // all the other particles will be flagged as NOT good. If the exclusive flag
+  // is FALSE than we will take into account all the previous applied criterias
+  // and we will flag as good only particles in this list which match also the mass, width and
+  // charmness criteria.
+  // Note: In order for the exclusive=FALSE to be effective, this function must be called after
+  // calling all the width, mass and charmness criteria functions.
+  void UseThisListOfParticles(Char_t *filename, Bool_t exclusive = kTRUE);
+  
+  Char_t* GetParticleFilename() {return fParticleFilename;}
+  Char_t* GetDecayFilename() {return fDecayFilename;}
+  Int_t GetNParticles(Bool_t all = kFALSE);      // true - no. of all particles; false - no. of good status particles
+  ParticlePDG* GetPDGParticleByIndex(Int_t index);
+  Bool_t GetPDGParticleStatusByIndex(Int_t index);
+  ParticlePDG* GetPDGParticle(Int_t pdg);
+  Bool_t GetPDGParticleStatus(Int_t pdg);
+  ParticlePDG* GetPDGParticle(Char_t *name);
+  Bool_t GetPDGParticleStatus(Char_t *name);
+  Bool_t GetUseCharmParticles() {return fUseCharmParticles;};
+  Double_t GetMinimumWidth() {return fMinimumWidth;};
+  Double_t GetMaximumWidth() {return fMaximumWidth;};
+  Double_t GetMinimumMass() {return fMinimumMass;};
+  Double_t GetMaximumMass() {return fMaximumMass;};
+  void DumpData(Bool_t dumpAll = kFALSE); // print the PDG information in the console
+  Int_t CheckImpossibleDecays(Bool_t dump = kFALSE);   // print all impossible decays included in the database
+  Bool_t IsChannelAllowed(DecayChannel *channel, Double_t motherMass);
+  Int_t GetNAllowedChannels(ParticlePDG *particle, Double_t motherMass);
+  void SetStable(Int_t pdg, Bool_t value) {GetPDGParticle(pdg)->SetStable(value);}
+  Bool_t GetStableStatus(Int_t pdg) {return GetPDGParticle(pdg)->GetStableStatus();}
+};
+
+#endif