Changeset 2482 for XIOS3/dev/XIOS_ATTACHED/src/transformation/scalar_algorithm/scalar_algorithm_redistribute.cpp

Timestamp:

03/28/23 16:42:11 (15 months ago)

Author:

ymipsl

Message:

First guess in supression of attached mode replaced by online reader and write filters

YM

File:

• XIOS3/dev/XIOS_ATTACHED/src/transformation/scalar_algorithm/scalar_algorithm_redistribute.cpp (modified) (2 diffs)

XIOS3/dev/XIOS_ATTACHED/src/transformation/scalar_algorithm/scalar_algorithm_redistribute.cpp

@@ -8 +8 @@
 #include "grid.hpp"
 #include "grid_transformation_factory_impl.hpp"
+#include "context.hpp"
 
 namespace xios
@@ -47 +48 @@
   {
 
+    CContext* context = CContext::getCurrent(); 
+
+    scalarDestination->n.reset();
+    scalarDestination->mask.reset();
+    scalarDestination->value.reset();
+    scalarDestination->bounds.reset();
+
+    redistributeScalar->checkValid(scalarSource);
+
+    // define the global index of the new domain
+
+    CArray<size_t,1> globalIndex ;
+
+    if (    redistributeScalar->type == CRedistributeScalar::type_attr::bands 
+         || redistributeScalar->type == CRedistributeScalar::type_attr::column 
+         || redistributeScalar->type == CRedistributeScalar::type_attr::root )
+    {
+      if (context->getIntraCommRank()==0)
+      {
+        scalarDestination->n = 1;
+        globalIndex.resize(1) ;
+        globalIndex(0)=0 ;
+      }
+      else
+      {
+        scalarDestination->n = 0;
+        globalIndex.resize(0) ;
+      }
+    }
+    else if (redistributeScalar->type == CRedistributeScalar::type_attr::full)
+    {
+      scalarDestination->n = 1;
+      globalIndex.resize(1) ;
+      globalIndex(0)=0 ;
+    }
+    
+    auto& transMap = this->transformationMapping_;
+    for(int i=0; i<globalIndex.numElements(); i++) transMap[globalIndex(i)]=globalIndex(i) ;
+
+    auto elementDst=shared_ptr<CLocalElement>(new CLocalElement(context->getIntraCommRank(), 1, globalIndex)) ;
+    elementDst->addFullView() ;
+
+    auto transformConnector = make_shared<CTransformConnector>(scalarSource->getLocalView(CElementView::FULL), elementDst->getView(CElementView::FULL),context->getIntraComm()) ;
+    transformConnector->computeConnector() ;
+
+    CArray<double,1> valSrc, valDst ;
+    valSrc.resize(scalarSource->getLocalView(CElementView::FULL)->getSize()) ;
+    valDst.resize(scalarDestination->getLocalView(CElementView::FULL)->getSize()) ;
+    
+    if (scalarSource->hasValue())
+    {
+      if (valSrc.numElements()>0)  valSrc(0)=scalarSource->value ;
+      transformConnector->transfer(valSrc, valDst) ;
+      if (valDst.numElements()>0)  scalarDestination->value = valDst(0) ;
+    }
+
+    if (scalarSource->hasBounds()) 
+    {
+      transformConnector->transfer(2, scalarSource->bounds, scalarDestination->bounds) ;
+    }
+    
+    if (scalarSource->hasLabel())
+    {
+//ym transferTransformConnector.transfer(axisSrc->label, axisDestination->label) ;  -> probably not supported now, see later
+    }
+
+    // transfer the mask
+    auto transformMask = make_shared<CTransformConnector>(scalarSource->getLocalView(CElementView::WORKFLOW), elementDst->getView(CElementView::FULL),context->getIntraComm()) ;
+    transformMask->computeConnector() ;
+
+    CArray<bool,1> workflow(scalarSource->getLocalView(CElementView::WORKFLOW)->getSize()) ;
+    CArray<bool,1> mask ;
+    mask.resize(scalarSource->getLocalView(CElementView::FULL)->getSize()) ;
+    workflow=true ;
+    
+    transformMask->transfer(workflow, mask, false) ;   
+    scalarDestination->mask = mask(0) ;
 
     scalarDestination->checkAttributes() ;