101 lines
3.0 KiB
C++
101 lines
3.0 KiB
C++
#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;
|
|
}
|