MDI_Library/html/mdi__global_8h_source.html

#ifndef MDI_GLOBAL

#define MDI_GLOBAL


#include <mpi.h>


#ifdef _WIN32

  #include <winsock2.h>

  #define sock_t SOCKET

#else

  #define sock_t int

#endif


#ifdef _WIN32

  #define mdi_strdup _strdup

#else

  #define mdi_strdup strdup

#endif


// Hard-coded values

#define PLUGIN_PATH_LENGTH 2048


// Defined languages

#define MDI_LANGUAGE_C 1

#define MDI_LANGUAGE_FORTRAN 2

#define MDI_LANGUAGE_PYTHON 3


// MDI version numbers

#define MDI_MAJOR_VERSION_ 1

#define MDI_MINOR_VERSION_ 4

#define MDI_PATCH_VERSION_ 19


// length of an MDI command in characters

#define MDI_COMMAND_LENGTH_ 256


// length of an MDI name in characters

#define MDI_NAME_LENGTH_ 256


// length of an MDI label in characters

#define MDI_LABEL_LENGTH_ 64


// value of a null communicator

#define MDI_COMM_NULL_ 0


// MDI data types

#define MDI_INT_ 1

#define MDI_INT8_T_ 7

#define MDI_INT16_T_ 8

#define MDI_INT32_T_ 9

#define MDI_INT64_T_ 10

#define MDI_UINT8_T_ 11

#define MDI_UINT16_T_ 12

#define MDI_UINT32_T_ 13

#define MDI_UINT64_T_ 14

#define MDI_DOUBLE_ 2

#define MDI_CHAR_ 3

#define MDI_FLOAT_ 4

#define MDI_BYTE_ 6


// MDI communication types

#define MDI_TCP_ 1

#define MDI_MPI_ 2

#define MDI_LINK_ 3

#define MDI_TEST_ 4


// MDI role types

#define MDI_DRIVER_ 1

#define MDI_ENGINE_ 2


// MDI Typedefs

typedef int MDI_Comm_Type;

typedef int MDI_Datatype_Type;

typedef int (*MDI_execute_command_type)(void*, MDI_Comm_Type, void*);


/*

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

!!!!!!!!!!!!!!!!! ABI WARNING !!!!!!!!!!!!!!!!!!!!

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!


Changing this structure will affect the plugin ABI


*/

typedef struct dynamic_array_struct {

  unsigned char* data;

  size_t stride;

  size_t capacity;

  size_t size; //number of elements actually stored

  size_t current_key; //number of elements actually stored

  int initialized;

} vector;


/*

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

!!!!!!!!!!!!!!!!! ABI WARNING !!!!!!!!!!!!!!!!!!!!

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!


Changing this structure will affect the plugin ABI


*/

typedef struct element_struct {

  void* data;

  size_t id;

} element;


typedef struct method_struct {

  int (*on_selection)();

  int (*on_accept_communicator)();

  int (*on_send_command)(const char*, MDI_Comm_Type, int* skip_flag);

  int (*after_send_command)(const char*, MDI_Comm_Type);

  int (*on_recv_command)(MDI_Comm_Type);

  int id;

  int method_id;

} method;


typedef struct communicator_struct {

  int mdi_version[3];

  vector* nodes;

  void* method_data;

  int (*send)(const void*, int, MDI_Datatype_Type, MDI_Comm_Type, int);

  int (*recv)(void*, int, MDI_Datatype_Type, MDI_Comm_Type, int);

  int (*delete)(void*);

  size_t code_id;

  sock_t sockfd;

  int method_id;

  MDI_Comm_Type id;

  int is_accepted;

  int name_length;

  int command_length;

} communicator;


typedef struct node_struct {

  char name[MDI_COMMAND_LENGTH_];

  vector* commands;

  vector* callbacks;

} node;


typedef struct code_struct {

  char name[MDI_NAME_LENGTH_];

  char role[MDI_NAME_LENGTH_];

  void* execute_command_obj;

  vector* nodes;

  vector* comms;

  vector* methods;

  char* plugin_path;

  int* plugin_argc_ptr;

  char*** plugin_argv_ptr;

  char** plugin_unedited_options_ptr;

  void* shared_state_from_driver;

  char* hostname;

  int (*driver_callback_f90)(void*);

  int (*execute_command_wrapper)(const char*, MDI_Comm_Type, void*);

  MDI_execute_command_type execute_command;

  int (*language_on_destroy)(int);

  int (*mpi4py_recv_callback)(void*, int, int, int, MDI_Comm_Type);

  int (*mpi4py_send_callback)(void*, int, int, int, MDI_Comm_Type);

  int (*mpi4py_allgather_callback)(void*, void*);

  int (*mpi4py_gather_names_callback)(void*, void*, int*, int*);

  int (*mpi4py_split_callback)(int, int, MDI_Comm_Type, int);

  int (*mpi4py_rank_callback)(int);

  int (*mpi4py_size_callback)(int);

  int (*mpi4py_barrier_callback)(int);

  int (*py_launch_plugin_callback)(void*, void*, void*, int);

  MPI_Comm intra_MPI_comm;

  sock_t tcp_socket;

  int is_library;

  int world_size;

  int world_rank;

  size_t id;

  int returned_comms;

  int next_comm;

  int language;

  int intra_rank;

  int called_set_execute_command_func;

  int selected_method_id;

  int ipi_compatibility;

  int initialized_mpi;

  int tcp_initialized;

  int mpi_initialized;

  int test_initialized;

  int port;

  int debug_mode;

} code;


extern vector codes;


int vector_init(vector* v, size_t stride);

int vector_push_back(vector* v, void* element);

int vector_get(vector* v, int index, void** element);

int vector_delete(vector* v, int index);

int vector_free(vector* v);


int get_node_index(vector* v, const char* node_name, int* node_index);

int get_command_index(node* n, const char* command_name, int* command_index);

int get_callback_index(node* n, const char* callback_name, int* callback_index);

int free_node_vector(vector* v);


int new_communicator(size_t code_id, int method, MDI_Comm_Type* comm_id_ptr);

int get_communicator(size_t code_id, MDI_Comm_Type comm_id, communicator** comm_ptr);

int delete_communicator(size_t code_id, MDI_Comm_Type comm_id);


int new_code(size_t* code_id);

int get_code(size_t code_id, code** ret_code);

int delete_code(size_t code_id);

int get_current_code(code** this_code_ptr);


int new_method(size_t code_id, int method_id, int* id_ptr);

int get_method(size_t code_id, int method_id, method** method_ptr);

int delete_method(size_t code_id, int method_id);

int free_methods_vector(vector* v);


int file_exists(const char* file_name, int* flag);


int communicator_delete(void* comm);


void mdi_error(const char* message);

void mdi_warning(const char* message);

int mdi_debug(const char* message, ...);


int datatype_info(MDI_Datatype_Type datatype, size_t* size, MDI_Datatype_Type* base);


int datatype_mpitype(MDI_Datatype_Type datatype, MPI_Datatype* mpitype);


int convert_buf_datatype(void* recvbuf_in, MDI_Datatype_Type recvtype,

             void* sendbuf_in, MDI_Datatype_Type sendtype,

             int count);


#endif