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

   \brief Definitions of MPI test function: MPI_Test, MPI_Testsome, MPI_Testany, MPI_Testall
 */

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

using namespace std;


namespace ep_lib {

  int MPI_Test(MPI_Request *request, int *flag, MPI_Status *status)
  {
    Debug("MPI_Test with EP");

    *flag = false;
    

    if((*request)->type == 1)      // isend
    {
      ::MPI_Status mpi_status;
      ::MPI_Test(static_cast< ::MPI_Request*>((*request)->mpi_request), flag, &mpi_status);
      
      if(*flag) 
      {
        status->mpi_status = new ::MPI_Status(mpi_status);
        status->ep_src = (*request)->ep_src;
        status->ep_tag = (*request)->ep_tag;
        status->ep_datatype = (*request)->ep_datatype;
        (*request)->state = 2;
        delete (*request)->mpi_request;
        delete *request;
      }

      return 0;
    }

    if((*request)->type == 2)   // irecv message not probed
    {
      Request_Check();
      
      #pragma omp flush
      
    }

    if((*request)->type == 3)  // imrecv
    {
      if((*request)->state == 2)
      {
        return *flag = true;
      }
      ::MPI_Status mpi_status;
      
      ::MPI_Test(static_cast< ::MPI_Request*>((*request)->mpi_request), flag, &mpi_status);
      
      
      if(*flag)
      {
        status->mpi_status = new ::MPI_Status(mpi_status);
        status->ep_src = (*request)->ep_src;
        status->ep_tag = (*request)->ep_tag;
        status->ep_datatype = (*request)->ep_datatype;
        
        (*request)->state = 2;
        
        delete (*request)->mpi_request;
        delete *request;
        //int count;
        //MPI_Get_count(status, request->ep_datatype, &count);
        //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2);
      }  

      return Request_Check();
    }

  }


  int MPI_Testall(int count, MPI_Request *array_of_requests, int *flag, MPI_Status *array_of_statuses)
  {
    Debug("MPI_Testall with EP");

    ::MPI_Request* mpi_request = new ::MPI_Request[count];
    ::MPI_Status* mpi_status = new ::MPI_Status[count];


    for(int i=0; i<count; i++)
    {
      mpi_request[i] = *static_cast< ::MPI_Request*>(array_of_requests[i]->mpi_request);
    }

    ::MPI_Testall(count, mpi_request, flag, mpi_status);

    if(*flag)
    {
      for(int i=0; i<count; i++)
      {
        array_of_statuses[i].mpi_status = &mpi_status[i];
        array_of_statuses[i].ep_src = array_of_requests[i]->ep_src;
        array_of_statuses[i].ep_tag = array_of_requests[i]->ep_tag;
        array_of_statuses[i].ep_datatype = array_of_requests[i]->ep_datatype;
        
        array_of_requests[i]->state = 2;
      
        memcheck("delete "<< array_of_requests[i]->mpi_request <<" : in ep_lib::MPI_Waitall, array_of_requests["<<i<<"]->mpi_request");
        delete array_of_requests[i]->mpi_request;
        delete array_of_requests[i]; 
      }
      
    }

    return Request_Check();
  }


}