source: XIOS/trunk/src/transformation/axis_algorithm_reduce_axis.cpp @ 1537

Last change on this file since 1537 was 1314, checked in by ymipsl, 6 years ago

Add 2 new spatial transformations :

  • reduce_scalar_to_scalar : global reduction between scalar
  • duplicate_scalar_to_axis : a scalar value is duplicated on each level of the axis.

YM

  • Property svn:eol-style set to native
File size: 4.5 KB
Line 
1/*!
2   \file axis_algorithm_reduce_axis.cpp
3   \author Ha NGUYEN
4   \since 23 June 2016
5   \date 23 June 2016
6
7   \brief Algorithm for reduce a axis to an axis
8 */
9#include "axis_algorithm_reduce_axis.hpp"
10#include "reduce_axis_to_axis.hpp"
11#include "axis.hpp"
12#include "grid.hpp"
13#include "grid_transformation_factory_impl.hpp"
14#include "reduction.hpp"
15
16namespace xios {
17CGenericAlgorithmTransformation* CAxisAlgorithmReduceAxis::create(CGrid* gridDst, CGrid* gridSrc,
18                                                                   CTransformation<CAxis>* transformation,
19                                                                   int elementPositionInGrid,
20                                                                   std::map<int, int>& elementPositionInGridSrc2ScalarPosition,
21                                                                   std::map<int, int>& elementPositionInGridSrc2AxisPosition,
22                                                                   std::map<int, int>& elementPositionInGridSrc2DomainPosition,
23                                                                   std::map<int, int>& elementPositionInGridDst2ScalarPosition,
24                                                                   std::map<int, int>& elementPositionInGridDst2AxisPosition,
25                                                                   std::map<int, int>& elementPositionInGridDst2DomainPosition)
26{
27  std::vector<CAxis*> axisListDestP = gridDst->getAxis();
28  std::vector<CAxis*> axisListSrcP  = gridSrc->getAxis();
29
30  CReduceAxisToAxis* reduceAxis = dynamic_cast<CReduceAxisToAxis*> (transformation);
31  int axisDstIndex   = elementPositionInGridDst2AxisPosition[elementPositionInGrid];
32  int axisSrcIndex = elementPositionInGridSrc2AxisPosition[elementPositionInGrid];
33
34  return (new CAxisAlgorithmReduceAxis(axisListDestP[axisDstIndex], axisListSrcP[axisSrcIndex], reduceAxis));
35}
36
37bool CAxisAlgorithmReduceAxis::registerTrans()
38{
39  CGridTransformationFactory<CAxis>::registerTransformation(TRANS_REDUCE_AXIS_TO_AXIS, create);
40}
41
42
43CAxisAlgorithmReduceAxis::CAxisAlgorithmReduceAxis(CAxis* axisDestination, CAxis* axisSource, CReduceAxisToAxis* algo)
44 : CAxisAlgorithmTransformation(axisDestination, axisSource), reduction_(0)
45{
46  eliminateRedondantSrc_= false ;
47  algo->checkValid(axisDestination, axisSource);
48  StdString op;
49  switch (algo->operation)
50  {
51    case CReduceAxisToAxis::operation_attr::sum:
52      op = "sum";
53      break;
54    case CReduceAxisToAxis::operation_attr::min:
55      op = "min";
56      break;
57    case CReduceAxisToAxis::operation_attr::max:
58      op = "max";
59      break;
60    case CReduceAxisToAxis::operation_attr::average:
61      op = "average";
62      break;
63    default:
64        ERROR("CAxisAlgorithmReduceAxis::CAxisAlgorithmReduceAxis(CAxis* axisDestination, CAxis* axisSource, CReduceAxisToAxis* algo)",
65         << "Operation is wrongly defined. Supported operations: sum, min, max, average." << std::endl
66         << "Axis source " <<axisSource->getId() << std::endl
67         << "Axis destination " << axisDestination->getId());
68
69  }
70
71  reduction_ = CReductionAlgorithm::createOperation(CReductionAlgorithm::ReductionOperations[op]);
72}
73
74void CAxisAlgorithmReduceAxis::apply(const std::vector<std::pair<int,double> >& localIndex,
75                                       const double* dataInput,
76                                       CArray<double,1>& dataOut,
77                                       std::vector<bool>& flagInitial,                     
78                                       bool ignoreMissingValue, bool firstPass)
79{
80  reduction_->apply(localIndex, dataInput, dataOut, flagInitial, ignoreMissingValue, firstPass);
81}
82
83void CAxisAlgorithmReduceAxis::updateData(CArray<double,1>& dataOut)
84{
85  reduction_->updateData(dataOut);
86}
87
88CAxisAlgorithmReduceAxis::~CAxisAlgorithmReduceAxis()
89{
90  if (0 != reduction_) delete reduction_;
91}
92
93void CAxisAlgorithmReduceAxis::computeIndexSourceMapping_(const std::vector<CArray<double,1>* >& dataAuxInputs)
94{
95  this->transformationMapping_.resize(1);
96  this->transformationWeight_.resize(1);
97
98  TransformationIndexMap& transMap = this->transformationMapping_[0];
99  TransformationWeightMap& transWeight = this->transformationWeight_[0];
100  CArray<int,1>& axisDstIndex = axisDest_->index;
101  int nbAxisIdx = axisDstIndex.numElements();
102  for (int idxAxis = 0; idxAxis < nbAxisIdx; ++idxAxis)
103  {
104    int globalAxisIdx = axisDstIndex(idxAxis);
105    transMap[globalAxisIdx].resize(1);
106    transWeight[globalAxisIdx].resize(1);
107    transMap[globalAxisIdx][0]=globalAxisIdx ;     
108    transWeight[globalAxisIdx][0] = 1.0 ;     
109  }
110}
111
112}
Note: See TracBrowser for help on using the repository browser.