source: XIOS/dev/branch_openmp/src/transformation/scalar_algorithm_reduce_domain.cpp @ 1545

Last change on this file since 1545 was 1545, checked in by yushan, 6 years ago

branch_openmp merged with trunk r1544

File size: 5.7 KB
Line 
1/*!
2   \file scalar_algorithm_reduce_domain.cpp
3   \author Ha NGUYEN
4   \since 23 June 2016
5   \date 23 June 2016
6
7   \brief Algorithm for reduce a domain to a scalar
8 */
9#include "scalar_algorithm_reduce_domain.hpp"
10#include "domain.hpp"
11#include "scalar.hpp"
12#include "reduce_domain_to_scalar.hpp"
13#include "grid.hpp"
14#include "grid_transformation_factory_impl.hpp"
15
16
17namespace xios {
18CGenericAlgorithmTransformation* CScalarAlgorithmReduceDomain::create(CGrid* gridDst, CGrid* gridSrc,
19                                                                     CTransformation<CScalar>* transformation,
20                                                                     int elementPositionInGrid,
21                                                                     std::map<int, int>& elementPositionInGridSrc2ScalarPosition,
22                                                                     std::map<int, int>& elementPositionInGridSrc2AxisPosition,
23                                                                     std::map<int, int>& elementPositionInGridSrc2DomainPosition,
24                                                                     std::map<int, int>& elementPositionInGridDst2ScalarPosition,
25                                                                     std::map<int, int>& elementPositionInGridDst2AxisPosition,
26                                                                     std::map<int, int>& elementPositionInGridDst2DomainPosition)
27{
28  std::vector<CScalar*> scalarListDestP = gridDst->getScalars();
29  std::vector<CDomain*> domainListSrcP  = gridSrc->getDomains();
30
31  CReduceDomainToScalar* reduceDomain = dynamic_cast<CReduceDomainToScalar*> (transformation);
32  int scalarDstIndex = elementPositionInGridDst2ScalarPosition[elementPositionInGrid];
33  int domainSrcIndex = elementPositionInGridSrc2DomainPosition[elementPositionInGrid];
34
35  return (new CScalarAlgorithmReduceDomain(scalarListDestP[scalarDstIndex], domainListSrcP[domainSrcIndex], reduceDomain));
36}
37
38bool CScalarAlgorithmReduceDomain::registerTrans()
39{
40  CGridTransformationFactory<CScalar>::registerTransformation(TRANS_REDUCE_DOMAIN_TO_SCALAR, create);
41}
42
43CScalarAlgorithmReduceDomain::CScalarAlgorithmReduceDomain(CScalar* scalarDestination, CDomain* domainSource, CReduceDomainToScalar* algo)
44 : CScalarAlgorithmTransformation(scalarDestination, domainSource),
45   reduction_(0)
46{
47  algo->checkValid(scalarDestination, domainSource);
48 
49  StdString op;
50  switch (algo->operation)
51  {
52    case CReduceDomainToScalar::operation_attr::sum:
53      op = "sum";
54      break;
55    case CReduceDomainToScalar::operation_attr::min:
56      op = "min";
57      break;
58    case CReduceDomainToScalar::operation_attr::max:
59      op = "max";
60      break;
61    case CReduceDomainToScalar::operation_attr::average:
62      op = "average";
63      break;
64    default:
65        ERROR("CScalarAlgorithmReduceDomain::CScalarAlgorithmReduceDomain(CDomain* domainDestination, CDomain* domainSource, CReduceDomainToScalar* algo)",
66         << "Operation must be defined."
67         << "Domain source " <<domainSource->getId() << std::endl
68         << "Scalar destination " << scalarDestination->getId());
69
70  }
71 
72  if(CReductionAlgorithm::ReductionOperations_ptr == 0) 
73  {
74    CReductionAlgorithm::initReductionOperation();
75  }
76  if (CReductionAlgorithm::ReductionOperations_ptr->end() == CReductionAlgorithm::ReductionOperations_ptr->find(op))
77    ERROR("CScalarAlgorithmReduceDomain::CScalarAlgorithmReduceDomain(CDomain* domainDestination, CDomain* domainSource, CReduceDomainToScalar* algo)",
78       << "Operation '" << op << "' not found. Please make sure to use a supported one"
79       << "Domain source " <<domainSource->getId() << std::endl
80       << "Scalar destination " << scalarDestination->getId());
81
82  reduction_ = CReductionAlgorithm::createOperation(CReductionAlgorithm::ReductionOperations_ptr->at(op));
83  local = algo->local ;
84}
85
86void CScalarAlgorithmReduceDomain::apply(const std::vector<std::pair<int,double> >& localIndex,
87                                         const double* dataInput,
88                                         CArray<double,1>& dataOut,
89                                         std::vector<bool>& flagInitial,                     
90                                         bool ignoreMissingValue, bool firstPass)
91{
92  reduction_->apply(localIndex, dataInput, dataOut, flagInitial, ignoreMissingValue, firstPass);
93}
94
95void CScalarAlgorithmReduceDomain::updateData(CArray<double,1>& dataOut)
96{
97  reduction_->updateData(dataOut);
98}
99
100CScalarAlgorithmReduceDomain::~CScalarAlgorithmReduceDomain()
101{
102  if (0 != reduction_) delete reduction_;
103}
104
105void CScalarAlgorithmReduceDomain::computeIndexSourceMapping_(const std::vector<CArray<double,1>* >& dataAuxInputs)
106{
107  this->transformationMapping_.resize(1);
108  this->transformationWeight_.resize(1);
109
110  TransformationIndexMap& transMap = this->transformationMapping_[0];
111  TransformationWeightMap& transWeight = this->transformationWeight_[0];
112
113  int ni_glo = domainSrc_->ni_glo ;
114  int nj_glo = domainSrc_->nj_glo ;
115  int nbDomainIdx ;
116 
117  if (local)
118  {
119      const CArray<int, 1>& i_index = domainSrc_-> i_index.getValue() ;
120      const CArray<int, 1>& j_index = domainSrc_-> j_index.getValue() ;
121      const CArray<bool,1>& localMask = domainSrc_-> localMask ;
122      int nbDomainIdx = i_index.numElements();
123
124      for (int idxDomain = 0; idxDomain < nbDomainIdx; ++idxDomain)
125      {
126        if (localMask(idxDomain))
127        { 
128          transMap[0].push_back(j_index(idxDomain)* ni_glo + i_index(idxDomain));
129          transWeight[0].push_back(1.0) ;
130        }
131      }
132  }
133  else
134  { 
135    nbDomainIdx = ni_glo * nj_glo;
136    transMap[0].resize(nbDomainIdx);
137    transWeight[0].resize(nbDomainIdx, 1.0);
138    for (int idxDomain = 0; idxDomain < nbDomainIdx; ++idxDomain) transMap[0][idxDomain] = idxDomain;   
139  }
140 
141}
142
143}
Note: See TracBrowser for help on using the repository browser.