resize_2d.h

Go to the documentation of this file.

#pragma once
 
#include <cmath>
#include <sstream>
#include <iomanip>
 
#include "tools/math.h"
#include "element.h"
#include "resize.h" // for InterpolationMethod and InterpolationMethodToString
 
namespace dnf_composer::element
{
    struct Resize2DParameters final : ElementSpecificParameters
    {
        InterpolationMethod method;        
        ElementDimensions inputDimensions; 
 
        explicit Resize2DParameters(const InterpolationMethod method = InterpolationMethod::LINEAR,
            const ElementDimensions& inputDimensions = ElementDimensions{ 100, 100, 1.0, 1.0 })
            : method(method), inputDimensions(inputDimensions)
        {}
 
        bool operator==(const Resize2DParameters& other) const
        {
            return method == other.method &&
                inputDimensions == other.inputDimensions;
        }
 
        [[nodiscard]] std::string toString() const override
        {
            std::ostringstream result;
            result << "Parameters: ["
                << "Interpolation method: " << InterpolationMethodToString.at(method) << ", "
                << "Input field dimensions: " << inputDimensions.toString()
                << "]";
            return result.str();
        }
    };
 
    class Resize2D final : public Element
    {
    private:
        Resize2DParameters parameters;
        std::vector<double> scratch;  
        std::vector<double> rowIn;    
        std::vector<double> rowOut;   
        std::vector<double> colIn;    
        std::vector<double> colOut;   
    public:
        Resize2D(const ElementCommonParameters& elementCommonParameters,
            const Resize2DParameters& 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 Resize2DParameters& parameters);
        [[nodiscard]] Resize2DParameters getParameters() const;
    };
}