JPDB.hh

Go to the documentation of this file.

#ifndef __JAANET__JPDB__
#define __JAANET__JPDB__

#include <string>
#include <ostream>
#include <iomanip>
#include <vector>
#include <algorithm>

#include "JAAnet/JParticleTypes.hh"
#include "JPhysics/JConstants.hh"

#include "JLang/JPredicate.hh"
#include "JLang/JException.hh"
#include "JLang/JManip.hh"

#include "TDatabasePDG.h"
#include "TParticlePDG.h"
#include "TCollection.h"
#include "THashList.h"


/**
 * \author mdejong
 */

namespace JAANET {}
namespace JPP { using namespace JAANET; }

namespace JAANET {

  using JLANG::JException;

  /**
   * Auxiliary class to handle particle name, codes and mass.
   */
  struct JParticle {
    /**
     * Default constructor.
     */
    JParticle()
    {
      this->name   = "";
      this->pdg    = 0;
      this->geant  = 0;
      this->mass   = 0.0;
      this->charge = 0;
    }

    
    /**
     * Constructor.
     *
     * \param  name              name       of particle
     * \param  pdg               PDG   code of particle
     * \param  geant             GEANT code of particle
     * \param  mass              mass       of particle
     * \param  charge            charge     of particle
     */
    JParticle(const std::string& name,
              const int          pdg,
              const int          geant,
              const double       mass,
              const int          charge)
    {
      this->name   = name;
      this->pdg    = pdg;
      this->geant  = geant;
      this->mass   = mass;
      this->charge = charge;
    }


    /**
     * Print particle.
     *
     * \param  out               output stream
     * \param  particle          particle
     * \return                   output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JParticle& particle) 
    {
      using namespace std;
      using namespace JPP;
      
      return out << setw(32) << left  << particle.name  << "  "
                 << setw( 8) << right << particle.pdg   << "  "
                 << setw( 6) << right << particle.geant << "  "
                 << FIXED(16,10)      << particle.mass  << "  "
                 << setw( 5) << right << particle.charge;
    }

    
    std::string name;     //!< name       of particle
    int         pdg;      //!< PDG   code of particle
    int         geant;    //!< GEANT code of particle
    double      mass;     //!< mass       of particle [GeV]
    int         charge;   //!< charge     of particle [|e|/3]
  };


  /**
   * Auxiliary macro for particle creation.
   *
   * \param  PDG               PDG   code
   * \param  GEANT             GEANT code
   * \param  MASS              mass
   */
#define MAKE_PARTICLE(PDG, GEANT, MASS, CHARGE) JParticle(#PDG, PDG, GEANT, MASS, CHARGE)

  
  /**
   * Collection of particles.
   */
  struct JPDB :
    public std::vector<JParticle>
  {
    /**
     * Default constructor.
     */
    JPDB()
    {}


    /**
     * Get particle data book.
     * This particle data book contains all known GEANT particle codes.
     *
     * \return                   particle data book
     */
    static const JPDB& getInstance()
    {
      using namespace JPP;
      
      static JPDB pdb;

      if (pdb.empty()) {

        TDatabasePDG::Instance()->ReadPDGTable();
  
        TIter next(TDatabasePDG::Instance()->ParticleList());

        while (TParticlePDG* p = (TParticlePDG*) next()) {

          const JParticle particle(p->GetName(), p->PdgCode(), convertPDGtoGEANT(p->PdgCode()), p->Mass(), p->Charge());
          
          pdb.push_back(particle);
        }
      }
      
      return pdb;
    }

    
    /**
     * Check if PDB has particle corresponding to given GEANT code.
     *
     * \param  geant             GEANT code
     * \return                   true if available; else false
     */
    bool hasGEANT(const int geant) const
    {
      return (find_if(this->begin(), this->end(), JLANG::make_predicate(&JParticle::geant, geant)) != this->end());
    }

    
    /**
     * Get particle corresponding to given GEANT code.\n
     * This method throws an error if the particle cannot be found.
     *
     * \param  geant             GEANT code
     * \return                   particle
     */
    const JParticle& getGEANT(const int geant) const
    {
      const_iterator p = find_if(this->begin(), this->end(), JLANG::make_predicate(&JParticle::geant, geant));

      if (p != this->end())
        return *p;
      else
        THROW(JException, "Invalid GEANT particle code " << geant);
    }


    /**
     * Retrieve GEANT code for a given PDG code.
     *
     * \param  pdg               PDG code
     * \return                   GEANT code
     */
    static int convertPDGtoGEANT(const int pdg)
    {
      switch (pdg) {
        
      case TRACK_TYPE_NUE:
      case TRACK_TYPE_NUMU:
      case TRACK_TYPE_NUTAU:
      case TRACK_TYPE_ANTINUE:
      case TRACK_TYPE_ANTINUMU:
      case TRACK_TYPE_ANTINUTAU:
        return GEANT4_TYPE_NEUTRINO;
        
      case TRACK_TYPE_ANTISIGMA_MINUS:
        return GEANT4_TYPE_ANTISIGMA_MINUS;

      case TRACK_TYPE_NEUTRAL_ANTISIGMA:
        return GEANT4_TYPE_NEUTRAL_ANTISIGMA;

      case TRACK_TYPE_ANTISIGMA_PLUS:
        return GEANT4_TYPE_ANTISIGMA_PLUS;

      case TRACK_TYPE_NEUTRAL_ANTIXI:
        return GEANT4_TYPE_NEUTRAL_ANTIXI;

      default:
        return TDatabasePDG::Instance()->ConvertPdgToGeant3(pdg);
      }
    }
    
    
    /**
     * Check if PDB has particle corresponding to given PDG code.
     *
     * \param  pdg               PDG code
     * \return                   true if available; else false
     */
    bool hasPDG(const int pdg) const
    {
      return (find_if(this->begin(), this->end(), JLANG::make_predicate(&JParticle::pdg, pdg)) != this->end());
    }

    
    /**
     * Get particle corresponding to given PDG code.\n
     * This method throws an error if the particle cannot be found.
     *
     * \param  pdg               PDG code
     * \return                   particle
     */
    const JParticle& getPDG(const int pdg) const
    {
      const_iterator p = find_if(this->begin(), this->end(), JLANG::make_predicate(&JParticle::pdg, pdg));

      if (p != this->end())
        return *p;
      else
        THROW(JException, "Invalid PDG particle code " << pdg);
    }


    /**
     * Print particle data book.
     *
     * \param  out               output stream
     * \param  pdb               particle data book
     * \return                   output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JPDB& pdb) 
    {
      using namespace std;

      out << setw(32) << left  << "particle"       << "  "
          << setw( 8) << right << "PDG"            << "  "
          << setw( 6) << right << "GEANT"          << "  "
          << setw(16) << right << "Mass [GeV]"     << "  "
          << setw( 5) << right << "Charge [|e|/3]" << endl;     

      for (JPDB::const_iterator i = pdb.begin(); i != pdb.end(); ++i) {
        out << *i << endl;
      }

      return out;
    }
  };
}

#endif