Changeset 9796d0a for python/lib/gio.cxx

Timestamp:

09/25/17 19:20:08 (7 years ago)

Author:

Hal Finkel <hfinkel@…>

Branches:

master, pympi

Children:

56b997e

Parents:

b02d091

git-author:

Hal Finkel <hfinkel@…> (09/25/17 19:20:08)

git-committer:

Hal Finkel <hfinkel@…> (09/25/17 19:20:08)

Message:

Add support for float4 (and similar) to Python bindings

File:

• python/lib/gio.cxx (modified) (4 diffs)

python/lib/gio.cxx

@@ -43 +43 @@
 #include <iostream>
 
-  void read_gio_float(char* file_name, char* var_name, float* data){
-    read_gio<float>(file_name,var_name,data);
+  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_double(char* file_name, char* var_name, double* data){
-    read_gio<double>(file_name,var_name,data);
+  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_int32(char* file_name, char* var_name, int* data){
-    read_gio<int>(file_name,var_name,data);
+  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_int64(char* file_name, char* var_name, int64_t* data){
-    read_gio<int64_t>(file_name,var_name,data);
+  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);
   }
   
@@ -77 +77 @@
       gio::GenericIO::VariableInfo vinfo = VI[i];
       if(vinfo.Name == var_name){
-        if(vinfo.IsFloat && vinfo.Size == 4)
+        if(vinfo.IsFloat && vinfo.ElementSize == 4)
           return float_type;
-        else if(vinfo.IsFloat && vinfo.Size == 8)
+        else if(vinfo.IsFloat && vinfo.ElementSize == 8)
           return double_type;
-        else if(!vinfo.IsFloat && vinfo.Size == 4)
+        else if(!vinfo.IsFloat && vinfo.ElementSize == 4)
           return int32_type;
-        else if(!vinfo.IsFloat && vinfo.Size == 8)
+        else if(!vinfo.IsFloat && vinfo.ElementSize == 8)
           return int64_type;
         else
@@ -91 +91 @@
     return var_not_found;
       
+  }
+
+  int get_variable_field_count(char* file_name,char* var_name){
+   gio::GenericIO reader(file_name);
+   std::vector<gio::GenericIO::VariableInfo> VI;
+   reader.openAndReadHeader(gio::GenericIO::MismatchAllowed);
+   reader.getVariableInfo(VI);
+
+   int num =VI.size();
+    for(int i =0;i<num;++i){
+      gio::GenericIO::VariableInfo vinfo = VI[i];
+      if(vinfo.Name == var_name) {
+        return vinfo.Size/vinfo.ElementSize;
+      }
+    }
+    return 0;
   }
 
@@ -110 +126 @@
     else
       std::cout<<"[i";
-    std::cout<<" "<<vinfo.Size*8<<"] ";
+    int NumElements = vinfo.Size/vinfo.ElementSize;
+    std::cout<<" "<<vinfo.ElementSize*8;
+    if (NumElements > 1)
+      std::cout<<"x"<<NumElements;
+    std::cout<<"] ";
     std::cout<<vinfo.Name<<std::endl;
   }