/* 
 * Copyright (c) 2004-2005 The Trustees of Indiana University.
 *                         All rights reserved.
 * Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
 *                         All rights reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */
/**
 * @file
 */

#ifndef MCA_PTL_MX_RECV_FRAG_H
#define MCA_PTL_MX_RECV_FRAG_H

#include "ptl_mx.h"
#include "mca/ptl/base/ptl_base_recvfrag.h"
#include "ptl_mx_sendfrag.h"

/**
 *  MX received fragment derived type.
 */
struct mca_ptl_mx_recv_frag_t {
    mca_ptl_base_recv_frag_t frag_recv;  /**< base receive fragment descriptor */
    mx_request_t frag_request;
    mx_segment_t frag_segments[2];
    uint32_t frag_segment_count;
    unsigned char frag_data[32*1024];
    size_t frag_size;
};
typedef struct mca_ptl_mx_recv_frag_t mca_ptl_mx_recv_frag_t;

OBJ_CLASS_DECLARATION(mca_ptl_mx_recv_frag_t);


#define MCA_PTL_MX_RECV_FRAG_ALLOC(frag, rc) \
    { \
    opal_list_item_t* item; \
    OMPI_FREE_LIST_GET(&mca_ptl_mx_component.mx_recv_frags, item, rc); \
    frag = (mca_ptl_mx_recv_frag_t*)item; \
    }

#define MCA_PTL_MX_RECV_FRAG_RETURN(frag) \
{ \
    if(frag->frag_recv.frag_is_buffered && \
       frag->frag_data != frag->frag_recv.frag_base.frag_addr) { \
        free(frag->frag_recv.frag_base.frag_addr); \
    } \
    OMPI_FREE_LIST_RETURN(&mca_ptl_mx_component.mx_recv_frags, (opal_list_item_t*)frag); \
}
    

/**
 *  Callback on receipt of a match fragment.
 */

#define MCA_PTL_MX_RECV_FRAG_MATCH(frag, hdr)                              \
do {                                                                       \
    if(hdr->hdr_common.hdr_flags & MCA_PTL_FLAGS_NBO) {                    \
         MCA_PTL_BASE_MATCH_HDR_NTOH(hdr->hdr_match);                      \
    }                                                                      \
    ptl->super.ptl_match(&ptl->super, &frag->frag_recv, &hdr->hdr_match);  \
} while(0) 


/**
 *  Callback on receipt of a rendezvous fragment.
 */

#define MCA_PTL_MX_RECV_FRAG_RNDV(frag, hdr)                               \
do {                                                                       \
    if(hdr->hdr_common.hdr_flags & MCA_PTL_FLAGS_NBO) {                    \
         MCA_PTL_BASE_RNDV_HDR_NTOH(hdr->hdr_rndv);                        \
    }                                                                      \
    ptl->super.ptl_match(&ptl->super, &frag->frag_recv, &hdr->hdr_match);  \
} while(0) 


/**
 * Process a fragment that completed.
 */

#define MCA_PTL_MX_RECV_FRAG_FRAG(frag) \
do { \
    /* copy into user space */ \
    if(frag->frag_recv.frag_is_buffered) { \
        struct iovec iov; \
        uint32_t iov_count; \
        size_t max_data; \
        int32_t free_after; \
  \
        iov.iov_base = frag->frag_recv.frag_base.frag_addr; \
        iov.iov_len = frag->frag_recv.frag_base.frag_size; \
        iov_count = 1; \
        max_data = iov.iov_len; \
        ompi_convertor_unpack( &frag->frag_recv.frag_base.frag_convertor, \
                               &iov, &iov_count, &max_data, &free_after ); \
        frag->frag_recv.frag_base.frag_size = max_data; \
    } \
 \
    /* progress the request */ \
    frag->frag_recv.frag_base.frag_owner->ptl_recv_progress( \
        frag->frag_recv.frag_base.frag_owner, \
        frag->frag_recv.frag_request, \
        frag->frag_size, \
        frag->frag_recv.frag_base.frag_size); \
 \
    MCA_PTL_MX_RECV_FRAG_RETURN(frag); \
} while(0)


#endif