source: XIOS/dev/branch_yushan_merged/src/transformation/Functions/sum_reduction.cpp @ 1205

Last change on this file since 1205 was 1205, checked in by yushan, 7 years ago

branch merged with trunk @1200

File size: 2.0 KB
Line 
1/*!
2   \file sum.cpp
3   \author Ha NGUYEN
4   \since 27 June 2016
5   \date 9 Jan 2017
6
7   \brief sum reduction
8 */
9#include "sum_reduction.hpp"
10#include "utils.hpp"
11
12namespace xios {
13
14CSumReductionAlgorithm::CSumReductionAlgorithm()
15  : CReductionAlgorithm()
16{
17}
18
19CReductionAlgorithm* CSumReductionAlgorithm::create()
20{
21  return (new CSumReductionAlgorithm());
22}
23
24bool CSumReductionAlgorithm::registerTrans()
25{
26  return registerOperation(TRANS_REDUCE_SUM, CSumReductionAlgorithm::create);
27}
28
29void CSumReductionAlgorithm::apply(const std::vector<std::pair<int,double> >& localIndex,
30                                   const double* dataInput,
31                                   CArray<double,1>& dataOut,
32                                   std::vector<bool>& flagInitial,
33                                   bool ignoreMissingValue)
34{
35  if (ignoreMissingValue)
36  {
37    int nbLocalIndex = localIndex.size();
38    int currentlocalIndex = 0;   
39 
40    for (int idx = 0; idx < nbLocalIndex; ++idx)
41    {
42      currentlocalIndex = localIndex[idx].first;   
43      if (!NumTraits<double>::isnan(*(dataInput + idx)))
44      {   
45        if (flagInitial[currentlocalIndex])
46        {
47          dataOut(currentlocalIndex) = *(dataInput + idx);
48          flagInitial[currentlocalIndex] = false;
49        }
50        else
51        {
52          dataOut(currentlocalIndex) += *(dataInput + idx);
53        }
54      }
55      else
56      {
57        if (flagInitial[currentlocalIndex]) 
58          dataOut(currentlocalIndex) = std::numeric_limits<double>::quiet_NaN();
59      }
60    }   
61  }
62  else
63  {
64    int nbLocalIndex = localIndex.size();
65    int currentlocalIndex = 0;   
66    for (int idx = 0; idx < nbLocalIndex; ++idx)
67    {
68      currentlocalIndex = localIndex[idx].first;     
69      if (flagInitial[currentlocalIndex])
70      {
71        dataOut(currentlocalIndex) = *(dataInput + idx);
72        flagInitial[currentlocalIndex] = false;
73      }
74      else
75      {
76        dataOut(currentlocalIndex) += *(dataInput + idx);
77      }
78    }
79  }
80}
81
82}
Note: See TracBrowser for help on using the repository browser.