/home/brandes/workspace/LAMA/src/lama/pgas/functor/PGASSumFunctor.hpp

Go to the documentation of this file.

#ifndef LAMA_PGASSUMFUNCTOR_HPP_
#define LAMA_PGASSUMFUNCTOR_HPP_

#include <lama/pgas/PGASInterface.hpp>
#include <lama/pgas/PGASFunctor.hpp>

namespace lama{
template <typename T>
class PGASSumFunctor: public PGASFunctor
{
    T* mResult;
    T* mWork;
    const PGASInterface* const mInterface;

public:
    PGASSumFunctor(T val);
    virtual ~PGASSumFunctor();
    virtual void iteration(int partner, bool active);
    T getResult();
};

template <typename T>
PGASSumFunctor<T>::PGASSumFunctor(T val): mInterface(PGASInterface::getInstance())
{
    mResult = static_cast<T*>(mInterface->allocate(sizeof(T)));
    mWork = static_cast<T*>(mInterface->allocate(sizeof(T)));
    //*mWork = val;
    *mResult = val;
}
template <typename T>
PGASSumFunctor<T>::~PGASSumFunctor()
{
    mInterface->free(mWork,sizeof(T));
    mInterface->free(mResult,sizeof(T));
}

template <typename T>
T PGASSumFunctor<T>::getResult()
{
    return *mResult;
}

template <typename T>
void PGASSumFunctor<T>::iteration(int partner, bool active)
{
    mInterface->syncronizeAll();
    *mWork = *mResult;
    mInterface->syncronizeAll();
    if(active)
    {
        mInterface->get(mResult,mWork,sizeof(T),partner);
        *mResult += *mWork;
    }
}
}//namespace lama

#endif // LAMA_PGASSUMFUNCTOR_HPP_