pico/pico/FieldExchanger.h

Go to the documentation of this file.

// -*- C++ -*-
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
//                                   Fehmi Cirak
//                        California Institute of Technology
//                           (C) 2003 All Rights Reserved
//
//  <LicenseText>
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//

#ifndef _FieldExchanger_h_
#define _FieldExchanger_h_

#include "definitions.h"
#include "DomainCoupler.h"

#include <vector>
#include <functional>
#include <cassert>


namespace pico {
    class FieldExchanger;
}


class pico::FieldExchanger{
public:    
    FieldExchanger(pico::DomainCoupler* coupler, 
                   std::vector<int>::iterator lGBegin, int nodes);
    ~FieldExchanger() {}
   
    template <int NVAR, typename FT, template <typename> class OP>
    void exchange(FT *field, int fsize);
 
// copy and equality constructor
private:
    FieldExchanger(const FieldExchanger &);
    const FieldExchanger & operator=(const FieldExchanger &);
   
private:
    // mapping from imported node id's to local node id's
    std::vector<int>        _importedNodeIDs;

    // domain coupler instance which handles the communication
    pico::DomainCoupler     *_coupler;
};



// templated function
namespace pico {

    template <int NVAR, typename FT, template <typename> class OP>
    void FieldExchanger::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(pico::RED, field, fsize);
        _coupler->exchangeSubdomainDataIntra(pico::RED); 
        
        const FT notUsed(0);
        FT *tmp = _coupler->recvBuffer(pico::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]);
            }
        }       
        return;
    }
}

#endif

//  End of file

---