BeamBGTypes.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/dataobjects/BackgroundMetaData.h>
#include <string>
#include <map>
 
namespace Belle2 {
  namespace background {
 
    class BeamBGTypes {
 
    public:
      BeamBGTypes()
      {
        m_tags["Coulomb_LER"] = BackgroundMetaData::bg_Coulomb_LER;
        m_tags["Coulomb_HER"] = BackgroundMetaData::bg_Coulomb_HER;
        m_tags["RBB_LER"] = BackgroundMetaData::bg_RBB_LER;
        m_tags["RBB_HER"] = BackgroundMetaData::bg_RBB_HER;
        m_tags["Touschek_LER"] = BackgroundMetaData::bg_Touschek_LER;
        m_tags["Touschek_HER"] = BackgroundMetaData::bg_Touschek_HER;
        m_tags["twoPhoton"] = BackgroundMetaData::bg_twoPhoton;
        m_tags["RBB_gamma"] = BackgroundMetaData::bg_RBB_gamma;
        m_tags["RBB_LER_far"] = BackgroundMetaData::bg_RBB_LER_far;
        m_tags["RBB_HER_far"] = BackgroundMetaData::bg_RBB_HER_far;
        m_tags["Touschek_LER_far"] = BackgroundMetaData::bg_Touschek_LER_far;
        m_tags["Touschek_HER_far"] = BackgroundMetaData::bg_Touschek_HER_far;
        m_tags["SynchRad_LER"] = BackgroundMetaData::bg_SynchRad_LER;
        m_tags["SynchRad_HER"] = BackgroundMetaData::bg_SynchRad_HER;
        m_tags["BHWide_LER"] = BackgroundMetaData::bg_BHWide_LER;
        m_tags["BHWide_HER"] = BackgroundMetaData::bg_BHWide_HER;
        m_tags["RBB"] = BackgroundMetaData::bg_RBB;
        m_tags["BHWide"] = BackgroundMetaData::bg_BHWide;
        m_tags["BHWideLargeAngle"] = BackgroundMetaData::bg_BHWideLargeAngle;
        m_tags["InjectionLER"] = BackgroundMetaData::bg_InjectionLER;
        m_tags["InjectionHER"] = BackgroundMetaData::bg_InjectionHER;
        m_tags["Brems_LER"] = BackgroundMetaData::bg_Brems_LER;
        m_tags["Brems_HER"] = BackgroundMetaData::bg_Brems_HER;
        m_tags["other"] = BackgroundMetaData::bg_other;
      }
 
      BackgroundMetaData::BG_TAG getTag(const std::string& bgType)
      {
        return m_tags[bgType];
      }
 
      std::string getType(BackgroundMetaData::BG_TAG bgTag) const
      {
        if (bgTag == 0) return "";
        for (auto tag : m_tags) {
          if (bgTag == tag.second) return tag.first;
        }
        return "";
      }
 
      std::string getType(unsigned int bgTag) const
      {
        if (bgTag == 0) return "";
        for (auto tag : m_tags) {
          if (bgTag == tag.second) return tag.first;
        }
        return "";
      }
 
      std::string getBGTypes() const
      {
        std::string types;
        for (auto tag : m_tags) {
          if (tag.first != "other" && tag.second != 0) types += tag.first + ", ";
        }
        types += "other";
        return types;
      }
 
    private:
 
      std::map<std::string, BackgroundMetaData::BG_TAG> m_tags; 
    };
 
  } // background
} // Belle2