StridedZAxisMap.h

Go to the documentation of this file.

#ifndef included_AMP_StridedZAxisMap
#define included_AMP_StridedZAxisMap
 
#include "AMP/operators/map/Map3to1to3.h"
#include "AMP/operators/map/Map3to1to3Parameters.h"
#include "AMP/operators/map/ScalarZAxisMap.h"
 
 
namespace AMP::Operator {
 
typedef AMP::Operator::Map3to1to3Parameters ScalarZAxisMapParameters;
 
class StridedZAxisMap : public AMP::Operator::ScalarZAxisMap
{
public:
    explicit StridedZAxisMap( std::shared_ptr<const AMP::Operator::OperatorParameters> params );
 
    virtual ~StridedZAxisMap();
 
    std::string type() const override { return "StridedZAxisMap"; }
 
    static bool validMapType( const std::string &s );
 
    // Overload the apply
    void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
    void setVector( AMP::LinearAlgebra::Vector::shared_ptr p ) override;
 
protected:
    int d_inpDofs, d_inpStride, d_outDofs, d_outStride;
};
 
 
} // namespace AMP::Operator
 
#endif