source: XIOS/dev/dev_trunk_omp/src/transformation/Functions/reduction.hpp @ 1601

Last change on this file since 1601 was 1601, checked in by yushan, 5 years ago

branch_openmp merged with trunk r1597

File size: 2.6 KB
Line 
1/*!
2   \file reduction.hpp
3   \author Ha NGUYEN
4   \since 27 June 2016
5   \date 27 June 2016
6
7   \brief Parent class for all reduction
8 */
9#ifndef __XIOS_REDUCTION_ALGORITHM_HPP__
10#define __XIOS_REDUCTION_ALGORITHM_HPP__
11
12#include <vector>
13#include "array_new.hpp"
14#include "reduction_types.hpp"
15
16namespace xios {
17
18/*!
19  \class CReductionAlgorithm
20  Interface for all reduction alogrithms.
21*/
22class CReductionAlgorithm
23{
24public:
25  static std::map<StdString,EReductionType> *ReductionOperations_ptr;
26  #pragma omp threadprivate(ReductionOperations_ptr)
27public:
28  CReductionAlgorithm() {}
29
30  /*!
31    Create an operation (sum, max, min) based on type
32    \param [in] reduceType type to create
33    return pointer to base class
34  */
35  static CReductionAlgorithm* createOperation(EReductionType reduceType);
36
37  /*!
38    Apply a reduction operation on local data.
39    \param [in] localIndex vector contains local index of local data output and the corresponding weight
40    \param [in] dataInput Pointer to the first element of data input array (in form of buffer)
41    \param [in/out] dataOut Array contains local data
42    \param [in/out] flagInitial vector of boolean to mark the local index already initialized. True means there is a need for initialization
43    \param [in] firstPass indicate if it is the first time the apply funtion is called for a same transformation, in order to make a clean initialization
44  */
45  virtual void apply(const std::vector<std::pair<int,double> >& localIndex,
46                     const double* dataInput,
47                     CArray<double,1>& dataOut,
48                     std::vector<bool>& flagInitial,                     
49                     bool ignoreMissingValue, bool firstPass) = 0;
50  /*!
51    Update local data
52    In some case (e.g average) we need global information (e.g weights) then update data with this information
53    \param [in] dataOut local data output
54  */
55  virtual void updateData(CArray<double,1>& dataOut) {}
56
57  virtual ~CReductionAlgorithm() {}
58
59protected:
60  typedef CReductionAlgorithm* (*CreateOperationCallBack)();
61  typedef std::map<EReductionType, CreateOperationCallBack> CallBackMap;
62  static CallBackMap* reductionCreationCallBacks_;
63  #pragma omp threadprivate(reductionCreationCallBacks_)
64
65  static bool registerOperation(EReductionType reduceType, CreateOperationCallBack createFn);
66  static bool unregisterOperation(EReductionType reduceType);
67
68protected:
69  static bool initReductionOperation(std::map<StdString,EReductionType>& m);
70  static bool initReductionOperation();
71  static bool _dummyInit;
72  #pragma omp threadprivate(_dummyInit)
73};
74
75}
76#endif // __XIOS_REDUCTION_ALGORITHM_HPP__
Note: See TracBrowser for help on using the repository browser.