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

#ifndef MAPREDUCE_HPP
#define MAPREDUCE_HPP

#include <omp.h>

template<class KeyA, class A, class KeyB, class B, class C>
class MapMP_MapReduce : public MapReduce<KeyA,A,KeyB,B,C>
{
public:
        MapMP_MapReduce():numThreads(1){};

        virtual vector< pair<KeyB,C> > run( Mapper<KeyA,A,KeyB,B,C> & mapper, Reducer<KeyA,A,KeyB,B,C> & reducer, vector< pair<KeyA,A> > & as)
        {       
                //cout << "Input  :\t" << as << endl;
                omp_set_num_threads(getNumThreads());
#ifndef MRI_USE_MULTIMAP
                vector< pair<KeyB,B> > bs = mapper.run( as );
#else
                multimap<KeyB,B> bs = mapper.run( as );
#endif
                //cout << "Mapped :\t" << bs << endl;
                omp_set_num_threads(getNumThreads());
                vector< pair<KeyB,C> > cs = reducer.run( bs );
                //cout << "Reduced:\t" << cs << endl;
                return cs;
        };

        void setNumThreads(int n) { numThreads = n; };
        int  getNumThreads()  { return numThreads; };
private:
        int numThreads;
};

class MapMP_StrMapReduce : public MapMP_MapReduce<string,string,string,string,string> {};

#endif /* MAPREDUCE_HPP */