source: XIOS/dev/dev_trunk_omp/src/transformation/Functions/sum_reduction.cpp @ 1677

/*!
   \file sum.cpp
   \author Ha NGUYEN
   \since 27 June 2016
   \date 9 Jan 2017

   \brief sum reduction
 */
#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)
{
  std::cout<<"================================ CSumReductionAlgorithm::apply"<<std::endl;
  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);
      }
    }
  }
}

}