openmpi/ompi/message/message.h

/*
 * Copyright (c) 2011-2012 Sandia National Laboratories. All rights reserved.
 * Copyright (c) 2012      Cisco Systems, Inc.  All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */

#ifndef OMPI_MESSAGE_H
#define OMPI_MESSAGE_H

#include "mpi.h"
#include "opal/class/opal_free_list.h"
#include "opal/class/opal_pointer_array.h"

BEGIN_C_DECLS

struct ompi_communicator_t;

struct ompi_message_t {
    opal_free_list_item_t super;        /**< Base type */
    int m_f_to_c_index;                 /**< Fortran handle for this message */
    struct ompi_communicator_t *comm;   /**< communicator used in probe */
    void* req_ptr;                      /**< PML data */
    int peer;                           /**< peer, same as status.MPI_SOURCE */
    size_t count;                       /**< same value as status._ucount */
};
typedef struct ompi_message_t ompi_message_t;
OMPI_DECLSPEC OBJ_CLASS_DECLARATION(ompi_message_t);

/**
 * Padded struct to maintain back compatibiltiy.
 * See ompi/communicator/communicator.h comments with struct ompi_communicator_t
 * for full explanation why we chose the following padding construct for predefines.
 */
#define PREDEFINED_MESSAGE_PAD (sizeof(void*) * 32)

struct ompi_predefined_message_t {
    struct ompi_message_t message;
    char padding[PREDEFINED_MESSAGE_PAD - sizeof(ompi_message_t)];
};

typedef struct ompi_predefined_message_t ompi_predefined_message_t;

int ompi_message_init(void);

int ompi_message_finalize(void);

OMPI_DECLSPEC extern opal_free_list_t ompi_message_free_list;
OMPI_DECLSPEC extern opal_pointer_array_t  ompi_message_f_to_c_table;
OMPI_DECLSPEC extern ompi_predefined_message_t  ompi_message_no_proc;

static inline
ompi_message_t*
ompi_message_alloc(void)
{
    int rc;
    opal_free_list_item_t *tmp;
    OPAL_FREE_LIST_GET(&ompi_message_free_list,
                       tmp,
                       rc);
    (void)rc;  /* prevent "set but not used" compiler complaints */
    return (ompi_message_t*) tmp;
}

static inline
void
ompi_message_return(ompi_message_t* msg)
{
    if (MPI_UNDEFINED != msg->m_f_to_c_index) {
        opal_pointer_array_set_item(&ompi_message_f_to_c_table,
                                    msg->m_f_to_c_index, NULL);
        msg->m_f_to_c_index = MPI_UNDEFINED;
    }

    OPAL_FREE_LIST_RETURN(&ompi_message_free_list, 
                          &msg->super);
}


END_C_DECLS

#endif