resize.h

Go to the documentation of this file.

#pragma once
 
#include <cmath>
#include <map>
#include <sstream>
#include <iomanip>
 
#include "tools/math.h"
#include "element.h"
 
namespace dnf_composer::element
{
    enum class InterpolationMethod : int
    {
        LINEAR,  
        NEAREST, 
        CUBIC    
    };
 
    inline const std::map<InterpolationMethod, std::string> InterpolationMethodToString = {
        {InterpolationMethod::LINEAR, "linear"},
        {InterpolationMethod::NEAREST, "nearest"},
        {InterpolationMethod::CUBIC, "cubic"}
    };
 
    struct ResizeParameters final : ElementSpecificParameters
    {
        InterpolationMethod method;        
        ElementDimensions inputDimensions; 
 
        explicit ResizeParameters(const InterpolationMethod method = InterpolationMethod::LINEAR,
            const ElementDimensions& inputDimensions = ElementDimensions{})
            : method(method), inputDimensions(inputDimensions)
        {}
 
        bool operator==(const ResizeParameters& 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 Resize final : public Element
    {
    private:
        ResizeParameters parameters;
    public:
        Resize(const ElementCommonParameters& elementCommonParameters,
            const ResizeParameters& 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 ResizeParameters& parameters);
        [[nodiscard]] ResizeParameters getParameters() const;
    };
}