/* dtkDistributedCommunicator.h ---
 *
 * Author: Julien Wintz
 * Copyright (C) 2008 - Julien Wintz, Inria.
 * Created: Mon Feb 15 09:18:40 2010 (+0100)
 */

/* Commentary:
 *
 */

/* Change log:
 *
 */

#ifndef DTKDISTRIBUTEDCOMMUNICATOR_H
#define DTKDISTRIBUTEDCOMMUNICATOR_H

#include <QtCore>

#include <dtkDistributedSupportExport.h>
#include <dtkCoreSupport/dtkAbstractData.h>

class dtkDistributedCommunicatorPrivate;

class dtkDistributedMessage;
class dtkDistributedCommunicatorStatusPrivate;

class DTKDISTRIBUTEDSUPPORT_EXPORT dtkDistributedCommunicatorStatus
{

public:
    dtkDistributedCommunicatorStatus(void);
    virtual ~dtkDistributedCommunicatorStatus(void);

public:
    dtkDistributedCommunicatorStatus(const dtkDistributedCommunicatorStatus& other);
    dtkDistributedCommunicatorStatus& operator = (const dtkDistributedCommunicatorStatus& other);

public:
    int       tag(void) const;
    qint64  count(void) const;
    qint16 source(void) const;
    int     error(void) const;

public:
    void    setTag(int tag);
    void  setCount(qint64 count);
    void setSource(qint16 source);
    void  setError(int error);

private:
    dtkDistributedCommunicatorStatusPrivate *d;
};

class DTKDISTRIBUTEDSUPPORT_EXPORT dtkDistributedCommunicatorRequest
{

public:
    dtkDistributedCommunicatorRequest(void);
    virtual ~dtkDistributedCommunicatorRequest(void);

public:
    virtual void wait(void);

};


class DTKDISTRIBUTEDSUPPORT_EXPORT dtkDistributedCommunicator : public QObject
{
    Q_OBJECT

public:
    dtkDistributedCommunicator(void);
    virtual ~dtkDistributedCommunicator(void);

public:
    dtkDistributedCommunicator(const dtkDistributedCommunicator& other);
    dtkDistributedCommunicator& operator = (const dtkDistributedCommunicator& other);

public:

    // use the same value as defined in mpi.h from openmpi
    static const int    ANY_TAG    = -1;
    static const qint16 ANY_SOURCE = -1;
    static const qint16 ROOT       = -4;
    static const qint16 PROC_NULL  = -2;

    enum DataType {
        dtkDistributedCommunicatorBool,
        dtkDistributedCommunicatorChar,
        dtkDistributedCommunicatorInt,
        dtkDistributedCommunicatorLong,
        dtkDistributedCommunicatorInt64,
        dtkDistributedCommunicatorFloat,
        dtkDistributedCommunicatorDouble
    };

    enum OperationType {
        dtkDistributedCommunicatorMin,
        dtkDistributedCommunicatorMax,
        dtkDistributedCommunicatorSum,
        dtkDistributedCommunicatorProduct,
        dtkDistributedCommunicatorBitwiseAnd,
        dtkDistributedCommunicatorBitwiseOr,
        dtkDistributedCommunicatorBitwiseXor,
        dtkDistributedCommunicatorLogicalAnd,
        dtkDistributedCommunicatorLogicalOr,
        dtkDistributedCommunicatorLogicalXor
    };

    enum MessageType {
        dtkDistributedCommunicatorBarrier = 1
    };

    enum MessageTag {
        dtkDistributedCommunicatorSend    = 2001,
        dtkDistributedCommunicatorReceive = 2002
    };

    virtual void   initialize(void);
    virtual bool  initialized(void);
    virtual void uninitialize(void);
    virtual dtkDistributedCommunicator *spawn(QString cmd, qlonglong np);

    virtual int rank(void);
    virtual int size(void);

    virtual void      send(void   *data, qint64 size, DataType dataType, qint16 target, int tag);
    virtual void      send(bool   *data, qint64 size, qint16 target, int tag);
    virtual void      send(char   *data, qint64 size, qint16 target, int tag);
    virtual void      send(int    *data, qint64 size, qint16 target, int tag);
    virtual void      send(long   *data, qint64 size, qint16 target, int tag);
    virtual void      send(qint64 *data, qint64 size, qint16 target, int tag);
    virtual void      send(float  *data, qint64 size, qint16 target, int tag);
    virtual void      send(double *data, qint64 size, qint16 target, int tag);
    virtual void      send(dtkAbstractData *data,     qint16 target, int tag);
    virtual void      send(const QString& s,   qint16 target, int tag);
    virtual void      send(QByteArray& array,  qint16 target, int tag);

    virtual void isend(void   *data, qint64 size, DataType dataType, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(bool   *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(char   *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(int    *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(long   *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(qint64 *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(float  *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(double *data, qint64 size, qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(dtkAbstractData *data,     qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(const QString& s,   qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);
    virtual void isend(QByteArray& array,  qint16 target, int tag, dtkDistributedCommunicatorRequest *req = NULL);

    virtual void   receive(void   *data, qint64 size, DataType dataType, qint16 source, int tag);
    virtual void   receive(bool   *data, qint64 size, qint16 source, int tag);
    virtual void   receive(char   *data, qint64 size, qint16 source, int tag);
    virtual void   receive(int    *data, qint64 size, qint16 source, int tag);
    virtual void   receive(long   *data, qint64 size, qint16 source, int tag);
    virtual void   receive(qint64 *data, qint64 size, qint16 source, int tag);
    virtual void   receive(float  *data, qint64 size, qint16 source, int tag);
    virtual void   receive(double *data, qint64 size, qint16 source, int tag);
    virtual void   receive(dtkAbstractData *&data, qint16 source, int tag);
    virtual void   receive(QString& s,   qint16 source, int tag);
    virtual void   receive(QByteArray& v,  qint16 source, int tag);

    virtual void   receive(void   *data, qint64 size, DataType dataType, qint16 source, int tag, dtkDistributedCommunicatorStatus& status);
    virtual void   receive(QByteArray& v,  qint16 source, int tag, dtkDistributedCommunicatorStatus& status);

    virtual void   ireceive(void   *data, qint64 size, DataType dataType, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(bool   *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(char   *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(int    *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(long   *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(qint64 *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(float  *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(double *data, qint64 size, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(dtkAbstractData *&data, qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(QString& s,   qint16 source, int tag, dtkDistributedCommunicatorRequest *req);
    virtual void   ireceive(QByteArray& v,  qint16 source, int tag, dtkDistributedCommunicatorRequest *req);


    virtual void   barrier(void);

    virtual void broadcast(void   *data, qint64 size, DataType dataType, qint16 source);
    virtual void broadcast(bool   *data, qint64 size, qint16 source);
    virtual void broadcast(char   *data, qint64 size, qint16 source);
    virtual void broadcast(int    *data, qint64 size, qint16 source);
    virtual void broadcast(long   *data, qint64 size, qint16 source);
    virtual void broadcast(qint64 *data, qint64 size, qint16 source);
    virtual void broadcast(float  *data, qint64 size, qint16 source);
    virtual void broadcast(double *data, qint64 size, qint16 source);
    virtual void broadcast(QString& s,   qint16 source);
    virtual void broadcast(QByteArray& v, qint16 source);
    virtual void broadcast(dtkAbstractData *&data, qint16 source);

    virtual void    gather(void   *send, void   *recv, qint64 size, DataType dataType, qint16 target, bool all = false);
    virtual void    gather(bool   *send, bool   *recv, qint64 size, qint16 target, bool all = false);
    virtual void    gather(char   *send, char   *recv, qint64 size, qint16 target, bool all = false);
    virtual void    gather(int    *send, int    *recv, qint64 size, qint16 target, bool all = false);
    virtual void    gather(float  *send, float  *recv, qint64 size, qint16 target, bool all = false);
    virtual void    gather(double *send, double *recv, qint64 size, qint16 target, bool all = false);

    virtual void   scatter(void   *send, void   *recv, qint64 size, DataType dataType, qint16 source);
    virtual void   scatter(bool   *send, bool   *recv, qint64 size, qint16 source);
    virtual void   scatter(char   *send, char   *recv, qint64 size, qint16 source);
    virtual void   scatter(int    *send, int    *recv, qint64 size, qint16 source);
    virtual void   scatter(float  *send, float  *recv, qint64 size, qint16 source);
    virtual void   scatter(double *send, double *recv, qint64 size, qint16 source);

    virtual void    reduce(void      *send, void      *recv, qint64 size, DataType dataType, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(bool      *send, bool      *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(char      *send, char      *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(int       *send, int       *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(qlonglong *send, qlonglong *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(float     *send, float     *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);
    virtual void    reduce(double    *send, double    *recv, qint64 size, OperationType operationType, qint16 target, bool all = false);

public:
    QMap<qlonglong, dtkDistributedMessage *> *msgBuffer(void);

private:
    dtkDistributedCommunicatorPrivate *d;
};

Q_DECLARE_METATYPE(dtkDistributedCommunicator);
Q_DECLARE_METATYPE(dtkDistributedCommunicator *);

#endif