binding framework: add a TS_LEVEL type

Signed-off-by: Howard Pritchard <[email protected]>

Author: hppritcha

ompi/mpi/bindings/ompi_bindings/c_type.py

@@ -1111,6 +1111,50 @@ def type_text(self, enable_count=False):
         type_name = self.mangle_name('MPI_Message')
         return f'{type_name} *'
 
+@Type.add_type('TS_LEVEL', abi_type=['ompi'])
+class TypeTSLevel(Type):
+
+    def type_text(self, enable_count=False):
+        return 'int'
+
+
+@Type.add_type('TS_LEVEL', abi_type=['standard'])
+class TypeTSLevelStandard(StandardABIType):
+
+    @property
+    def init_code(self):
+        return [f'int {self.tmpname} = {ConvertFuncs.TS_LEVEL}({self.name});']
+
+    def tmp_type_text(self, enable_count=False):
+        return 'int'
+
+    def return_code(self, name):
+        return [f'return {ConvertOMPIToStandard.TS_LEVEL}({name});']
+
+    def type_text(self, enable_count=False):
+        return 'int'
+
+
+@Type.add_type('TS_LEVEL_OUT', abi_type=['ompi'])
+class TypeTSLevelOut(Type):
+
+    def type_text(self, enable_count=False):
+        return 'int *'
+
+
+@Type.add_type('TS_LEVEL_OUT', abi_type=['standard'])
+class TypeTSLevelOutStandard(StandardABIType):
+
+    @property
+    def final_code(self):
+        return [f'*{self.name} = {ConvertOMPIToStandard.TS_LEVEL}((int) *{self.name});']
+
+    def type_text(self, enable_count=False):
+        return f'int *'
+
+    @property
+    def argument(self):
+        return f'{self.name}'
 
 @Type.add_type('COMM_ERRHANDLER_FUNCTION', abi_type=['ompi'])
 class TypeCommErrhandlerFunction(Type):

ompi/mpi/bindings/ompi_bindings/consts.py

@@ -274,6 +274,7 @@ class ConvertFuncs:
     WIN = 'ompi_convert_abi_win_intern_win'
     INFO = 'ompi_convert_abi_info_intern_info'
     FILE = 'ompi_convert_abi_file_intern_file'
+    TS_LEVEL = 'ompi_convert_abi_ts_level_intern_ts_level'
 
 
 class ConvertOMPIToStandard:
@@ -292,6 +293,7 @@ class ConvertOMPIToStandard:
     WIN = 'ompi_convert_win_ompi_to_standard'
     REQUEST = 'ompi_convert_ompi_request_abi_request'
     INFO = 'ompi_convert_info_ompi_to_standard'
+    TS_LEVEL = 'ompi_convert_ts_level_ompi_to_standard'
 
 
 # Inline function attributes

ompi/mpi/c/abi_converters.h

@@ -561,6 +561,34 @@ __opal_attribute_always_inline__ static inline void ompi_convert_intern_status_a
     memcpy(ptr, &inp->_ucount,sizeof(inp->_ucount));
 }
 
+__opal_attribute_always_inline__ static inline int ompi_convert_abi_ts_level_intern_ts_level(int ts_level)
+{
+    if (MPI_THREAD_SINGLE_ABI_INTERNAL == ts_level) {
+        return MPI_THREAD_SINGLE;
+    } else if (MPI_THREAD_FUNNELED_ABI_INTERNAL == ts_level) {
+        return MPI_THREAD_FUNNELED;
+    } else if (MPI_THREAD_SERIALIZED_ABI_INTERNAL == ts_level) {
+        return MPI_THREAD_SERIALIZED;
+    } else if (MPI_THREAD_MULTIPLE_ABI_INTERNAL  == ts_level) {
+        return MPI_THREAD_MULTIPLE;
+    }
+    return ts_level;
+}
+
+__opal_attribute_always_inline__ static inline int ompi_convert_ts_level_ompi_to_standard(int ts_level)
+{
+    if (MPI_THREAD_SINGLE == ts_level) {
+        return MPI_THREAD_SINGLE_ABI_INTERNAL;
+    } else if (MPI_THREAD_FUNNELED == ts_level) {
+        return MPI_THREAD_FUNNELED_ABI_INTERNAL;
+    } else if (MPI_THREAD_SERIALIZED == ts_level) {
+        return MPI_THREAD_SERIALIZED_ABI_INTERNAL;
+    } else if (MPI_THREAD_MULTIPLE  == ts_level) {
+        return MPI_THREAD_MULTIPLE_ABI_INTERNAL;
+    }
+    return ts_level;
+}
+
 #if defined(c_plusplus) || defined(__cplusplus)
 }
 #endif

ompi/mpi/c/init_thread.c.in

@@ -37,8 +37,8 @@
 #include "ompi/constants.h"
 #include "ompi/mca/hook/base/base.h"
 
-PROTOTYPE ERROR_CLASS init_thread(INT_OUT argc, ARGV argv, INT required,
-                                  INT_OUT provided)
+PROTOTYPE ERROR_CLASS init_thread(INT_OUT argc, ARGV argv, TS_LEVEL required,
+                                  TS_LEVEL_OUT provided)
 {
     int err, safe_required = MPI_THREAD_SERIALIZED;
     bool err_arg_required = false;