1 | #include "algorithm_transformation_weight.hpp" |
---|
2 | #include "context.hpp" |
---|
3 | |
---|
4 | namespace xios |
---|
5 | { |
---|
6 | |
---|
7 | |
---|
8 | void CAlgorithmTransformationWeight::computeAlgorithm(CLocalView* srcView, CLocalView* dstView) |
---|
9 | { |
---|
10 | auto& srcMap = transformationMapping_ ; |
---|
11 | set<size_t> srcIndex ; |
---|
12 | for(auto& it : srcMap) |
---|
13 | for(size_t index : it.second) srcIndex.insert(index) ; |
---|
14 | |
---|
15 | CArray<size_t,1> srcArrayIndex(srcIndex.size()) ; |
---|
16 | int i=0 ; |
---|
17 | for(size_t index : srcIndex) { srcArrayIndex(i) = index ; i++ ;} |
---|
18 | CLocalElement recvElement(CContext::getCurrent()->getIntraCommRank(), srcView->getGlobalSize(), srcArrayIndex) ; |
---|
19 | recvElement.addFullView() ; |
---|
20 | |
---|
21 | transformConnector_ = new CTransformConnector(srcView, recvElement.getView(CElementView::FULL), CContext::getCurrent()->getIntraComm()) ; |
---|
22 | transformConnector_->computeConnector() ; |
---|
23 | weightTransformConnector_ = new CWeightTransformConnector( recvElement.getView(CElementView::FULL), dstView, transformationMapping_, transformationWeight_) ; |
---|
24 | } |
---|
25 | |
---|
26 | |
---|
27 | void CAlgorithmTransformationWeight::apply(int dimBefore, int dimAfter, const CArray<double,1>& dataIn, CArray<double,1>& dataOut) |
---|
28 | { |
---|
29 | CArray<double,1> dataOutTmp ; |
---|
30 | transformConnector_->transfer(dimBefore, dimAfter, dataIn, dataOutTmp) ; |
---|
31 | weightTransformConnector_ -> transfer(dimBefore, dimAfter, dataOutTmp, dataOut) ; |
---|
32 | } |
---|
33 | |
---|
34 | } |
---|