如何“适应”仿函数与map / multimap一起使用?

时间:2009-01-27 01:38:22

标签: c++ stl map functor multimap

我很难将std :: for_each和其他算法与multimap一起使用,并且想知道是否有人可以帮我开发一个可以将相应参数传递给“通用”函数的仿函数。

我对map / multimap的具体问题是它们的迭代器计算为std :: pair而不是我需要使用的包含值(我的意思是mapped_type)。所以,我的问题是,有没有办法将正确的值传递给一个设计用于处理包含的map / multimap类型之一的函数?

以下是我的示例代码:

// the key type for the multimap
typedef enum { ... } Module;

// my class as the mapped type for the multimap
struct Patch
{
    void Apply(bool enable);
}

// I have some functors designed to work directly with Patch pointers
// because in other places I use set<Patch*> or other containers
struct ApplyOnCondtion: public unary_function<Patch*, void>
{
    void operator() (Patch* patch)
    {
        if(some_condition) patch->Apply(enable_or_not);
    }
}

// somewhere I have a collection of patches attributed to some module
multimap<Module, Patch*> patches;

// the problem is that using for_each with a map or a multimap results in
// a `pair<Module,Patch*>` as argument to the functor, not the Patch* as desired.
for_each(patches.begin(), patches.end(), ApplyOnCondition(...));

我认为也许bind1st或bind2nd与mem_fun结合可以解决这个问题,或者我能想到的另一种方式是创建一个存储原始仿函数并传递对的正确成员的新仿函数,但是我我没有得到任何好结果。任何有STL经验的人都可以给出一些建议吗?

EDIT 1

好的,我能得到的最好的,不使用boost或额外的临时容器如下:

#include <functional>
#include <utility>
using namespace std;


//////////////////////////////////////////////////////////////////////////
// any functor to be called must be derived from unary_function or
// have defined result_type and argument_type.
// template 'First' should be set to pair::first_type
template<typename First, typename Func>
class passSecond_t: public unary_function<
                        pair<First,typename Func::argument_type>,
                        typename Func::result_type>
{
    Func* func;

public:
    passSecond_t(Func &functor): func(&functor) {}

    result_type operator()(argument_type value)
    {
        return (*func)(value.second);
    }
};

// construction helper, unfortunately 'First' must be explicitly specified
template <typename First, typename Func>
passSecond_t<First, Func> passSecond(Func& functor)
{
    return passSecond_t<First, Func> (functor);
}


// the following is a sample
#include <map>
#include <algorithm>
#include <iostream>

struct SampleClass 
{
    void execute(char* text)
    {
        cout << "this: " << this << ", text: " << text << endl;
    }
};

struct SampleFunctor: public unary_function<SampleClass*,void>
{
    char* text;
    SampleFunctor(char* text_): text(text_) {}

    result_type operator() (argument_type argu)
    {
        argu->execute(text);
    }
};

void main()
{
    map<int,SampleClass*> mymap;
    SampleClass s1, s2;
    mymap[0] = &s1;
    mymap[1] = &s2;

    SampleFunctor myfunctor("my text");

    for_each(mymap.begin(), mymap.end(), passSecond<int>(myfunctor));
}

3 个答案:

答案 0 :(得分:1)

我在开始时有一个提升解决方案:

for_each(patches.begin(), patches.end(), 
         boost::bind(ApplyOnCondition(...),
                     boost::bind(&map_type::value_type::second, _1)));

这将获取该对的第二个成员,并将其推送到ApplyOnCondition的operator()。 map_type是地图的类型(当然multimap<Module, Patch*>)。

答案 1 :(得分:1)

您可以使用std::transform抓取所有值:

std::multimap<Module, Patch *> patches;

std::vector<Patch *> values(patches.size());
std::transform(patches.begin(), patches.end(), values.begin(),
    std::select2nd<std::multimap<Module, Patch *>::value_type>());

然后std::for_each在该向量上:

std::for_each(values.begin(), values.end(), ApplyOnCondition(...));

答案 2 :(得分:0)

这是一个解决方案,更像是如何在函数式语言中执行此操作:

  /**
   * Applies a function to the values in a map, producing a map with the same
   * keys but containing the values produced by the function.
   * In Haskell, the type of this function would be something like
   * mapply :: (a -> b) -> Map c a -> Map c b
   */
  template<typename Function, typename Key, typename InValue, typename OutValue>
  static std::map<Key, OutValue> mapply(Function function, const std::map<Key, InValue>& inMap) {
    typedef typename std::map<Key, InValue>::const_iterator InIterator;
    const InIterator end(inMap.end());
    std::map<Key, OutValue> outMap;
    for (InIterator i(inMap.begin()); i != end; ++i) {
      outMap.insert(std::make_pair(i->first, function(i->second)));
    }
    return outMap;
  }

此代码符合C ++ 98标准。在C ++ 11中,它可以简化。

另请注意,该函数按值返回地图。如果你没有迹象表明这是你性能的瓶颈(过早的优化是所有邪恶的根源),那就没关系了,但如果你有一个巨大的地图并且不相信你的编译器进行优化,你可能宁愿拥有“mapply”函数通过引用获取out参数的解决方案:填充键值对的映射。