Roundrobin.hpp

#pragma once

#include <vector>    /* std::vector */

namespace graybat {

    namespace mapping {
    struct Roundrobin {
    
        template<typename T_Graph>
        std::vector<typename T_Graph::Vertex> operator()(const unsigned processID, const unsigned processCount, T_Graph &graph){
        typedef typename T_Graph::Vertex Vertex;


        // Distribute and announce vertices
        unsigned vertexCount   = graph.getVertices().size();
        unsigned maxVertex     = ceil((float)vertexCount / processCount);

        std::vector<Vertex> myVertices;
        if(processID <= vertexCount){
            for(unsigned i = 0; i < maxVertex; ++i){
            unsigned vertex_i = processID + (i * processCount);
            if(vertex_i >= vertexCount){
                break;
            }
            else {
                myVertices.push_back(graph.getVertex(vertex_i));
            }
    
            }
        }
        return myVertices;
    
        }

    };

    } /* mapping */
    
} /* graybat */