Multithreading.h

/*
    Hyperspace Explorer - visualizing higher-dimensional geometry
    Copyright (C) 2012  Lene Preuss <lene.preuss@gmail.com>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License along
    with this program; if not, write to the Free Software Foundation, Inc.,
    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

*/

#ifndef PARTITIONEDMULTITHREADEDMAP_H
#define PARTITIONEDMULTITHREADEDMAP_H

#include "MultithreadingPrivate.h"

#include <QtConcurrentMap>


namespace Multithreading {

    template <class RandomAccessContainer, typename Function> 
    RandomAccessContainer mapped(const RandomAccessContainer &container, Function function) {
        return QtConcurrent::blockingMapped<RandomAccessContainer>(container, function);
    }
    

    template <class RandomAccessContainer, typename Function> 
    void partitionedMap(
        const RandomAccessContainer &container, RandomAccessContainer &output, Function function, 
        unsigned tasks_per_processor = 1
    ) {
        auto tasks = Private::Kernel<RandomAccessContainer, Function>::generateKernels(container, output, function, tasks_per_processor);

        Private::Kernel<RandomAccessContainer, Function>::startTasks(tasks);
        
        Private::Kernel<RandomAccessContainer, Function>::waitForTasks(tasks);
    }


    template <class RandomAccessContainer, typename Function> 
    void partitionedMap(
        RandomAccessContainer &container, Function function, unsigned tasks_per_processor = 1
    ) {
        partitionedMap(container, container, function, tasks_per_processor);
    }


    template <class RandomAccessContainer, typename Function> 
    RandomAccessContainer partitionedMapped(
        const RandomAccessContainer &container, Function function, unsigned tasks_per_processor = 1
    ) {
        RandomAccessContainer output(container.size());
        
        partitionedMap(container, output, function, tasks_per_processor);

        return output;
    }
    
}

#endif  /* PARTITIONEDMULTITHREADEDMAP_H */