THMEnums.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include <algorithm>
#include "MooseEnum.h"

namespace THM
{

template <typename T>
T stringToEnum(const std::string & s);

template <typename T>
T stringToEnum(const std::string & s, const std::map<std::string, T> & enum_map);

template <typename T>
MooseEnum getMooseEnum(const std::string & default_key, const std::map<std::string, T> & enum_map);
}

template <typename T>
T
THM::stringToEnum(const std::string & s, const std::map<std::string, T> & enum_map)
{
  std::string upper(s);
  std::transform(upper.begin(), upper.end(), upper.begin(), ::toupper);

  if (!enum_map.count(upper))
    return static_cast<T>(-100);
  else
    return enum_map.at(upper);
}

template <typename T>
MooseEnum
THM::getMooseEnum(const std::string & default_key, const std::map<std::string, T> & enum_map)
{
  std::string keys_string;
  for (typename std::map<std::string, T>::const_iterator it = enum_map.begin();
       it != enum_map.end();
       it++)
    if (it == enum_map.begin())
      keys_string += it->first;
    else
      keys_string += " " + it->first;

  return MooseEnum(keys_string, default_key, true);
}