Changeset 1984 for XIOS/dev/dev_ym/XIOS_COUPLING/src/distribution/gatherer_connector.hpp

Timestamp:

12/06/20 22:14:56 (4 years ago)

Author:

ymipsl

Message:

intermediate commit for new tranformation engine?
YM

File:

• XIOS/dev/dev_ym/XIOS_COUPLING/src/distribution/gatherer_connector.hpp (modified) (7 diffs)

XIOS/dev/dev_ym/XIOS_COUPLING/src/distribution/gatherer_connector.hpp

@@ -29 +29 @@
       
       template<typename T>
-      void transfer(int sizeT, map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut)
+      void transfer(int repeat, int sizeT, map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut)
       {
         // for future, make a specific transfer function for sizeT=1 to avoid multiplication (increasing performance)
-        dataOut.resize(dstSize_*sizeT) ;  
-        T* output = dataOut.dataFirst() ;
+        size_t dstSlice = dstSize_*sizeT ;
+        dataOut.resize(repeat* dstSlice) ;  
+        
         for(auto& data : dataIn)
         {
+          T* output = dataOut.dataFirst() ;
           int rank=data.first ;
           auto input  = data.second.dataFirst() ;
@@ -41 +43 @@
           auto& mask=mask_[rank] ;
           int size=mask.size() ;
-
-          for(int i=0, j=0 ;i<size;i++)
-          {
-            if (mask[i]) 
+          size_t srcSlice = size * sizeT ;
+          for(int l=0; l<repeat; l++)
+          {   
+            for(int i=0, j=0 ;i<size;i++)
             {
-              int cj = connector[j]*sizeT ;
-              int ci = i*sizeT ;
-              for (int k=0;k<sizeT;k++) output[cj+k] = input[ci+k] ;
-              j++ ;
+              if (mask[i]) 
+              {
+                int cj = connector[j]*sizeT ;
+                int ci = i*sizeT ;
+                for (int k=0;k<sizeT;k++) output[cj+k] = input[ci+k] ;
+                j++ ;
+              }
             }
+            input+=srcSlice ;
+            output+=dstSlice ;
           }
         }
+      }
+
+      template<typename T>
+      void transfer(int sizeT, map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut)
+      {
+        transfer(1, 1, dataIn, dataOut) ;
       }
     
@@ -101 +114 @@
       void transfer(map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut, T missingValue)
       {
-        transfer(1, dataIn, dataout, missingValue)
+        transfer(1, 1, dataIn, dataout, missingValue)
       }
       
@@ -107 +120 @@
       void transfer(int sizeT, map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut, T missingValue)
       {
-        dataOut.resize(dstSize_*sizeT) ;
+         transfer(1, sizeT, dataIn, dataOut, missingValue) ;
+      }
+
+      template<typename T>
+      void transfer(int repeat , int sizeT, map<int, CArray<T,1>>& dataIn, CArray<T,1>& dataOut, T missingValue)
+      {
+        dataOut.resize(repeat*dstSize_*sizeT) ;
         dataOut=missingValue ;
-        transfer(sizeT, dataIn, dataOut) ;
+        transfer(1, sizeT, dataIn, dataOut) ;
       }
 
@@ -121 +140 @@
           (*subEvent.buffer) >> data ;
         }
-        transfer(sizeT, dataIn, dataOut) ;
+        transfer(1, sizeT, dataIn, dataOut) ;
       }
       
@@ -139 +158 @@
           (*subEvent.buffer) >> data ;
         }
-        transfer(sizeT, dataIn, dataOut, missingValue) ;
+        transfer(1, sizeT, dataIn, dataOut, missingValue) ;
       }
 
@@ -151 +170 @@
           (*subEvent.buffer) >> data ;
         }
-        transfer(1, dataIn, dataOut, missingValue) ;
+        transfer(1, 1, dataIn, dataOut, missingValue) ;
       }