partmap.cpp

Go to the documentation of this file.

//   GAMBIT: Global and Modular BSM Inference Tool
//   *********************************************


#include <sstream>

#include "gambit/Models/partmap.hpp"
#include "gambit/Utils/standalone_error_handlers.hpp"
#include "gambit/Utils/stream_overloads.hpp"

namespace Gambit
{

  namespace Models
  {

    partmap& ParticleDB()
    {
      static partmap local;
      return local;
    }

    extern void define_particles(partmap*);

    partmap::partmap() { define_particles(this); }

    void partmap::add(str long_name, std::pair<int, int> pdgpr)
    {
      if (has_particle(long_name))
      {
        model_error().raise(LOCAL_INFO,"Particle "+long_name+" is multiply defined.");
      }
      long_name_to_pdg_pair[long_name] = pdgpr;
      pdg_pair_to_long_name[pdgpr] = long_name;
    }

    void partmap::add_SM(str long_name, std::pair<int, int> pdgpr)
    {
      add(long_name, pdgpr);
      SM.push_back(pdgpr);
    }

    void partmap::add_generic(str long_name, std::pair<int, int> pdgpr)
    {
      add(long_name, pdgpr);
      generic.push_back(pdgpr);
    }

    void partmap::add_with_short_pair(str long_name, std::pair<int, int> pdgpr, std::pair<str, int> shortpr)
    {
      add(long_name, pdgpr);
      short_name_pair_to_pdg_pair[shortpr] = pdgpr;
      short_name_pair_to_long_name[shortpr] = long_name;
      pdg_pair_to_short_name_pair[pdgpr] = shortpr;
      long_name_to_short_name_pair[long_name] = shortpr;
    }

    void partmap::add_SM_with_short_pair(str long_name, std::pair<int, int> pdgpr, std::pair<str, int> shortpr)
    {
      add_with_short_pair(long_name, pdgpr, shortpr);
      SM.push_back(pdgpr);
    }

    std::pair<int, int> partmap::pdg_pair(str long_name) const
    {
      if (not has_particle(long_name))
      {
        model_error().raise(LOCAL_INFO,"Particle long name "+long_name+" is not in the particle database.");
      }
      return long_name_to_pdg_pair.at(long_name);
    }

    std::pair<int, int> partmap::pdg_pair(std::pair<str,int> shortpr) const
    {
      return pdg_pair(shortpr.first,shortpr.second);
    }

    std::pair<int, int> partmap::pdg_pair(str short_name, int i) const
    {
      std::pair<str, int> shortpr(short_name, i);
      if (not has_particle(shortpr))
      {
        std::ostringstream ss;
        ss << "Short name " << short_name << " and index " << i << " are not in the particle database.";
        model_error().raise(LOCAL_INFO,ss.str());
      }
      return short_name_pair_to_pdg_pair.at(shortpr);
    }

    str partmap::long_name(str short_name, int i) const
    {
      std::pair<str, int> shortpr(short_name, i);
      if (not has_particle(std::pair<str, int>(short_name, i)))
      {
        std::ostringstream ss;
        ss << "Short name " << short_name << " and index " << i << " are not in the particle database.";
        model_error().raise(LOCAL_INFO,ss.str());
      }
      return short_name_pair_to_long_name.at(shortpr);
    }

    str partmap::long_name(std::pair<int, int> pdgpr) const
    {
      if (not has_particle(pdgpr))
      {
        std::ostringstream ss;
        ss << "Particle with PDG code " << pdgpr.first << " and context integer " << pdgpr.second << " is not in the particle database.";
        model_error().raise(LOCAL_INFO,ss.str());
      }
      return pdg_pair_to_long_name.at(pdgpr);
    }

    str partmap::long_name(int pdg_code, int context) const
    {
       return long_name(std::make_pair(pdg_code,context));
    }

    std::pair<str, int> partmap::short_name_pair(str long_name) const
    {
      if (not has_particle(long_name))
      {
        model_error().raise(LOCAL_INFO,"Particle "+long_name+" is not in the particle database.");
      }
      if (not has_short_name(long_name))
      {
        model_error().raise(LOCAL_INFO,"Particle "+long_name+" does not have a short name.");
      }
      return long_name_to_short_name_pair.at(long_name);
    }

    std::pair<str, int> partmap::short_name_pair(std::pair<int, int> pdgpr) const
    {
      if (not has_particle(pdgpr))
      {
        std::ostringstream ss;
        ss << "Particle with PDG code " << pdgpr.first << " and context integer " << pdgpr.second << " is not in the particle database.";
        model_error().raise(LOCAL_INFO,ss.str());
      }
      if (not has_short_name(pdgpr))
      {
        std::ostringstream ss;
        ss << "Particle with PDG code " << pdgpr.first << " and context integer " << pdgpr.second << " does not have a short name.";
        model_error().raise(LOCAL_INFO,ss.str());
      }
      return pdg_pair_to_short_name_pair.at(pdgpr);
    }

    std::pair<str, int> partmap::short_name_pair(int pdg_code, int context) const
    {
      return short_name_pair(std::make_pair(pdg_code,context));
    }

    const std::vector<std::pair<int, int> >& partmap::get_SM_particles() const
    {
      return SM;
    }

    const std::vector<std::pair<int, int> >& partmap::get_generic_particles() const
    {
      return generic;
    }

    bool partmap::has_particle(str long_name) const
    {
      return (long_name_to_pdg_pair.find(long_name) != long_name_to_pdg_pair.end());
    }

    bool partmap::has_particle(str short_name, int i) const
    {
      return has_particle(std::make_pair(short_name,i));
    }
    bool partmap::has_particle(std::pair<str, int> shortpr) const
    {
      return (short_name_pair_to_pdg_pair.find(shortpr) != short_name_pair_to_pdg_pair.end());
    }

    bool partmap::has_particle(std::pair<int, int> pdgpr) const
    {
      return (pdg_pair_to_long_name.find(pdgpr) != pdg_pair_to_long_name.end());
    }

    bool partmap::has_short_name(str long_name) const
    {
      return (long_name_to_short_name_pair.find(long_name) != long_name_to_short_name_pair.end());
    }

    bool partmap::has_short_name(std::pair<int, int> pdgpr) const
    {
      return (pdg_pair_to_short_name_pair.find(pdgpr) != pdg_pair_to_short_name_pair.end());
    }

    str partmap::get_antiparticle(str lname) const
    {
      return long_name(get_antiparticle(pdg_pair(lname)));
    }

    std::pair<str, int> partmap::get_antiparticle(std::pair<str, int> shortpr) const
    {
      return short_name_pair(get_antiparticle(pdg_pair(shortpr)));
    }
    std::pair<str, int> partmap::get_antiparticle(str name, int index) const
    {
      return get_antiparticle(std::make_pair(name,index));
    }

    std::pair<int, int> partmap::get_antiparticle(std::pair<int, int> pdgpr) const
    {
      if (has_antiparticle(pdgpr))
      {
        pdgpr.first = -pdgpr.first;
      }
      return pdgpr;
    }
    std::pair<int, int> partmap::get_antiparticle(int pdgcode, int context) const
    {
      return get_antiparticle(std::make_pair(pdgcode,context));
    }

    bool partmap::has_antiparticle(str long_name) const
    {
      return has_antiparticle(pdg_pair(long_name));
    }

    bool partmap::has_antiparticle(std::pair<str, int> shortpr) const
    {
      return has_antiparticle(pdg_pair(shortpr));
    }
    bool partmap::has_antiparticle(str name, int index) const
    {
      return has_antiparticle(std::make_pair(name,index));
    }

    bool partmap::has_antiparticle(std::pair<int, int> pdgpr) const
    {
      pdgpr.first = -pdgpr.first;
      return has_particle(pdgpr);
    }
    bool partmap::has_antiparticle(int pdgcode, int context) const
    {
      return has_antiparticle(std::make_pair(pdgcode,context));
    }

    void partmap::check_contents() const
    {
       // Check that long and short names retrieve same information (when short name exists)
       typedef std::map<str, std::pair<int, int> >::const_iterator                 it_long_name_to_pdg_pair;
       typedef std::map<std::pair<int, int>, str>::const_iterator                  it_pdg_pair_to_long_name;
       typedef std::map<std::pair<str, int>, str>::const_iterator                  it_short_name_pair_to_long_name;
       //typedef std::map<std::pair<str, int>, std::pair<int, int> >::const_iterator it_short_name_pair_to_pdg_pair;
       //typedef std::map<std::pair<int, int>, std::pair<str, int> >::const_iterator it_pdg_pair_to_short_name_pair;
       //typedef std::map<str, std::pair<str, int> >::const_iterator                 it_long_name_to_short_name_pair;

       cout << "PDB: long name as key" << endl;
       for(it_long_name_to_pdg_pair it = long_name_to_pdg_pair.begin(); it != long_name_to_pdg_pair.end(); it++) {
           cout   << "  long_name_to_pdg_pair       [" << it->first << "] => " << it->second << endl;
           if(has_short_name(it->first))
           { cout << "  long_name_to_short_name_pair[" << it->first << "] => " << long_name_to_short_name_pair.at(it->first) << endl; }
           else
           { cout << "  long_name_to_short_name_pair[" << it->first << "] => " << "Has no short name!" << endl; }
       }
       cout << endl << "PDB: pdg_pair as key" << endl;
       for(it_pdg_pair_to_long_name it = pdg_pair_to_long_name.begin(); it != pdg_pair_to_long_name.end(); it++) {
           cout   << "  pdg_pair_to_long_name [" << it->first << "] => " << it->second << endl;
           if(has_short_name(it->second))
           { cout << "  pdg_pair_to_short_name[" << it->first << "] => " << pdg_pair_to_short_name_pair.at(it->first) << endl; }
           else
           { cout << "  pdg_pair_to_short_name[" << it->first << "] => " << "Has no short name!" << endl; }
       }
       cout << endl << "PDB: short name pair as key" << endl;
       for(it_short_name_pair_to_long_name it = short_name_pair_to_long_name.begin(); it != short_name_pair_to_long_name.end(); it++) {
           cout << "  short_name_pair_to_long_name[" << it->first << "] => " << it->second << endl;
           cout << "  short_name_pair_to_pdg_pair [" << it->first << "] => " << short_name_pair_to_pdg_pair.at(it->first) << endl;
       }
    }


  }

}