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

#ifndef MAPPER_HPP
#define MAPPER_HPP
#include <iterator>

template<class KeyA, class A, class KeyB, class B, class C>
class SeqMR_Mapper : public Mapper<KeyA,A,KeyB,B,C>
{ 
public:
        SeqMR_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;

                typename vector< pair<KeyA,A> >::iterator itAs;
                for (itAs = as.begin() ; itAs != as.end() ; ++itAs)
                {
                        vector< pair<KeyB,B> > bss = map( *itAs );
                        typename vector< pair<KeyB,B> >::iterator itBss;
                        for (itBss = bss.begin() ; itBss != bss.end() ; ++itBss) bs.push_back( *itBss );
                }
                return bs;
        };
#else
        virtual multimap<KeyB,B> run( vector< pair<KeyA,A> > & as )
        {
                multimap<KeyB,B> bs;

                typename vector< pair<KeyA,A> >::iterator itAs;
                for (itAs = as.begin() ; itAs != as.end() ; ++itAs) 
                {
                        vector< pair<KeyB,B> > bss = map( *itAs );
                        typename vector< pair<KeyB,B> >::iterator itBss;
                        for (itBss = bss.begin() ; itBss != bss.end() ; ++itBss) bs.insert( *itBss );
                }
                return bs;
        };
#endif

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

class SeqMR_StrMapper : public SeqMR_Mapper<string,string,string,string,string> 
{
public:
        SeqMR_StrMapper(MapReduce<string,string,string,string,string> * _mapReduce)
                :SeqMR_Mapper<string,string,string,string,string>(_mapReduce) {};       
};

#endif /* MAPPER_HPP */