CubicSDR/src/Gradient.cpp

#include "Gradient.h"

Gradient::Gradient() {

}

void Gradient::addColor(GradientColor c) {
    colors.push_back(c);
}

std::vector<float> &Gradient::getRed() {
    return r_val;
}

std::vector<float> &Gradient::getGreen() {
    return g_val;
}

std::vector<float> &Gradient::getBlue() {
    return b_val;
}

void Gradient::generate(unsigned int len) {
    int chunk_size = len / (colors.size() - 1);

    int p = 0;
    r_val.resize(len);
    g_val.resize(len);
    b_val.resize(len);

    for (unsigned int j = 0, jMax = colors.size() - 1; j < jMax; j++) {
        if (chunk_size * (jMax + 1) < len && j == jMax - 1) {
            chunk_size += len - chunk_size * (jMax + 1);
        }

        for (unsigned int i = 0; i < chunk_size; i++) {
            float idx = (float) (i) / (float) chunk_size;

            float r1 = colors[j].r;
            float g1 = colors[j].g;
            float b1 = colors[j].b;

            float r2 = colors[j + 1].r;
            float g2 = colors[j + 1].g;
            float b2 = colors[j + 1].b;

            float r = r1 + (r2 - r1) * idx;
            float g = g1 + (g2 - g1) * idx;
            float b = b1 + (b2 - b1) * idx;

            if (r < 0.0)
                r = 0.0;
            if (r > 1.0)
                r = 1.0;
            if (g < 0.0)
                g = 0.0;
            if (g > 1.0)
                g = 1.0;
            if (b < 0.0)
                b = 0.0;
            if (b > 1.0)
                b = 1.0;

            r_val[p] = r;
            g_val[p] = g;
            b_val[p] = b;

            p++;
        }
    }
}

Gradient::~Gradient() {

}