SMPI over SIMIX_network in a two days rush.

[simgrid.git] / src / smpi / smpi_group.c

diff --git a/src/smpi/smpi_group.c b/src/smpi/smpi_group.c
new file mode 100644 (file)
index 0000000..0d7b4f8
--- /dev/null
+++ b/src/smpi/smpi_group.c
@@ -0,0 +1,109 @@
+#include "private.h"
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_group, smpi,
+                                "Logging specific to SMPI (group)");
+
+typedef struct s_smpi_mpi_group {
+  int size;
+  int* rank_to_index_map;
+  int* index_to_rank_map;
+  int refcount;
+} s_smpi_mpi_group_t;
+
+static s_smpi_mpi_group_t mpi_MPI_GROUP_EMPTY = {
+  0,    /* size */
+  NULL, /* rank_to_index_map */
+  NULL, /* index_to_rank_map */
+  1,    /* refcount: start > 0 so that this group never gets freed */
+};
+MPI_Group MPI_GROUP_EMPTY = &mpi_MPI_GROUP_EMPTY;
+
+MPI_Group smpi_group_new(int size) {
+  MPI_Group group;
+  int i, count;
+
+  count = smpi_process_count();
+  group = xbt_new(s_smpi_mpi_group_t, 1);
+  group->size = size;
+  group->rank_to_index_map = xbt_new(int, size);
+  group->index_to_rank_map = xbt_new(int, count);
+  group->refcount = 0;
+  for(i = 0; i < size; i++) {
+    group->rank_to_index_map[i] = MPI_UNDEFINED;
+  }
+  for(i = 0; i < count; i++) {
+    group->index_to_rank_map[i] = MPI_UNDEFINED;
+  }
+  return group;
+}
+
+void smpi_group_destroy(MPI_Group group) {
+  if(smpi_group_unuse(group) <= 0) {
+    xbt_free(group->rank_to_index_map);
+    xbt_free(group->index_to_rank_map);
+    xbt_free(group);
+  }
+}
+
+void smpi_group_set_mapping(MPI_Group group, int index, int rank) {
+  if(rank < group->size && index < smpi_process_count()) {
+    group->rank_to_index_map[rank] = index;
+    group->index_to_rank_map[index] = rank;
+  }
+}
+
+int smpi_group_index(MPI_Group group, int rank) {
+  int index = MPI_UNDEFINED;
+
+  if(rank < group->size) {
+    index = group->rank_to_index_map[rank];
+  }
+  return index;
+}
+
+int smpi_group_rank(MPI_Group group, int index) {
+  int rank = MPI_UNDEFINED;
+
+  if(index < smpi_process_count()) {
+    rank = group->index_to_rank_map[index];
+  }
+  return rank;
+}
+
+int smpi_group_use(MPI_Group group) {
+  group->refcount++;
+  return group->refcount;
+}
+
+int smpi_group_unuse(MPI_Group group) {
+  group->refcount--;
+  return group->refcount;
+}
+
+int smpi_group_size(MPI_Group group) {
+  return group->size;
+}
+
+int smpi_group_compare(MPI_Group group1, MPI_Group group2) {
+  int result;
+  int i, index, rank, size;
+
+  result = MPI_IDENT;
+  if(smpi_group_size(group1) != smpi_group_size(group2)) {
+    result = MPI_UNEQUAL;
+  } else {
+    size = smpi_group_size(group2);
+    for(i = 0; i < size; i++) {
+      index = smpi_group_index(group1, i);
+      rank = smpi_group_rank(group2, index);
+      if(rank == MPI_UNDEFINED) {
+        result = MPI_UNEQUAL;
+        break;
+      }
+      if(rank != i) {
+        result = MPI_SIMILAR;
+      }
+    }
+  }
+  return result;
+}