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

#pragma once

#include "imgui/imgui.h"
#include <cstdint>
#include <vector>
#include <mutex>
#include "common/colormaps.h"

namespace widgets
{
    class WaterfallPlot
    {
    private:
        float *values;
        int fft_max_size;
        int fft_size;
        int fft_lines;
        const int resolution = 2000; // Number of colors
        unsigned int texture_id;
        uint32_t *raw_img_buffer;

        std::vector<uint32_t> palette;

        std::mutex work_mtx;

        bool first_run = true;

    public:
        float scale_min, scale_max;

    public:
        WaterfallPlot(float *v, int size, int lines);
        ~WaterfallPlot();
        void draw(ImVec2 size, bool active = true);

        void set_size(int size)
        {
            work_mtx.lock();
            if (size <= fft_max_size)
                fft_size = size;
            work_mtx.unlock();
        }

        void set_palette(colormaps::Map palette, bool mtx = true);
    };
}