2010-01-28 24 views
13

Sınıfı göz önünde bulundurun:Bir std :: vector sort işlevine bir parametre iletebilir miyim?

MyClass { 
    int varA; 
    int varB; 
}; 

MyClass nesnelerine bir işaretçi vektörüne sahibim:

std::vector<MyClass*> Vec; 

Aynı sıralama işlevini kullanarak varA veya varB'ye göre vektörü sıralamak istiyorum, örneğin:

bool SortFunction(const MyClass* obj1, const MyClass* obj2, const short type) { 
    if(type == VARA_ID) 
     return obj1->varA < obj2->varA; 
    else if(type == VARB_ID) 
     return obj1->varB < obj2->varB; 
} 

AFAICT bu mümkün değil. Dış kütüphaneleri kullanmadan en şık yolu ne olurdu?

+1

İşaretçi nesnelerin vektörünü neden kullanıyorsunuz? –

cevap

21
class sorter { 
     short type_; 
public: 
     sorter(short type) : type_(type) {} 
     bool operator()(MyClass const* o1, MyClass const* o2) const { 
      return SortFunction(o1, o2, type_); 
     } 
}; 

std::sort(Vec.begin(), Vec.end(), sorter(MY_TYPE)); 
+1

+1 Çok güzel, beni cevapıma dövdün. Ben şahsen 'std :: binary_function ' türetici türetmek istiyorum, ama bu isteğe bağlı. –

+0

Evet, iç içe yazılan yazım hataları yalnızca std :: bind1st veya std :: bind2nd biçiminde gereklidir, dolayısıyla onu atladım^_ ^. – OwnWaterloo

+0

Mükemmel. Teşekkürler. –

2

type değeri her bir karşılaştırma ile değişiyor mu? Göründüğü gibi olamaz. Bu durumda, bir curried işlevi kullanın - bkz. boost::bind.

std::sort(v.begin(), v.end(), boost::bind(SortFunction, _1, _2, type)); 
+0

Anlamıyor musunuz? Bağlama (SortFunction, _1, _2, type) '? –

+0

Evet, çok tembel ... – dirkgently

2

bir daha genel bir çözüm de kullanılması işaretçiler-to-üyelere yapmak olabilir: type bir şablon parametresi yapmak Neredeyse geldin

#include <vector> 
#include <algorithm> 
#include <functional> 

struct MyClass { 
    int varA; 
    int varB; 
}; 

template <class Object, class VarType> 
class CompareMemberT: public std::binary_function<bool, const Object*, const Object*> 
{ 
    VarType Object::*p; 
public: 
    CompareMemberT(VarType Object::*p): p(p) {} 
    bool operator()(const Object* a, const Object* b) const 
    { 
     return a->*p < b->*p; 
    } 
}; 

//helper to deduce template arguments 
template <class Object, class VarType> 
CompareMemberT<Object, VarType> CompareMember(VarType Object::*p) 
{ 
    return CompareMemberT<Object, VarType>(p); 
} 

int main() 
{ 
    std::vector<MyClass*> vec; 
    std::sort(vec.begin(), vec.end(), CompareMember(&MyClass::varA)); 
    std::sort(vec.begin(), vec.end(), CompareMember(&MyClass::varB)); 
} 
6

ve imza Tamam :

template<int type> 
bool SortFunction(const MyClass* obj1, const MyClass* obj2) { 
    if(type == VARA_ID) 
     return obj1->varA < obj2->varA; 
    else // if(type == VARB_ID) -- A sort function must have a default. 
     return obj1->varB < obj2->varB; 
} 

std::sort(Vec.begin(), Vec.end(), &SortFunction<VARA_ID>); 

iyileştirici (type == VARA_ID) bir derleme zamanı sabiti olduğunu nokta olacaktır. Sağ Eğer sıralama ihtiyaç kodunuzda herhangi bir özel sıralama işlevi olmadan Boost.Lambda ve kullanma

+3

+1 fikri için, 'template 'yerine' template ' demeniz gerektiğini düşünüyorum. –

+0

Oops, evet. Sabit. – MSalters

3

:

Kısaca

// sort with VARA_ID 
sort(Vec.begin(), Vec.end(), bind(&MyClass::varA, _1)<bind(&MyClass::varA, _2)); 

// sort with VARB_ID 
sort(Vec.begin(), Vec.end(), bind(&MyClass::varB, _1)<bind(&MyClass::varB, _2)); 

için tam bir örnek

#include <iostream> 
#include <vector> 
#include <boost/lambda/lambda.hpp> 
#include <boost/lambda/bind.hpp> 
#include <algorithm> 

using namespace boost::lambda; 

class MyClass { 
public: 
    int varA; 
    int varB; 
}; 

enum SorterType { 
    VARA_ID, 
    VARB_ID 
}; 

int main() 
{ 

    std::vector<MyClass*> Vec; 

    Vec.push_back(new MyClass()); 
    Vec.push_back(new MyClass()); 
    Vec.push_back(new MyClass()); 

    Vec[0]->varA = 1; Vec[0]->varB = 8; 
    Vec[1]->varA = 2; Vec[1]->varB = 7; 
    Vec[2]->varA = 3; Vec[2]->varB = 6; 

    // sort with VARA_ID 
    std::sort(Vec.begin(), Vec.end(), 
       bind(&MyClass::varA, _1) < bind(&MyClass::varA, _2)); 

    // VARB_ID 
    std::sort(Vec.begin(), Vec.end(), 
       bind(&MyClass::varB, _1) < bind(&MyClass::varB, _2)); 

    return 0; 
} 
İlgili konular