| 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 | |
|---|
| 16 | namespace xios { |
|---|
| 17 | |
|---|
| 18 | /*! |
|---|
| 19 | \class CReductionAlgorithm |
|---|
| 20 | Interface for all reduction alogrithms. |
|---|
| 21 | */ |
|---|
| 22 | class CReductionAlgorithm |
|---|
| 23 | { |
|---|
| 24 | public: |
|---|
| 25 | //static std::map<StdString,EReductionType> ReductionOperations; |
|---|
| 26 | static std::map<StdString,EReductionType> *ReductionOperations_ptr; |
|---|
| 27 | #pragma omp threadprivate(ReductionOperations_ptr) |
|---|
| 28 | |
|---|
| 29 | public: |
|---|
| 30 | CReductionAlgorithm() {} |
|---|
| 31 | |
|---|
| 32 | /*! |
|---|
| 33 | Create an operation (sum, max, min) based on type |
|---|
| 34 | \param [in] reduceType type to create |
|---|
| 35 | return pointer to base class |
|---|
| 36 | */ |
|---|
| 37 | static CReductionAlgorithm* createOperation(EReductionType reduceType); |
|---|
| 38 | |
|---|
| 39 | /*! |
|---|
| 40 | Apply a reduction operation on local data. |
|---|
| 41 | \param [in] localIndex vector contains local index of local data output and the corresponding weight |
|---|
| 42 | \param [in] dataInput Pointer to the first element of data input array (in form of buffer) |
|---|
| 43 | \param [in/out] dataOut Array contains local data |
|---|
| 44 | \param [in/out] flagInitial vector of boolean to mark the local index already initialized. True means there is a need for initialization |
|---|
| 45 | */ |
|---|
| 46 | virtual void apply(const std::vector<std::pair<int,double> >& localIndex, |
|---|
| 47 | const double* dataInput, |
|---|
| 48 | CArray<double,1>& dataOut, |
|---|
| 49 | std::vector<bool>& flagInitial) = 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 | |
|---|
| 59 | protected: |
|---|
| 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 | |
|---|
| 68 | protected: |
|---|
| 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__ |
|---|
