如何摆脱此代码中的重复?

时间:2017-01-11 21:44:55

标签: c++ reflection enums

作为概念验证的一部分,我正在尝试构建可用于构建GLSL脚本的代码。我做的第一件事就是创建对GLSL类型的引用,并创建一个系统,我可以根据枚举(这是GLSL数据类型的类型)查询在GLSL中表示它的字符串文字码。这是它的样子:

#define VEC_EXPAND(name) g ## name ## 2, g ## name ## 3, g ## name ## 4
#define MAT_EXPAND(name) VEC_EXPAND(name), VEC_EXPAND(name ## 2x), VEC_EXPAND(name ## 3x), VEC_EXPAND(name ## 4x)

enum class gtype : uint32_t {
    gvoid,
    gbool, gint, guint, gfloat, gdouble,
    VEC_EXPAND(bvec), VEC_EXPAND(ivec), VEC_EXPAND(uvec), VEC_EXPAND(vec), VEC_EXPAND(dvec),
    MAT_EXPAND(mat), MAT_EXPAND(dmat),
    gsampler2d
};

#undef VEC_EXPAND
#undef MAT_EXPAND

#define GLSL_EXPAND(name) #name
#define VEC_EXPAND(name) GLSL_EXPAND(name ## 2), GLSL_EXPAND(name ## 3), GLSL_EXPAND(name ## 4)
#define MAT_EXPAND(name) VEC_EXPAND(name), VEC_EXPAND(name ## 2x), VEC_EXPAND(name ## 3x), VEC_EXPAND(name ## 4x)

template<class T, class... Tail, class Elem = typename std::decay<T>::type>
constexpr std::array<Elem, 1 + sizeof...(Tail)> make_array(T&& head, Tail&&... values)
{
    return { std::forward<T>(head), std::forward<Tail>(values)... };
}


constexpr auto glsl_string_array = make_array(
    "void", "bool", "int", "uint", "float", "double",
    VEC_EXPAND(bvec), VEC_EXPAND(ivec), VEC_EXPAND(uvec), VEC_EXPAND(vec), VEC_EXPAND(dvec),
    MAT_EXPAND(mat), MAT_EXPAND(dmat),
    "sampler2d"
);
constexpr const char * to_string_literal(gtype type) {
    return glsl_string_array[uint32_t(type)];
}

std::string to_string(gtype type) {
    return to_string_literal(type);
}

#undef GLSL_EXPAND
#undef VEC_EXPAND
#undef MAT_EXPAND

到目前为止,我对代码的功能没有任何问题,但我将代码定义为枚举的代码重复,然后将它们再次写为字符串文字对我来说无疑是令人不安的。我将不得不添加更多类型(这些不是唯一的GLSL类型!)我想编写类似的代码来编写OpenCL内核代码(这将取决于类似的语义),所以我该如何减少这个代码向下,以便我只需要每种类型的一个声明?

此外,任何可以减少甚至消除我使用宏的建议都将受到赞赏。

1 个答案:

答案 0 :(得分:1)

我不知道这是否适合您的情况,但我可以提出类似的建议:

// things.inc
#ifndef GTYPE_VOID
#define GTYPE_VOID GTYPE_DECL(gvoid)
#endif
GTYPE_VOID

#undef GTYPE_VOID
#ifndef GTYPE_INT
#define GTYPE_INT GTYPE_DECL(gint)
#endif
GTYPE_INT
#undef GTYPE_INT

#ifndef GTYPE_FLOAT
#define GTYPE_FLOAT GTYPE_DECL(gfloat)
#endif
GTYPE_FLOAT
#undef GTYPE_FLOAT

然后,您可以按照自己喜欢的方式定义宏,并将其包含在您需要的位置:

// things.cpp
enum class gtypes {
#define GTYPE_DECL(Thing) Thing,
#include "things.inc"
#undef GTYPE_DECL
};
const char *as_string(gtype t) {
  switch (t) {
  #define GTYPE_DECL(Thing) case gtype::Thing: return #Thing;
  #include "things.inc"
  #undef GTYPE_DECL
  }
  // or unreachable
  return "(unknown)";
}

此外,您可以尝试使用constexpr地图。