src/smpi/smpi_datatype.hpp

   1 /* Copyright (c) 2009-2010, 2012-2014. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #ifndef SMPI_DATATYPE_HPP
   8 #define SMPI_DATATYPE_HPP
   9
  10 #include <xbt/base.h>
  11
  12 #include "private.h"
  13
  14 #define DT_FLAG_DESTROYED     0x0001  /**< user destroyed but some other layers still have a reference */
  15 #define DT_FLAG_COMMITED      0x0002  /**< ready to be used for a send/recv operation */
  16 #define DT_FLAG_CONTIGUOUS    0x0004  /**< contiguous datatype */
  17 #define DT_FLAG_OVERLAP       0x0008  /**< datatype is unpropper for a recv operation */
  18 #define DT_FLAG_USER_LB       0x0010  /**< has a user defined LB */
  19 #define DT_FLAG_USER_UB       0x0020  /**< has a user defined UB */
  20 #define DT_FLAG_PREDEFINED    0x0040  /**< cannot be removed: initial and predefined datatypes */
  21 #define DT_FLAG_NO_GAPS       0x0080  /**< no gaps around the datatype */
  22 #define DT_FLAG_DATA          0x0100  /**< data or control structure */
  23 #define DT_FLAG_ONE_SIDED     0x0200  /**< datatype can be used for one sided operations */
  24 #define DT_FLAG_UNAVAILABLE   0x0400  /**< datatypes unavailable on the build (OS or compiler dependant) */
  25 #define DT_FLAG_DERIVED       0x0800  /**< is the datatype derived ? */
  26 /*
  27  * We should make the difference here between the predefined contiguous and non contiguous
  28  * datatypes. The DT_FLAG_BASIC is held by all predefined contiguous datatypes.
  29  */
  30 #define DT_FLAG_BASIC      (DT_FLAG_PREDEFINED | DT_FLAG_CONTIGUOUS | DT_FLAG_NO_GAPS | DT_FLAG_DATA | DT_FLAG_COMMITED)
  31
  32 extern const MPI_Datatype MPI_PTR;
  33
  34 //The following are datatypes for the MPI functions MPI_MAXLOC and MPI_MINLOC.
  35 typedef struct {
  36   float value;
  37   int index;
  38 } float_int;
  39 typedef struct {
  40   float value;
  41   float index;
  42 } float_float;
  43 typedef struct {
  44   long value;
  45   long index;
  46 } long_long;
  47 typedef struct {
  48   double value;
  49   double index;
  50 } double_double;
  51 typedef struct {
  52   long value;
  53   int index;
  54 } long_int;
  55 typedef struct {
  56   double value;
  57   int index;
  58 } double_int;
  59 typedef struct {
  60   short value;
  61   int index;
  62 } short_int;
  63 typedef struct {
  64   int value;
  65   int index;
  66 } int_int;
  67 typedef struct {
  68   long double value;
  69   int index;
  70 } long_double_int;
  71 typedef struct {
  72   int64_t value;
  73   int64_t index;
  74 } integer128_t;
  75
  76
  77 namespace simgrid{
  78 namespace smpi{
  79
  80 class Datatype{
  81   protected:
  82     char* name_;
  83     size_t size_;
  84     MPI_Aint lb_;
  85     MPI_Aint ub_;
  86     int flags_;
  87     xbt_dict_t attributes_;
  88     int in_use_;
  89
  90   public:
  91     Datatype(int size,MPI_Aint lb, MPI_Aint ub, int flags);
  92     Datatype(char* name, int size,MPI_Aint lb, MPI_Aint ub, int flags);
  93     Datatype(Datatype *datatype, int* ret);
  94     virtual ~Datatype();
  95     void use();
  96     void unuse();
  97     void commit();
  98     bool is_valid();
  99     size_t size();
 100     int flags();
 101     void addflag(int flag);
 102     MPI_Aint lb();
 103     MPI_Aint ub();
 104     int extent(MPI_Aint * lb, MPI_Aint * extent);
 105     MPI_Aint get_extent();
 106     char* name();
 107     void get_name(char* name, int* length);
 108     void set_name(char* name);
 109     static int copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 110                     void *recvbuf, int recvcount, MPI_Datatype recvtype);
 111     virtual void serialize( void* noncontiguous, void *contiguous,
 112                             int count);
 113     virtual void unserialize( void* contiguous, void *noncontiguous,
 114                               int count, MPI_Op op);
 115     int attr_delete(int keyval);
 116     int attr_get(int keyval, void* attr_value, int* flag);
 117     int attr_put(int keyval, void* attr_value);
 118     static int keyval_create(MPI_Type_copy_attr_function* copy_fn, MPI_Type_delete_attr_function* delete_fn, int* keyval, void* extra_state);
 119     static int keyval_free(int* keyval);
 120     int pack(void* inbuf, int incount, void* outbuf, int outcount, int* position, MPI_Comm comm);
 121     int unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Comm comm);
 122
 123
 124     static int create_contiguous(int count, MPI_Datatype old_type, MPI_Aint lb, MPI_Datatype* new_type);
 125     static int create_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type);
 126     static int create_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type);
 127     static int create_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type);
 128     static int create_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type);
 129     static int create_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type);
 130 };
 131
 132 class Type_Contiguous: public Datatype{
 133   private:
 134     int block_count_;
 135     MPI_Datatype old_type_;
 136   public:
 137     Type_Contiguous(int size, MPI_Aint lb, MPI_Aint ub, int flags, int block_count, MPI_Datatype old_type);
 138     ~Type_Contiguous();
 139     void use();
 140     void serialize( void* noncontiguous, void *contiguous,
 141                             int count);
 142     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 143                               int count, MPI_Op op);
 144 };
 145
 146 class Type_Vector: public Datatype{
 147   private:
 148     int block_count_;
 149     int block_length_;
 150     int block_stride_;
 151     MPI_Datatype old_type_;
 152   public:
 153     Type_Vector(int size,MPI_Aint lb, MPI_Aint ub, int flags, int count, int blocklen, int stride, MPI_Datatype old_type);
 154     ~Type_Vector();
 155     void use();
 156     void serialize( void* noncontiguous, void *contiguous,
 157                             int count);
 158     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 159                               int count, MPI_Op op);
 160 };
 161
 162 class Type_Hvector: public Datatype{
 163   private:
 164     int block_count_;
 165     int block_length_;
 166     MPI_Aint block_stride_;
 167     MPI_Datatype old_type_;
 168   public:
 169     Type_Hvector(int size,MPI_Aint lb, MPI_Aint ub, int flags, int block_count, int block_length, MPI_Aint block_stride, MPI_Datatype old_type);
 170     ~Type_Hvector();
 171     void use();
 172     void serialize( void* noncontiguous, void *contiguous,
 173                             int count);
 174     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 175                               int count, MPI_Op op);
 176 };
 177
 178 class Type_Indexed: public Datatype{
 179   private:
 180     int block_count_;
 181     int* block_lengths_;
 182     int* block_indices_;
 183     MPI_Datatype old_type_;
 184   public:
 185     Type_Indexed(int size,MPI_Aint lb, MPI_Aint ub, int flags, int block_count, int* block_lengths, int* block_indices, MPI_Datatype old_type);
 186     ~Type_Indexed();
 187     void use();
 188     void serialize( void* noncontiguous, void *contiguous,
 189                             int count);
 190     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 191                               int count, MPI_Op op);
 192 };
 193
 194 class Type_Hindexed: public Datatype{
 195   private:
 196     int block_count_;
 197     int* block_lengths_;
 198     MPI_Aint* block_indices_;
 199     MPI_Datatype old_type_;
 200   public:
 201     Type_Hindexed(int size,MPI_Aint lb, MPI_Aint ub, int flags, int block_count, int* block_lengths, MPI_Aint* block_indices, MPI_Datatype old_type);
 202     ~Type_Hindexed();
 203     void use();
 204     void serialize( void* noncontiguous, void *contiguous,
 205                             int count);
 206     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 207                               int count, MPI_Op op);
 208 };
 209
 210 class Type_Struct: public Datatype{
 211   private:
 212     int block_count_;
 213     int* block_lengths_;
 214     MPI_Aint* block_indices_;
 215     MPI_Datatype* old_types_;
 216   public:
 217     Type_Struct(int size,MPI_Aint lb, MPI_Aint ub, int flags, int block_count, int* block_lengths, MPI_Aint* block_indices, MPI_Datatype* old_types);
 218     ~Type_Struct();
 219     void use();
 220     void serialize( void* noncontiguous, void *contiguous,
 221                             int count);
 222     void unserialize( void* contiguous_vector, void *noncontiguous_vector,
 223                               int count, MPI_Op op);
 224 };
 225
 226
 227 }
 228 }
 229
 230 #endif