SubchannelToCladMap.h

Go to the documentation of this file.

#ifndef included_AMP_SubchannelToCladMap
#define included_AMP_SubchannelToCladMap
 
#include "AMP/operators/map/ScalarZAxisMap.h"
 
namespace AMP::Operator {
 
 
typedef AsyncMapOperatorParameters SubchannelToCladMapParameters;
 
 
class SubchannelToCladMap : public AMP::Operator::AsyncMapOperator
{
public:
    static bool validMapType( const std::string &s );
 
    typedef SubchannelToCladMapParameters Parameters;
 
    explicit SubchannelToCladMap( std::shared_ptr<const AMP::Operator::OperatorParameters> params );
 
    virtual ~SubchannelToCladMap();
 
    std::string type() const override { return "SubchannelToCladMap"; }
 
    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:
    // Return the list of local MeshElements in each subchannel
    virtual std::vector<std::vector<AMP::Mesh::MeshElementID>>
    getElementsInSubchannel( const std::vector<double> &x,
                             const std::vector<double> &y,
                             AMP::Mesh::MeshIterator iterator );
 
    // For a given subchannel, fill the elements of interest using the coordinates
    virtual void fillReturnVector( AMP::LinearAlgebra::Vector::shared_ptr vec,
                                   double range[4],
                                   std::shared_ptr<AMP::Mesh::Mesh> mesh,
                                   const std::vector<AMP::Mesh::MeshElementID> &ids,
                                   const std::vector<double> &z,
                                   const std::vector<double> &f );
 
    // Iterators over the mesh elemens of interest
    AMP::Mesh::MeshIterator d_iterator1;
    AMP::Mesh::MeshIterator d_iterator2;
 
private:
    // Function to fill the subchannel data for all processors
    void fillSubchannelGrid( std::shared_ptr<AMP::Mesh::Mesh> );
    // The total number of subchannels
    size_t N_subchannels;
    // The x, y, z grid for the subchannel
    std::vector<double> d_x, d_y, d_z;
    // Which subchannels do I own (multple procs my own a subchannel)
    std::vector<bool> d_ownSubChannel;
    // The processors that own each x-y point in the subchannel
    std::vector<std::vector<int>> d_subchannelRanks;
    // The processors that are recieving data from each subchannel
    std::vector<std::vector<int>> d_subchannelRecv;
 
    // Function to get the iterator for the subchannel mesh
    static AMP::Mesh::MeshIterator getSubchannelIterator( std::shared_ptr<AMP::Mesh::Mesh> );
 
    // The list of local MeshElements in each subchannel
    std::vector<std::vector<AMP::Mesh::MeshElementID>> d_elem;
 
    // Buffers to send/recv the data
    std::vector<AMP_MPI::Request> d_currRequests;
    std::vector<std::vector<double>> d_sendBuffer;
 
    int getSubchannelIndex( double x, double y );
};
 
 
} // namespace AMP::Operator
 
#endif