// -*- mode: c++; c-indent-level: 4; c++-member-init-indent: 8; comment-column: 35; -*-

//-----------------------------------------------------------------------------
// eoApply.h
// (c) Maarten Keijzer 2000
/* 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    Contact: todos@geneura.ugr.es, http://geneura.ugr.es
             mak@dhi.dk
 */
//-----------------------------------------------------------------------------

#ifndef _apply_h
#define _apply_h

#include <utils/eoParallel.h>
#include <utils/eoParser.h>
#include <eoFunctor.h>
#include <vector>
#include <omp.h>

/**
  Applies a unary function to a std::vector of things.

  @ingroup Utilities
*/
template <class EOT>
void apply(eoUF<EOT&, void>& _proc, std::vector<EOT>& _pop)
{
#ifdef _OPENMP

    omp_set_num_threads(eo::parallel.nthreads());

    size_t size = _pop.size();

    double t1 = omp_get_wtime();

    if (!eo::parallel.isDynamic())
	{
#pragma omp parallel for if(eo::parallel.isEnabled()) //default(none) shared(_proc, _pop, size)
	    for (size_t i = 0; i < size; ++i) { _proc(_pop[i]); }
	}
    else
	{
#pragma omp parallel for schedule(dynamic) if(eo::parallel.isEnabled())
	    //doesnot work with gcc 4.1.2
	    //default(none) shared(_proc, _pop, size)
	    for (size_t i = 0; i < size; ++i) { _proc(_pop[i]); }
	}

    double t2 = omp_get_wtime();

    eoLogger log;
    log << eo::file(eo::parallel.prefix()) << t2 - t1 << ' ';

#else // _OPENMP

    for (size_t i = 0; i < size; ++i) { _proc(_pop[i]); }

#endif // !_OPENMP
}

/**
  This is a variant of apply<EOT> which is called in parallel
  thanks to OpenMP.

  @ingroup Utilities
*/
template <class EOT>
void omp_apply(eoUF<EOT&, void>& _proc, std::vector<EOT>& _pop)
{
    size_t size = _pop.size();
#pragma omp parallel for if(eo::parallel.isEnabled())
    //doesnot work with gcc 4.1.2
    //default(none) shared(_proc, _pop, size)
    for (size_t i = 0; i < size; ++i)
    {
        _proc(_pop[i]);
    }
}

/**
  And now we are using the dynamic scheduling.

  @ingroup Utilities
*/
template <class EOT>
void omp_dynamic_apply(eoUF<EOT&, void>& _proc, std::vector<EOT>& _pop)
{
    size_t size = _pop.size();
#pragma omp parallel for if(eo::parallel.isEnabled()) schedule(dynamic)
    //doesnot work with gcc 4.1.2
    //default(none) shared(_proc, _pop, size)
    for (size_t i = 0; i < size; ++i)
    {
        _proc(_pop[i]);
    }
}

#endif