sum_reduction.cpp

Aller à la documentation de ce fichier.

#include "sum_reduction.hpp"
#include "utils.hpp"

namespace xios {

CSumReductionAlgorithm::CSumReductionAlgorithm()
  : CReductionAlgorithm()
{
}

CReductionAlgorithm* CSumReductionAlgorithm::create()
{
  return (new CSumReductionAlgorithm());
}

bool CSumReductionAlgorithm::registerTrans()
{
  return registerOperation(TRANS_REDUCE_SUM, CSumReductionAlgorithm::create);
}

void CSumReductionAlgorithm::apply(const std::vector<std::pair<int,double> >& localIndex,
                                   const double* dataInput,
                                   CArray<double,1>& dataOut,
                                   std::vector<bool>& flagInitial,
                                   bool ignoreMissingValue, bool firstPass)
{
  if (ignoreMissingValue)
  {
    int nbLocalIndex = localIndex.size();
    int currentlocalIndex = 0;

    if (firstPass) dataOut=std::numeric_limits<double>::quiet_NaN();
  
    for (int idx = 0; idx < nbLocalIndex; ++idx)
    {
      currentlocalIndex = localIndex[idx].first;   
      if (!NumTraits<double>::isNan(*(dataInput + idx)))
      {   
        if (flagInitial[currentlocalIndex])
        {
          dataOut(currentlocalIndex) = *(dataInput + idx);
          flagInitial[currentlocalIndex] = false;
        }
        else
        {
          dataOut(currentlocalIndex) += *(dataInput + idx);
        }
      }
    }    
  }
  else
  {
    int nbLocalIndex = localIndex.size();
    int currentlocalIndex = 0;    
    for (int idx = 0; idx < nbLocalIndex; ++idx)
    {
      currentlocalIndex = localIndex[idx].first;      
      if (flagInitial[currentlocalIndex])
      {
        dataOut(currentlocalIndex) = *(dataInput + idx);
        flagInitial[currentlocalIndex] = false;
      }
      else
      {
        dataOut(currentlocalIndex) += *(dataInput + idx);
      }
    }
  }
}

}