Changeset d409b8c for python/lib

Timestamp:

05/02/19 12:12:31 (6 years ago)

Author:

Thomas Uram <turam@…>

Branches:

pympi

Children:

45bb293

Parents:

1d6ab66

git-author:

Thomas Uram <turam@…> (05/02/19 12:12:31)

git-committer:

Thomas Uram <turam@…> (05/02/19 12:12:31)

Message:

added branch for mpi-based python api

Location:

python/lib

Files:

• gio.cxx (modified) (4 diffs)
• gio.h (modified) (3 diffs)

python/lib/gio.cxx

@@ -43 +43 @@
 #include <iostream>
 
-  void read_gio_float(char* file_name, char* var_name, float* data, int field_count){
-    read_gio<float>(file_name,var_name,data,field_count);
+  void read_gio_float(const MPI_Comm &C, char* file_name, char* var_name, float* data, int field_count){
+    read_gio<float>(C, file_name,var_name,data,field_count);
   }
-  void read_gio_double(char* file_name, char* var_name, double* data, int field_count){
-    read_gio<double>(file_name,var_name,data,field_count);
+  void read_gio_double(const MPI_Comm &C, char* file_name, char* var_name, double* data, int field_count){
+    read_gio<double>(C, file_name,var_name,data,field_count);
   }
-  void read_gio_uint16(char* file_name, char* var_name, uint16_t* data, int field_count){
-    read_gio<uint16_t>(file_name,var_name,data,field_count);
+  void read_gio_uint16(const MPI_Comm &C, char* file_name, char* var_name, uint16_t* data, int field_count){
+    read_gio<uint16_t>(C, file_name,var_name,data,field_count);
   }
-  void read_gio_int32(char* file_name, char* var_name, int* data, int field_count){
-    read_gio<int>(file_name,var_name,data,field_count);
+  void read_gio_int32(const MPI_Comm &C, char* file_name, char* var_name, int* data, int field_count){
+    read_gio<int>(C, file_name,var_name,data,field_count);
   }
-  void read_gio_int64(char* file_name, char* var_name, int64_t* data, int field_count){
-    read_gio<int64_t>(file_name,var_name,data,field_count);
+  void read_gio_int64(const MPI_Comm &C, char* file_name, char* var_name, int64_t* data, int field_count){
+    read_gio<int64_t>(C, file_name,var_name,data,field_count);
   }
   
-  int64_t get_elem_num(char* file_name){
-    gio::GenericIO reader(file_name);
+  int64_t get_elem_num(const MPI_Comm &C, char* file_name){
+    gio::GenericIO reader(C, file_name);
     reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
     int num_ranks = reader.readNRanks();
@@ -70 +70 @@
   }
 
-  var_type get_variable_type(char* file_name,char* var_name){
-   gio::GenericIO reader(file_name);
+  var_type get_variable_type(const MPI_Comm &C, char* file_name,char* var_name){
+   gio::GenericIO reader(C, file_name);
    std::vector<gio::GenericIO::VariableInfo> VI;
    reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
@@ -98 +98 @@
   }
 
-  int get_variable_field_count(char* file_name,char* var_name){
-   gio::GenericIO reader(file_name);
+  int get_variable_field_count(const MPI_Comm &C, char* file_name,char* var_name){
+   gio::GenericIO reader(C, file_name);
    std::vector<gio::GenericIO::VariableInfo> VI;
    reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
@@ -114 +114 @@
   }
 
-extern "C" void inspect_gio(char* file_name){
-  int64_t size = get_elem_num(file_name);
-  gio::GenericIO reader(file_name);
+#ifndef GENERICIO_NO_MPI
+extern "C" void inspect_gio(const MPI_Comm &C, char* file_name){
+  gio::GenericIO reader(C, file_name);
+#else
+extern "C" void inspect_gio(const MPI_Comm &C, char* file_name){
+  gio::GenericIO reader(C, file_name);
+#endif
+  int64_t size = get_elem_num(C,file_name);
   std::vector<gio::GenericIO::VariableInfo> VI;
   reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);

python/lib/gio.h

@@ -46 +46 @@
 
 template <class T>
+#ifndef GENERICIO_NO_MPI
+void read_gio(const MPI_Comm &C, char* file_name, std::string var_name, T*& data, int field_count){
+  gio::GenericIO reader(C, file_name);
+#else
 void read_gio(char* file_name, std::string var_name, T*& data, int field_count){
   gio::GenericIO reader(file_name);
+#endif
   reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
   int num_ranks = reader.readNRanks();
@@ -71 +76 @@
 extern "C" int64_t get_elem_num(char* file_name);
 
-extern "C" void read_gio_float (char* file_name, char* var_name, float* data, int field_count);
-extern "C" void read_gio_double(char* file_name, char* var_name, double* data, int field_count);
-extern "C" void read_gio_uint16 (char* file_name, char* var_name, uint16_t* data, int field_count); 
-extern "C" void read_gio_int32 (char* file_name, char* var_name, int* data, int field_count); 
-extern "C" void read_gio_int64 (char* file_name, char* var_name, int64_t* data, int field_count);
+extern "C" void read_gio_float (const MPI_Comm &C, char* file_name, char* var_name, float* data, int field_count);
+extern "C" void read_gio_double(const MPI_Comm &C, char* file_name, char* var_name, double* data, int field_count);
+extern "C" void read_gio_uint16 (const MPI_Comm &C, char* file_name, char* var_name, uint16_t* data, int field_count); 
+extern "C" void read_gio_int32 (const MPI_Comm &C, char* file_name, char* var_name, int* data, int field_count); 
+extern "C" void read_gio_int64 (const MPI_Comm &C, char* file_name, char* var_name, int64_t* data, int field_count);
 enum var_type{
   float_type=0,
@@ -87 +92 @@
 extern "C" var_type get_variable_type(char* file_name,char* var_name);
 extern "C" int get_variable_field_count(char* file_name,char* var_name);
-extern "C" void inspect_gio(char* file_name);
+extern "C" void inspect_gio(const MPI_Comm &C, char* file_name);