Map3to1to3.h

Go to the documentation of this file.

#ifndef included_AMP_Map3to1to3
#define included_AMP_Map3to1to3
 
#include <map>
#include <vector>
 
#include "AMP/mesh/MeshIterator.h"
#include "AMP/operators/map/AsyncMapOperator.h"
#include "AMP/operators/map/AsyncMapOperatorParameters.h"
 
 
namespace AMP::Operator {
 
class Map3to1to3 : public AsyncMapOperator
{
public:
    explicit Map3to1to3( std::shared_ptr<const OperatorParameters> params );
 
    virtual ~Map3to1to3();
 
    std::string type() const override { return "Map3to1to3"; }
 
    void setVector( AMP::LinearAlgebra::Vector::shared_ptr result ) override;
 
    void applyStart( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                     AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
    void applyFinish( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                      AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
 
protected:
    void addTo1DMap( std::multimap<double, double> &map, double z, double val );
 
    void addTo1DMap( std::multimap<double, double> &map,
                     const std::vector<double> &z,
                     const std::vector<double> &val );
 
    virtual std::multimap<double, double>
    buildMap( AMP::LinearAlgebra::Vector::const_shared_ptr vec,
              const std::shared_ptr<AMP::Mesh::Mesh> mesh,
              const AMP::Mesh::MeshIterator &it );
 
    virtual void buildReturn( AMP::LinearAlgebra::Vector::shared_ptr vec,
                              const std::shared_ptr<AMP::Mesh::Mesh> mesh,
                              const AMP::Mesh::MeshIterator &it,
                              const std::map<double, double> &map );
 
    AMP::Mesh::MeshIterator d_srcIterator1;
    AMP::Mesh::MeshIterator d_srcIterator2;
    AMP::Mesh::MeshIterator d_dstIterator1;
    AMP::Mesh::MeshIterator d_dstIterator2;
 
private:
    AMP::LinearAlgebra::Vector::shared_ptr d_ResultVector;
 
    std::vector<bool> d_own_mesh1;
    std::vector<bool> d_own_mesh2;
 
    int d_commTag;
 
    struct comm_data {
        int N;
        double z;
        double sum;
        comm_data()
        {
            N   = 0;
            z   = 0.0;
            sum = 0.0;
        }
    };
 
    std::vector<comm_data> d_SendBuf1;
    std::vector<comm_data> d_SendBuf2;
 
    // Function to unpack the recv buffer
    static void unpackBuffer( const std::vector<comm_data> &,
                              std::map<double, std::pair<int, double>> & );
};
 
 
} // namespace AMP::Operator
 
#endif