colormap.h

Go to the documentation of this file.

 
 
#pragma once
 
#include <scimesh/types.h>
#include <vector>
#include <cstdint>
#include <cmath>
#include <cstddef>
 
namespace scimesh {
 
// =========================================================================
// ColorMap
// =========================================================================
 
struct ColorMap {
    std::vector<Color> colors;
 
    Color sample(float t) const;
 
    size_t size() const { return colors.size(); }
 
    bool empty() const { return colors.empty(); }
 
    // ---------------------------------------------------------------------
    //  Factory: scibar interop (duck-typed, no scibar dependency)
    // ---------------------------------------------------------------------
 
    template<typename T>
    static ColorMap from_uint8_colors(const std::vector<T>& cmap) {
        ColorMap out;
        out.colors.reserve(cmap.size());
        for (const auto& c : cmap) {
            out.colors.emplace_back(
                c.r / 255.0f, c.g / 255.0f,
                c.b / 255.0f, c.a / 255.0f);
        }
        return out;
    }
 
    static ColorMap from_interleaved(const std::vector<uint8_t>& data,
                                     int channels = 3) {
        ColorMap out;
        size_t n = data.size() / static_cast<size_t>(channels);
        out.colors.reserve(n);
        for (size_t i = 0; i < n; i++) {
            float r = data[i * channels] / 255.0f;
            float g = data[i * channels + 1] / 255.0f;
            float b = data[i * channels + 2] / 255.0f;
            float a = (channels >= 4) ? data[i * channels + 3] / 255.0f : 1.0f;
            out.colors.emplace_back(r, g, b, a);
        }
        return out;
    }
 
    static const ColorMap& viridis();
};
 
// =========================================================================
// ApplyColormapResult — single dataset
// =========================================================================
 
struct ApplyColormapResult {
    std::vector<Color> colors;
 
    float data_min = NAN;
 
    float data_max = NAN;
 
    float raw_min = NAN;
 
    float raw_max = NAN;
 
    float winsor_lo = NAN;
 
    float winsor_hi = NAN;
 
    size_t nan_count = 0;
};
 
// =========================================================================
// MultiApplyColormapResult — multiple datasets
// =========================================================================
 
struct MultiApplyColormapResult {
    std::vector<ApplyColormapResult> per_dataset;
 
    float pooled_data_min = NAN;
 
    float pooled_data_max = NAN;
 
    float pooled_raw_min = NAN;
 
    float pooled_raw_max = NAN;
 
    float pooled_winsor_lo = NAN;
 
    float pooled_winsor_hi = NAN;
 
    size_t total_nan_count = 0;
};
 
// =========================================================================
// apply_colormap — function declarations
// =========================================================================
 
ApplyColormapResult apply_colormap(
    const std::vector<float>& data,
    const ColorMap& colormap,
    float vmin       = NAN,
    float vmax       = NAN,
    const Color& nan_color = Color{0.5f, 0.5f, 0.5f, 1.0f},
    float lo_pct     = 0.0f,
    float hi_pct     = 100.0f
);
 
MultiApplyColormapResult apply_colormap(
    const std::vector<std::vector<float>>& datasets,
    const ColorMap& colormap,
    float vmin       = NAN,
    float vmax       = NAN,
    const Color& nan_color = Color{0.5f, 0.5f, 0.5f, 1.0f},
    float lo_pct     = 0.0f,
    float hi_pct     = 100.0f,
    bool  global_range = false
);
 
} // namespace scimesh