BeamabortGeo.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <map>
#include <string>
#include <TObject.h>
#include <framework/gearbox/GearDir.h>
#include <framework/logging/Logger.h>
#include <iostream>
 
namespace Belle2 {
  class BeamabortGeo:  public TObject {
 
  public:
 
    BeamabortGeo()
    {}
 
    double getParameter(const std::string& name) const
    {
      std::map<std::string, double>::const_iterator it = m_params.find(name);
      if (it != m_params.end()) return (*it).second;
      B2FATAL("Requested parameter from BEAMABORT database not found: " << name);
    }
 
    double getParameter(const std::string& name, double def) const
    {
      std::map<std::string, double>::const_iterator it = m_params.find(name);
      if (it != m_params.end()) return (*it).second;
      return def;
    }
 
    std::vector<double> getParArray(const std::string& name) const
    {
      std::map<std::string, std::vector<double>>::const_iterator it = m_par_arrays.find(name);
      if (it != m_par_arrays.end()) return (*it).second;
      B2FATAL("Requested parameter from BEAMABORT database not found: " << name);
    }
 
    std::vector<double> getParArray(const std::string& name, std::vector<double> def) const
    {
      std::map<std::string, std::vector<double>>::const_iterator it = m_par_arrays.find(name);
      if (it != m_par_arrays.end()) return (*it).second;
      return def;
    }
 
    void addParameter(const std::string& name, double val)
    {
      if (m_params.insert(std::pair<std::string, double>(name, val)).second) return;
      else {
        m_params.find(name)->second = val;
      }
    }
 
    void addArray(const std::string& name, std::vector<double> arr)
    {
      if (m_par_arrays.insert(std::pair<std::string, std::vector<double>>(name, arr)).second) return;
      else {
        m_par_arrays.find(name)->second = arr;
      }
    }
 
    const std::map<std::string, double>&  getParameters() const { return m_params;}
 
    const std::map<std::string, std::vector<double>>&  getParArrays() const { return m_par_arrays;}
 
    void print() const
    {
      for (std::pair<std::string, double> element : m_params) {
        std::cout << element.first << " " << element.second << std::endl;
      }
    }
 
    void initialize(const GearDir& content);
 
  private:
 
    ClassDef(BeamabortGeo, 1); 
  protected:
 
    std::map<std::string, double>  m_params;   
    std::map<std::string, std::vector<double>>  m_par_arrays;   
  };
 
} // end namespace Belle2