implot_metrics_sink.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <unordered_map>
#include <vector>
 
#include "pulsatrix/metrics_sink.hpp"
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
struct ScalarSeries {
    std::vector<int> steps;
    std::vector<double> values;
};
 
class ImPlotMetricsSink : public MetricsSink {
public:
    void log_scalar(const std::string& tag, double value, int step) override;
    void log_histogram(const std::string& tag, const Tensor& values, int step) override;
 
    [[nodiscard]] const std::unordered_map<std::string, ScalarSeries>& scalar_series() const {
        return scalar_series_;
    }
 
    [[nodiscard]] const std::unordered_map<std::string, std::vector<float>>& latest_histograms() const {
        return latest_histograms_;
    }
 
    void Draw() const;
 
private:
    std::unordered_map<std::string, ScalarSeries> scalar_series_;
    std::unordered_map<std::string, std::vector<float>> latest_histograms_;
};
 
}  // namespace pulsatrix