#include "complimentary_colour_extractor.h"

#include <math.h>

#include <stdio.h>

// CONSTRUCTORS
ComplimentaryColourExtractor::ComplimentaryColourExtractor(struct ColourHSL hsl,
	size_t h_levels, size_t s_levels, size_t v_levels)
:ColourExtractor(h_levels, s_levels, v_levels), m_main_colour(hsl)
{
}

// PROTECTED FUNCS
size_t ComplimentaryColourExtractor::evaluate_colour(struct ColourHSL hsl,
	double weight) const
{
	size_t result = 0;
	double angle_delta;
	// double additional_colour_weight;

	if ((0 == (1.0 - fabs(2 * m_main_colour.l - 1.0))) ||
		(0 == (1.0 - fabs(2 * hsl.l - 1.0))) ||
		(0 == m_main_colour.s) ||
		(0 == hsl.s))
	{
		angle_delta = 90.0;
		// additional_colour_weight = 0.5;
	}
	else
	{
		// additional_colour_weight = 1.0;
		/* SECTION BEGIN */
		angle_delta = fabs(hsl.h - m_main_colour.h);
		if (angle_delta > 180.0F)
		{
			angle_delta = 360.0F - angle_delta;
		}
		/* SECTION END */
	}

	// double main_x = cos(m_main_colour.h * M_PI / 180.0) * m_main_colour.s;
	// double main_y = sin(m_main_colour.h * M_PI / 180.0) * m_main_colour.s;
	// double main_z = 1.0 - 2 * m_main_colour.l;
	// double eval_x = cos(hsl.h * M_PI / 180.0) * hsl.s;
	// double eval_y = sin(hsl.h * M_PI / 180.0) * hsl.s;
	// double eval_z = 1.0 - 2 * hsl.l;
	// // double total_x = main_x - eval_x;
	// // double total_y = main_y - eval_y;
	// // double total_z = main_z - eval_z;
	// //double vec_magn = sqrt(pow(total_x, 2) + pow(total_y, 2) + pow(total_z, 2));
	// double angle_diff = -(((main_x * eval_x + main_y * eval_y + main_z * eval_z) /
	// 	(sqrt(pow(main_x, 2) + pow(main_y, 2) + pow(main_z, 2)) *
	// 		sqrt(pow(eval_x, 2) + pow(eval_y, 2) + pow(eval_z, 2)))) - 1.0) / 2;


	// static double max_angle_diff = 0.0;
	// printf("Main - %f %f %f\n", m_main_colour.h, m_main_colour.s, m_main_colour.l);
	// printf("Eval - %f %f %f\n", hsl.h, hsl.s, hsl.l);
	// printf("Main Vec - %f %f %f\n", main_x, main_y, main_z);
	// printf("Eval Vec - %f %f %f\n", eval_x, eval_y, eval_z);
	// // printf("Res  Vec - %f %f %f\n", total_x, total_y, total_z);
	// printf("Ang diff - %f\n", angle_diff);
	// if (angle_diff > max_angle_diff)
	// {
	// 	max_angle_diff = angle_diff;
	// 	printf("Max Angle Diff\n");
	// }

	// angle_delta = fabs(hsl.h - m_main_colour.h);
	// if (angle_delta > 180.0F)
	// {
	// 	angle_delta = 360.0F - angle_delta;
	// }

	// result = static_cast<size_t>(round(1000.0 * pow(weight, 0.3) * hsv.v *
	//     fmin(1.0, hsv.v + hsv.s) * additional_colour_weight * angle_delta));

	// result = static_cast<size_t>(round(1000.0 * pow(weight, 0.3) * hsl.l *
	// 	additional_colour_weight * angle_delta));

	result = static_cast<size_t>(round(1000.0 * pow(weight, 0.3) *
		pow(((1.0 - fabs(2 * hsl.l - 1.0)) * (1 - 1.0 / m_l_levels) + 1.0 / m_l_levels), 0.3) *
		hsl.s *
		// additional_colour_weight *
		angle_delta));

	return result;
}

// PUBLIC FUNCS
struct ColourHSL ComplimentaryColourExtractor::get_main_colour() const
{
	return m_main_colour;
}

void ComplimentaryColourExtractor::set_main_colour(struct ColourHSL hsl)
{
	m_main_colour = hsl;
}