#pragma once
#include "node.h"
#include "parameter_factory.h"
#include "parameter_vx.h"
#include "graph.h"

class WarpAffineNode : public Node
{
public:
    void create(std::shared_ptr<Graph> graph) override;
    WarpAffineNode(const std::vector<Image*>& inputs, const std::vector<Image*>& outputs);
    WarpAffineNode() = delete;
    void init(float x0, float x1, float y0, float y1, float o0, float o1);
    void init(FloatParam* x0, FloatParam* x1, FloatParam* y0, FloatParam* y1, FloatParam* o0, FloatParam* o1);
    void update_parameters() override;
private:
    ParameterVX<float> _x0;
    ParameterVX<float> _x1;
    ParameterVX<float> _y0;
    ParameterVX<float> _y1;
    ParameterVX<float> _o0;
    ParameterVX<float> _o1;
    constexpr static unsigned COEFFICIENT_X0_OVX_PARAM_IDX = 4;
    constexpr static unsigned COEFFICIENT_X1_OVX_PARAM_IDX = 7;
    constexpr static unsigned COEFFICIENT_Y0_OVX_PARAM_IDX = 5;
    constexpr static unsigned COEFFICIENT_Y1_OVX_PARAM_IDX = 8;
    constexpr static unsigned COEFFICIENT_O0_OVX_PARAM_IDX = 6;
    constexpr static unsigned COEFFICIENT_O1_OVX_PARAM_IDX = 9;
    constexpr static float COEFFICIENT_RANGE_0 [2] = {-0.35, 0.35};
    constexpr static float COEFFICIENT_RANGE_1 [2] = {0.65, 1.35};
    constexpr static float COEFFICIENT_RANGE_OFFSET [2] = {-10.0, 10.0};
};