Changeset 2599

Timestamp:

11/20/23 14:07:25 (6 months ago)

Author:

jderouillat

Message:

Add a field attribute, conversion_by_NetCDF, to operate type conversion in XIOS, and not in NetCDF

Location:

XIOS2/trunk/src

Files:

• config/field_attribute.conf (modified) (1 diff)
• io/nc4_data_output.cpp (modified) (4 diffs)
• io/nc4_data_output.hpp (modified) (1 diff)
• io/netCdfInterface.cpp (modified) (2 diffs)
• io/netCdfInterface_decl.cpp (modified) (1 diff)
• io/onetcdf4.cpp (modified) (1 diff)
• io/onetcdf4_decl.cpp (modified) (1 diff)

XIOS2/trunk/src/config/field_attribute.conf

@@ -13 +13 @@
 DECLARE_ATTRIBUTE(int,       level)
 DECLARE_ATTRIBUTE(int,       prec)
+DECLARE_ATTRIBUTE(bool,      conversion_by_NetCDF)
 DECLARE_ATTRIBUTE(StdString, expr)
 

XIOS2/trunk/src/io/nc4_data_output.cpp

@@ -13 +13 @@
 #include "uuid.hpp"
 #include "mesh_values.hpp"
+
+#include <limits.h>
+#define X_FLOAT_MAX     FLT_MAX
+#define X_FLOAT_MIN     FLT_MIN
+#define X_SHORT_MAX     SHRT_MAX
+#define X_SHORT_MIN     SHRT_MIN
+
 namespace xios
 {
@@ -2515 +2522 @@
               {
                  CTimer::get("Files : writing data").resume();
-                 SuperClassWriter::writeData(fieldData, fieldid, isCollective, field->getNStep() - 1);
+                 writeAndConvertData(field, fieldData, field->getNStep() - 1);
                  CTimer::get("Files : writing data").suspend();
                  if (wtime)
@@ -2655 +2662 @@
 
                 CTimer::get("Files : writing data").resume();
-                SuperClassWriter::writeData(fieldData, fieldid, isCollective, field->getNStep() - 1, &start, &count);
+                writeAndConvertData(field, fieldData, field->getNStep() - 1, &start, &count);
                 CTimer::get("Files : writing data").suspend();
 
@@ -2688 +2695 @@
            ERROR("CNc4DataOutput::writeFieldData_ (CField*  field)", << msg);
          }
+      }
+
+      void CNc4DataOutput::writeAndConvertData(CField* field, const CArray<double,1>& data, StdSize record, const std::vector<StdSize> *start, const std::vector<StdSize> *count)
+      {
+        StdString fieldid = field->getFieldOutputName();
+        nc_type type ;
+        if (field->prec.isEmpty()) type =  NC_FLOAT ;
+        else
+        {
+          if (field->prec==2) type = NC_SHORT ;
+          else if (field->prec==4)  type =  NC_FLOAT ;
+          else if (field->prec==8)   type =  NC_DOUBLE ;
+        }
+
+        bool conversionByNetCDF = true; // default : conversion operated by NetCDF for now (legacy behaviour)
+        if (!field->conversion_by_NetCDF.isEmpty())
+        {
+          // use conversion_by_NetCDF = ".false." to  bypass NetCDF conversion
+          //   poor performances from NC_DOUBLE to NC_FLOAT with isgreater/isless in recent NetCDF available at TGCC
+          conversionByNetCDF = field->conversion_by_NetCDF;
+        }
+        if ( ( type == NC_DOUBLE ) || ( conversionByNetCDF ) )
+        {
+          SuperClassWriter::writeData(data, fieldid, isCollective, record, start, count);
+        }
+        else if ( type == NC_FLOAT )
+        {
+          CArray<float,1> f_data;
+          f_data.resize( data.numElements() );
+          for (int i = 0; i < data.numElements(); i++) 
+          {
+            if (data(i) <= X_FLOAT_MAX || data(i) >= X_FLOAT_MIN)
+              f_data(i) = data(i);
+            else if (data(i) > X_FLOAT_MAX)
+              f_data(i) = X_FLOAT_MAX;
+            else if (data(i) < X_FLOAT_MIN)
+              f_data(i) = X_FLOAT_MIN;
+          }
+          SuperClassWriter::writeData(f_data, fieldid, isCollective, record, start, count);
+        }
+        else if ( type == NC_SHORT )
+        {
+          CArray<short,1> s_data;
+          s_data.resize( data.numElements() );
+          for (int i = 0; i < data.numElements(); i++) 
+          {
+            if (data(i) <= X_SHORT_MAX || data(i) >= X_SHORT_MIN)
+              s_data(i) = data(i);
+            else if (data(i) > X_SHORT_MAX)
+              s_data(i) = X_SHORT_MAX;
+            else if (data(i) < X_SHORT_MIN)
+              s_data(i) = X_SHORT_MIN;
+          }
+          SuperClassWriter::writeData(s_data, fieldid, isCollective, record, start, count);
+        }
+        else
+        {
+            ERROR("CNc4DataOutput::writeAndConvertData(...)", << "Type conversion not managed for " << fieldid ); 
+        }
+        
       }
 

XIOS2/trunk/src/io/nc4_data_output.hpp

@@ -57 +57 @@
             virtual void writeAttribute_(CVariable* var, const string& fieldId);
             virtual void writeFieldData_(CField* field);
+            void writeAndConvertData(CField* field, const CArray<double,1>& data, StdSize record,
+                                     const std::vector<StdSize> *start = NULL, const std::vector<StdSize> *count = NULL);
             virtual void writeFile_     (CFile* file);
             virtual void closeFile_     (void);

XIOS2/trunk/src/io/netCdfInterface.cpp

@@ -991 +991 @@
 
 template<>
+int CNetCdfInterface::ncGetVaraType(int ncid, int varid, const StdSize* start, const StdSize* count, short* data)
+{
+  return nc_get_vara_short(ncid, varid, start, count, data);
+}
+
+template<>
 int CNetCdfInterface::ncGetVaraType(int ncid, int varid, const StdSize* start, const StdSize* count, char* data)
 {
@@ -1013 +1019 @@
 {
   return nc_put_vara_int(ncid, varid, start, count, data);
+}
+
+template<>
+int CNetCdfInterface::ncPutVaraType(int ncid, int varid, const StdSize* start, const StdSize* count, const short* data)
+{
+  return nc_put_vara_short(ncid, varid, start, count, data);
 }
 

XIOS2/trunk/src/io/netCdfInterface_decl.cpp

@@ -33 +33 @@
   macroPutVar(float)
   macroPutVar(int)
+  macroPutVar(short)
   macroPutVar(char)
   

XIOS2/trunk/src/io/onetcdf4.cpp

@@ -554 +554 @@
           CNetCdfInterface::putVaraType(grpid, varid, &sstart[0], &scount[0], data);
       }
-
+      
+      template <>
+      void CONetCDF4::writeData_(int grpid, int varid,
+                                 const std::vector<StdSize>& sstart,
+                                 const std::vector<StdSize>& scount, const short* data)
+      {
+          CNetCdfInterface::putVaraType(grpid, varid, &sstart[0], &scount[0], data);
+      }
       //---------------------------------------------------------------
 

XIOS2/trunk/src/io/onetcdf4_decl.cpp

@@ -14 +14 @@
   macro(double, 2)
   macro(double, 3)
+  macro(float, 1)
+  macro(short, 1)
   macro(StdString, 1)