Move Keyval handling to a new file, as it is shared between Comm and Datatype (and...

[simgrid.git] / src / smpi / smpi_keyvals.hpp

diff --git a/src/smpi/smpi_keyvals.hpp b/src/smpi/smpi_keyvals.hpp
new file mode 100644 (file)
index 0000000..0508f33
--- /dev/null
+++ b/src/smpi/smpi_keyvals.hpp
@@ -0,0 +1,144 @@
+/* Copyright (c) 2010, 2013-2017. The SimGrid Team.
+ * All rights reserved.                                                     */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#ifndef SMPI_KEYVALS_HPP_INCLUDED
+#define SMPI_KEYVALS_HPP_INCLUDED
+
+#include "private.h"
+#include <unordered_map>
+#include <xbt/ex.hpp>
+
+namespace simgrid{
+namespace smpi{
+
+
+typedef union smpi_delete_fn{
+  MPI_Comm_delete_attr_function          *comm_delete_fn;
+  MPI_Type_delete_attr_function          *type_delete_fn;
+  MPI_Win_delete_attr_function           *win_delete_fn;
+} smpi_delete_fn;
+
+typedef union smpi_copy_fn{
+  MPI_Comm_copy_attr_function          *comm_copy_fn;
+  MPI_Type_copy_attr_function          *type_copy_fn;
+  MPI_Win_copy_attr_function           *win_copy_fn;
+} smpi_copy_fn;
+
+typedef struct s_smpi_key_elem {
+  smpi_copy_fn copy_fn;
+  smpi_delete_fn delete_fn;
+  int refcount;
+} s_smpi_mpi_key_elem_t; 
+typedef struct s_smpi_key_elem *smpi_key_elem;
+
+class Keyval{
+  protected:
+    std::unordered_map<int, void*> attributes_;
+  public:
+// Each subclass should have two members, as we want to separate the ones for Win, Comm, and Datatypes :  
+//    static std::unordered_map<int, smpi_key_elem> keyvals_;
+//    static int keyval_id_;
+    template <typename T> static int keyval_create(smpi_copy_fn copy_fn, smpi_delete_fn delete_fn, int* keyval, void* extra_statee);
+    template <typename T> static int keyval_free(int* keyval);
+    template <typename T> int attr_delete(int keyval);
+    template <typename T> int attr_get(int keyval, void* attr_value, int* flag);
+    template <typename T> int attr_put(int keyval, void* attr_value);
+    template <typename T> static int call_deleter(T* obj, smpi_key_elem elem, int keyval, void * value, int* flag);
+};
+
+template <typename T> int Keyval::keyval_create(smpi_copy_fn copy_fn, smpi_delete_fn delete_fn, int* keyval, void* extra_state){
+
+  smpi_key_elem value = (smpi_key_elem) xbt_new0(s_smpi_mpi_key_elem_t,1);
+
+  value->copy_fn=copy_fn;
+  value->delete_fn=delete_fn;
+  value->refcount=1;
+
+  *keyval = T::keyval_id_;
+  T::keyvals_.insert({*keyval, value});
+  T::keyval_id_++;
+  return MPI_SUCCESS;
+}
+
+template <typename T> int Keyval::keyval_free(int* keyval){
+/* See MPI-1, 5.7.1.  Freeing the keyval does not remove it if it
+         * is in use in an attribute */
+  smpi_key_elem elem = T::keyvals_.at(*keyval);
+  if(elem==0){
+    return MPI_ERR_ARG;
+  }
+  if(elem->refcount==1){
+    T::keyvals_.erase(*keyval);
+    xbt_free(elem);
+  }else{
+    elem->refcount--;
+  }
+  return MPI_SUCCESS;
+}
+
+//specialized in smpi_keyvals.cpp
+template <typename T> int Keyval::call_deleter(T* obj, smpi_key_elem elem, int keyval, void * value, int* flag){
+  return MPI_SUCCESS;
+}
+
+template <typename T> int Keyval::attr_delete(int keyval){
+  smpi_key_elem elem = T::keyvals_.at(keyval);
+  if(elem==nullptr)
+    return MPI_ERR_ARG;
+  elem->refcount--;
+  void * value = nullptr;
+  int flag;
+  if(this->attr_get<T>(keyval, &value, &flag)==MPI_SUCCESS){
+    int ret = call_deleter<T>((T*)this, elem, keyval,value,&flag);
+    if(ret!=MPI_SUCCESS)
+        return ret;
+  }
+  if(attributes_.empty())
+    return MPI_ERR_ARG;
+  attributes_.erase(keyval);
+  return MPI_SUCCESS;
+}
+
+
+template <typename T> int Keyval::attr_get(int keyval, void* attr_value, int* flag){
+  smpi_key_elem elem = T::keyvals_.at(keyval);
+  if(elem==nullptr)
+    return MPI_ERR_ARG;
+  if(attributes_.empty()){
+    *flag=0;
+    return MPI_SUCCESS;
+  }
+  try {
+    *static_cast<void**>(attr_value) = attributes_.at(keyval);
+    *flag=1;
+  }
+  catch (const std::out_of_range& oor) {
+    *flag=0;
+  }
+  return MPI_SUCCESS;
+}
+
+template <typename T> int Keyval::attr_put(int keyval, void* attr_value){
+  smpi_key_elem elem = T::keyvals_.at(keyval);
+  if(elem==nullptr)
+    return MPI_ERR_ARG;
+  elem->refcount++;
+  void * value = nullptr;
+  int flag;
+  this->attr_get<T>(keyval, &value, &flag);
+  if(flag!=0){
+    int ret = call_deleter<T>((T*)this, elem, keyval,value,&flag);
+    if(ret!=MPI_SUCCESS)
+        return ret;
+  }
+  attributes_.insert({keyval, attr_value});
+  return MPI_SUCCESS;
+}
+
+}
+}
+
+#endif