Files
IQ.Pilot/iqpilot/tools/cabana/ui/chart/chart.h
2026-09-02 16:46:45 -05:00

143 lines
4.6 KiB
C++

#pragma once
#include <functional>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
#include "imgui.h"
#include "imgui_internal.h"
#include "implot.h"
#include "tools/cabana/ui/chart/tiplabel.h"
#include "tools/cabana/dbc/dbcmanager.h"
#include "tools/cabana/streams/abstractstream.h"
#include "tools/cabana/utils/util.h"
enum class SeriesType {
Line = 0,
StepLine,
Scatter
};
inline constexpr const char *SERIES_TYPE_NAMES[] = {"Line", "Step Line", "Scatter"};
inline std::string msgLabel(const MessageId &id) { return " " + msgName(id) + " " + id.toString(); }
class ChartsWidget;
class ChartView {
public:
struct SigItem {
MessageId msg_id;
const cabana::Signal *sig = nullptr;
CabanaColor color;
bool visible = true;
std::vector<ImPlotPoint> vals;
std::vector<ImPlotPoint> step_vals;
ImPlotPoint track_pt{};
SegmentTree segment_tree;
double min = 0;
double max = 0;
};
ChartView(const std::pair<double, double> &x_range, ChartsWidget *parent);
void addSignal(const MessageId &msg_id, const cabana::Signal *sig);
bool hasSignal(const MessageId &msg_id, const cabana::Signal *sig) const;
void updateSeries(const cabana::Signal *sig = nullptr, const MessageEventsMap *msg_new_events = nullptr);
void updatePlot(double cur, double min, double max);
void setSeriesType(SeriesType type) { series_type_ = type; }
void showTip(double sec);
void hideTip();
void draw(float width);
void drawGhost(float width);
void removeIf(std::function<bool(const SigItem &)> predicate);
void takeSignalsFrom(ChartView *source);
std::vector<SigItem> takeExtraSignals();
void adoptSignal(SigItem s);
void setDropHighlight(bool highlight) { can_drop_ = highlight; }
const std::vector<SigItem> &signals() const { return sigs_; }
const ImRect &rect() const { return layout_.rect; }
bool plotHovered() const { return layout_.plot_hovered; }
double secondsAtPoint(const ImVec2 &pt) const {
return x_min_ + (pt.x - layout_.plot_area.Min.x) * (x_max_ - x_min_) / std::max(layout_.plot_area.GetWidth(), 1.0f);
}
private:
using PointIter = std::vector<ImPlotPoint>::const_iterator;
void signalUpdated(const cabana::Signal *sig);
void manageSignals();
void msgRemoved(MessageId id) { removeIf([=](auto &s) { return s.msg_id.address == id.address && !dbc()->msg(id); }); }
void signalRemoved(const cabana::Signal *sig) { removeIf([=](auto &s) { return s.sig == sig; }); }
void appendCanEvents(const cabana::Signal *sig, const std::vector<const CanEvent *> &events,
std::vector<ImPlotPoint> &vals, std::vector<ImPlotPoint> &step_vals);
void createToolButtons();
void drawContextMenu();
void handleMousePress();
void handleMouseMove();
void handleMouseRelease();
void updateLayout();
void updateAxisY();
void paint();
void drawStaticLayer();
void drawAxes();
void drawLegend();
void drawSeries();
void drawForeground();
void drawSignalValue();
void drawTimeline();
void drawRubberBandTimeRange();
void drawMenuActions();
int xAxisPrecision() const;
std::tuple<double, double, int> getNiceAxisNumbers(double min, double max, int tick_count);
double niceNumber(double x, bool ceiling);
CabanaColor uniqueColor(CabanaColor color, const cabana::Signal *exclude = nullptr) const;
const ImPlotPoint *lastPointBefore(const SigItem &s, double sec) const;
std::pair<PointIter, PointIter> visibleRange(const std::vector<ImPlotPoint> &points) const;
inline void clearTrackPoints() { for (auto &s : sigs_) s.track_pt = {}; }
inline float xPos(double sec) const { return layout_.plot_area.Min.x + (sec - x_min_) / (x_max_ - x_min_) * layout_.plot_area.GetWidth(); }
inline float yPos(double val) const { return layout_.plot_area.Max.y - (val - y_min_) / (y_max_ - y_min_) * layout_.plot_area.GetHeight(); }
struct Layout {
ImRect rect;
ImRect plot_area;
ImRect move_icon_rect;
ImRect close_btn_rect;
ImRect manage_btn_rect;
std::vector<ImRect> legend_rects;
float header_bottom = 0;
bool plot_hovered = false;
} layout_;
double x_min_;
double x_max_;
double y_min_ = 0;
double y_max_ = 1;
int y_tick_count_ = 3;
int y_precision_ = 0;
std::string y_unit_;
enum class MouseMode { None, Rubber, Scrub };
MouseMode mouse_mode_ = MouseMode::None;
ImVec2 press_pos_;
ImRect rubber_rect_;
bool resume_after_scrub_ = false;
bool drawing_ghost_ = false;
ImGuiID context_menu_id_ = 0;
TipLabel tip_label_;
std::vector<SigItem> sigs_;
double cur_sec_ = 0;
SeriesType series_type_ = SeriesType::Line;
bool can_drop_ = false;
double tooltip_x_ = -1;
ChartsWidget *charts_widget_;
Connections connections_;
};