MPIProxy.hpp

/* Copyright (c) 2013 David Sichau <mail"at"sichau"dot"eu>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 */

#ifndef UTILIB_MPIPROXY_HPP_
#define UTILIB_MPIPROXY_HPP_
#include <UtilLib/config.hpp>
#include <UtilLib/include/Log.hpp>
#include <UtilLib/include/Exception.hpp>
#include <UtilLib/include/Singleton.hpp>
#ifdef ENABLE_MPI
#include <boost/mpi/collectives.hpp>
#include <boost/mpi/communicator.hpp>
#include <boost/mpi/nonblocking.hpp>
#include <boost/mpi/request.hpp>
namespace mpi = boost::mpi;
#endif
#include <vector>


namespace UtilLib {
class MPIProxy_ {
 public:
    virtual ~MPIProxy_();

    int getRank() const;

    int getSize() const;

    void barrier();

    void waitAll();

    template <typename T>
    void broadcast(
            T& value,
            int root);

    template <typename T>
    void irecv(
            int source,
            int tag,
            T& value) const;

    template <typename T>
    void isend(
            int dest,
            int tag,
            const T& value) const;

 private:
    friend class Singleton<MPIProxy_>;
    MPIProxy_();

#ifdef ENABLE_MPI

    static std::vector<boost::mpi::request> _mpiStatus;
#endif

    static int _rank;

    static int _size;
};

template <typename T>
void MPIProxy_::broadcast(
        T& value,
        int root) {
#ifdef ENABLE_MPI
    mpi::communicator world;
    boost::mpi::broadcast(world, value, root);
#endif
}

template <typename T>
void MPIProxy_::irecv(
        int source,
        int tag,
        T& value) const {
#ifdef ENABLE_MPI
    mpi::communicator world;
    _mpiStatus.push_back(world.irecv(source, tag, value));
    LOG(utilities::logDEBUG4) << "recv source: " << source << "; tag: " <<
    tag << "; value: " << value;
#else
    UtilLib::Exception("MPI Code called from serial code in irecv");
#endif
}

template <typename T>
void MPIProxy_::isend(
        int dest,
        int tag,
        const T& value) const {
#ifdef ENABLE_MPI
    mpi::communicator world;
    _mpiStatus.push_back(world.isend(dest, tag, value));
    LOG(utilities::logDEBUG4) << "send destination: " << dest << "; tag: " <<
    tag << "; value: " << value;
#else
    UtilLib::Exception("MPI Code called from serial code in isend");
#endif
}

typedef Singleton<MPIProxy_> MPIProxySingleton;

inline MPIProxy_& MPIProxy() {
    return MPIProxySingleton::instance();
}
} /* end namespace */
#endif /* UTILIB_MPIPROXY_HPP_ */