/*
 * 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_PML_PROC_H
#define MCA_PML_PROC_H

#include "opal/threads/mutex.h"
#include "communicator/communicator.h"
#include "group/group.h"
#include "proc/proc.h"
#include "pml_ob1_endpoint.h"

#if defined(c_plusplus) || defined(__cplusplus)
extern "C" {
#endif
/**
 *  Structure associated w/ ompi_proc_t that contains data specific
 *  to the PML.
 */
struct mca_pml_proc_t {
   opal_object_t super;
   ompi_proc_t *proc_ompi;             /**< back-pointer to ompi_proc_t */
   opal_mutex_t proc_lock;             /**< lock to protect against concurrent access */
   int proc_flags;                     /**< prefered method of accessing this peer */
   mca_pml_ob1_ep_array_t btl_eager;   /**< array of endpoints to use for first fragments */
   mca_pml_ob1_ep_array_t btl_send;    /**< array of endpoints to use for remaining fragments */
   mca_pml_ob1_ep_array_t btl_rdma;    /**< array of endpoints that support (prefer) rdma */
   size_t proc_rdma_offset;            /**< max of min rdma size for available rmda btls */
};
typedef struct mca_pml_proc_t mca_pml_ob1_proc_t;

OMPI_COMP_EXPORT OBJ_CLASS_DECLARATION(mca_pml_ob1_proc_t);


/**
 * Return the mca_pml_proc_t instance cached in the communicators local group.
 * 
 * @param comm   Communicator
 * @param rank   Peer rank
 * @return       mca_pml_proc_t instance
 */

static inline mca_pml_ob1_proc_t* mca_pml_ob1_proc_lookup_local(ompi_communicator_t* comm, int rank)
{
    return comm->c_local_group->grp_proc_pointers[rank]->proc_pml;
}

/**
 * Return the mca_pml_proc_t instance cached on the communicators remote group.
 * 
 * @param comm   Communicator
 * @param rank   Peer rank
 * @return       mca_pml_proc_t instance
 */

static inline mca_pml_ob1_proc_t* mca_pml_ob1_proc_lookup_remote(ompi_communicator_t* comm, int rank)
{
    return comm->c_pml_procs[rank];
}

/**
 * Return the mca_btl_peer_t instance corresponding to the process/btl combination.
 * 
 * @param comm   Communicator
 * @param rank   Peer rank
 * @return       mca_pml_proc_t instance
 */

static inline struct mca_btl_base_endpoint_t* mca_pml_ob1_proc_lookup_remote_endpoint(
    ompi_communicator_t* comm, 
    int rank, 
    struct mca_btl_base_module_t* btl)
{
    mca_pml_ob1_proc_t* proc = comm->c_pml_procs[rank];
    size_t i, size = mca_pml_ob1_ep_array_get_size(&proc->btl_eager);
    mca_pml_ob1_endpoint_t* endpoint = proc->btl_eager.arr_endpoints;
    for(i = 0; i < size; i++) {
        if(endpoint->btl == btl) {
            return endpoint->btl_endpoint;
        }
        endpoint++;
    }
    return NULL;
}


#if defined(c_plusplus) || defined(__cplusplus)
}
#endif
#endif