source: XIOS/dev/branch_openmp/extern/ep_dev/ep_recv.cpp @ 1501

/*!
   \file ep_recv.cpp
   \since 2 may 2016

   \brief Definitions of MPI receive functions: MPI_Recv, MPI_Mrecv, MPI_Irecv, MPI_Imrecv
 */


#include "ep_lib.hpp"
#include <mpi.h>
#include "ep_declaration.hpp"
#include "ep_mpi.hpp"

using namespace std;

extern std::list< ep_lib::MPI_Request* > * EP_PendingRequests;
#pragma omp threadprivate(EP_PendingRequests)

namespace ep_lib 
{
  

  int MPI_Recv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status *status)
  {

    if(!comm->is_ep)
    {
      ::MPI_Status mpi_status;
      ::MPI_Recv(buf, count, to_mpi_type(datatype), src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), &mpi_status);

      status->ep_src = src;
      status->ep_tag = tag;
      status->ep_datatype = datatype;

      return 0;  
    }

    //Message_Check(comm);

    MPI_Request request;
    MPI_Irecv(buf, count, datatype, src, tag, comm, &request);
    MPI_Wait(&request, status);

    return 0;
  }




  int MPI_Irecv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Request *request)
  {

    Debug("MPI_Irecv with EP");
    int dest_rank;
    MPI_Comm_rank(comm, &dest_rank);
    

    if(!comm->is_ep)
    {
      ::MPI_Request mpi_request;
      ::MPI_Irecv(buf, count, to_mpi_type(datatype), src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm->mpi_comm), &mpi_request);

      (*request)->mpi_request = new ::MPI_Request(mpi_request);
      (*request)->ep_src = src;
      (*request)->ep_datatype = datatype;
      (*request)->ep_tag = tag;
    }
    
    *request = new ep_request;
    printf("new %p : in ep_lib::MPI_Irecv, *request = new ep_request\n", (*request));

    (*request)->mpi_request = new ::MPI_Request;
    printf("new %p : in ep_lib::MPI_Irecv, (*request)->mpi_request = new ::MPI_Request\n", (*request)->mpi_request);
    
    (*request)->buf = buf;
    (*request)->comm = comm;
    (*request)->type = 2;
    (*request)->state = 0;
    

    (*request)->ep_src = src;
    (*request)->ep_tag = tag;
    (*request)->ep_datatype = datatype;
    
    
    /* With Improbe*/

    if(EP_PendingRequests == 0 ) 
    {
      EP_PendingRequests = new std::list< MPI_Request* >;
    }

    EP_PendingRequests->push_back(request);                                                                
                                          
    Request_Check();

    return 0;
  }

  int MPI_Imrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Request *request)
  {
    Debug("MPI_Imrecv");

    (*request)->type = 3;
        
    ::MPI_Imrecv(buf, count, to_mpi_type(datatype), to_mpi_message_ptr(*message), to_mpi_request_ptr(*request));                

    (*request)->ep_datatype = datatype;
    (*request)->ep_tag = (*message)->ep_tag;
    (*request)->ep_src = (*message)->ep_src;
    
    
    return 0;
  }


  int MPI_Mrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Status *status)
  {
    Debug("EP Mrecv called\n");

    ::MPI_Status mpi_status;
    ::MPI_Mrecv(buf, count, to_mpi_type(datatype), static_cast< ::MPI_Message* >((*message)->mpi_message), &mpi_status);

    status->mpi_status = new ::MPI_Status(mpi_status);
    status->ep_src = (*message)->ep_src;
    status->ep_datatype = datatype;
    status->ep_tag = (*message)->ep_tag;

    delete (*message)->mpi_message;

    //check_sum_recv(buf, count, datatype, message->ep_src, message->ep_tag);

    return 0;
  }

}