Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Remove unused state.
[simgrid.git] / src / smpi / include / smpi_keyvals.hpp
1 /* Copyright (c) 2010-2020. 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_KEYVALS_HPP_INCLUDED
7 #define SMPI_KEYVALS_HPP_INCLUDED
8
9 #include "smpi/smpi.h"
10
11 #include <unordered_map>
12
13 struct smpi_delete_fn {
14   MPI_Comm_delete_attr_function          *comm_delete_fn;
15   MPI_Type_delete_attr_function          *type_delete_fn;
16   MPI_Win_delete_attr_function           *win_delete_fn;
17   MPI_Comm_delete_attr_function_fort     *comm_delete_fn_fort;
18   MPI_Type_delete_attr_function_fort     *type_delete_fn_fort;
19   MPI_Win_delete_attr_function_fort      *win_delete_fn_fort;
20 };
21
22 struct smpi_copy_fn {
23   MPI_Comm_copy_attr_function          *comm_copy_fn;
24   MPI_Type_copy_attr_function          *type_copy_fn;
25   MPI_Win_copy_attr_function           *win_copy_fn;
26   MPI_Comm_copy_attr_function_fort     *comm_copy_fn_fort;
27   MPI_Type_copy_attr_function_fort     *type_copy_fn_fort;
28   MPI_Win_copy_attr_function_fort      *win_copy_fn_fort;
29 };
30
31 struct s_smpi_key_elem_t {
32   smpi_copy_fn copy_fn;
33   smpi_delete_fn delete_fn;
34   void* extra_state;
35   int refcount;
36 };
37
38 typedef s_smpi_key_elem_t* smpi_key_elem;
39
40 namespace simgrid{
41 namespace smpi{
42
43 class Keyval{
44   private:
45     std::unordered_map<int, void*> attributes_;
46   protected:
47     std::unordered_map<int, void*>* attributes();
48   public:
49 // Each subclass should have two members, as we want to separate the ones for Win, Comm, and Datatypes :
50 //    static std::unordered_map<int, smpi_key_elem> keyvals_;
51 //    static int keyval_id_;
52     template <typename T>
53     static int keyval_create(const smpi_copy_fn& copy_fn, const smpi_delete_fn& delete_fn, int* keyval,
54                              void* extra_state);
55     template <typename T> static int keyval_free(int* keyval);
56     template <typename T> int attr_delete(int keyval);
57     template <typename T> int attr_get(int keyval, void* attr_value, int* flag);
58     template <typename T> int attr_put(int keyval, void* attr_value);
59     template <typename T>
60     static int call_deleter(T* obj, const s_smpi_key_elem_t* elem, int keyval, void* value, int* flag);
61     template <typename T> void cleanup_attr();
62 };
63
64 template <typename T>
65 int Keyval::keyval_create(const smpi_copy_fn& copy_fn, const smpi_delete_fn& delete_fn, int* keyval, void* extra_state)
66 {
67   smpi_key_elem value = new s_smpi_key_elem_t;
68
69   value->copy_fn=copy_fn;
70   value->delete_fn=delete_fn;
71   value->extra_state=extra_state;
72   value->refcount=1;
73
74   *keyval = T::keyval_id_;
75   T::keyvals_.insert({*keyval, value});
76   T::keyval_id_++;
77   return MPI_SUCCESS;
78 }
79
80 template <typename T> int Keyval::keyval_free(int* keyval){
81 /* See MPI-1, 5.7.1.  Freeing the keyval does not remove it if it
82          * is in use in an attribute */
83   smpi_key_elem elem = T::keyvals_.at(*keyval);
84   if(elem==0){
85     return MPI_ERR_ARG;
86   }
87   if(elem->refcount==1){
88     T::keyvals_.erase(*keyval);
89     delete elem;
90   }else{
91     elem->refcount--;
92   }
93   return MPI_SUCCESS;
94 }
95
96 template <typename T> int Keyval::attr_delete(int keyval){
97   smpi_key_elem elem = T::keyvals_.at(keyval);
98   if(elem==nullptr)
99     return MPI_ERR_ARG;
100   elem->refcount--;
101   void * value = nullptr;
102   int flag=0;
103   if(this->attr_get<T>(keyval, &value, &flag)==MPI_SUCCESS){
104     int ret = call_deleter<T>((T*)this, elem, keyval,value,&flag);
105     if(ret!=MPI_SUCCESS)
106         return ret;
107   }
108   if(attributes()->empty())
109     return MPI_ERR_ARG;
110   attributes()->erase(keyval);
111   return MPI_SUCCESS;
112 }
113
114
115 template <typename T> int Keyval::attr_get(int keyval, void* attr_value, int* flag){
116   const s_smpi_key_elem_t* elem = T::keyvals_.at(keyval);
117   if(elem==nullptr)
118     return MPI_ERR_ARG;
119   if(attributes()->empty()){
120     *flag=0;
121     return MPI_SUCCESS;
122   }
123   const auto& attribs = attributes();
124   auto attr           = attribs->find(keyval);
125   if (attr != attribs->end()) {
126     *static_cast<void**>(attr_value) = attr->second;
127     *flag=1;
128   } else {
129     *flag=0;
130   }
131   return MPI_SUCCESS;
132 }
133
134 template <typename T> int Keyval::attr_put(int keyval, void* attr_value){
135   smpi_key_elem elem = T::keyvals_.at(keyval);
136   if(elem==nullptr)
137     return MPI_ERR_ARG;
138   elem->refcount++;
139   int flag=0;
140   auto p = attributes()->insert({keyval, attr_value});
141   if (!p.second) {
142     int ret = call_deleter<T>((T*)this, elem, keyval,p.first->second,&flag);
143     // overwrite previous value
144     p.first->second = attr_value;
145     if(ret!=MPI_SUCCESS)
146       return ret;
147   }
148   return MPI_SUCCESS;
149 }
150
151 template <typename T> void Keyval::cleanup_attr(){
152   if (not attributes()->empty()) {
153     int flag=0;
154     for (auto const& it : attributes_) {
155       auto elm = T::keyvals_.find(it.first);
156       if (elm != T::keyvals_.end()) {
157         smpi_key_elem elem = elm->second;
158         if(elem != nullptr){
159           call_deleter<T>((T*)this, elem, it.first,it.second,&flag);
160         }
161       } else {
162         // already deleted, not a problem
163         flag=0;
164       }
165     }
166   }
167 }
168
169 }
170 }
171
172 #endif