#ifndef _BGC_AFFINE2_H_INCLUDED_
#define _BGC_AFFINE2_H_INCLUDED_

#include "vector2.h"
#include "matrixes.h"
#include "matrix2x2.h"

// ==================== Types ==================== //

typedef struct {
    BGC_FP32_Matrix2x2 distortion;
    BGC_FP32_Vector2 shift;
} BGC_FP32_Affine2;

typedef struct {
    BGC_FP64_Matrix2x2 distortion;
    BGC_FP64_Vector2 shift;
} BGC_FP64_Affine2;

// ==================== Reset ==================== //

inline void bgc_fp32_affine2_reset(BGC_FP32_Affine2 * affine)
{
    bgc_fp32_matrix2x2_make_identity(&affine->distortion);
    bgc_fp32_vector2_reset(&affine->shift);
}

inline void bgc_fp64_affine2_reset(BGC_FP64_Affine2 * affine)
{
    bgc_fp64_matrix2x2_make_identity(&affine->distortion);
    bgc_fp64_vector2_reset(&affine->shift);
}

// ==================== Make ===================== //

inline void bgc_fp32_affine2_make(const BGC_FP32_Matrix2x2 * distortion, const BGC_FP32_Vector2 * shift, BGC_FP32_Affine2 * affine)
{
    bgc_fp32_matrix2x2_copy(distortion, &affine->distortion);
    bgc_fp32_vector2_copy(shift, &affine->shift);
}

inline void bgc_fp64_affine2_make(const BGC_FP64_Matrix2x2 * distortion, const BGC_FP64_Vector2 * shift, BGC_FP64_Affine2 * affine)
{
    bgc_fp64_matrix2x2_copy(distortion, &affine->distortion);
    bgc_fp64_vector2_copy(shift, &affine->shift);
}


// ==================== Copy ===================== //

inline void bgc_fp32_affine2_copy(const BGC_FP32_Affine2 * source, BGC_FP32_Affine2 * destination)
{
    bgc_fp32_matrix2x2_copy(&source->distortion, &destination->distortion);
    bgc_fp32_vector2_copy(&source->shift, &destination->shift);
}

inline void bgc_fp64_affine2_copy(const BGC_FP64_Affine2 * source, BGC_FP64_Affine2 * destination)
{
    bgc_fp64_matrix2x2_copy(&source->distortion, &destination->distortion);
    bgc_fp64_vector2_copy(&source->shift, &destination->shift);
}

// ==================== Swap ===================== //

inline void bgc_fp32_affine2_swap(BGC_FP32_Affine2 * first, BGC_FP32_Affine2 * second)
{
    bgc_fp32_matrix2x2_swap(&first->distortion, &second->distortion);
    bgc_fp32_vector2_swap(&first->shift, &second->shift);
}

inline void bgc_fp64_affine2_swap(BGC_FP64_Affine2 * first, BGC_FP64_Affine2 * second)
{
    bgc_fp64_matrix2x2_swap(&first->distortion, &second->distortion);
    bgc_fp64_vector2_swap(&first->shift, &second->shift);
}

// =================== Convert =================== //

inline void bgc_fp64_affine2_convert_to_fp32(const BGC_FP64_Affine2 * source, BGC_FP32_Affine2 * destination)
{
    bgc_fp64_matrix2x2_convert_to_fp32(&source->distortion, &destination->distortion);
    bgc_fp64_vector2_convert_to_fp32(&source->shift, &destination->shift);
}

inline void bgc_fp32_affine2_convert_to_fp64(const BGC_FP32_Affine2 * source, BGC_FP64_Affine2 * destination)
{
    bgc_fp32_matrix2x2_convert_to_fp64(&source->distortion, &destination->distortion);
    bgc_fp32_vector2_convert_to_fp64(&source->shift, &destination->shift);
}

// =================== Invert ==================== //

inline int bgc_fp32_affine2_invert(BGC_FP32_Affine2 * affine)
{
    if (!bgc_fp32_matrix2x2_invert(&affine->distortion)) {
        return 0;
    }

    bgc_fp32_multiply_matrix2x2_by_vector2(&affine->distortion, &affine->shift, &affine->shift);
    bgc_fp32_vector2_revert(&affine->shift);

    return 1;
}

inline int bgc_fp64_affine2_invert(BGC_FP64_Affine2 * affine)
{
    if (!bgc_fp64_matrix2x2_invert(&affine->distortion)) {
        return 0;
    }

    bgc_fp64_multiply_matrix2x2_by_vector2(&affine->distortion, &affine->shift, &affine->shift);
    bgc_fp64_vector2_revert(&affine->shift);

    return 1;
}

// ================= Get Inverse ================= //

inline int bgc_fp32_affine2_get_inverse(const BGC_FP32_Affine2 * source, BGC_FP32_Affine2 * destination)
{
    if (!bgc_fp32_matrix2x2_get_inverse(&source->distortion, &destination->distortion)) {
        return 0;
    }

    bgc_fp32_multiply_matrix2x2_by_vector2(&destination->distortion, &source->shift, &destination->shift);
    bgc_fp32_vector2_revert(&destination->shift);

    return 1;
}

inline int bgc_fp64_affine2_get_inverse(const BGC_FP64_Affine2 * source, BGC_FP64_Affine2 * destination)
{
    if (!bgc_fp64_matrix2x2_get_inverse(&source->distortion, &destination->distortion)) {
        return 0;
    }

    bgc_fp64_multiply_matrix2x2_by_vector2(&destination->distortion, &source->shift, &destination->shift);
    bgc_fp64_vector2_revert(&destination->shift);

    return 1;
}

// =================== Combine =================== //

inline void bgc_fp32_affine2_combine(const BGC_FP32_Affine2 * first, const BGC_FP32_Affine2 * second, BGC_FP32_Affine2 * combination)
{
    BGC_FP32_Vector2 first_shift;
    bgc_fp32_multiply_matrix2x2_by_vector2(&second->distortion, &first->shift, &first_shift);
    bgc_fp32_multiply_matrix2x2_by_matrix2x2(&second->distortion, &first->distortion, &combination->distortion);
    bgc_fp32_vector2_add(&first_shift, &second->shift, &combination->shift);
}

inline void bgc_fp64_affine2_combine(const BGC_FP64_Affine2 * first, const BGC_FP64_Affine2 * second, BGC_FP64_Affine2 * combination)
{
    BGC_FP64_Vector2 first_shift;
    bgc_fp64_multiply_matrix2x2_by_vector2(&second->distortion, &first->shift, &first_shift);
    bgc_fp64_multiply_matrix2x2_by_matrix2x2(&second->distortion, &first->distortion, &combination->distortion);
    bgc_fp64_vector2_add(&first_shift, &second->shift, &combination->shift);
}

// =============== Transform Point =============== //

inline void bgc_fp32_affine2_transform_point(const BGC_FP32_Affine2 * affine, const BGC_FP32_Vector2 * initial_point, BGC_FP32_Vector2 * transformed_point)
{
    BGC_FP32_Vector2 distorted;
    bgc_fp32_multiply_matrix2x2_by_vector2(&affine->distortion, initial_point, &distorted);
    bgc_fp32_vector2_add(&affine->shift, &distorted, transformed_point);
}

inline void bgc_fp64_affine2_transform_point(const BGC_FP64_Affine2 * affine, const BGC_FP64_Vector2 * initial_point, BGC_FP64_Vector2 * transformed_point)
{
    BGC_FP64_Vector2 distorted;
    bgc_fp64_multiply_matrix2x2_by_vector2(&affine->distortion, initial_point, &distorted);
    bgc_fp64_vector2_add(&affine->shift, &distorted, transformed_point);
}

// ============== Transform Vector =============== //

inline void bgc_fp32_affine2_transform_vector(const BGC_FP32_Affine2 * affine, const BGC_FP32_Vector2 * initial_vector, BGC_FP32_Vector2 * transformed_vector)
{
    bgc_fp32_multiply_matrix2x2_by_vector2(&affine->distortion, initial_vector, transformed_vector);
}

inline void bgc_fp64_affine2_transform_vector(const BGC_FP64_Affine2 * affine, const BGC_FP64_Vector2 * initial_vector, BGC_FP64_Vector2 * transformed_vector)
{
    bgc_fp64_multiply_matrix2x2_by_vector2(&affine->distortion, initial_vector, transformed_vector);
}

#endif