add SOURCE_ARRAY type

to help out MPI_Group_translate_ranks

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

Author: hppritcha

ompi/mpi/bindings/ompi_bindings/c_type.py

@@ -892,6 +892,42 @@ def init_code(self):
     def type_text(self, enable_count=False):
         return 'int'
 
+@Type.add_type('SOURCE_ARRAY', abi_type=['ompi'])
+class TypeSourceArray(Type):
+
+    def type_text(self, enable_count=False):
+        return 'const int*'
+
+    def parameter(self, enable_count=False, **kwargs):
+        return f'const int {self.name}[]'
+
+@Type.add_type('SOURCE_ARRAY', abi_type=['standard'])
+class TypeSourceArrayStandard(StandardABIType):
+
+    @property
+    def init_code(self):
+        code = [(f'int *{self.tmpname} = NULL;')]
+        code.append('if('+f'{self.name}' + '!= NULL)' + '{')
+        code.append(f'{self.tmpname} = (int *)malloc(sizeof(int) * {self.count_param});')
+        code.append(f'for(int i=0;i<{self.count_param};i++){{')
+        code.append(f'{self.tmpname}[i] = {ConvertFuncs.SOURCE}({self.name}[i]);')
+        code.append('}')
+        code.append('}')
+        return code
+
+    @property
+    def final_code(self):
+        code = [f'if({self.tmpname} != NULL){{']
+        code.append(f'free({self.tmpname});')
+        code.append('}')
+        return code
+
+    def type_text(self, enable_count=False):
+        return 'int *'
+
+    def parameter(self, enable_count=False, **kwargs):
+        return f'const int {self.name}[]'
+
 @Type.add_type('SOURCE_OUT', abi_type=['ompi'])
 class TypeSourceOut(Type):
 
@@ -912,6 +948,39 @@ def type_text(self, enable_count=False):
     def argument(self):
         return f'(int *) {self.name}'
 
+@Type.add_type('SOURCE_ARRAY_OUT', abi_type=['ompi'])
+class TypeSourceArrayOut(Type):
+
+    def type_text(self, enable_count=False):
+        return 'int *'
+
+    def parameter(self, enable_count=False, **kwargs):
+        return f'int {self.name}[]'
+
+@Type.add_type('SOURCE_ARRAY_OUT', abi_type=['standard'])
+class TypeSourceArrayOutStandard(StandardABIType):
+        
+    @property
+    def init_code(self):
+        code = [f'int *{self.tmpname} = (int*)malloc({self.count_param} * sizeof(int));']
+        return code
+    
+    @property
+    def final_code(self):
+        code = [f'if (NULL != {self.name}){{']
+        code.append(f'for(int i=0;i<{self.count_param};i++){{')
+        code.append(f'{self.name}[i] = {ConvertOMPIToStandard.SOURCE}({self.tmpname}[i]);')
+        code.append('}')
+        code.append('}')
+        code.append(f'free({self.tmpname});')
+        return code
+        
+    def type_text(self, enable_count=False):
+        return 'int *'
+
+    def parameter(self, enable_count=False, **kwargs):
+        return f'int {self.name}[]'
+
 @Type.add_type('COMM', abi_type=['ompi'])
 class TypeCommunicator(Type):
 

ompi/mpi/c/group_translate_ranks.c.in

@@ -32,8 +32,8 @@
 #include "ompi/errhandler/errhandler.h"
 #include "ompi/group/group.h"
 
-PROTOTYPE ERROR_CLASS group_translate_ranks(GROUP group1, INT n_ranks, INT_ARRAY ranks1,
-                                            GROUP group2, INT_OUT ranks2)
+PROTOTYPE ERROR_CLASS group_translate_ranks(GROUP group1, INT n_ranks, SOURCE_ARRAY ranks1:n_ranks,
+                                            GROUP group2, SOURCE_ARRAY_OUT ranks2:n_ranks)
 {
     int err;