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 #include <unordered_map>
  14
  15 #define DT_FLAG_DESTROYED     0x0001  /**< user destroyed but some other layers still have a reference */
  16 #define DT_FLAG_COMMITED      0x0002  /**< ready to be used for a send/recv operation */
  17 #define DT_FLAG_CONTIGUOUS    0x0004  /**< contiguous datatype */
  18 #define DT_FLAG_OVERLAP       0x0008  /**< datatype is unpropper for a recv operation */
  19 #define DT_FLAG_USER_LB       0x0010  /**< has a user defined LB */
  20 #define DT_FLAG_USER_UB       0x0020  /**< has a user defined UB */
  21 #define DT_FLAG_PREDEFINED    0x0040  /**< cannot be removed: initial and predefined datatypes */
  22 #define DT_FLAG_NO_GAPS       0x0080  /**< no gaps around the datatype */
  23 #define DT_FLAG_DATA          0x0100  /**< data or control structure */
  24 #define DT_FLAG_ONE_SIDED     0x0200  /**< datatype can be used for one sided operations */
  25 #define DT_FLAG_UNAVAILABLE   0x0400  /**< datatypes unavailable on the build (OS or compiler dependant) */
  26 #define DT_FLAG_DERIVED       0x0800  /**< is the datatype derived ? */
  27 /*
  28  * We should make the difference here between the predefined contiguous and non contiguous
  29  * datatypes. The DT_FLAG_BASIC is held by all predefined contiguous datatypes.
  30  */
  31 #define DT_FLAG_BASIC      (DT_FLAG_PREDEFINED | DT_FLAG_CONTIGUOUS | DT_FLAG_NO_GAPS | DT_FLAG_DATA | DT_FLAG_COMMITED)
  32
  33 extern const MPI_Datatype MPI_PTR;
  34
  35 typedef struct s_smpi_mpi_type_key_elem {
  36   MPI_Type_copy_attr_function* copy_fn;
  37   MPI_Type_delete_attr_function* delete_fn;
  38 } s_smpi_mpi_type_key_elem_t;
  39 typedef struct s_smpi_mpi_type_key_elem *smpi_type_key_elem;
  40
  41 //The following are datatypes for the MPI functions MPI_MAXLOC and MPI_MINLOC.
  42 typedef struct {
  43   float value;
  44   int index;
  45 } float_int;
  46 typedef struct {
  47   float value;
  48   float index;
  49 } float_float;
  50 typedef struct {
  51   long value;
  52   long index;
  53 } long_long;
  54 typedef struct {
  55   double value;
  56   double index;
  57 } double_double;
  58 typedef struct {
  59   long value;
  60   int index;
  61 } long_int;
  62 typedef struct {
  63   double value;
  64   int index;
  65 } double_int;
  66 typedef struct {
  67   short value;
  68   int index;
  69 } short_int;
  70 typedef struct {
  71   int value;
  72   int index;
  73 } int_int;
  74 typedef struct {
  75   long double value;
  76   int index;
  77 } long_double_int;
  78 typedef struct {
  79   int64_t value;
  80   int64_t index;
  81 } integer128_t;
  82
  83
  84 namespace simgrid{
  85 namespace smpi{
  86
  87 class Datatype : public F2C{
  88   protected:
  89     char* name_;
  90     size_t size_;
  91     MPI_Aint lb_;
  92     MPI_Aint ub_;
  93     int flags_;
  94     xbt_dict_t attributes_;
  95     int refcount_;
  96
  97     static std::unordered_map<int, smpi_type_key_elem> keyvals_;
  98     static int keyval_id_;
  99
 100   public:
 101     static MPI_Datatype null_id_;
 102
 103     Datatype(int size,MPI_Aint lb, MPI_Aint ub, int flags);
 104     Datatype(char* name, int size,MPI_Aint lb, MPI_Aint ub, int flags);
 105     Datatype(Datatype *datatype, int* ret);
 106     virtual ~Datatype();
 107     void ref();
 108     static void unref(MPI_Datatype datatype);
 109     void commit();
 110     bool is_valid();
 111     size_t size();
 112     int flags();
 113     void addflag(int flag);
 114     MPI_Aint lb();
 115     MPI_Aint ub();
 116     int extent(MPI_Aint * lb, MPI_Aint * extent);
 117     MPI_Aint get_extent();
 118     char* name();
 119     void get_name(char* name, int* length);
 120     void set_name(char* name);
 121     static int copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 122                     void *recvbuf, int recvcount, MPI_Datatype recvtype);
 123     virtual void serialize( void* noncontiguous, void *contiguous,
 124                             int count);
 125     virtual void unserialize( void* contiguous, void *noncontiguous,
 126                               int count, MPI_Op op);
 127     int attr_delete(int keyval);
 128     int attr_get(int keyval, void* attr_value, int* flag);
 129     int attr_put(int keyval, void* attr_value);
 130     static int keyval_create(MPI_Type_copy_attr_function* copy_fn, MPI_Type_delete_attr_function* delete_fn, int* keyval, void* extra_state);
 131     static int keyval_free(int* keyval);
 132     int pack(void* inbuf, int incount, void* outbuf, int outcount, int* position, MPI_Comm comm);
 133     int unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Comm comm);
 134
 135
 136     static int create_contiguous(int count, MPI_Datatype old_type, MPI_Aint lb, MPI_Datatype* new_type);
 137     static int create_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type);
 138     static int create_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type);
 139     static int create_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type);
 140     static int create_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type);
 141     static int create_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type);
 142
 143     static Datatype* f2c(int id);
 144 };
 145
 146 }
 147 }
 148
 149 #endif