src/smpi/smpi_mpi_dt.cpp

   1 /* smpi_mpi_dt.c -- MPI primitives to handle datatypes                        */
   2 /* FIXME: a very incomplete implementation                                    */
   3
   4 /* Copyright (c) 2009-2015. The SimGrid Team.
   5  * All rights reserved.                                                     */
   6
   7 /* This program is free software; you can redistribute it and/or modify it
   8  * under the terms of the license (GNU LGPL) which comes with this package. */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <limits.h>
  14 #include "private.h"
  15 #include "smpi_mpi_dt_private.h"
  16 #include "mc/mc.h"
  17 #include "xbt/replay.h"
  18 #include "simgrid/modelchecker.h"
  19
  20 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_mpi_dt, smpi, "Logging specific to SMPI (datatype)");
  21
  22 xbt_dict_t smpi_type_keyvals = NULL;
  23 int type_keyval_id=0;//avoid collisions
  24
  25 #define CREATE_MPI_DATATYPE(name, type)               \
  26   static s_smpi_mpi_datatype_t mpi_##name = {         \
  27     (char*) # name,                                   \
  28     sizeof(type),   /* size */                        \
  29     0,              /*was 1 sizeof_substruct*/             \
  30     0,              /* lb */                          \
  31     sizeof(type),   /* ub = lb + size */              \
  32     DT_FLAG_BASIC,  /* flags */                       \
  33     NULL,           /* attributes */                  \
  34     NULL,           /* pointer on extended struct*/   \
  35     0               /* in_use counter */              \
  36   };                                                  \
  37 const MPI_Datatype name = &mpi_##name;
  38
  39 #define CREATE_MPI_DATATYPE_NULL(name)                \
  40   static s_smpi_mpi_datatype_t mpi_##name = {         \
  41     (char*) # name,                                   \
  42     0,              /* size */                        \
  43     0,              /* was 1 sizeof_substruct*/            \
  44     0,              /* lb */                          \
  45     0,              /* ub = lb + size */              \
  46     DT_FLAG_BASIC,  /* flags */                       \
  47     NULL,           /* attributes */                  \
  48     NULL,           /* pointer on extended struct*/   \
  49     0               /* in_use counter */              \
  50   };                                                  \
  51 const MPI_Datatype name = &mpi_##name;
  52
  53 //The following are datatypes for the MPI functions MPI_MAXLOC and MPI_MINLOC.
  54 typedef struct {
  55   float value;
  56   int index;
  57 } float_int;
  58 typedef struct {
  59   float value;
  60   float index;
  61 } float_float;
  62 typedef struct {
  63   long value;
  64   long index;
  65 } long_long;
  66 typedef struct {
  67   double value;
  68   double index;
  69 } double_double;
  70 typedef struct {
  71   long value;
  72   int index;
  73 } long_int;
  74 typedef struct {
  75   double value;
  76   int index;
  77 } double_int;
  78 typedef struct {
  79   short value;
  80   int index;
  81 } short_int;
  82 typedef struct {
  83   int value;
  84   int index;
  85 } int_int;
  86 typedef struct {
  87   long double value;
  88   int index;
  89 } long_double_int;
  90 typedef struct {
  91   int64_t value;
  92   int64_t index;
  93 } integer128_t;
  94 // Predefined data types
  95 CREATE_MPI_DATATYPE(MPI_CHAR, char);
  96 CREATE_MPI_DATATYPE(MPI_SHORT, short);
  97 CREATE_MPI_DATATYPE(MPI_INT, int);
  98 CREATE_MPI_DATATYPE(MPI_LONG, long);
  99 CREATE_MPI_DATATYPE(MPI_LONG_LONG, long long);
 100 CREATE_MPI_DATATYPE(MPI_SIGNED_CHAR, signed char);
 101 CREATE_MPI_DATATYPE(MPI_UNSIGNED_CHAR, unsigned char);
 102 CREATE_MPI_DATATYPE(MPI_UNSIGNED_SHORT, unsigned short);
 103 CREATE_MPI_DATATYPE(MPI_UNSIGNED, unsigned int);
 104 CREATE_MPI_DATATYPE(MPI_UNSIGNED_LONG, unsigned long);
 105 CREATE_MPI_DATATYPE(MPI_UNSIGNED_LONG_LONG, unsigned long long);
 106 CREATE_MPI_DATATYPE(MPI_FLOAT, float);
 107 CREATE_MPI_DATATYPE(MPI_DOUBLE, double);
 108 CREATE_MPI_DATATYPE(MPI_LONG_DOUBLE, long double);
 109 CREATE_MPI_DATATYPE(MPI_WCHAR, wchar_t);
 110 CREATE_MPI_DATATYPE(MPI_C_BOOL, bool);
 111 CREATE_MPI_DATATYPE(MPI_BYTE, int8_t);
 112 CREATE_MPI_DATATYPE(MPI_INT8_T, int8_t);
 113 CREATE_MPI_DATATYPE(MPI_INT16_T, int16_t);
 114 CREATE_MPI_DATATYPE(MPI_INT32_T, int32_t);
 115 CREATE_MPI_DATATYPE(MPI_INT64_T, int64_t);
 116 CREATE_MPI_DATATYPE(MPI_UINT8_T, uint8_t);
 117 CREATE_MPI_DATATYPE(MPI_UINT16_T, uint16_t);
 118 CREATE_MPI_DATATYPE(MPI_UINT32_T, uint32_t);
 119 CREATE_MPI_DATATYPE(MPI_UINT64_T, uint64_t);
 120 CREATE_MPI_DATATYPE(MPI_C_FLOAT_COMPLEX, float _Complex);
 121 CREATE_MPI_DATATYPE(MPI_C_DOUBLE_COMPLEX, double _Complex);
 122 CREATE_MPI_DATATYPE(MPI_C_LONG_DOUBLE_COMPLEX, long double _Complex);
 123 CREATE_MPI_DATATYPE(MPI_AINT, MPI_Aint);
 124 CREATE_MPI_DATATYPE(MPI_OFFSET, MPI_Offset);
 125
 126 CREATE_MPI_DATATYPE(MPI_FLOAT_INT, float_int);
 127 CREATE_MPI_DATATYPE(MPI_LONG_INT, long_int);
 128 CREATE_MPI_DATATYPE(MPI_DOUBLE_INT, double_int);
 129 CREATE_MPI_DATATYPE(MPI_SHORT_INT, short_int);
 130 CREATE_MPI_DATATYPE(MPI_2INT, int_int);
 131 CREATE_MPI_DATATYPE(MPI_2FLOAT, float_float);
 132 CREATE_MPI_DATATYPE(MPI_2DOUBLE, double_double);
 133 CREATE_MPI_DATATYPE(MPI_2LONG, long_long);
 134
 135 CREATE_MPI_DATATYPE(MPI_REAL, float);
 136 CREATE_MPI_DATATYPE(MPI_REAL4, float);
 137 CREATE_MPI_DATATYPE(MPI_REAL8, float);
 138 CREATE_MPI_DATATYPE(MPI_REAL16, double);
 139 CREATE_MPI_DATATYPE_NULL(MPI_COMPLEX8);
 140 CREATE_MPI_DATATYPE_NULL(MPI_COMPLEX16);
 141 CREATE_MPI_DATATYPE_NULL(MPI_COMPLEX32);
 142 CREATE_MPI_DATATYPE(MPI_INTEGER1, int);
 143 CREATE_MPI_DATATYPE(MPI_INTEGER2, int16_t);
 144 CREATE_MPI_DATATYPE(MPI_INTEGER4, int32_t);
 145 CREATE_MPI_DATATYPE(MPI_INTEGER8, int64_t);
 146 CREATE_MPI_DATATYPE(MPI_INTEGER16, integer128_t);
 147
 148 CREATE_MPI_DATATYPE(MPI_LONG_DOUBLE_INT, long_double_int);
 149
 150 CREATE_MPI_DATATYPE_NULL(MPI_UB);
 151 CREATE_MPI_DATATYPE_NULL(MPI_LB);
 152 CREATE_MPI_DATATYPE(MPI_PACKED, char);
 153 // Internal use only
 154 CREATE_MPI_DATATYPE(MPI_PTR, void*);
 155
 156 /** Check if the datatype is usable for communications */
 157 int is_datatype_valid(MPI_Datatype datatype) {
 158     return datatype != MPI_DATATYPE_NULL && (datatype->flags & DT_FLAG_COMMITED);
 159 }
 160
 161 size_t smpi_datatype_size(MPI_Datatype datatype)
 162 {
 163   return datatype->size;
 164 }
 165
 166 MPI_Aint smpi_datatype_lb(MPI_Datatype datatype)
 167 {
 168   return datatype->lb;
 169 }
 170
 171 MPI_Aint smpi_datatype_ub(MPI_Datatype datatype)
 172 {
 173   return datatype->ub;
 174 }
 175
 176 int smpi_datatype_dup(MPI_Datatype datatype, MPI_Datatype* new_t)
 177 {
 178   int ret=MPI_SUCCESS;
 179   *new_t= xbt_new(s_smpi_mpi_datatype_t,1);
 180   memcpy(*new_t, datatype, sizeof(s_smpi_mpi_datatype_t));
 181   if (datatype->sizeof_substruct){
 182     (*new_t)->substruct=xbt_malloc(datatype->sizeof_substruct);
 183     memcpy((*new_t)->substruct, datatype->substruct, datatype->sizeof_substruct);
 184   }
 185   if(datatype->name)
 186     (*new_t)->name = xbt_strdup(datatype->name);
 187   if(datatype->attributes !=NULL){
 188       (*new_t)->attributes=xbt_dict_new();
 189       xbt_dict_cursor_t cursor = NULL;
 190       int *key;
 191       int flag;
 192       void* value_in;
 193       void* value_out;
 194       xbt_dict_foreach(datatype->attributes, cursor, key, value_in){
 195         smpi_type_key_elem elem =
 196           static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals,  (const char*)key, sizeof(int)));
 197         if(elem && elem->copy_fn!=MPI_NULL_COPY_FN){
 198           ret = elem->copy_fn(datatype, *key, NULL, value_in, &value_out, &flag );
 199           if(ret!=MPI_SUCCESS){
 200             *new_t=MPI_DATATYPE_NULL;
 201             return ret;
 202           }
 203           if(flag)
 204             xbt_dict_set_ext((*new_t)->attributes, (const char*)key, sizeof(int),value_out, NULL);
 205         }
 206       }
 207     }
 208   return ret;
 209 }
 210
 211 int smpi_datatype_extent(MPI_Datatype datatype, MPI_Aint * lb, MPI_Aint * extent)
 212 {
 213   if(datatype == MPI_DATATYPE_NULL){
 214     *lb=0;
 215     *extent=0;
 216     return MPI_SUCCESS;
 217   }
 218   *lb = datatype->lb;
 219   *extent = datatype->ub - datatype->lb;
 220   return MPI_SUCCESS;
 221 }
 222
 223 MPI_Aint smpi_datatype_get_extent(MPI_Datatype datatype){
 224   if(datatype == MPI_DATATYPE_NULL){
 225     return 0;
 226   }
 227   return datatype->ub - datatype->lb;
 228 }
 229
 230 void smpi_datatype_get_name(MPI_Datatype datatype, char* name, int* length){
 231   *length = strlen(datatype->name);
 232   strcpy(name, datatype->name);
 233 }
 234
 235 void smpi_datatype_set_name(MPI_Datatype datatype, char* name){
 236   datatype->name = xbt_strdup(name);;
 237 }
 238
 239 int smpi_datatype_copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 240                        void *recvbuf, int recvcount, MPI_Datatype recvtype)
 241 {
 242   int count;
 243   if(smpi_privatize_global_variables){
 244     smpi_switch_data_segment(smpi_process_index());
 245   }
 246   /* First check if we really have something to do */
 247   if (recvcount > 0 && recvbuf != sendbuf) {
 248     /* FIXME: treat packed cases */
 249     sendcount *= smpi_datatype_size(sendtype);
 250     recvcount *= smpi_datatype_size(recvtype);
 251     count = sendcount < recvcount ? sendcount : recvcount;
 252
 253     if(sendtype->sizeof_substruct == 0 && recvtype->sizeof_substruct == 0) {
 254       if(!smpi_process_get_replaying()) memcpy(recvbuf, sendbuf, count);
 255     }
 256     else if (sendtype->sizeof_substruct == 0)
 257     {
 258       s_smpi_subtype_t *subtype =  static_cast<s_smpi_subtype_t*>(recvtype->substruct);
 259       subtype->unserialize( sendbuf, recvbuf, recvcount/smpi_datatype_size(recvtype), subtype, MPI_REPLACE);
 260     }
 261     else if (recvtype->sizeof_substruct == 0)
 262     {
 263       s_smpi_subtype_t *subtype =  static_cast<s_smpi_subtype_t*>(sendtype->substruct);
 264       subtype->serialize(sendbuf, recvbuf, sendcount/smpi_datatype_size(sendtype), subtype);
 265     }else{
 266       s_smpi_subtype_t *subtype =  static_cast<s_smpi_subtype_t*>(sendtype->substruct);
 267
 268       void * buf_tmp = xbt_malloc(count);
 269
 270       subtype->serialize( sendbuf, buf_tmp,count/smpi_datatype_size(sendtype), subtype);
 271       subtype =  static_cast<s_smpi_subtype_t*>(recvtype->substruct);
 272       subtype->unserialize( buf_tmp, recvbuf,count/smpi_datatype_size(recvtype), subtype, MPI_REPLACE);
 273
 274       free(buf_tmp);
 275     }
 276   }
 277
 278   return sendcount > recvcount ? MPI_ERR_TRUNCATE : MPI_SUCCESS;
 279 }
 280
 281 /*
 282  *  Copies noncontiguous data into contiguous memory.
 283  *  @param contiguous_vector - output vector
 284  *  @param noncontiguous_vector - input vector
 285  *  @param type - pointer contening :
 286  *      - stride - stride of between noncontiguous data
 287  *      - block_length - the width or height of blocked matrix
 288  *      - count - the number of rows of matrix
 289  */
 290 void serialize_vector( const void *noncontiguous_vector, void *contiguous_vector, int count, void *type)
 291 {
 292   s_smpi_mpi_vector_t* type_c = (s_smpi_mpi_vector_t*)type;
 293   int i;
 294   char* contiguous_vector_char = (char*)contiguous_vector;
 295   char* noncontiguous_vector_char = (char*)noncontiguous_vector;
 296
 297   for (i = 0; i < type_c->block_count * count; i++) {
 298       if (type_c->old_type->sizeof_substruct == 0)
 299         memcpy(contiguous_vector_char, noncontiguous_vector_char, type_c->block_length * type_c->size_oldtype);
 300       else
 301         ((s_smpi_subtype_t*)type_c->old_type->substruct)->serialize( noncontiguous_vector_char,
 302                                                                      contiguous_vector_char,
 303                                                                      type_c->block_length, type_c->old_type->substruct);
 304
 305     contiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 306     if((i+1)%type_c->block_count ==0)
 307       noncontiguous_vector_char += type_c->block_length*smpi_datatype_get_extent(type_c->old_type);
 308     else
 309       noncontiguous_vector_char += type_c->block_stride*smpi_datatype_get_extent(type_c->old_type);
 310   }
 311 }
 312
 313 /*
 314  *  Copies contiguous data into noncontiguous memory.
 315  *  @param noncontiguous_vector - output vector
 316  *  @param contiguous_vector - input vector
 317  *  @param type - pointer contening :
 318  *      - stride - stride of between noncontiguous data
 319  *      - block_length - the width or height of blocked matrix
 320  *      - count - the number of rows of matrix
 321  */
 322 void unserialize_vector( const void *contiguous_vector, void *noncontiguous_vector, int count, void *type, MPI_Op op)
 323 {
 324   s_smpi_mpi_vector_t* type_c = (s_smpi_mpi_vector_t*)type;
 325   int i;
 326
 327   char* contiguous_vector_char = (char*)contiguous_vector;
 328   char* noncontiguous_vector_char = (char*)noncontiguous_vector;
 329
 330   for (i = 0; i < type_c->block_count * count; i++) {
 331     if (type_c->old_type->sizeof_substruct == 0)
 332       smpi_op_apply(op, contiguous_vector_char, noncontiguous_vector_char, &type_c->block_length,
 333           &type_c->old_type);
 334      /* memcpy(noncontiguous_vector_char,
 335              contiguous_vector_char, type_c->block_length * type_c->size_oldtype);*/
 336     else
 337       ((s_smpi_subtype_t*)type_c->old_type->substruct)->unserialize(contiguous_vector_char, noncontiguous_vector_char,
 338                                                                     type_c->block_length,type_c->old_type->substruct,
 339                                                                     op);
 340     contiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 341     if((i+1)%type_c->block_count ==0)
 342       noncontiguous_vector_char += type_c->block_length*smpi_datatype_get_extent(type_c->old_type);
 343     else
 344       noncontiguous_vector_char += type_c->block_stride*smpi_datatype_get_extent(type_c->old_type);
 345   }
 346 }
 347
 348 /* Create a Sub type vector to be able to serialize and unserialize it the structure s_smpi_mpi_vector_t is derived
 349  * from s_smpi_subtype which required the functions unserialize and serialize */
 350 s_smpi_mpi_vector_t* smpi_datatype_vector_create( int block_stride, int block_length, int block_count,
 351                                                   MPI_Datatype old_type, int size_oldtype){
 352   s_smpi_mpi_vector_t *new_t= xbt_new(s_smpi_mpi_vector_t,1);
 353   new_t->base.serialize = &serialize_vector;
 354   new_t->base.unserialize = &unserialize_vector;
 355   new_t->base.subtype_free = &free_vector;
 356   new_t->base.subtype_use = &use_vector;
 357   new_t->block_stride = block_stride;
 358   new_t->block_length = block_length;
 359   new_t->block_count = block_count;
 360   smpi_datatype_use(old_type);
 361   new_t->old_type = old_type;
 362   new_t->size_oldtype = size_oldtype;
 363   return new_t;
 364 }
 365
 366 void smpi_datatype_create(MPI_Datatype* new_type, int size,int lb, int ub, int sizeof_substruct, void *struct_type,
 367                           int flags){
 368   MPI_Datatype new_t= xbt_new(s_smpi_mpi_datatype_t,1);
 369   new_t->name = NULL;
 370   new_t->size = size;
 371   new_t->sizeof_substruct = size>0? sizeof_substruct:0;
 372   new_t->lb = lb;
 373   new_t->ub = ub;
 374   new_t->flags = flags;
 375   new_t->substruct = struct_type;
 376   new_t->in_use=1;
 377   new_t->attributes=NULL;
 378   *new_type = new_t;
 379
 380 #if HAVE_MC
 381   if(MC_is_active())
 382     MC_ignore(&(new_t->in_use), sizeof(new_t->in_use));
 383 #endif
 384 }
 385
 386 void smpi_datatype_free(MPI_Datatype* type){
 387   xbt_assert((*type)->in_use >= 0);
 388   if((*type)->attributes !=NULL){
 389     xbt_dict_cursor_t cursor = NULL;
 390     int* key;
 391     void * value;
 392     int flag;
 393     xbt_dict_foreach((*type)->attributes, cursor, key, value){
 394       smpi_type_key_elem elem =
 395           static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals, (const char*)key, sizeof(int)));
 396       if(elem &&  elem->delete_fn)
 397         elem->delete_fn(*type,*key, value, &flag);
 398     }
 399   }
 400
 401   if((*type)->flags & DT_FLAG_PREDEFINED)return;
 402
 403   //if still used, mark for deletion
 404   if((*type)->in_use!=0){
 405       (*type)->flags |=DT_FLAG_DESTROYED;
 406       return;
 407   }
 408
 409   if ((*type)->sizeof_substruct != 0){
 410     //((s_smpi_subtype_t *)(*type)->substruct)->subtype_free(type);
 411     xbt_free((*type)->substruct);
 412   }
 413   if ((*type)->name != NULL){
 414     xbt_free((*type)->name);
 415   }
 416   *type = MPI_DATATYPE_NULL;
 417 }
 418
 419 void smpi_datatype_use(MPI_Datatype type){
 420
 421   if(type)type->in_use++;
 422
 423   if(type->sizeof_substruct!=0){
 424     ((s_smpi_subtype_t *)(type)->substruct)->subtype_use(&type);
 425   }
 426 #if HAVE_MC
 427   if(MC_is_active())
 428     MC_ignore(&(type->in_use), sizeof(type->in_use));
 429 #endif
 430 }
 431
 432 void smpi_datatype_unuse(MPI_Datatype type){
 433   if (type->in_use > 0)
 434     type->in_use--;
 435
 436   if(type->sizeof_substruct!=0){
 437     ((s_smpi_subtype_t *)(type)->substruct)->subtype_free(&type);
 438   }
 439
 440   if(type && type->in_use == 0){
 441     MPI_Datatype t = type;
 442     if (!(type->flags & DT_FLAG_DESTROYED))
 443       smpi_datatype_free(&type);
 444     if(t->flags & DT_FLAG_PREDEFINED) return;
 445     xbt_free(t);
 446   }
 447 #if HAVE_MC
 448   if(MC_is_active())
 449     MC_ignore(&(type->in_use), sizeof(type->in_use));
 450 #endif
 451 }
 452
 453 /*Contiguous Implementation*/
 454
 455 /* Copies noncontiguous data into contiguous memory.
 456  *  @param contiguous_hvector - output hvector
 457  *  @param noncontiguous_hvector - input hvector
 458  *  @param type - pointer contening :
 459  *      - stride - stride of between noncontiguous data, in bytes
 460  *      - block_length - the width or height of blocked matrix
 461  *      - count - the number of rows of matrix
 462  */
 463 void serialize_contiguous( const void *noncontiguous_hvector, void *contiguous_hvector, int count, void *type)
 464 {
 465   s_smpi_mpi_contiguous_t* type_c = (s_smpi_mpi_contiguous_t*)type;
 466   char* contiguous_vector_char = (char*)contiguous_hvector;
 467   char* noncontiguous_vector_char = (char*)noncontiguous_hvector+type_c->lb;
 468   memcpy(contiguous_vector_char, noncontiguous_vector_char, count* type_c->block_count * type_c->size_oldtype);
 469 }
 470 /* Copies contiguous data into noncontiguous memory.
 471  *  @param noncontiguous_vector - output hvector
 472  *  @param contiguous_vector - input hvector
 473  *  @param type - pointer contening :
 474  *      - stride - stride of between noncontiguous data, in bytes
 475  *      - block_length - the width or height of blocked matrix
 476  *      - count - the number of rows of matrix
 477  */
 478 void unserialize_contiguous(const void *contiguous_vector, void *noncontiguous_vector, int count, void *type, MPI_Op op)
 479 {
 480   s_smpi_mpi_contiguous_t* type_c = (s_smpi_mpi_contiguous_t*)type;
 481   char* contiguous_vector_char = (char*)contiguous_vector;
 482   char* noncontiguous_vector_char = (char*)noncontiguous_vector+type_c->lb;
 483   int n= count* type_c->block_count;
 484   smpi_op_apply(op, contiguous_vector_char, noncontiguous_vector_char, &n, &type_c->old_type);
 485   /*memcpy(noncontiguous_vector_char, contiguous_vector_char, count*  type_c->block_count * type_c->size_oldtype);*/
 486 }
 487
 488 void free_contiguous(MPI_Datatype* d){
 489   smpi_datatype_unuse(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 490 }
 491
 492 void use_contiguous(MPI_Datatype* d){
 493   smpi_datatype_use(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 494 }
 495
 496 /* Create a Sub type contiguous to be able to serialize and unserialize it the structure s_smpi_mpi_contiguous_t is
 497  * erived from s_smpi_subtype which required the functions unserialize and serialize */
 498 s_smpi_mpi_contiguous_t* smpi_datatype_contiguous_create( MPI_Aint lb, int block_count, MPI_Datatype old_type,
 499                                                   int size_oldtype){
 500   s_smpi_mpi_contiguous_t *new_t= xbt_new(s_smpi_mpi_contiguous_t,1);
 501   new_t->base.serialize = &serialize_contiguous;
 502   new_t->base.unserialize = &unserialize_contiguous;
 503   new_t->base.subtype_free = &free_contiguous;
 504   new_t->base.subtype_use = &use_contiguous;
 505   new_t->lb = lb;
 506   new_t->block_count = block_count;
 507   new_t->old_type = old_type;
 508   smpi_datatype_use(old_type);
 509   new_t->size_oldtype = size_oldtype;
 510   smpi_datatype_use(old_type);
 511   return new_t;
 512 }
 513
 514 int smpi_datatype_contiguous(int count, MPI_Datatype old_type, MPI_Datatype* new_type, MPI_Aint lb)
 515 {
 516   int retval;
 517   if(old_type->sizeof_substruct){
 518     //handle this case as a hvector with stride equals to the extent of the datatype
 519     return smpi_datatype_hvector(count, 1, smpi_datatype_get_extent(old_type), old_type, new_type);
 520   }
 521
 522   s_smpi_mpi_contiguous_t* subtype = smpi_datatype_contiguous_create( lb, count, old_type,smpi_datatype_size(old_type));
 523
 524   smpi_datatype_create(new_type, count * smpi_datatype_size(old_type),lb,lb + count * smpi_datatype_size(old_type),
 525             sizeof(s_smpi_mpi_contiguous_t),subtype, DT_FLAG_CONTIGUOUS);
 526   retval=MPI_SUCCESS;
 527   return retval;
 528 }
 529
 530 int smpi_datatype_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type)
 531 {
 532   int retval;
 533   if (blocklen<0) return MPI_ERR_ARG;
 534   MPI_Aint lb = 0;
 535   MPI_Aint ub = 0;
 536   if(count>0){
 537     lb=smpi_datatype_lb(old_type);
 538     ub=((count-1)*stride+blocklen-1)*smpi_datatype_get_extent(old_type)+smpi_datatype_ub(old_type);
 539   }
 540   if(old_type->sizeof_substruct || stride != blocklen){
 541
 542     s_smpi_mpi_vector_t* subtype = smpi_datatype_vector_create(stride, blocklen, count, old_type,
 543                                                                 smpi_datatype_size(old_type));
 544     smpi_datatype_create(new_type, count * (blocklen) * smpi_datatype_size(old_type), lb, ub, sizeof(s_smpi_mpi_vector_t), subtype,
 545                          DT_FLAG_VECTOR);
 546     retval=MPI_SUCCESS;
 547   }else{
 548     /* in this situation the data are contignous thus it's not required to serialize and unserialize it*/
 549     smpi_datatype_create(new_type, count * blocklen * smpi_datatype_size(old_type), 0, ((count -1) * stride + blocklen)*
 550                          smpi_datatype_size(old_type), 0, NULL, DT_FLAG_VECTOR|DT_FLAG_CONTIGUOUS);
 551     retval=MPI_SUCCESS;
 552   }
 553   return retval;
 554 }
 555
 556 void free_vector(MPI_Datatype* d){
 557   smpi_datatype_unuse(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 558 }
 559
 560 void use_vector(MPI_Datatype* d){
 561   smpi_datatype_use(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 562 }
 563
 564 /* Hvector Implementation - Vector with stride in bytes */
 565
 566 /* Copies noncontiguous data into contiguous memory.
 567  *  @param contiguous_hvector - output hvector
 568  *  @param noncontiguous_hvector - input hvector
 569  *  @param type - pointer contening :
 570  *      - stride - stride of between noncontiguous data, in bytes
 571  *      - block_length - the width or height of blocked matrix
 572  *      - count - the number of rows of matrix
 573  */
 574 void serialize_hvector( const void *noncontiguous_hvector, void *contiguous_hvector, int count, void *type)
 575 {
 576   s_smpi_mpi_hvector_t* type_c = (s_smpi_mpi_hvector_t*)type;
 577   int i;
 578   char* contiguous_vector_char = (char*)contiguous_hvector;
 579   char* noncontiguous_vector_char = (char*)noncontiguous_hvector;
 580
 581   for (i = 0; i < type_c->block_count * count; i++) {
 582     if (type_c->old_type->sizeof_substruct == 0)
 583       memcpy(contiguous_vector_char, noncontiguous_vector_char, type_c->block_length * type_c->size_oldtype);
 584     else
 585       ((s_smpi_subtype_t*)type_c->old_type->substruct)->serialize( noncontiguous_vector_char,
 586                                                                    contiguous_vector_char,
 587                                                                    type_c->block_length, type_c->old_type->substruct);
 588
 589     contiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 590     if((i+1)%type_c->block_count ==0)
 591       noncontiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 592     else
 593       noncontiguous_vector_char += type_c->block_stride;
 594   }
 595 }
 596 /* Copies contiguous data into noncontiguous memory.
 597  *  @param noncontiguous_vector - output hvector
 598  *  @param contiguous_vector - input hvector
 599  *  @param type - pointer contening :
 600  *      - stride - stride of between noncontiguous data, in bytes
 601  *      - block_length - the width or height of blocked matrix
 602  *      - count - the number of rows of matrix
 603  */
 604 void unserialize_hvector( const void *contiguous_vector, void *noncontiguous_vector, int count, void *type, MPI_Op op)
 605 {
 606   s_smpi_mpi_hvector_t* type_c = (s_smpi_mpi_hvector_t*)type;
 607   int i;
 608
 609   char* contiguous_vector_char = (char*)contiguous_vector;
 610   char* noncontiguous_vector_char = (char*)noncontiguous_vector;
 611
 612   for (i = 0; i < type_c->block_count * count; i++) {
 613     if (type_c->old_type->sizeof_substruct == 0)
 614       smpi_op_apply(op, contiguous_vector_char, noncontiguous_vector_char, &type_c->block_length, &type_c->old_type);
 615              /*memcpy(noncontiguous_vector_char,
 616            contiguous_vector_char, type_c->block_length * type_c->size_oldtype);*/
 617     else
 618       ((s_smpi_subtype_t*)type_c->old_type->substruct)->unserialize( contiguous_vector_char, noncontiguous_vector_char,
 619                                                                      type_c->block_length, type_c->old_type->substruct,
 620                                                                      op);
 621     contiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 622     if((i+1)%type_c->block_count ==0)
 623       noncontiguous_vector_char += type_c->block_length*type_c->size_oldtype;
 624     else
 625       noncontiguous_vector_char += type_c->block_stride;
 626   }
 627 }
 628
 629 /* Create a Sub type vector to be able to serialize and unserialize it the structure s_smpi_mpi_vector_t is derived
 630  * from s_smpi_subtype which required the functions unserialize and serialize
 631  *
 632  */
 633 s_smpi_mpi_hvector_t* smpi_datatype_hvector_create( MPI_Aint block_stride, int block_length, int block_count,
 634                                                   MPI_Datatype old_type, int size_oldtype){
 635   s_smpi_mpi_hvector_t *new_t= xbt_new(s_smpi_mpi_hvector_t,1);
 636   new_t->base.serialize = &serialize_hvector;
 637   new_t->base.unserialize = &unserialize_hvector;
 638   new_t->base.subtype_free = &free_hvector;
 639   new_t->base.subtype_use = &use_hvector;
 640   new_t->block_stride = block_stride;
 641   new_t->block_length = block_length;
 642   new_t->block_count = block_count;
 643   new_t->old_type = old_type;
 644   new_t->size_oldtype = size_oldtype;
 645   smpi_datatype_use(old_type);
 646   return new_t;
 647 }
 648
 649 //do nothing for vector types
 650 void free_hvector(MPI_Datatype* d){
 651   smpi_datatype_unuse(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 652 }
 653
 654 void use_hvector(MPI_Datatype* d){
 655   smpi_datatype_use(((s_smpi_mpi_indexed_t *)(*d)->substruct)->old_type);
 656 }
 657
 658 int smpi_datatype_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type)
 659 {
 660   int retval;
 661   if (blocklen<0) return MPI_ERR_ARG;
 662   MPI_Aint lb = 0;
 663   MPI_Aint ub = 0;
 664   if(count>0){
 665     lb=smpi_datatype_lb(old_type);
 666     ub=((count-1)*stride)+(blocklen-1)*smpi_datatype_get_extent(old_type)+smpi_datatype_ub(old_type);
 667   }
 668   if(old_type->sizeof_substruct || stride != blocklen*smpi_datatype_get_extent(old_type)){
 669     s_smpi_mpi_hvector_t* subtype = smpi_datatype_hvector_create( stride, blocklen, count, old_type,
 670                                                                   smpi_datatype_size(old_type));
 671
 672     smpi_datatype_create(new_type, count * blocklen * smpi_datatype_size(old_type), lb,ub, sizeof(s_smpi_mpi_hvector_t), subtype, DT_FLAG_VECTOR);
 673     retval=MPI_SUCCESS;
 674   }else{
 675     smpi_datatype_create(new_type, count * blocklen * smpi_datatype_size(old_type),0,count * blocklen *
 676                                              smpi_datatype_size(old_type), 0, NULL, DT_FLAG_VECTOR|DT_FLAG_CONTIGUOUS);
 677     retval=MPI_SUCCESS;
 678   }
 679   return retval;
 680 }
 681
 682 /* Indexed Implementation */
 683
 684 /* Copies noncontiguous data into contiguous memory.
 685  *  @param contiguous_indexed - output indexed
 686  *  @param noncontiguous_indexed - input indexed
 687  *  @param type - pointer contening :
 688  *      - block_lengths - the width or height of blocked matrix
 689  *      - block_indices - indices of each data, in element
 690  *      - count - the number of rows of matrix
 691  */
 692 void serialize_indexed( const void *noncontiguous_indexed, void *contiguous_indexed, int count, void *type)
 693 {
 694   s_smpi_mpi_indexed_t* type_c = (s_smpi_mpi_indexed_t*)type;
 695   int i,j;
 696   char* contiguous_indexed_char = (char*)contiguous_indexed;
 697   char* noncontiguous_indexed_char = (char*)noncontiguous_indexed+type_c->block_indices[0] * type_c->size_oldtype;
 698   for(j=0; j<count;j++){
 699     for (i = 0; i < type_c->block_count; i++) {
 700       if (type_c->old_type->sizeof_substruct == 0)
 701         memcpy(contiguous_indexed_char, noncontiguous_indexed_char, type_c->block_lengths[i] * type_c->size_oldtype);
 702       else
 703         ((s_smpi_subtype_t*)type_c->old_type->substruct)->serialize( noncontiguous_indexed_char,
 704                                                                      contiguous_indexed_char,
 705                                                                      type_c->block_lengths[i],
 706                                                                      type_c->old_type->substruct);
 707
 708       contiguous_indexed_char += type_c->block_lengths[i]*type_c->size_oldtype;
 709       if (i<type_c->block_count-1)
 710         noncontiguous_indexed_char =
 711           (char*)noncontiguous_indexed + type_c->block_indices[i+1]*smpi_datatype_get_extent(type_c->old_type);
 712       else
 713         noncontiguous_indexed_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_type);
 714     }
 715     noncontiguous_indexed=(void*)noncontiguous_indexed_char;
 716   }
 717 }
 718 /* Copies contiguous data into noncontiguous memory.
 719  *  @param noncontiguous_indexed - output indexed
 720  *  @param contiguous_indexed - input indexed
 721  *  @param type - pointer contening :
 722  *      - block_lengths - the width or height of blocked matrix
 723  *      - block_indices - indices of each data, in element
 724  *      - count - the number of rows of matrix
 725  */
 726 void unserialize_indexed( const void *contiguous_indexed, void *noncontiguous_indexed, int count, void *type, MPI_Op op)
 727 {
 728   s_smpi_mpi_indexed_t* type_c = (s_smpi_mpi_indexed_t*)type;
 729   int i,j;
 730   char* contiguous_indexed_char = (char*)contiguous_indexed;
 731   char* noncontiguous_indexed_char =
 732     (char*)noncontiguous_indexed+type_c->block_indices[0]*smpi_datatype_get_extent(type_c->old_type);
 733   for(j=0; j<count;j++){
 734     for (i = 0; i < type_c->block_count; i++) {
 735       if (type_c->old_type->sizeof_substruct == 0)
 736         smpi_op_apply(op, contiguous_indexed_char, noncontiguous_indexed_char, &type_c->block_lengths[i],
 737                     &type_c->old_type);
 738                /*memcpy(noncontiguous_indexed_char ,
 739              contiguous_indexed_char, type_c->block_lengths[i] * type_c->size_oldtype);*/
 740       else
 741         ((s_smpi_subtype_t*)type_c->old_type->substruct)->unserialize( contiguous_indexed_char,
 742                                                                        noncontiguous_indexed_char,
 743                                                                        type_c->block_lengths[i],
 744                                                                        type_c->old_type->substruct, op);
 745
 746       contiguous_indexed_char += type_c->block_lengths[i]*type_c->size_oldtype;
 747       if (i<type_c->block_count-1)
 748         noncontiguous_indexed_char =
 749           (char*)noncontiguous_indexed + type_c->block_indices[i+1]*smpi_datatype_get_extent(type_c->old_type);
 750       else
 751         noncontiguous_indexed_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_type);
 752     }
 753     noncontiguous_indexed=(void*)noncontiguous_indexed_char;
 754   }
 755 }
 756
 757 void free_indexed(MPI_Datatype* type){
 758   if((*type)->in_use==0){
 759     xbt_free(((s_smpi_mpi_indexed_t *)(*type)->substruct)->block_lengths);
 760     xbt_free(((s_smpi_mpi_indexed_t *)(*type)->substruct)->block_indices);
 761   }
 762   smpi_datatype_unuse(((s_smpi_mpi_indexed_t *)(*type)->substruct)->old_type);
 763 }
 764
 765 void use_indexed(MPI_Datatype* type){
 766   smpi_datatype_use(((s_smpi_mpi_indexed_t *)(*type)->substruct)->old_type);
 767 }
 768
 769
 770 /* Create a Sub type indexed to be able to serialize and unserialize it the structure s_smpi_mpi_indexed_t is derived
 771  * from s_smpi_subtype which required the functions unserialize and serialize */
 772 s_smpi_mpi_indexed_t* smpi_datatype_indexed_create( int* block_lengths, int* block_indices, int block_count,
 773                                                   MPI_Datatype old_type, int size_oldtype){
 774   s_smpi_mpi_indexed_t *new_t= xbt_new(s_smpi_mpi_indexed_t,1);
 775   new_t->base.serialize = &serialize_indexed;
 776   new_t->base.unserialize = &unserialize_indexed;
 777   new_t->base.subtype_free = &free_indexed;
 778   new_t->base.subtype_use = &use_indexed;
 779  //TODO : add a custom function for each time to clean these
 780   new_t->block_lengths= xbt_new(int, block_count);
 781   new_t->block_indices= xbt_new(int, block_count);
 782   int i;
 783   for(i=0;i<block_count;i++){
 784     new_t->block_lengths[i]=block_lengths[i];
 785     new_t->block_indices[i]=block_indices[i];
 786   }
 787   new_t->block_count = block_count;
 788   smpi_datatype_use(old_type);
 789   new_t->old_type = old_type;
 790   new_t->size_oldtype = size_oldtype;
 791   return new_t;
 792 }
 793
 794 int smpi_datatype_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type)
 795 {
 796   int i;
 797   int retval;
 798   int size = 0;
 799   int contiguous=1;
 800   MPI_Aint lb = 0;
 801   MPI_Aint ub = 0;
 802   if(count>0){
 803     lb=indices[0]*smpi_datatype_get_extent(old_type);
 804     ub=indices[0]*smpi_datatype_get_extent(old_type) + blocklens[0]*smpi_datatype_ub(old_type);
 805   }
 806
 807   for(i=0; i< count; i++){
 808     if   (blocklens[i]<0)
 809       return MPI_ERR_ARG;
 810     size += blocklens[i];
 811
 812     if(indices[i]*smpi_datatype_get_extent(old_type)+smpi_datatype_lb(old_type)<lb)
 813       lb = indices[i]*smpi_datatype_get_extent(old_type)+smpi_datatype_lb(old_type);
 814     if(indices[i]*smpi_datatype_get_extent(old_type)+blocklens[i]*smpi_datatype_ub(old_type)>ub)
 815       ub = indices[i]*smpi_datatype_get_extent(old_type)+blocklens[i]*smpi_datatype_ub(old_type);
 816
 817     if ( (i< count -1) && (indices[i]+blocklens[i] != indices[i+1]) )contiguous=0;
 818   }
 819   if (old_type->sizeof_substruct != 0)
 820     contiguous=0;
 821
 822   if(!contiguous){
 823     s_smpi_mpi_indexed_t* subtype = smpi_datatype_indexed_create( blocklens, indices, count, old_type,
 824                                                                   smpi_datatype_size(old_type));
 825      smpi_datatype_create(new_type,  size * smpi_datatype_size(old_type),lb,ub,sizeof(s_smpi_mpi_indexed_t), subtype, DT_FLAG_DATA);
 826   }else{
 827     s_smpi_mpi_contiguous_t* subtype = smpi_datatype_contiguous_create( lb, size, old_type,
 828                                                                   smpi_datatype_size(old_type));
 829     smpi_datatype_create(new_type, size * smpi_datatype_size(old_type), lb, ub, sizeof(s_smpi_mpi_contiguous_t), subtype,
 830                          DT_FLAG_DATA|DT_FLAG_CONTIGUOUS);
 831   }
 832   retval=MPI_SUCCESS;
 833   return retval;
 834 }
 835 /* Hindexed Implementation - Indexed with indices in bytes */
 836
 837 /* Copies noncontiguous data into contiguous memory.
 838  *  @param contiguous_hindexed - output hindexed
 839  *  @param noncontiguous_hindexed - input hindexed
 840  *  @param type - pointer contening :
 841  *      - block_lengths - the width or height of blocked matrix
 842  *      - block_indices - indices of each data, in bytes
 843  *      - count - the number of rows of matrix
 844  */
 845 void serialize_hindexed( const void *noncontiguous_hindexed, void *contiguous_hindexed, int count, void *type)
 846 {
 847   s_smpi_mpi_hindexed_t* type_c = (s_smpi_mpi_hindexed_t*)type;
 848   int i,j;
 849   char* contiguous_hindexed_char = (char*)contiguous_hindexed;
 850   char* noncontiguous_hindexed_char = (char*)noncontiguous_hindexed+ type_c->block_indices[0];
 851   for(j=0; j<count;j++){
 852     for (i = 0; i < type_c->block_count; i++) {
 853       if (type_c->old_type->sizeof_substruct == 0)
 854         memcpy(contiguous_hindexed_char, noncontiguous_hindexed_char, type_c->block_lengths[i] * type_c->size_oldtype);
 855       else
 856         ((s_smpi_subtype_t*)type_c->old_type->substruct)->serialize( noncontiguous_hindexed_char,
 857                                                                      contiguous_hindexed_char,
 858                                                                      type_c->block_lengths[i],
 859                                                                      type_c->old_type->substruct);
 860
 861       contiguous_hindexed_char += type_c->block_lengths[i]*type_c->size_oldtype;
 862       if (i<type_c->block_count-1)
 863         noncontiguous_hindexed_char = (char*)noncontiguous_hindexed + type_c->block_indices[i+1];
 864       else
 865         noncontiguous_hindexed_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_type);
 866     }
 867     noncontiguous_hindexed=(void*)noncontiguous_hindexed_char;
 868   }
 869 }
 870 /* Copies contiguous data into noncontiguous memory.
 871  *  @param noncontiguous_hindexed - output hindexed
 872  *  @param contiguous_hindexed - input hindexed
 873  *  @param type - pointer contening :
 874  *      - block_lengths - the width or height of blocked matrix
 875  *      - block_indices - indices of each data, in bytes
 876  *      - count - the number of rows of matrix
 877  */
 878 void unserialize_hindexed( const void *contiguous_hindexed, void *noncontiguous_hindexed, int count, void *type,
 879                          MPI_Op op)
 880 {
 881   s_smpi_mpi_hindexed_t* type_c = (s_smpi_mpi_hindexed_t*)type;
 882   int i,j;
 883
 884   char* contiguous_hindexed_char = (char*)contiguous_hindexed;
 885   char* noncontiguous_hindexed_char = (char*)noncontiguous_hindexed+ type_c->block_indices[0];
 886   for(j=0; j<count;j++){
 887     for (i = 0; i < type_c->block_count; i++) {
 888       if (type_c->old_type->sizeof_substruct == 0)
 889         smpi_op_apply(op, contiguous_hindexed_char, noncontiguous_hindexed_char, &type_c->block_lengths[i],
 890                             &type_c->old_type);
 891         /*memcpy(noncontiguous_hindexed_char,contiguous_hindexed_char,type_c->block_lengths[i]*type_c->size_oldtype);*/
 892       else
 893         ((s_smpi_subtype_t*)type_c->old_type->substruct)->unserialize( contiguous_hindexed_char,
 894                                                                        noncontiguous_hindexed_char,
 895                                                                        type_c->block_lengths[i],
 896                                                                        type_c->old_type->substruct, op);
 897
 898       contiguous_hindexed_char += type_c->block_lengths[i]*type_c->size_oldtype;
 899       if (i<type_c->block_count-1)
 900         noncontiguous_hindexed_char = (char*)noncontiguous_hindexed + type_c->block_indices[i+1];
 901       else
 902         noncontiguous_hindexed_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_type);
 903     }
 904     noncontiguous_hindexed=(void*)noncontiguous_hindexed_char;
 905   }
 906 }
 907
 908 void free_hindexed(MPI_Datatype* type){
 909   if((*type)->in_use==0){
 910     xbt_free(((s_smpi_mpi_hindexed_t *)(*type)->substruct)->block_lengths);
 911     xbt_free(((s_smpi_mpi_hindexed_t *)(*type)->substruct)->block_indices);
 912   }
 913   smpi_datatype_unuse(((s_smpi_mpi_indexed_t *)(*type)->substruct)->old_type);
 914 }
 915
 916 void use_hindexed(MPI_Datatype* type){
 917   smpi_datatype_use(((s_smpi_mpi_indexed_t *)(*type)->substruct)->old_type);
 918 }
 919
 920 /* Create a Sub type hindexed to be able to serialize and unserialize it the structure s_smpi_mpi_hindexed_t is derived
 921  * from s_smpi_subtype which required the functions unserialize and serialize
 922  */
 923 s_smpi_mpi_hindexed_t* smpi_datatype_hindexed_create( int* block_lengths, MPI_Aint* block_indices, int block_count,
 924                                                   MPI_Datatype old_type, int size_oldtype){
 925   s_smpi_mpi_hindexed_t *new_t= xbt_new(s_smpi_mpi_hindexed_t,1);
 926   new_t->base.serialize = &serialize_hindexed;
 927   new_t->base.unserialize = &unserialize_hindexed;
 928   new_t->base.subtype_free = &free_hindexed;
 929   new_t->base.subtype_use = &use_hindexed;
 930  //TODO : add a custom function for each time to clean these
 931   new_t->block_lengths= xbt_new(int, block_count);
 932   new_t->block_indices= xbt_new(MPI_Aint, block_count);
 933   int i;
 934   for(i=0;i<block_count;i++){
 935     new_t->block_lengths[i]=block_lengths[i];
 936     new_t->block_indices[i]=block_indices[i];
 937   }
 938   new_t->block_count = block_count;
 939   new_t->old_type = old_type;
 940   smpi_datatype_use(old_type);
 941   new_t->size_oldtype = size_oldtype;
 942   return new_t;
 943 }
 944
 945 int smpi_datatype_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type)
 946 {
 947   int i;
 948   int retval;
 949   int size = 0;
 950   int contiguous=1;
 951   MPI_Aint lb = 0;
 952   MPI_Aint ub = 0;
 953   if(count>0){
 954     lb=indices[0] + smpi_datatype_lb(old_type);
 955     ub=indices[0] + blocklens[0]*smpi_datatype_ub(old_type);
 956   }
 957   for(i=0; i< count; i++){
 958     if   (blocklens[i]<0)
 959       return MPI_ERR_ARG;
 960     size += blocklens[i];
 961
 962     if(indices[i]+smpi_datatype_lb(old_type)<lb) lb = indices[i]+smpi_datatype_lb(old_type);
 963     if(indices[i]+blocklens[i]*smpi_datatype_ub(old_type)>ub) ub = indices[i]+blocklens[i]*smpi_datatype_ub(old_type);
 964
 965     if ( (i< count -1) && (indices[i]+blocklens[i]*static_cast<int>(smpi_datatype_size(old_type)) != indices[i+1]) )
 966       contiguous=0;
 967   }
 968   if (old_type->sizeof_substruct != 0 || lb!=0)
 969     contiguous=0;
 970
 971   if(!contiguous){
 972     s_smpi_mpi_hindexed_t* subtype = smpi_datatype_hindexed_create( blocklens, indices, count, old_type,
 973                                                                   smpi_datatype_size(old_type));
 974     smpi_datatype_create(new_type,  size * smpi_datatype_size(old_type), lb, ub ,sizeof(s_smpi_mpi_hindexed_t), subtype, DT_FLAG_DATA);
 975   }else{
 976     s_smpi_mpi_contiguous_t* subtype = smpi_datatype_contiguous_create(lb,size, old_type, smpi_datatype_size(old_type));
 977     smpi_datatype_create(new_type,  size * smpi_datatype_size(old_type), 0,size * smpi_datatype_size(old_type),
 978                1, subtype, DT_FLAG_DATA|DT_FLAG_CONTIGUOUS);
 979   }
 980   retval=MPI_SUCCESS;
 981   return retval;
 982 }
 983
 984 /* struct Implementation - Indexed with indices in bytes */
 985
 986 /* Copies noncontiguous data into contiguous memory.
 987  *  @param contiguous_struct - output struct
 988  *  @param noncontiguous_struct - input struct
 989  *  @param type - pointer contening :
 990  *      - stride - stride of between noncontiguous data
 991  *      - block_length - the width or height of blocked matrix
 992  *      - count - the number of rows of matrix
 993  */
 994 void serialize_struct( const void *noncontiguous_struct, void *contiguous_struct, int count, void *type)
 995 {
 996   s_smpi_mpi_struct_t* type_c = (s_smpi_mpi_struct_t*)type;
 997   int i,j;
 998   char* contiguous_struct_char = (char*)contiguous_struct;
 999   char* noncontiguous_struct_char = (char*)noncontiguous_struct+ type_c->block_indices[0];
1000   for(j=0; j<count;j++){
1001     for (i = 0; i < type_c->block_count; i++) {
1002       if (type_c->old_types[i]->sizeof_substruct == 0)
1003         memcpy(contiguous_struct_char, noncontiguous_struct_char,
1004                type_c->block_lengths[i] * smpi_datatype_size(type_c->old_types[i]));
1005       else
1006         ((s_smpi_subtype_t*)type_c->old_types[i]->substruct)->serialize( noncontiguous_struct_char,
1007                                                                          contiguous_struct_char,
1008                                                                          type_c->block_lengths[i],
1009                                                                          type_c->old_types[i]->substruct);
1010
1011
1012       contiguous_struct_char += type_c->block_lengths[i]*smpi_datatype_size(type_c->old_types[i]);
1013       if (i<type_c->block_count-1)
1014         noncontiguous_struct_char = (char*)noncontiguous_struct + type_c->block_indices[i+1];
1015       else //let's hope this is MPI_UB ?
1016         noncontiguous_struct_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_types[i]);
1017     }
1018     noncontiguous_struct=(void*)noncontiguous_struct_char;
1019   }
1020 }
1021
1022 /* Copies contiguous data into noncontiguous memory.
1023  *  @param noncontiguous_struct - output struct
1024  *  @param contiguous_struct - input struct
1025  *  @param type - pointer contening :
1026  *      - stride - stride of between noncontiguous data
1027  *      - block_length - the width or height of blocked matrix
1028  *      - count - the number of rows of matrix
1029  */
1030 void unserialize_struct( const void *contiguous_struct, void *noncontiguous_struct, int count, void *type, MPI_Op op)
1031 {
1032   s_smpi_mpi_struct_t* type_c = (s_smpi_mpi_struct_t*)type;
1033   int i,j;
1034
1035   char* contiguous_struct_char = (char*)contiguous_struct;
1036   char* noncontiguous_struct_char = (char*)noncontiguous_struct+ type_c->block_indices[0];
1037   for(j=0; j<count;j++){
1038     for (i = 0; i < type_c->block_count; i++) {
1039       if (type_c->old_types[i]->sizeof_substruct == 0)
1040         smpi_op_apply(op, contiguous_struct_char, noncontiguous_struct_char, &type_c->block_lengths[i],
1041            & type_c->old_types[i]);
1042       /*memcpy(noncontiguous_struct_char,
1043                contiguous_struct_char, type_c->block_lengths[i] * smpi_datatype_size(type_c->old_types[i]));*/
1044       else
1045         ((s_smpi_subtype_t*)type_c->old_types[i]->substruct)->unserialize( contiguous_struct_char,
1046                                                                            noncontiguous_struct_char,
1047                                                                            type_c->block_lengths[i],
1048                                                                            type_c->old_types[i]->substruct, op);
1049
1050       contiguous_struct_char += type_c->block_lengths[i]*smpi_datatype_size(type_c->old_types[i]);
1051       if (i<type_c->block_count-1)
1052         noncontiguous_struct_char =  (char*)noncontiguous_struct + type_c->block_indices[i+1];
1053       else
1054         noncontiguous_struct_char += type_c->block_lengths[i]*smpi_datatype_get_extent(type_c->old_types[i]);
1055     }
1056     noncontiguous_struct=(void*)noncontiguous_struct_char;
1057   }
1058 }
1059
1060 void free_struct(MPI_Datatype* type){
1061   int i=0;
1062   for (i = 0; i < ((s_smpi_mpi_struct_t *)(*type)->substruct)->block_count; i++)
1063     smpi_datatype_unuse(((s_smpi_mpi_struct_t *)(*type)->substruct)->old_types[i]);
1064   if((*type)->in_use==0){
1065     xbt_free(((s_smpi_mpi_struct_t *)(*type)->substruct)->block_lengths);
1066     xbt_free(((s_smpi_mpi_struct_t *)(*type)->substruct)->block_indices);
1067     xbt_free(((s_smpi_mpi_struct_t *)(*type)->substruct)->old_types);
1068   }
1069 }
1070
1071 void use_struct(MPI_Datatype* type){
1072   int i=0;
1073   for (i = 0; i < ((s_smpi_mpi_struct_t *)(*type)->substruct)->block_count; i++)
1074     smpi_datatype_use(((s_smpi_mpi_struct_t *)(*type)->substruct)->old_types[i]);
1075 }
1076
1077 /* Create a Sub type struct to be able to serialize and unserialize it the structure s_smpi_mpi_struct_t is derived
1078  * from s_smpi_subtype which required the functions unserialize and serialize
1079  */
1080 s_smpi_mpi_struct_t* smpi_datatype_struct_create( int* block_lengths, MPI_Aint* block_indices, int block_count,
1081                                                   MPI_Datatype* old_types){
1082   s_smpi_mpi_struct_t *new_t= xbt_new(s_smpi_mpi_struct_t,1);
1083   new_t->base.serialize = &serialize_struct;
1084   new_t->base.unserialize = &unserialize_struct;
1085   new_t->base.subtype_free = &free_struct;
1086   new_t->base.subtype_use = &use_struct;
1087  //TODO : add a custom function for each time to clean these
1088   new_t->block_lengths= xbt_new(int, block_count);
1089   new_t->block_indices= xbt_new(MPI_Aint, block_count);
1090   new_t->old_types=  xbt_new(MPI_Datatype, block_count);
1091   int i;
1092   for(i=0;i<block_count;i++){
1093     new_t->block_lengths[i]=block_lengths[i];
1094     new_t->block_indices[i]=block_indices[i];
1095     new_t->old_types[i]=old_types[i];
1096     smpi_datatype_use(new_t->old_types[i]);
1097   }
1098   //new_t->block_lengths = block_lengths;
1099   //new_t->block_indices = block_indices;
1100   new_t->block_count = block_count;
1101   //new_t->old_types = old_types;
1102   return new_t;
1103 }
1104
1105 int smpi_datatype_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type)
1106 {
1107   int i;
1108   size_t size = 0;
1109   int contiguous=1;
1110   size = 0;
1111   MPI_Aint lb = 0;
1112   MPI_Aint ub = 0;
1113   if(count>0){
1114     lb=indices[0] + smpi_datatype_lb(old_types[0]);
1115     ub=indices[0] + blocklens[0]*smpi_datatype_ub(old_types[0]);
1116   }
1117   int forced_lb=0;
1118   int forced_ub=0;
1119   for(i=0; i< count; i++){
1120     if (blocklens[i]<0)
1121       return MPI_ERR_ARG;
1122     if (old_types[i]->sizeof_substruct != 0)
1123       contiguous=0;
1124
1125     size += blocklens[i]*smpi_datatype_size(old_types[i]);
1126     if (old_types[i]==MPI_LB){
1127       lb=indices[i];
1128       forced_lb=1;
1129     }
1130     if (old_types[i]==MPI_UB){
1131       ub=indices[i];
1132       forced_ub=1;
1133     }
1134
1135     if(!forced_lb && indices[i]+smpi_datatype_lb(old_types[i])<lb) lb = indices[i];
1136     if(!forced_ub &&  indices[i]+blocklens[i]*smpi_datatype_ub(old_types[i])>ub)
1137       ub = indices[i]+blocklens[i]*smpi_datatype_ub(old_types[i]);
1138
1139     if ( (i< count -1) && (indices[i]+blocklens[i]*static_cast<int>(smpi_datatype_size(old_types[i])) != indices[i+1]) )
1140       contiguous=0;
1141   }
1142
1143   if(!contiguous){
1144     s_smpi_mpi_struct_t* subtype = smpi_datatype_struct_create( blocklens, indices, count, old_types);
1145
1146     smpi_datatype_create(new_type,  size, lb, ub,sizeof(s_smpi_mpi_struct_t), subtype, DT_FLAG_DATA);
1147   }else{
1148     s_smpi_mpi_contiguous_t* subtype = smpi_datatype_contiguous_create( lb, size, MPI_CHAR, 1);
1149     smpi_datatype_create(new_type,  size, lb, ub,1, subtype, DT_FLAG_DATA|DT_FLAG_CONTIGUOUS);
1150   }
1151   return MPI_SUCCESS;
1152 }
1153
1154 void smpi_datatype_commit(MPI_Datatype *datatype)
1155 {
1156   (*datatype)->flags=  ((*datatype)->flags | DT_FLAG_COMMITED);
1157 }
1158
1159 typedef struct s_smpi_mpi_op {
1160   MPI_User_function *func;
1161   int is_commute;
1162 } s_smpi_mpi_op_t;
1163
1164 #define MAX_OP(a, b)  (b) = (a) < (b) ? (b) : (a)
1165 #define MIN_OP(a, b)  (b) = (a) < (b) ? (a) : (b)
1166 #define SUM_OP(a, b)  (b) += (a)
1167 #define PROD_OP(a, b) (b) *= (a)
1168 #define LAND_OP(a, b) (b) = (a) && (b)
1169 #define LOR_OP(a, b)  (b) = (a) || (b)
1170 #define LXOR_OP(a, b) (b) = (!(a) && (b)) || ((a) && !(b))
1171 #define BAND_OP(a, b) (b) &= (a)
1172 #define BOR_OP(a, b)  (b) |= (a)
1173 #define BXOR_OP(a, b) (b) ^= (a)
1174 #define MAXLOC_OP(a, b)  (b) = (a.value) < (b.value) ? (b) : (a)
1175 #define MINLOC_OP(a, b)  (b) = (a.value) < (b.value) ? (a) : (b)
1176
1177 #define APPLY_FUNC(a, b, length, type, func) \
1178 {                                          \
1179   int i;                                   \
1180   type* x = (type*)(a);                    \
1181   type* y = (type*)(b);                    \
1182   for(i = 0; i < *(length); i++) {         \
1183     func(x[i], y[i]);                      \
1184   }                                        \
1185 }
1186
1187 static void max_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1188 {
1189   if (*datatype == MPI_CHAR) {
1190     APPLY_FUNC(a, b, length, char, MAX_OP);
1191   } else if (*datatype == MPI_SHORT) {
1192     APPLY_FUNC(a, b, length, short, MAX_OP);
1193   } else if (*datatype == MPI_INT) {
1194     APPLY_FUNC(a, b, length, int, MAX_OP);
1195   } else if (*datatype == MPI_LONG) {
1196     APPLY_FUNC(a, b, length, long, MAX_OP);
1197   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1198     APPLY_FUNC(a, b, length, unsigned short, MAX_OP);
1199   } else if (*datatype == MPI_UNSIGNED) {
1200     APPLY_FUNC(a, b, length, unsigned int, MAX_OP);
1201   } else if (*datatype == MPI_UNSIGNED_LONG) {
1202     APPLY_FUNC(a, b, length, unsigned long, MAX_OP);
1203   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1204     APPLY_FUNC(a, b, length, unsigned char, MAX_OP);
1205   } else if (*datatype == MPI_FLOAT) {
1206     APPLY_FUNC(a, b, length, float, MAX_OP);
1207   } else if (*datatype == MPI_DOUBLE) {
1208     APPLY_FUNC(a, b, length, double, MAX_OP);
1209   } else if (*datatype == MPI_LONG_DOUBLE) {
1210     APPLY_FUNC(a, b, length, long double, MAX_OP);
1211   }
1212 }
1213
1214 static void min_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1215 {
1216   if (*datatype == MPI_CHAR) {
1217     APPLY_FUNC(a, b, length, char, MIN_OP);
1218   } else if (*datatype == MPI_SHORT) {
1219     APPLY_FUNC(a, b, length, short, MIN_OP);
1220   } else if (*datatype == MPI_INT) {
1221     APPLY_FUNC(a, b, length, int, MIN_OP);
1222   } else if (*datatype == MPI_LONG) {
1223     APPLY_FUNC(a, b, length, long, MIN_OP);
1224   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1225     APPLY_FUNC(a, b, length, unsigned short, MIN_OP);
1226   } else if (*datatype == MPI_UNSIGNED) {
1227     APPLY_FUNC(a, b, length, unsigned int, MIN_OP);
1228   } else if (*datatype == MPI_UNSIGNED_LONG) {
1229     APPLY_FUNC(a, b, length, unsigned long, MIN_OP);
1230   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1231     APPLY_FUNC(a, b, length, unsigned char, MIN_OP);
1232   } else if (*datatype == MPI_FLOAT) {
1233     APPLY_FUNC(a, b, length, float, MIN_OP);
1234   } else if (*datatype == MPI_DOUBLE) {
1235     APPLY_FUNC(a, b, length, double, MIN_OP);
1236   } else if (*datatype == MPI_LONG_DOUBLE) {
1237     APPLY_FUNC(a, b, length, long double, MIN_OP);
1238   }
1239 }
1240
1241 static void sum_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1242 {
1243   if (*datatype == MPI_CHAR) {
1244     APPLY_FUNC(a, b, length, char, SUM_OP);
1245   } else if (*datatype == MPI_SHORT) {
1246     APPLY_FUNC(a, b, length, short, SUM_OP);
1247   } else if (*datatype == MPI_INT) {
1248     APPLY_FUNC(a, b, length, int, SUM_OP);
1249   } else if (*datatype == MPI_LONG) {
1250     APPLY_FUNC(a, b, length, long, SUM_OP);
1251   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1252     APPLY_FUNC(a, b, length, unsigned short, SUM_OP);
1253   } else if (*datatype == MPI_UNSIGNED) {
1254     APPLY_FUNC(a, b, length, unsigned int, SUM_OP);
1255   } else if (*datatype == MPI_UNSIGNED_LONG) {
1256     APPLY_FUNC(a, b, length, unsigned long, SUM_OP);
1257   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1258     APPLY_FUNC(a, b, length, unsigned char, SUM_OP);
1259   } else if (*datatype == MPI_FLOAT) {
1260     APPLY_FUNC(a, b, length, float, SUM_OP);
1261   } else if (*datatype == MPI_DOUBLE) {
1262     APPLY_FUNC(a, b, length, double, SUM_OP);
1263   } else if (*datatype == MPI_LONG_DOUBLE) {
1264     APPLY_FUNC(a, b, length, long double, SUM_OP);
1265   } else if (*datatype == MPI_C_FLOAT_COMPLEX) {
1266     APPLY_FUNC(a, b, length, float _Complex, SUM_OP);
1267   } else if (*datatype == MPI_C_DOUBLE_COMPLEX) {
1268     APPLY_FUNC(a, b, length, double _Complex, SUM_OP);
1269   } else if (*datatype == MPI_C_LONG_DOUBLE_COMPLEX) {
1270     APPLY_FUNC(a, b, length, long double _Complex, SUM_OP);
1271   }
1272 }
1273
1274 static void prod_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1275 {
1276   if (*datatype == MPI_CHAR) {
1277     APPLY_FUNC(a, b, length, char, PROD_OP);
1278   } else if (*datatype == MPI_SHORT) {
1279     APPLY_FUNC(a, b, length, short, PROD_OP);
1280   } else if (*datatype == MPI_INT) {
1281     APPLY_FUNC(a, b, length, int, PROD_OP);
1282   } else if (*datatype == MPI_LONG) {
1283     APPLY_FUNC(a, b, length, long, PROD_OP);
1284   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1285     APPLY_FUNC(a, b, length, unsigned short, PROD_OP);
1286   } else if (*datatype == MPI_UNSIGNED) {
1287     APPLY_FUNC(a, b, length, unsigned int, PROD_OP);
1288   } else if (*datatype == MPI_UNSIGNED_LONG) {
1289     APPLY_FUNC(a, b, length, unsigned long, PROD_OP);
1290   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1291     APPLY_FUNC(a, b, length, unsigned char, PROD_OP);
1292   } else if (*datatype == MPI_FLOAT) {
1293     APPLY_FUNC(a, b, length, float, PROD_OP);
1294   } else if (*datatype == MPI_DOUBLE) {
1295     APPLY_FUNC(a, b, length, double, PROD_OP);
1296   } else if (*datatype == MPI_LONG_DOUBLE) {
1297     APPLY_FUNC(a, b, length, long double, PROD_OP);
1298   } else if (*datatype == MPI_C_FLOAT_COMPLEX) {
1299     APPLY_FUNC(a, b, length, float _Complex, PROD_OP);
1300   } else if (*datatype == MPI_C_DOUBLE_COMPLEX) {
1301     APPLY_FUNC(a, b, length, double _Complex, PROD_OP);
1302   } else if (*datatype == MPI_C_LONG_DOUBLE_COMPLEX) {
1303     APPLY_FUNC(a, b, length, long double _Complex, PROD_OP);
1304   }
1305 }
1306
1307 static void land_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1308 {
1309   if (*datatype == MPI_CHAR) {
1310     APPLY_FUNC(a, b, length, char, LAND_OP);
1311   } else if (*datatype == MPI_SHORT) {
1312     APPLY_FUNC(a, b, length, short, LAND_OP);
1313   } else if (*datatype == MPI_INT) {
1314     APPLY_FUNC(a, b, length, int, LAND_OP);
1315   } else if (*datatype == MPI_LONG) {
1316     APPLY_FUNC(a, b, length, long, LAND_OP);
1317   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1318     APPLY_FUNC(a, b, length, unsigned short, LAND_OP);
1319   } else if (*datatype == MPI_UNSIGNED) {
1320     APPLY_FUNC(a, b, length, unsigned int, LAND_OP);
1321   } else if (*datatype == MPI_UNSIGNED_LONG) {
1322     APPLY_FUNC(a, b, length, unsigned long, LAND_OP);
1323   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1324     APPLY_FUNC(a, b, length, unsigned char, LAND_OP);
1325   } else if (*datatype == MPI_C_BOOL) {
1326     APPLY_FUNC(a, b, length, bool, LAND_OP);
1327   }
1328 }
1329
1330 static void lor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1331 {
1332   if (*datatype == MPI_CHAR) {
1333     APPLY_FUNC(a, b, length, char, LOR_OP);
1334   } else if (*datatype == MPI_SHORT) {
1335     APPLY_FUNC(a, b, length, short, LOR_OP);
1336   } else if (*datatype == MPI_INT) {
1337     APPLY_FUNC(a, b, length, int, LOR_OP);
1338   } else if (*datatype == MPI_LONG) {
1339     APPLY_FUNC(a, b, length, long, LOR_OP);
1340   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1341     APPLY_FUNC(a, b, length, unsigned short, LOR_OP);
1342   } else if (*datatype == MPI_UNSIGNED) {
1343     APPLY_FUNC(a, b, length, unsigned int, LOR_OP);
1344   } else if (*datatype == MPI_UNSIGNED_LONG) {
1345     APPLY_FUNC(a, b, length, unsigned long, LOR_OP);
1346   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1347     APPLY_FUNC(a, b, length, unsigned char, LOR_OP);
1348   } else if (*datatype == MPI_C_BOOL) {
1349     APPLY_FUNC(a, b, length, bool, LOR_OP);
1350   }
1351 }
1352
1353 static void lxor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1354 {
1355   if (*datatype == MPI_CHAR) {
1356     APPLY_FUNC(a, b, length, char, LXOR_OP);
1357   } else if (*datatype == MPI_SHORT) {
1358     APPLY_FUNC(a, b, length, short, LXOR_OP);
1359   } else if (*datatype == MPI_INT) {
1360     APPLY_FUNC(a, b, length, int, LXOR_OP);
1361   } else if (*datatype == MPI_LONG) {
1362     APPLY_FUNC(a, b, length, long, LXOR_OP);
1363   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1364     APPLY_FUNC(a, b, length, unsigned short, LXOR_OP);
1365   } else if (*datatype == MPI_UNSIGNED) {
1366     APPLY_FUNC(a, b, length, unsigned int, LXOR_OP);
1367   } else if (*datatype == MPI_UNSIGNED_LONG) {
1368     APPLY_FUNC(a, b, length, unsigned long, LXOR_OP);
1369   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1370     APPLY_FUNC(a, b, length, unsigned char, LXOR_OP);
1371   } else if (*datatype == MPI_C_BOOL) {
1372     APPLY_FUNC(a, b, length, bool, LXOR_OP);
1373   }
1374 }
1375
1376 static void band_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1377 {
1378   if (*datatype == MPI_CHAR) {
1379     APPLY_FUNC(a, b, length, char, BAND_OP);
1380   }else if (*datatype == MPI_SHORT) {
1381     APPLY_FUNC(a, b, length, short, BAND_OP);
1382   } else if (*datatype == MPI_INT) {
1383     APPLY_FUNC(a, b, length, int, BAND_OP);
1384   } else if (*datatype == MPI_LONG) {
1385     APPLY_FUNC(a, b, length, long, BAND_OP);
1386   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1387     APPLY_FUNC(a, b, length, unsigned short, BAND_OP);
1388   } else if (*datatype == MPI_UNSIGNED) {
1389     APPLY_FUNC(a, b, length, unsigned int, BAND_OP);
1390   } else if (*datatype == MPI_UNSIGNED_LONG) {
1391     APPLY_FUNC(a, b, length, unsigned long, BAND_OP);
1392   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1393     APPLY_FUNC(a, b, length, unsigned char, BAND_OP);
1394   } else if (*datatype == MPI_BYTE) {
1395     APPLY_FUNC(a, b, length, uint8_t, BAND_OP);
1396   }
1397 }
1398
1399 static void bor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1400 {
1401   if (*datatype == MPI_CHAR) {
1402     APPLY_FUNC(a, b, length, char, BOR_OP);
1403   } else if (*datatype == MPI_SHORT) {
1404     APPLY_FUNC(a, b, length, short, BOR_OP);
1405   } else if (*datatype == MPI_INT) {
1406     APPLY_FUNC(a, b, length, int, BOR_OP);
1407   } else if (*datatype == MPI_LONG) {
1408     APPLY_FUNC(a, b, length, long, BOR_OP);
1409   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1410     APPLY_FUNC(a, b, length, unsigned short, BOR_OP);
1411   } else if (*datatype == MPI_UNSIGNED) {
1412     APPLY_FUNC(a, b, length, unsigned int, BOR_OP);
1413   } else if (*datatype == MPI_UNSIGNED_LONG) {
1414     APPLY_FUNC(a, b, length, unsigned long, BOR_OP);
1415   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1416     APPLY_FUNC(a, b, length, unsigned char, BOR_OP);
1417   } else if (*datatype == MPI_BYTE) {
1418     APPLY_FUNC(a, b, length, uint8_t, BOR_OP);
1419   }
1420 }
1421
1422 static void bxor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1423 {
1424   if (*datatype == MPI_CHAR) {
1425     APPLY_FUNC(a, b, length, char, BXOR_OP);
1426   } else if (*datatype == MPI_SHORT) {
1427     APPLY_FUNC(a, b, length, short, BXOR_OP);
1428   } else if (*datatype == MPI_INT) {
1429     APPLY_FUNC(a, b, length, int, BXOR_OP);
1430   } else if (*datatype == MPI_LONG) {
1431     APPLY_FUNC(a, b, length, long, BXOR_OP);
1432   } else if (*datatype == MPI_UNSIGNED_SHORT) {
1433     APPLY_FUNC(a, b, length, unsigned short, BXOR_OP);
1434   } else if (*datatype == MPI_UNSIGNED) {
1435     APPLY_FUNC(a, b, length, unsigned int, BXOR_OP);
1436   } else if (*datatype == MPI_UNSIGNED_LONG) {
1437     APPLY_FUNC(a, b, length, unsigned long, BXOR_OP);
1438   } else if (*datatype == MPI_UNSIGNED_CHAR) {
1439     APPLY_FUNC(a, b, length, unsigned char, BXOR_OP);
1440   } else if (*datatype == MPI_BYTE) {
1441     APPLY_FUNC(a, b, length, uint8_t, BXOR_OP);
1442   }
1443 }
1444
1445 static void minloc_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1446 {
1447   if (*datatype == MPI_FLOAT_INT) {
1448     APPLY_FUNC(a, b, length, float_int, MINLOC_OP);
1449   } else if (*datatype == MPI_LONG_INT) {
1450     APPLY_FUNC(a, b, length, long_int, MINLOC_OP);
1451   } else if (*datatype == MPI_DOUBLE_INT) {
1452     APPLY_FUNC(a, b, length, double_int, MINLOC_OP);
1453   } else if (*datatype == MPI_SHORT_INT) {
1454     APPLY_FUNC(a, b, length, short_int, MINLOC_OP);
1455   } else if (*datatype == MPI_2LONG) {
1456     APPLY_FUNC(a, b, length, long_long, MINLOC_OP);
1457   } else if (*datatype == MPI_2INT) {
1458     APPLY_FUNC(a, b, length, int_int, MINLOC_OP);
1459   } else if (*datatype == MPI_LONG_DOUBLE_INT) {
1460     APPLY_FUNC(a, b, length, long_double_int, MINLOC_OP);
1461   } else if (*datatype == MPI_2FLOAT) {
1462     APPLY_FUNC(a, b, length, float_float, MINLOC_OP);
1463   } else if (*datatype == MPI_2DOUBLE) {
1464     APPLY_FUNC(a, b, length, double_double, MINLOC_OP);
1465   }
1466 }
1467
1468 static void maxloc_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1469 {
1470   if (*datatype == MPI_FLOAT_INT) {
1471     APPLY_FUNC(a, b, length, float_int, MAXLOC_OP);
1472   } else if (*datatype == MPI_LONG_INT) {
1473     APPLY_FUNC(a, b, length, long_int, MAXLOC_OP);
1474   } else if (*datatype == MPI_DOUBLE_INT) {
1475     APPLY_FUNC(a, b, length, double_int, MAXLOC_OP);
1476   } else if (*datatype == MPI_SHORT_INT) {
1477     APPLY_FUNC(a, b, length, short_int, MAXLOC_OP);
1478   } else if (*datatype == MPI_2LONG) {
1479     APPLY_FUNC(a, b, length, long_long, MAXLOC_OP);
1480   } else if (*datatype == MPI_2INT) {
1481     APPLY_FUNC(a, b, length, int_int, MAXLOC_OP);
1482   } else if (*datatype == MPI_LONG_DOUBLE_INT) {
1483     APPLY_FUNC(a, b, length, long_double_int, MAXLOC_OP);
1484   } else if (*datatype == MPI_2FLOAT) {
1485     APPLY_FUNC(a, b, length, float_float, MAXLOC_OP);
1486   } else if (*datatype == MPI_2DOUBLE) {
1487     APPLY_FUNC(a, b, length, double_double, MAXLOC_OP);
1488   }
1489 }
1490
1491 static void replace_func(void *a, void *b, int *length, MPI_Datatype * datatype)
1492 {
1493   memcpy(b, a, *length * smpi_datatype_size(*datatype));
1494 }
1495
1496 #define CREATE_MPI_OP(name, func)                             \
1497   static s_smpi_mpi_op_t mpi_##name = { &(func) /* func */, true }; \
1498 MPI_Op name = &mpi_##name;
1499
1500 CREATE_MPI_OP(MPI_MAX, max_func);
1501 CREATE_MPI_OP(MPI_MIN, min_func);
1502 CREATE_MPI_OP(MPI_SUM, sum_func);
1503 CREATE_MPI_OP(MPI_PROD, prod_func);
1504 CREATE_MPI_OP(MPI_LAND, land_func);
1505 CREATE_MPI_OP(MPI_LOR, lor_func);
1506 CREATE_MPI_OP(MPI_LXOR, lxor_func);
1507 CREATE_MPI_OP(MPI_BAND, band_func);
1508 CREATE_MPI_OP(MPI_BOR, bor_func);
1509 CREATE_MPI_OP(MPI_BXOR, bxor_func);
1510 CREATE_MPI_OP(MPI_MAXLOC, maxloc_func);
1511 CREATE_MPI_OP(MPI_MINLOC, minloc_func);
1512 CREATE_MPI_OP(MPI_REPLACE, replace_func);
1513
1514 MPI_Op smpi_op_new(MPI_User_function * function, int commute)
1515 {
1516   MPI_Op op;
1517   op = xbt_new(s_smpi_mpi_op_t, 1);
1518   op->func = function;
1519   op-> is_commute = commute;
1520   return op;
1521 }
1522
1523 int smpi_op_is_commute(MPI_Op op)
1524 {
1525   return (op==MPI_OP_NULL) ? 1 : op-> is_commute;
1526 }
1527
1528 void smpi_op_destroy(MPI_Op op)
1529 {
1530   xbt_free(op);
1531 }
1532
1533 void smpi_op_apply(MPI_Op op, void *invec, void *inoutvec, int *len, MPI_Datatype * datatype)
1534 {
1535   if(op==MPI_OP_NULL)
1536     return;
1537
1538   if(smpi_privatize_global_variables){//we need to switch as the called function may silently touch global variables
1539     XBT_DEBUG("Applying operation, switch to the right data frame ");
1540     smpi_switch_data_segment(smpi_process_index());
1541   }
1542
1543   if(!smpi_process_get_replaying())
1544   op->func(invec, inoutvec, len, datatype);
1545 }
1546
1547 int smpi_type_attr_delete(MPI_Datatype type, int keyval){
1548   smpi_type_key_elem elem =
1549     static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals, (const char*)&keyval, sizeof(int)));
1550   if(!elem)
1551     return MPI_ERR_ARG;
1552   if(elem->delete_fn!=MPI_NULL_DELETE_FN){
1553     void * value;
1554     int flag;
1555     if(smpi_type_attr_get(type, keyval, &value, &flag)==MPI_SUCCESS){
1556       int ret = elem->delete_fn(type, keyval, value, &flag);
1557       if(ret!=MPI_SUCCESS) return ret;
1558     }
1559   }
1560   if(type->attributes==NULL)
1561     return MPI_ERR_ARG;
1562
1563   xbt_dict_remove_ext(type->attributes, (const char*)&keyval, sizeof(int));
1564   return MPI_SUCCESS;
1565 }
1566
1567 int smpi_type_attr_get(MPI_Datatype type, int keyval, void* attr_value, int* flag){
1568   smpi_type_key_elem elem =
1569     static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals, (const char*)&keyval, sizeof(int)));
1570   if(!elem)
1571     return MPI_ERR_ARG;
1572   xbt_ex_t ex;
1573   if(type->attributes==NULL){
1574     *flag=0;
1575     return MPI_SUCCESS;
1576   }
1577   TRY {
1578     *(void**)attr_value = xbt_dict_get_ext(type->attributes, (const char*)&keyval, sizeof(int));
1579     *flag=1;
1580   }
1581   CATCH(ex) {
1582     *flag=0;
1583     xbt_ex_free(ex);
1584   }
1585   return MPI_SUCCESS;
1586 }
1587
1588 int smpi_type_attr_put(MPI_Datatype type, int keyval, void* attr_value){
1589   if(!smpi_type_keyvals)
1590   smpi_type_keyvals = xbt_dict_new();
1591   smpi_type_key_elem elem =
1592      static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals, (const char*)&keyval, sizeof(int)));
1593   if(!elem )
1594     return MPI_ERR_ARG;
1595   int flag;
1596   void* value;
1597   smpi_type_attr_get(type, keyval, &value, &flag);
1598   if(flag && elem->delete_fn!=MPI_NULL_DELETE_FN){
1599     int ret = elem->delete_fn(type, keyval, value, &flag);
1600     if(ret!=MPI_SUCCESS) return ret;
1601   }
1602   if(type->attributes==NULL)
1603     type->attributes=xbt_dict_new();
1604
1605   xbt_dict_set_ext(type->attributes, (const char*)&keyval, sizeof(int), attr_value, NULL);
1606   return MPI_SUCCESS;
1607 }
1608
1609 int smpi_type_keyval_create(MPI_Type_copy_attr_function* copy_fn, MPI_Type_delete_attr_function* delete_fn, int* keyval,
1610                             void* extra_state){
1611   if(!smpi_type_keyvals)
1612   smpi_type_keyvals = xbt_dict_new();
1613
1614   smpi_type_key_elem value = (smpi_type_key_elem) xbt_new0(s_smpi_mpi_type_key_elem_t,1);
1615
1616   value->copy_fn=copy_fn;
1617   value->delete_fn=delete_fn;
1618
1619   *keyval = type_keyval_id;
1620   xbt_dict_set_ext(smpi_type_keyvals,(const char*)keyval, sizeof(int),(void*)value, NULL);
1621   type_keyval_id++;
1622   return MPI_SUCCESS;
1623 }
1624
1625 int smpi_type_keyval_free(int* keyval){
1626   smpi_type_key_elem elem =
1627     static_cast<smpi_type_key_elem>(xbt_dict_get_or_null_ext(smpi_type_keyvals, (const char*)keyval, sizeof(int)));
1628   if(!elem){
1629     return MPI_ERR_ARG;
1630   }
1631   xbt_dict_remove_ext(smpi_type_keyvals, (const char*)keyval, sizeof(int));
1632   xbt_free(elem);
1633   return MPI_SUCCESS;
1634 }
1635
1636 int smpi_mpi_pack(void* inbuf, int incount, MPI_Datatype type, void* outbuf, int outcount, int* position,MPI_Comm comm){
1637   size_t size = smpi_datatype_size(type);
1638   if (outcount - *position < incount*static_cast<int>(size))
1639     return MPI_ERR_BUFFER;
1640   smpi_datatype_copy(inbuf, incount, type, (char*)outbuf + *position, outcount, MPI_CHAR);
1641   *position += incount * size;
1642   return MPI_SUCCESS;
1643 }
1644
1645 int smpi_mpi_unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Datatype type,MPI_Comm comm){
1646   int size = static_cast<int>(smpi_datatype_size(type));
1647   if (outcount*size> insize)
1648     return MPI_ERR_BUFFER;
1649   smpi_datatype_copy((char*)inbuf + *position, insize, MPI_CHAR, outbuf, outcount, type);
1650   *position += outcount * size;
1651   return MPI_SUCCESS;
1652 }