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

namespace ep_lib
{

  int MPI_Comm_dup(MPI_Comm comm, MPI_Comm *newcomm)
  {
    if(!comm.is_ep)
    {
      Debug("Comm_dup MPI\n");
      newcomm = new MPI_Comm;
      newcomm->is_ep = comm.is_ep;

      ::MPI_Comm input = to_mpi_comm(comm.mpi_comm);
      ::MPI_Comm *output = new ::MPI_Comm;

      
      ::MPI_Comm_dup(input, output);

      newcomm->mpi_comm = output;

      return 0;
    }

    if(comm.is_intercomm) return MPI_Comm_dup_intercomm(comm, newcomm);

    // for intracomm
    if(comm.mpi_comm == static_cast< ::MPI_Comm* >(MPI_COMM_NULL.mpi_comm)) return 0;


    int my_rank = comm.ep_comm_ptr->size_rank_info[1].first;
    int num_ep  = comm.ep_comm_ptr->size_rank_info[1].second;


    if(0 == my_rank)
    {
      MPI_Info info;
      MPI_Comm *out_comm;
      ::MPI_Comm *mpi_dup = new ::MPI_Comm;

      ::MPI_Comm in_comm = to_mpi_comm(comm.mpi_comm);

      ::MPI_Comm_dup(in_comm, mpi_dup);

      MPI_Comm_create_endpoints(mpi_dup, num_ep, info, out_comm);
      comm.ep_comm_ptr->comm_list->mem_bridge = out_comm;
    }

    MPI_Barrier(comm);

    *newcomm = (comm.ep_comm_ptr->comm_list->mem_bridge[my_rank]);

    return MPI_SUCCESS;
  }

  int MPI_Comm_dup_intercomm(MPI_Comm comm, MPI_Comm *newcomm)
  {
    
    if(comm.mpi_comm == static_cast< ::MPI_Comm* >(MPI_COMM_NULL.mpi_comm)) return 0;

    int my_rank = comm.ep_comm_ptr->size_rank_info[1].first;
    int num_ep  = comm.ep_comm_ptr->size_rank_info[1].second;


    if(0 == my_rank)
    {
      MPI_Info info;
      MPI_Comm *out_comm;
      ::MPI_Comm *mpi_dup = new ::MPI_Comm;

      ::MPI_Comm in_comm = to_mpi_comm(comm.mpi_comm);

      ::MPI_Comm_dup(in_comm, mpi_dup);

      MPI_Comm_create_endpoints(mpi_dup, num_ep, info, out_comm);

      ::MPI_Comm *mpi_inter = new ::MPI_Comm;

      ::MPI_Comm_dup(to_mpi_comm(comm.ep_comm_ptr->intercomm->mpi_inter_comm), mpi_inter);
      
      for(int i=0; i<num_ep; i++)
      {
        out_comm[i].ep_comm_ptr->comm_label = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->comm_label;
        out_comm[i].ep_comm_ptr->intercomm = new ep_intercomm;

        out_comm[i].ep_comm_ptr->intercomm->mpi_inter_comm = mpi_inter;
        out_comm[i].is_intercomm = true;

        out_comm[i].ep_comm_ptr->intercomm->intercomm_rank_map = new RANK_MAP;
        out_comm[i].ep_comm_ptr->intercomm->local_rank_map = new RANK_MAP;
        out_comm[i].ep_comm_ptr->intercomm->remote_rank_map = new RANK_MAP;

        out_comm[i].ep_comm_ptr->intercomm->intercomm_rank_map = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->intercomm_rank_map;
        out_comm[i].ep_comm_ptr->intercomm->local_rank_map = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->local_rank_map;
        out_comm[i].ep_comm_ptr->intercomm->remote_rank_map = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->remote_rank_map;

        out_comm[i].ep_comm_ptr->intercomm->local_comm = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->local_comm;        
        out_comm[i].ep_comm_ptr->intercomm->intercomm_tag = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->intercomm_tag;

        for(int j =0; j<3; j++)
        {
          out_comm[i].ep_comm_ptr->intercomm->size_rank_info[j] = comm.ep_comm_ptr->comm_list[i].ep_comm_ptr->intercomm->size_rank_info[j];
        }

      }

      comm.ep_comm_ptr->comm_list->mem_bridge = out_comm;
    }

    MPI_Barrier(comm);

    *newcomm = comm.ep_comm_ptr->comm_list->mem_bridge[my_rank];
    

    return MPI_SUCCESS;
  }



}