2016-10-04 184 views
3

给定一个对象矢量,是否有一种优雅的方法来提取其成员?我目前只是使用for循环,但如果有办法做到这一点会很好。例如,从对象矢量中提取元素

class Object { 

int x; 
float y; 

} 

vector<Object> obj; 
// Fill up obj 

vector<int> all_x = obj.x // Won't work obviously 

回答

2

由于std::vector(或C++一般)不支持协变聚集,没有语法上相当的方式做你想做的。

如果你真的想初始化all_xobj元素x成员,那么你可以定义一个新的迭代器类,像:

class getx_iter : public vector<Object>::iterator 
{ 
public: 
    getx_iter(const vector<Object>::iterator &iter) : vector<Object>::iterator(iter) {} 
    int operator*() { return (*this)->x; } 
}; 

Working code example

如果您能够接受初始化一个空的vector然后填充它,std::transform与labmda是一个更清晰的选项(如@andars建议)。

您也可避免额外的初始化通过使用vector::reserve()back_inserter

xs.reserve(foos.size()); 
std::transform(foos.begin(), foos.end(), back_inserter(xs), [](Foo f){return f.x;}); 

还要注意的是,虽然xObject私有成员并没有干将,这将是十分困难的提取它。

1

我想不出一个很好的方法。

一种替代方案是使用std::transform与拉姆达。

#include <vector> 
#include <algorithm> 


class Foo { 
public: 
    Foo(int x_): x(x_) {} 
    int x; 
}; 

int main() { 
    std::vector<Foo> foos; 
    for (int i = 0; i<10; i++) { 
     foos.push_back(Foo(i)); 
    } 

    std::vector<int> xs; 
    xs.resize(foos.size()); 
    std::transform(foos.begin(), foos.end(), xs.begin(), [](Foo f){return f.x;}); 
} 
0

一些模板和宏观魔法,它的工作原理:

#include <vector> 
#include <algorithm> 
using namespace std; 


class Foo { 
public: 
    Foo(int x_): x(x_) {} 
    int x; 
}; 

#define GETFIELD(type, field) [](const type & obj){return obj.field;} 

template<typename T,typename U, typename TMapper> 
void MapVector(vector<T>& src, vector<U>& dst, TMapper mapper) { 
    for (const auto& it: src) { 
     dst.push_back(mapper(it)); 
    } 
} 

#define MapVectorField(src, dst, field) MapVector(src, dst, GETFIELD(decltype(src)::value_type, field)) 

int main() { 
    vector<Foo> vFoo; 
    for (int i = 0; i < 10; i++) { 
     vFoo.push_back(Foo(i)); 
    } 

    vector<int> vX; 
    MapVector(vFoo, vX, GETFIELD(Foo, x)); 
    MapVectorField(vFoo, vX, x); 

    for (int i = 0; i < vX.size(); i++) { 
     printf("%d\n", vX[i]); 
    } 
} 

当然,记住,这不是很好的名字宏观MapVectorField,因为它是功能,还是写在生产using namespace std

+0

您可以使用成员指针,而不是你的宏:'&富:: X '。 – Jarod42

3

随着range-v3,它仅仅是

std::vector<int> xs = objs | ranges::view::transform(&Object::x); 

或者仅使用视图:

auto xs = objs | ranges::view::transform(&Object::x); 

Demo