OptFrame/Parallel/MapReduce++/MapMP/mapper.hpp

#ifndef MAPPER_HPP
#define MAPPER_HPP
#include <iterator>

template<class KeyA, class A, class KeyB, class B, class C>
class MapMP_Mapper : public Mapper<KeyA,A,KeyB,B,C>
{ 
public:
        MapMP_Mapper(MapReduce<KeyA,A,KeyB,B,C> * _mapReduce):mapReduce(_mapReduce){};
#ifndef MRI_USE_MULTIMAP
        virtual vector< pair<KeyB,B> > run( vector< pair<KeyA,A> > & as )
        {
                vector< pair<KeyB,B> > bs;

                #pragma omp parallel for
                for (int i = 0 ; i < as.size() ; i++) 
                {
                        vector< pair<KeyB,B> > bss = map( as[i] );
                        #pragma omp critical
                        for (int j = 0 ; j < bss.size() ; j++) bs.push_back( bss[j] );
                }

                return bs;
        };
#else
        virtual multimap<KeyB,B> run( vector< pair<KeyA,A> > & as )
        {
                multimap<KeyB,B> bs;

                #pragma omp parallel for
                for (int i = 0 ; i < as.size() ; i++)
                {
                        vector< pair<KeyB,B> > bss = map( as[i] );
                        #pragma omp critical
                        for (int j = 0 ; j < bss.size() ; j++) bs.insert( bss[j] );
                }

                return bs;
        };
#endif

        virtual vector< pair<KeyB,B> > map( pair<KeyA,A> ) = 0 ;
protected:
        MapReduce<KeyA,A,KeyB,B,C> * mapReduce;
};

class MapMP_StrMapper : public MapMP_Mapper<string,string,string,string,string> 
{
public:
        MapMP_StrMapper(MapReduce<string,string,string,string,string> * _mapReduce)
                :MapMP_Mapper<string,string,string,string,string>(_mapReduce) {};       
};

#endif /* MAPPER_HPP */