56 lines
1.3 KiB
C++
56 lines
1.3 KiB
C++
#include "complimentary_colour_extractor.h"
|
|
|
|
#include <math.h>
|
|
|
|
// CONSTRUCTORS
|
|
ComplimentaryColourExtractor::ComplimentaryColourExtractor(struct ColourHSV hsv,
|
|
size_t h_levels, size_t s_levels, size_t v_levels)
|
|
:ColourExtractor(h_levels, s_levels, v_levels), m_main_colour(hsv)
|
|
{
|
|
}
|
|
|
|
// PROTECTED FUNCS
|
|
size_t ComplimentaryColourExtractor::evaluate_colour(struct ColourHSV hsv,
|
|
double weight) const
|
|
{
|
|
size_t result = 0;
|
|
double angle_delta;
|
|
double additional_colour_weight;
|
|
|
|
if ((0.0 == hsv.s) || (0.0 == m_main_colour.s))
|
|
{
|
|
angle_delta = 180.0;
|
|
additional_colour_weight = 0.5;
|
|
}
|
|
else
|
|
{
|
|
additional_colour_weight = 1.0;
|
|
/* SECTION BEGIN */
|
|
angle_delta = fabs(hsv.h - m_main_colour.h);
|
|
if (angle_delta > 180.0F)
|
|
{
|
|
angle_delta = 360.0F - angle_delta;
|
|
}
|
|
/* SECTION END */
|
|
}
|
|
|
|
// 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) * hsv.v *
|
|
additional_colour_weight * angle_delta));
|
|
|
|
return result;
|
|
}
|
|
|
|
// PUBLIC FUNCS
|
|
struct ColourHSV ComplimentaryColourExtractor::get_main_colour() const
|
|
{
|
|
return m_main_colour;
|
|
}
|
|
|
|
void ComplimentaryColourExtractor::set_main_colour(struct ColourHSV hsv)
|
|
{
|
|
m_main_colour = hsv;
|
|
}
|