StarParticleData.h

#ifndef __StarParticleData_h__
#define __StarParticleData_h__

#include "TParticlePDG.h"
#include "TObjectSet.h"

#include <map>

class StarParticleData : public TObjectSet
{
 public:

  StarParticleData( const Char_t *name="PDG_Database",       TDataSet *parent = 0 );
  ~StarParticleData();

  TParticlePDG *GetParticle( const Char_t *name ) ;
  TParticlePDG *GetParticle( const Int_t   id   );
  TParticlePDG *GetParticleG3( const Int_t id );

  void          AddParticle( const Char_t *name,  TParticlePDG *particle ); 
  void          AddAlias   ( const Char_t *alias, const Char_t *realname );

  TParticlePDG *AddParticle( const Char_t *name, const Char_t *title, Double_t mass, Bool_t Stable, Double_t Width, Double_t Charge3, const char* ParticleClass, Int_t PdgCode, Int_t Anti, Int_t geantCode );

  TParticlePDG *AddParticleToG3( const char*   name, const double mass, const double lifetime, const double charge, const int tracktype, const int pdgcode, const int g3code, const double* bratio=0, const int* mode=0 );
  TParticlePDG *AddParticleToG3( TParticlePDG *part, const int g3code );

  TParticlePDG *SetTrackingCode( const int pdgid, const int g3id );


  static StarParticleData &instance();

  TParticlePDG *operator()( const Char_t *name );
  TParticlePDG *operator()( const Int_t  id    );

  const TObjArray &GetParticles() const { return mParticleList; }

 private:
 protected:

  // Singleton instance of this class
  static StarParticleData *sInstance; 

  TObjArray                           mParticleList;      // my particles
  std::map< TString, TParticlePDG* >  mParticleNameMap;   // particle aliases
  std::map< Int_t,   TParticlePDG* >  mParticleIdMap;     // map by PDG id
  std::map< Int_t,   TParticlePDG* >  mParticleG3IdMap;   // map by G3 id

  ClassDef( StarParticleData, 1 );

};

#endif