/* -*- Mode: C; c-basic-offset:4 ; -*- */
/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  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$
 */
#include "ompi_config.h"
#include <errno.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <fcntl.h>
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif

#include "ompi/include/constants.h"
#include "ompi/class/ompi_bitmap.h"
#include "opal/event/event.h"
#include "opal/util/argv.h"
#include "opal/util/output.h"
#include "opal/mca/base/mca_base_param.h"
#include "ompi/datatype/datatype.h"
#include "ompi/mca/pml/pml.h"
#include "ompi/mca/ptl/ptl.h"
#include "ompi/mca/pml/base/pml_base_sendreq.h"
#include "ompi/mca/ptl/base/ptl_base_sendfrag.h"
#include "ompi/mca/ptl/base/ptl_base_recvfrag.h"
#include "ptl_self.h"
#include "ompi/communicator/communicator.h"

mca_ptl_base_module_t mca_ptl_self_module = {
    &mca_ptl_self_component.super,
    8, /* ptl_cache_size */
    sizeof(mca_ptl_base_recv_frag_t), /* ptl_cache_bytes */
    0, /* ptl_frag_first_size */
    0, /* ptl_frag_min_size */
    0, /* ptl_frag_max_size */
    65535, /* ptl_exclusivity */
    0, /* ptl_latency */
    0, /* ptl_bandwidth */
    MCA_PTL_PUT, /* ptl flags */
    mca_ptl_self_add_proc,
    mca_ptl_self_del_proc,
    mca_ptl_self_finalize,
    mca_ptl_self_send,  /* put */
    mca_ptl_self_send,  /* put */
    NULL, /* get */
    mca_ptl_self_matched, /* matched */
    mca_ptl_self_request_init,
    mca_ptl_self_request_fini,
    NULL, /* match */
    NULL,
    NULL
};

extern mca_ptl_self_component_t mca_ptl_self_component;


int mca_ptl_self_add_proc(struct mca_ptl_base_module_t* ptl, size_t nprocs, struct ompi_proc_t **ompi_proc, struct mca_ptl_base_peer_t** peer_ret, ompi_bitmap_t* reachable)
{
    size_t i, count;

    for( i = 0, count = 0; i < nprocs; i++ ) {
        if( ompi_proc[i] == mca_ptl_self_component.self_local ) {
            ompi_bitmap_set_bit( reachable, i );
            count++;
        }
    }
    return OMPI_SUCCESS;
}

int mca_ptl_self_del_proc(struct mca_ptl_base_module_t* ptl, size_t nprocs, struct ompi_proc_t **proc, struct mca_ptl_base_peer_t** ptl_peer)
{
    return OMPI_SUCCESS;
}

/* before the module is unloaded (called once)*/
int mca_ptl_self_finalize(struct mca_ptl_base_module_t* ptl)
{
    return OMPI_SUCCESS;
}

int mca_ptl_self_request_init(struct mca_ptl_base_module_t* ptl, mca_ptl_base_send_request_t* request)
{
    OBJ_CONSTRUCT(request+1, mca_ptl_base_recv_frag_t);
    return OMPI_SUCCESS;
}

void mca_ptl_self_request_fini(struct mca_ptl_base_module_t* ptl, mca_ptl_base_send_request_t* request)
{
    OBJ_DESTRUCT(request+1);
}

/*
 *  Initiate a send. If this is the first fragment, use the fragment
 *  descriptor allocated with the send requests, otherwise obtain
 *  one from the free list. Initialize the fragment and foward
 *  on to the peer.
 */

int mca_ptl_self_send( struct mca_ptl_base_module_t* ptl,
                       struct mca_ptl_base_peer_t* ptl_base_peer,
                       struct mca_ptl_base_send_request_t* request,
                       size_t offset,
                       size_t size,
                       int flags )
{
    mca_ptl_self_send_request_t* req = (mca_ptl_self_send_request_t*)request;
    mca_ptl_base_header_t* hdr = &(req->req_frag.frag_base.frag_header);
    bool match;

    hdr->hdr_common.hdr_type = MCA_PTL_HDR_TYPE_MATCH;
    hdr->hdr_common.hdr_flags = flags;
    hdr->hdr_match.hdr_contextid = request->req_send.req_base.req_comm->c_contextid;
    hdr->hdr_match.hdr_src = request->req_send.req_base.req_comm->c_my_rank;
    hdr->hdr_match.hdr_dst = request->req_send.req_base.req_peer;
    hdr->hdr_match.hdr_tag = request->req_send.req_base.req_tag;
    hdr->hdr_match.hdr_msg_length = request->req_send.req_bytes_packed;
    hdr->hdr_match.hdr_msg_seq = request->req_send.req_base.req_sequence;
    hdr->hdr_rndv.hdr_src_ptr.lval = 0;
    hdr->hdr_rndv.hdr_src_ptr.pval = request;
    req->req_frag.frag_base.frag_peer = ptl_base_peer;
    req->req_frag.frag_base.frag_size = request->req_send.req_bytes_packed;
    req->req_frag.frag_base.frag_owner = &mca_ptl_self_module;
    req->req_frag.frag_request = NULL;
    req->req_frag.frag_is_buffered = 0;
    match = ptl->ptl_match( ptl, &(req->req_frag), &(hdr->hdr_match) );
#if !OMPI_ENABLE_MPI_THREADS && 0
    /* If we are in a non threaded case and the send is blocking for MPI correctness
     * the receive should be already posted. Otherwise the program will lead to a deadlock.
     */
    if( (false == match) && (MCA_PML_REQUEST_SEND == req->req_send.req_base.req_type) ) {
        opal_output( 0, "OMPI reach a dead-lock situation. A send to self was posted without a proper receive\n" );
        return OMPI_ERROR;
    }
#endif  /* OMPI_ENABLE_MPI_THREADS */
    return OMPI_SUCCESS;
}


/*
 *  A posted receive has been matched - if required send an
 *  ack back to the peer and process the fragment.
 */
void mca_ptl_self_matched( mca_ptl_base_module_t* ptl,
                           mca_ptl_base_recv_frag_t* frag)
{
    mca_ptl_self_send_request_t* sendreq = (mca_ptl_self_send_request_t*)
        frag->frag_base.frag_header.hdr_rndv.hdr_src_ptr.pval;
    mca_ptl_base_recv_request_t* recvreq = frag->frag_request;

    if( (recvreq->req_recv.req_base.req_count != 0) &&
        (sendreq->req_ptl.req_send.req_base.req_count != 0) ) {
        /* Did we have the same datatype or not ? If yes we can use an optimized version
         * for the copy function, if not we have to use a temporary buffer to pack/unpack
         * 
         * Note that if this is a buffered send - the data has already been packed into
         * a contigous buffer and the convertor on the send request initialized to point
         * into this buffer.
         */
        if( sendreq->req_ptl.req_send.req_datatype == recvreq->req_recv.req_base.req_datatype ) {
            ompi_ddt_copy_content_same_ddt( recvreq->req_recv.req_base.req_datatype, 
                                            recvreq->req_recv.req_base.req_count > sendreq->req_ptl.req_send.req_count ?
                                            sendreq->req_ptl.req_send.req_count : recvreq->req_recv.req_base.req_count,
                                            (char *)recvreq->req_recv.req_base.req_addr, 
                                            (char *)sendreq->req_ptl.req_send.req_addr );
        } else {
            ompi_convertor_t *send_convertor, *recv_convertor;
            struct iovec iov[1];
            int32_t completed, freeAfter, length;
            uint32_t iov_count;
            size_t max_data;
            char* buf;
            
            /* We use a temporary buffer as it look to be faster on much architectures */
            length = 64 * 1024;
            buf = (char *)malloc( length * sizeof(char) );
            
            recv_convertor = &(recvreq->req_recv.req_convertor);
            send_convertor = &(sendreq->req_ptl.req_send.req_convertor);

            completed = 0;
            freeAfter = 0;
            while( !completed ) {
                iov[0].iov_base = buf;
                iov[0].iov_len = length;
                iov_count = 1;
                max_data = length;
                completed |= ompi_convertor_pack( send_convertor, iov, &iov_count,
                                                  &max_data, &freeAfter );
                assert( freeAfter == 0 );
                completed |= ompi_convertor_unpack( recv_convertor, iov, &iov_count,
                                                    &max_data, &freeAfter );
                assert( freeAfter == 0 );
            }
            free( buf );
        }
    }
    ptl->ptl_send_progress( ptl, &sendreq->req_ptl,
                            sendreq->req_ptl.req_send.req_bytes_packed );
    ptl->ptl_recv_progress( ptl, recvreq, 
                            frag->frag_base.frag_header.hdr_match.hdr_msg_length,
                            frag->frag_base.frag_size );
}