satdump/src-core/common/widgets/waterfall_plot.cpp

#include "waterfall_plot.h"
#include "imgui/imgui_internal.h"
#include <string>
#include "common/colormaps.h"
#include "imgui/imgui_image.h"
#include "resources.h"

namespace widgets
{
    WaterfallPlot::WaterfallPlot(float *v, int size, int lines)
    {
        values = v;
        fft_size = size;
        fft_lines = lines;
    }

    WaterfallPlot::~WaterfallPlot()
    {
        if (!first_run)
            delete[] raw_img_buffer;
    }

    void WaterfallPlot::draw(ImVec2 size)
    {
        if (first_run)
        {
            texture_id = makeImageTexture();
            first_run = false;
            raw_img_buffer = new uint32_t[fft_size * fft_lines];
            palette = colormaps::generatePalette(colormaps::loadMap(resources::getResourcePath("waterfall/classic.json")), resolution);
        }

        memmove(&raw_img_buffer[fft_size * 1], &raw_img_buffer[fft_size * 0], fft_size * (fft_lines - 1));

        for (int i = 0; i < fft_size; i++)
        {
            int v = ((values[i] - scale_min) / fabs(scale_max - scale_min)) * resolution;

            if (v < 0)
                v = 0;
            if (v > resolution)
                v = resolution;

            raw_img_buffer[i] = palette[v];
        }

        updateImageTexture(texture_id, raw_img_buffer, fft_size, fft_lines);

        ImGui::Image((void *)(intptr_t)texture_id, size);
    }
}