#ifndef _BGC_AFFINE2_H_INCLUDED_
#define _BGC_AFFINE2_H_INCLUDED_

#include "vector2.h"
#include "matrices.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(BGC_FP32_Affine2* affine, const BGC_FP32_Matrix2x2* distortion, const BGC_FP32_Vector2* shift)
{
    bgc_fp32_matrix2x2_copy(&affine->distortion, distortion);
    bgc_fp32_vector2_copy(&affine->shift, shift);
}

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


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

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

inline void bgc_fp64_affine2_copy(BGC_FP64_Affine2* destination, const BGC_FP64_Affine2* source)
{
    bgc_fp64_matrix2x2_copy(&destination->distortion, &source->distortion);
    bgc_fp64_vector2_copy(&destination->shift, &source->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_fp32_affine2_convert_to_fp64(BGC_FP64_Affine2* destination, const BGC_FP32_Affine2* source)
{
    bgc_fp32_matrix2x2_convert_to_fp64(&destination->distortion, &source->distortion);
    bgc_fp32_vector2_convert_to_fp64(&destination->shift, &source->shift);
}

inline void bgc_fp64_affine2_convert_to_fp32(BGC_FP32_Affine2* destination, const BGC_FP64_Affine2 * source)
{
    bgc_fp64_matrix2x2_convert_to_fp32(&destination->distortion, &source->distortion);
    bgc_fp64_vector2_convert_to_fp32(&destination->shift, &source->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->shift, &affine->distortion, &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->shift, &affine->distortion, &affine->shift);
    bgc_fp64_vector2_revert(&affine->shift);

    return 1;
}

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

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

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

    return 1;
}

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

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

    return 1;
}

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

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

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

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

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

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

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

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

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

#endif