src/smpi/smpi_datatype.hpp

   1 /* Copyright (c) 2009-2017. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef SMPI_DATATYPE_HPP
   7 #define SMPI_DATATYPE_HPP
   8
   9 #include "src/smpi/smpi_f2c.hpp"
  10 #include "src/smpi/smpi_keyvals.hpp"
  11
  12 #define DT_FLAG_DESTROYED     0x0001  /**< user destroyed but some other layers still have a reference */
  13 #define DT_FLAG_COMMITED      0x0002  /**< ready to be used for a send/recv operation */
  14 #define DT_FLAG_CONTIGUOUS    0x0004  /**< contiguous datatype */
  15 #define DT_FLAG_OVERLAP       0x0008  /**< datatype is unpropper for a recv operation */
  16 #define DT_FLAG_USER_LB       0x0010  /**< has a user defined LB */
  17 #define DT_FLAG_USER_UB       0x0020  /**< has a user defined UB */
  18 #define DT_FLAG_PREDEFINED    0x0040  /**< cannot be removed: initial and predefined datatypes */
  19 #define DT_FLAG_NO_GAPS       0x0080  /**< no gaps around the datatype */
  20 #define DT_FLAG_DATA          0x0100  /**< data or control structure */
  21 #define DT_FLAG_ONE_SIDED     0x0200  /**< datatype can be used for one sided operations */
  22 #define DT_FLAG_UNAVAILABLE   0x0400  /**< datatypes unavailable on the build (OS or compiler dependant) */
  23 #define DT_FLAG_DERIVED       0x0800  /**< is the datatype derived ? */
  24 /*
  25  * We should make the difference here between the predefined contiguous and non contiguous
  26  * datatypes. The DT_FLAG_BASIC is held by all predefined contiguous datatypes.
  27  */
  28 #define DT_FLAG_BASIC      (DT_FLAG_PREDEFINED | DT_FLAG_CONTIGUOUS | DT_FLAG_NO_GAPS | DT_FLAG_DATA | DT_FLAG_COMMITED)
  29
  30 extern const MPI_Datatype MPI_PTR;
  31
  32 //The following are datatypes for the MPI functions MPI_MAXLOC and MPI_MINLOC.
  33 typedef struct {
  34   float value;
  35   int index;
  36 } float_int;
  37 typedef struct {
  38   float value;
  39   float index;
  40 } float_float;
  41 typedef struct {
  42   long value;
  43   long index;
  44 } long_long;
  45 typedef struct {
  46   double value;
  47   double index;
  48 } double_double;
  49 typedef struct {
  50   long value;
  51   int index;
  52 } long_int;
  53 typedef struct {
  54   double value;
  55   int index;
  56 } double_int;
  57 typedef struct {
  58   short value;
  59   int index;
  60 } short_int;
  61 typedef struct {
  62   int value;
  63   int index;
  64 } int_int;
  65 typedef struct {
  66   long double value;
  67   int index;
  68 } long_double_int;
  69 typedef struct {
  70   int64_t value;
  71   int64_t index;
  72 } integer128_t;
  73
  74
  75 namespace simgrid{
  76 namespace smpi{
  77
  78 class Datatype : public F2C, public Keyval{
  79   private:
  80     char* name_;
  81     size_t size_;
  82     MPI_Aint lb_;
  83     MPI_Aint ub_;
  84     int flags_;
  85     int refcount_;
  86
  87   public:
  88     static std::unordered_map<int, smpi_key_elem> keyvals_;
  89     static int keyval_id_;
  90
  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
  96     char* name();
  97     size_t size();
  98     MPI_Aint lb();
  99     MPI_Aint ub();
 100     int flags();
 101     int refcount();
 102
 103     void ref();
 104     static void unref(MPI_Datatype datatype);
 105     void commit();
 106     bool is_valid();
 107     void addflag(int flag);
 108     int extent(MPI_Aint * lb, MPI_Aint * extent);
 109     MPI_Aint get_extent();
 110     void get_name(char* name, int* length);
 111     void set_name(char* name);
 112     static int copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 113                     void *recvbuf, int recvcount, MPI_Datatype recvtype);
 114     virtual void serialize( void* noncontiguous, void *contiguous,
 115                             int count);
 116     virtual void unserialize( void* contiguous, void *noncontiguous,
 117                               int count, MPI_Op op);
 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     static Datatype* f2c(int id);
 132 };
 133
 134 }
 135 }
 136
 137 #endif