如何在模板类函数中分配struct值？

Posted 2021-05-06

我有一堆结构需要添加到我的矢量并给它们一些默认值。

1. 结构是外部的，我无法改变它们。
2. 在分配这些值时，为了方便起见，我需要查看struct成员名称。
3. 为方便起见，我需要在同一个地方直观地显示所有默认值。
4. 编译器当前优化了myClass :: addStruct（）中未使用的if分支 - 这正是我想要的。

问：可以用更好/更简单的方式完成吗？

编辑我没有C ++ 17

//外部结构

typedef struct A {
    int member1;
    float member2;
    char *member3;
    // ...
} A;

typedef struct B {
    double member5;
    float member3;
    int *member4;
    // ...
} B;

typedef struct C {
    char* member5;
    char* member2;
    float *member3;
    // ...
} C;

...

template <class T>
void myClass::addStruct(std::vector<T> &vp)
{
    void *sp = nullptr;
    if(std::is_same<T, A>::value) {
        A s{};
        s.member1 = 2;
        s.member3 = "whatever";
        sp = &s;
    }
    else if(std::is_same<T, B>::value) {
        B s{};
        s.member4 = nullptr;
        s.member3 = 3.1f;
        sp = &s;
    }
    else if(std::is_same<T, C>::value) {
        C s{};
        s.member2 = "whenever";
        sp = &s;
    }
/*  else if() {
    }
    else if() {
    } ...
*/

    if(sp == nullptr) {
        // print error
        return;
    }

    vp.push_back(*(reinterpret_cast<T*>(sp)));
}

// usage

addStruct<A>(...);
addStruct<A>(...);
addStruct<B>(...);
addStruct<C>(...);

答案

创建工厂方法：

template <typename T> struct Tag {};

A MakeDefault(Tag<A>) {
    A a{};
    a.member1 = 2;
    a.member3 = "whatever";
    return a;
}

B MakeDefault(Tag<B>) {
    B b{};
    b.member4 = nullptr;
    b.member3 = 3.1f;
    return b;
}

C MakeDefault(Tag<C>) {
    C c{};
    c.member2 = "whenever";
    return c;
}

// ...

template <class T>
void myClass::addStruct(std::vector<T> &vp)
{
    vp.push_back(MakeDefault(tag<T>{}));
}

我使用标签调度而不是模板专业化，这在这里看起来更简单。

另一答案

1. 像这样的代码甚至不应该编译，除非你使用constexpr（又名“静态if”）。它只是“有效”，因为你破坏了类型系统（整个void *和reinterpret_cast舞蹈）。 Type aliasing rules适用。
2. 您正在获取已超出范围的对象的地址。这是Undefined Behavior，现在工作时可能会在更改代码或编译器或编译器版本时随时中断。
3. 如果你真的想要使用该代码，为什么不为不同的类型创建非模板重载？方式更容易。

另一答案

#include <vector>

typedef struct A {
    int member1;
    float member2;
    const char *member3;
    // ...
} A;

typedef struct B {
    double member5;
    float member3;
    int *member4;
    // ...
} B;

typedef struct C {
    char* member5;
    char* member2;
    float *member3;
    // ...
} C;

template <class T>
void addStruct(std::vector<T> &vp)
{
    // print error
}

template <>
void addStruct<A>(std::vector<A> &vp)
{
    A s{};
    s.member1 = 2;
    s.member3 = "whatever";
    vp.push_back(s);
}

template <>
void addStruct<B>(std::vector<B> &vp)
{
    B s{};
    s.member4 = nullptr;
    s.member3 = 3.1f;
    vp.push_back(s);
}

// ...

Trass3r之后的UPD：

#include <vector>

typedef struct A {
    int member1;
    float member2;
    const char *member3;
    // ...
} A;

typedef struct B {
    double member5;
    float member3;
    int *member4;
    // ...
} B;

typedef struct C {
    char* member5;
    char* member2;
    float *member3;
    // ...
} C;

void addStruct(std::vector<A> &vp)
{
    A s{};
    s.member1 = 2;
    s.member3 = "whatever";
    vp.push_back(s);
}

void addStruct(std::vector<B> &vp)
{
    B s{};
    s.member4 = nullptr;
    s.member3 = 3.1f;
    vp.push_back(s);
}

// ...

另一答案

问：可以用更好/更简单的方式完成吗？

是的：通过使用功能多态（具有相同名称和不同参数的函数），可以在没有模板的情况下完成。

typedef struct A {
    int member1;
    float member2;
    char *member3;
    // ...
} A;

typedef struct B {
    double member5;
    float member3;
    int *member4;
    // ...
} B;

...

void myClass::addStruct(std::vector<A>& v)
{
    v.push_back({0});
    v.back().member1 = 2;
    v.back().member3 = "whatever";
}

void myClass::addStruct(std::vector<B>& v)
{
    v.push_back({0});
    // v.back().member4 = nullptr; // this is unnecessary now
    v.back().member3 = 3.1f;
}

...

// usage

std::vector<A> va;
std::vector<B> vb;

myClass c;
c.addStruct(va);
c.addStruct(vb);