You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.4 KiB
34 lines
1.4 KiB
#pragma once
|
|
|
|
#include <QPainter>
|
|
#include <QPolygonF>
|
|
|
|
#include "selfdrive/ui/ui.h"
|
|
|
|
class ModelRenderer {
|
|
public:
|
|
ModelRenderer() {}
|
|
void setTransform(const Eigen::Matrix3f &transform) { car_space_transform = transform; }
|
|
void draw(QPainter &painter, const QRect &surface_rect);
|
|
|
|
private:
|
|
bool mapToScreen(float in_x, float in_y, float in_z, QPointF *out);
|
|
void mapLineToPolygon(const cereal::XYZTData::Reader &line, float y_off, float z_off,
|
|
QPolygonF *pvd, int max_idx, bool allow_invert = true);
|
|
void drawLead(QPainter &painter, const cereal::RadarState::LeadData::Reader &lead_data, const QPointF &vd, const QRect &surface_rect);
|
|
void update_leads(const cereal::RadarState::Reader &radar_state, const cereal::XYZTData::Reader &line);
|
|
void update_model(const cereal::ModelDataV2::Reader &model, const cereal::RadarState::LeadData::Reader &lead);
|
|
void drawLaneLines(QPainter &painter);
|
|
void drawPath(QPainter &painter, const cereal::ModelDataV2::Reader &model, int height);
|
|
|
|
bool longitudinal_control = false;
|
|
bool experimental_model = false;
|
|
float lane_line_probs[4] = {};
|
|
float road_edge_stds[2] = {};
|
|
QPolygonF track_vertices;
|
|
QPolygonF lane_line_vertices[4] = {};
|
|
QPolygonF road_edge_vertices[2] = {};
|
|
QPointF lead_vertices[2] = {};
|
|
Eigen::Matrix3f car_space_transform = Eigen::Matrix3f::Zero();
|
|
QRectF clip_region;
|
|
};
|
|
|