parallel/pico/FieldExchanger.cpp

// -*- C++ -*-
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
//                                   Fehmi Cirak
//                        California Institute of Technology
//                           (C) 2004 All Rights Reserved
//
//  <LicenseText>
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
// EDITED BY MATT GALLOWAY FOR USE WITH DAPTA

#include "DAPTA.hpp"

#include <cassert>

namespace DAPTA { // Define namespace DAPTA

namespace pico { // Define namespace pico

template <typename V>
FieldExchanger<V>::FieldExchanger(DomainCoupler<V>* coupler, std::vector<int>::iterator lGbegin, int nodes) {
   _coupler = coupler;
   //_coupler->registerSendBuffer<int>(RED, &(*lGbegin), nodes);
   _coupler->template registerSendBuffer<int>(RED, &(*lGbegin), nodes);
   _coupler->exchangeSubdomainDataIntra(RED); 
   
   //int importedNodes = _coupler->recvBufferSize<int>(RED);
   int importedNodes = _coupler->template recvBufferSize<int>(RED);
   assert(importedNodes>0);
   _importedNodeIDs.reserve(importedNodes);
   
   const int notUsed=0;
   int *tmp = _coupler->recvBuffer(RED, notUsed);
   
   for (int i=0; i<importedNodes; ++i) {
      std::vector<int>::iterator it = std::find(lGbegin, lGbegin+nodes, tmp[i]); 
      int localID = std::distance(lGbegin, it);
      if (localID<nodes) {
         _importedNodeIDs.push_back(localID);
      } else { 
         _importedNodeIDs.push_back(-1);
      }
   }
}

template <typename V>
template <int NVAR, typename FT, template <typename> class OP>
void FieldExchanger<V>::exchange(FT *field, int fsize) {    
   // NVAR - number of variables for each node
   // FT - type of the field (double/int)
   // OP - operation to be performend, e.g. std::plus<FT>
   
   _coupler->registerSendBuffer(RED, field, fsize);
   _coupler->exchangeSubdomainDataIntra(RED);
   
   const FT notUsed(0);
   FT *tmp = _coupler->recvBuffer(RED, notUsed);
   
   OP<FT> operation;
   std::vector<int>::iterator it = _importedNodeIDs.begin();
   std::vector<int>::iterator ite = _importedNodeIDs.end();
   for (int i=0; it!=ite; ++it, ++i) {
      if (*it==-1) continue;
      for (int j=0; j<NVAR; ++j) {
         assert((NVAR*(*it)+j)<fsize);
         field[NVAR*(*it)+j] = operation(field[NVAR*(*it)+j], tmp[NVAR*i+j]);
      }
   }
}
   
} // namespace pico

} // namespace DAPTA

---