2016-10-04 12 views
3

Gibt es einen Vektor von Objekten, gibt es eine elegante Möglichkeit, seinen Member zu extrahieren? Ich benutze gerade eine for-Schleife, aber es wäre schön, wenn es eine Möglichkeit gibt, es zu tun. Beispiel,Extrahieren Sie Elemente aus einem Vektor des Objekts

class Object { 

int x; 
float y; 

} 

vector<Object> obj; 
// Fill up obj 

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

Antwort

2

Als std::vector (oder C++ im Allgemeinen) nicht covariant Aggregation unterstützt, gibt es keine syntaktisch hübsche Weise zu tun, was Sie wollen.

Wenn Sie wirklich wollen initialisierenall_x mit x Mitglieder obj Elemente, dann können Sie einen neuen Iterator-Klasse definieren, wie folgt aus:

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

Wenn Sie in Ordnung sind mit initialisieren eine leere vector und dann füllen sie, std::transform mit einem labmda ist eine klarere option (wie @andars vorgeschlagen).

Sie könnten auch zusätzliche Initialisierung vermeiden, indem vector::reserve() und back_inserter mit:

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

Beachten Sie auch, dass, während x ein privates Mitglied Object und hat keine Getter, wird es sehr schwierig sein, sie zu extrahieren.

1

Ich kann nicht wirklich einen guten Weg denken.

Eine Alternative wäre, std::transform mit einem Lambda zu verwenden.

#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

Einige Vorlage und Makro-Magie, und es funktioniert:

#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]); 
    } 
} 

Natürlich, denken Sie daran, dass es nicht sehr gut Makro MapVectorField zu nennen, wie es Funktion ist, oder using namespace std in Produktion schreiben.

+0

Sie Mitglied verwenden können anstelle des Makros Zeiger: '& Foo :: x '. – Jarod42

3

Mit range-v3, wäre es einfach sein

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

oder einfach nur die Aussicht verwenden:

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

Demo

Verwandte Themen