c++ - 包装函数可以重构吗

Question

我有一组 C++ 函数，它们都接受来自同一组类型的类型(TypeA、TypeB 和 TypeC下面的示例)作为模板参数。为了简化这些函数对 python 的解释，我想为每个函数定义一个函数，该函数不将类型作为模板参数，而是作为字符串参数，如下所示:

template<typename dataType>
int function(int arg)
{
    ...
}

int function(int arg, string type)
{
    if (type == "type_A")
    {
        return function<TypeA>(arg);
    }
    else if (type == "type_B")
    {
        return function<TypeB>(arg);
    }
    else if (type == "type_C")
    {
        return function<TypeC>(arg);
    }
    else
    {
        std::cerr << "Invalid type!" << std::endl;
        exit(1);
    }
}

目前，我以这种方式包装所有函数，但这会导致大量代码重复，所以我想知道是否有更好的方法来做到这一点，也许使用预处理器指令？

Answer 1

减少 if/else 逻辑的一种方法是存储 std::function 对象的映射，并使用该映射进行正确的调用。

int function(int arg, std::string type)
{
   using FMap = std::map<std::string, std::function<int(int)>>;
   static const FMap fmap{{"type_A", [](int arg) { return function<TypeA>(arg); }},
                          {"type_B", [](int arg) { return function<TypeB>(arg); }},
                          {"type_C", [](int arg) { return function<TypeC>(arg); }}};

   auto iter = fmap.find(type);
   if ( iter != fmap.end() )
   {
      return iter->second(arg);
   }

   std::cerr << "Invalid type!" << std::endl;
   exit(1);
   return 0;
}

如果您愿意重命名函数模板，则可以简化构建函数映射的代码。

template <typename T>
int fun_2(int arg) { ... }

int function(int arg, std::string type)
{
   using FMap = std::map<std::string, std::function<int(int)>>;
   static const FMap fmap{{"type_A", fun_2<TypeA>},
                          {"type_B", fun_2<TypeB>},
                          {"type_C", fun_2<TypeC>}};

   auto iter = fmap.find(type);
   if ( iter != fmap.end() )
   {
      return iter->second(arg);
   }

   std::cerr << "Invalid type!" << std::endl;
   exit(1);
   return 0;
}