expand.h

Go to the documentation of this file.

#pragma once
 
#include <sstream>
 
#include "tools/math.h"
#include "element.h"
#include "collapse.h" // for ProjectionAxis and ProjectionAxisToString
 
namespace dnf_composer::element
{
    struct ExpandParameters final : ElementSpecificParameters
    {
        ProjectionAxis broadcastProfileAxis; 
        ElementDimensions inputDimensions;   
 
        explicit ExpandParameters(const ProjectionAxis broadcastProfileAxis = ProjectionAxis::X,
            const ElementDimensions& inputDimensions = ElementDimensions{ 100, 1.0 })
            : broadcastProfileAxis(broadcastProfileAxis), inputDimensions(inputDimensions)
        {}
 
        bool operator==(const ExpandParameters& other) const
        {
            return broadcastProfileAxis == other.broadcastProfileAxis &&
                inputDimensions == other.inputDimensions;
        }
 
        [[nodiscard]] std::string toString() const override
        {
            std::ostringstream result;
            result << "Parameters: ["
                << "Profile axis: " << ProjectionAxisToString.at(broadcastProfileAxis) << ", "
                << "Input field dimensions: " << inputDimensions.toString()
                << "]";
            return result.str();
        }
    };
 
    class Expand final : public Element
    {
    private:
        ExpandParameters parameters;
    public:
        Expand(const ElementCommonParameters& elementCommonParameters,
            const ExpandParameters& parameters);
 
        void init() override;
        void step(double t, double deltaT) override;
        void addInput(const std::shared_ptr<Element>& inputElement,
            const std::string& inputComponent = "output") override;
        std::string toString() const override;
        std::shared_ptr<Element> clone() const override;
 
        void changeInputDimensions(const ElementDimensions& newInputDimensions);
 
        void setParameters(const ExpandParameters& parameters);
        [[nodiscard]] ExpandParameters getParameters() const;
    };
}