Files
RCS-3000/examples/qt_gui_demo.cpp.backup
CaiXiang af65c2425d initial
2025-11-14 16:09:58 +08:00

639 lines
22 KiB
Plaintext

#include "path_tracker.h"
#include <QApplication>
#include <QMainWindow>
#include <QWidget>
#include <QPushButton>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QLabel>
#include <QComboBox>
#include <QDoubleSpinBox>
#include <QTableWidget>
#include <QGroupBox>
#include <QPainter>
#include <QTimer>
#include <QHeaderView>
#include <cmath>
#include <QFileDialog>
#include <QMessageBox>
#include <QInputDialog>
/**
* @brief AGV Path Visualization Widget
* Displays the reference path and AGV trajectory
*/
class PathVisualizationWidget : public QWidget {
Q_OBJECT
public:
explicit PathVisualizationWidget(QWidget* parent = nullptr)
: QWidget(parent), current_step_(0), show_animation_(false) {
setMinimumSize(600, 600);
setStyleSheet("background-color: white;");
}
void setPath(const PathCurve& path) {
path_ = path;
update();
}
void setControlSequence(const ControlSequence& sequence) {
sequence_ = sequence;
current_step_ = 0;
update();
}
void setCurrentStep(int step) {
current_step_ = step;
update();
}
void setShowAnimation(bool show) {
show_animation_ = show;
update();
}
protected:
void paintEvent(QPaintEvent* event) override {
QPainter painter(this);
painter.setRenderHint(QPainter::Antialiasing);
// Calculate coordinate transformation
const auto& path_points = path_.getPathPoints();
if (path_points.empty()) return;
// Find bounds
double min_x = 1e9, max_x = -1e9, min_y = 1e9, max_y = -1e9;
for (const auto& pt : path_points) {
min_x = std::min(min_x, pt.x);
max_x = std::max(max_x, pt.x);
min_y = std::min(min_y, pt.y);
max_y = std::max(max_y, pt.y);
}
// Add trajectory points
if (!sequence_.predicted_states.empty()) {
for (const auto& state : sequence_.predicted_states) {
min_x = std::min(min_x, state.x);
max_x = std::max(max_x, state.x);
min_y = std::min(min_y, state.y);
max_y = std::max(max_y, state.y);
}
}
// Add margin
double margin = 0.5;
min_x -= margin; max_x += margin;
min_y -= margin; max_y += margin;
double range_x = max_x - min_x;
double range_y = max_y - min_y;
double range = std::max(range_x, range_y);
// Center the view
double center_x = (min_x + max_x) / 2.0;
double center_y = (min_y + max_y) / 2.0;
// Scale to fit widget with padding
int padding = 40;
// Prevent division by zero if all points are at the same location
if (range < 1e-6) {
range = 1.0;
}
double scale = std::min(width() - 2 * padding, height() - 2 * padding) / range;
// Coordinate transformation: world to screen
auto toScreen = [&](double x, double y) -> QPointF {
double sx = (x - center_x) * scale + width() / 2.0;
double sy = height() / 2.0 - (y - center_y) * scale; // Flip Y axis
return QPointF(sx, sy);
};
// Draw grid
painter.setPen(QPen(QColor(220, 220, 220), 1));
int grid_lines = 10;
for (int i = 0; i <= grid_lines; ++i) {
double t = static_cast<double>(i) / grid_lines;
double x = min_x + t * range;
double y = min_y + t * range;
QPointF p1 = toScreen(x, min_y);
QPointF p2 = toScreen(x, min_y + range);
painter.drawLine(p1, p2);
p1 = toScreen(min_x, y);
p2 = toScreen(min_x + range, y);
painter.drawLine(p1, p2);
}
// Draw axes
painter.setPen(QPen(Qt::black, 2));
QPointF origin = toScreen(0, 0);
QPointF x_axis = toScreen(1, 0);
QPointF y_axis = toScreen(0, 1);
painter.drawLine(origin, x_axis);
painter.drawLine(origin, y_axis);
painter.drawText(x_axis + QPointF(5, 5), "X");
painter.drawText(y_axis + QPointF(5, 5), "Y");
// Draw reference path
painter.setPen(QPen(QColor(100, 100, 255), 3, Qt::DashLine));
for (size_t i = 1; i < path_points.size(); ++i) {
QPointF p1 = toScreen(path_points[i-1].x, path_points[i-1].y);
QPointF p2 = toScreen(path_points[i].x, path_points[i].y);
painter.drawLine(p1, p2);
}
// Draw path points
painter.setPen(Qt::NoPen);
painter.setBrush(QColor(100, 100, 255, 100));
for (const auto& pt : path_points) {
QPointF p = toScreen(pt.x, pt.y);
painter.drawEllipse(p, 3, 3);
}
// Draw predicted trajectory
if (!sequence_.predicted_states.empty()) {
painter.setPen(QPen(QColor(255, 100, 100), 2));
for (size_t i = 1; i < sequence_.predicted_states.size(); ++i) {
QPointF p1 = toScreen(sequence_.predicted_states[i-1].x,
sequence_.predicted_states[i-1].y);
QPointF p2 = toScreen(sequence_.predicted_states[i].x,
sequence_.predicted_states[i].y);
painter.drawLine(p1, p2);
}
// Draw current AGV position
if (show_animation_ && current_step_ < sequence_.predicted_states.size()) {
const auto& state = sequence_.predicted_states[current_step_];
QPointF pos = toScreen(state.x, state.y);
// Draw AGV body (rectangle)
painter.save();
painter.translate(pos);
painter.rotate(-state.theta * 180.0 / M_PI); // Rotate to heading
double agv_length = 0.3 * scale;
double agv_width = 0.2 * scale;
painter.setBrush(QColor(50, 200, 50));
painter.setPen(QPen(Qt::black, 2));
painter.drawRect(QRectF(-agv_length/2, -agv_width/2,
agv_length, agv_width));
// Draw heading indicator (arrow)
painter.setBrush(QColor(255, 50, 50));
painter.drawEllipse(QPointF(agv_length/2, 0),
agv_width/4, agv_width/4);
painter.restore();
// Draw position label
painter.setPen(Qt::black);
painter.drawText(pos + QPointF(10, -10),
QString("Step: %1").arg(current_step_));
}
}
// Draw legend
int legend_x = 10;
int legend_y = 10;
painter.fillRect(legend_x, legend_y, 150, 80, QColor(255, 255, 255, 200));
painter.setPen(Qt::black);
painter.drawRect(legend_x, legend_y, 150, 80);
// Reference path
painter.setPen(QPen(QColor(100, 100, 255), 3, Qt::DashLine));
painter.drawLine(legend_x + 10, legend_y + 20, legend_x + 40, legend_y + 20);
painter.setPen(Qt::black);
painter.drawText(legend_x + 50, legend_y + 25, "Reference Path");
// Trajectory
painter.setPen(QPen(QColor(255, 100, 100), 3));
painter.drawLine(legend_x + 10, legend_y + 40, legend_x + 40, legend_y + 40);
painter.setPen(Qt::black);
painter.drawText(legend_x + 50, legend_y + 45, "Trajectory");
// AGV
painter.fillRect(legend_x + 10, legend_y + 55, 30, 15, QColor(50, 200, 50));
painter.drawRect(legend_x + 10, legend_y + 55, 30, 15);
painter.drawText(legend_x + 50, legend_y + 65, "AGV");
}
private:
PathCurve path_;
ControlSequence sequence_;
int current_step_;
bool show_animation_;
};
/**
* @brief Main Window for AGV Path Tracking GUI
*/
class MainWindow : public QMainWindow {
Q_OBJECT
public:
MainWindow(QWidget* parent = nullptr) : QMainWindow(parent) {
setWindowTitle("AGV Path Tracking Control System - Qt GUI");
resize(1200, 800);
// Create central widget
QWidget* central = new QWidget(this);
setCentralWidget(central);
QHBoxLayout* main_layout = new QHBoxLayout(central);
// Left panel: Visualization
visualization_ = new PathVisualizationWidget(this);
main_layout->addWidget(visualization_, 2);
// Right panel: Controls and data
QWidget* right_panel = new QWidget(this);
QVBoxLayout* right_layout = new QVBoxLayout(right_panel);
// AGV Parameters Group
QGroupBox* param_group = new QGroupBox("AGV Parameters", this);
QVBoxLayout* param_layout = new QVBoxLayout(param_group);
wheelbase_spin_ = createParamRow("Wheelbase (m):", 0.5, 3.0, 1.0, param_layout);
max_vel_spin_ = createParamRow("Max Velocity (m/s):", 0.5, 5.0, 2.0, param_layout);
max_steer_spin_ = createParamRow("Max Steering (deg):", 10, 60, 45, param_layout);
right_layout->addWidget(param_group);
// Control Parameters Group
QGroupBox* control_group = new QGroupBox("Control Parameters", this);
QVBoxLayout* control_layout = new QVBoxLayout(control_group);
// Algorithm selection
QHBoxLayout* algo_layout = new QHBoxLayout();
algo_layout->addWidget(new QLabel("Algorithm:", this));
algorithm_combo_ = new QComboBox(this);
algorithm_combo_->addItem("Pure Pursuit");
algorithm_combo_->addItem("Stanley");
algo_layout->addWidget(algorithm_combo_);
control_layout->addLayout(algo_layout);
// Path type selection
QHBoxLayout* path_layout = new QHBoxLayout();
path_layout->addWidget(new QLabel("Path Type:", this));
path_combo_ = new QComboBox(this);
path_combo_->addItem("Circle Arc");
path_combo_->addItem("Straight Line");
path_combo_->addItem("S-Curve");
//自定义路径
path_combo_->addItem("Load from CSV");
path_combo_->addItem("Custom Spline");
path_layout->addWidget(path_combo_);
control_layout->addLayout(path_layout);
dt_spin_ = createParamRow("Time Step (s):", 0.01, 1.0, 0.1, control_layout);
horizon_spin_ = createParamRow("Horizon (s):", 1.0, 30.0, 10.0, control_layout);
right_layout->addWidget(control_group);
control_layout->addLayout(path_layout);
// 添加自定义路径按钮
QHBoxLayout* custom_btn_layout = new QHBoxLayout();
QPushButton* load_csv_btn = new QPushButton("Browse CSV...", this);
connect(load_csv_btn, &QPushButton::clicked, [this]() {
QString filename = QFileDialog::getOpenFileName(
this, "Open CSV Path File", "", "CSV Files (*.csv)");
if (!filename.isEmpty()) {
if (custom_path_.loadFromCSV(filename.toStdString(), true)) {
custom_path_loaded_ = true;
QMessageBox::information(this, "Success",
QString("Loaded %1 points from CSV!").arg(
custom_path_.getPathPoints().size()));
} else {
QMessageBox::warning(this, "Error", "Failed to load CSV file!");
}
}
});
custom_btn_layout->addWidget(load_csv_btn);
QPushButton* save_csv_btn = new QPushButton("Save Path...", this);
connect(save_csv_btn, &QPushButton::clicked, [this]() {
QString filename = QFileDialog::getSaveFileName(
this, "Save Path as CSV", "my_path.csv", "CSV Files (*.csv)");
if (!filename.isEmpty() && custom_path_loaded_) {
if (custom_path_.saveToCSV(filename.toStdString())) {
QMessageBox::information(this, "Success", "Path saved!");
}
}
});
custom_btn_layout->addWidget(save_csv_btn);
control_layout->addLayout(custom_btn_layout);
// Buttons
QHBoxLayout* button_layout = new QHBoxLayout();
QPushButton* generate_btn = new QPushButton("Generate Control", this);
connect(generate_btn, &QPushButton::clicked, this, &MainWindow::generateControl);
button_layout->addWidget(generate_btn);
start_btn_ = new QPushButton("Start Animation", this);
connect(start_btn_, &QPushButton::clicked, this, &MainWindow::toggleAnimation);
start_btn_->setEnabled(false);
button_layout->addWidget(start_btn_);
right_layout->addLayout(button_layout);
// Statistics Group
stats_group_ = new QGroupBox("Statistics", this);
QVBoxLayout* stats_layout = new QVBoxLayout(stats_group_);
stats_label_ = new QLabel("No data", this);
stats_label_->setWordWrap(true);
stats_layout->addWidget(stats_label_);
right_layout->addWidget(stats_group_);
// Control Sequence Table
table_ = new QTableWidget(this);
table_->setColumnCount(4);
table_->setHorizontalHeaderLabels({"Step", "Time(s)", "Velocity(m/s)", "Steering(deg)"});
table_->horizontalHeader()->setStretchLastSection(true);
table_->setAlternatingRowColors(true);
right_layout->addWidget(table_);
main_layout->addWidget(right_panel, 1);
// Initialize AGV model
updateAGVModel();
// Animation timer
animation_timer_ = new QTimer(this);
connect(animation_timer_, &QTimer::timeout, this, &MainWindow::updateAnimation);
}
private slots:
void generateControl() {
// Update AGV model
updateAGVModel();
// Create path based on selection
PathCurve path;
QString path_type = path_combo_->currentText();
if (path_type == "Load from CSV") {
if (!custom_path_loaded_) {
QMessageBox::warning(this, "Warning",
"Please load a CSV file first using 'Browse CSV...' button!");
return;
}
path = custom_path_;
}
else if (path_type == "Custom Spline") {
if (!custom_path_loaded_) {
// 如果没有预加载,让用户输入关键点
bool ok;
int num_points = QInputDialog::getInt(this, "Spline Input",
"Number of key points (2-10):", 4, 2, 10, 1, &ok);
if (!ok) return;
std::vector<PathPoint> key_points;
for (int i = 0; i < num_points; ++i) {
double x = QInputDialog::getDouble(this, "Key Point",
QString("Point %1 - X coordinate:").arg(i+1),
i * 3.0, -100, 100, 2, &ok);
if (!ok) return;
double y = QInputDialog::getDouble(this, "Key Point",
QString("Point %1 - Y coordinate:").arg(i+1),
(i % 2) * 3.0, -100, 100, 2, &ok);
if (!ok) return;
key_points.push_back(PathPoint(x, y));
}
path.generateSpline(key_points, 200, 0.5);
custom_path_ = path;
custom_path_loaded_ = true;
} else {
path = custom_path_;
}
}
else if (path_type == "Circle Arc") {
path.generateCircleArc(5.0, 0.0, 5.0, M_PI, M_PI / 2, 100);
} else if (path_type == "Straight Line") {
PathPoint start(0, 0, 0, 0);
PathPoint end(10, 0, 0, 0);
path.generateLine(start, end, 100);
} else if (path_type == "S-Curve") {
PathPoint p0(0, 0, 0, 0);
PathPoint p1(3, 2, 0, 0);
PathPoint p2(7, 2, 0, 0);
PathPoint p3(10, 0, 0, 0);
path.generateCubicBezier(p0, p1, p2, p3, 100);
}
// 验证路径
if (path.getPathPoints().empty()) {
QMessageBox::warning(this, "Error", "Invalid path!");
return;
}
// Set up tracker
tracker_->setReferencePath(path);
AGVModel::State initial_state(0.0, 0.0, 0.0);
tracker_->setInitialState(initial_state);
// Generate control sequence
QString algo = algorithm_combo_->currentText();
std::string algo_str = (algo == "Pure Pursuit") ? "pure_pursuit" : "stanley";
double dt = dt_spin_->value();
double horizon = horizon_spin_->value();
tracker_->generateControlSequence(algo_str, dt, horizon);
const ControlSequence& sequence = tracker_->getControlSequence();
// Update visualization
visualization_->setPath(path);
visualization_->setControlSequence(sequence);
visualization_->setCurrentStep(0);
visualization_->setShowAnimation(true);
// Update table
updateTable(sequence);
// Update statistics
updateStatistics(sequence);
// Enable animation button
start_btn_->setEnabled(true);
start_btn_->setText("Start Animation");
animation_running_ = false;
}
void toggleAnimation() {
const ControlSequence& sequence = tracker_->getControlSequence();
if (sequence.size() == 0) return;
if (animation_running_) {
// Pause the animation
animation_timer_->stop();
start_btn_->setText("Resume Animation");
animation_running_ = false;
} else {
// Resume or start animation
// Only reset to beginning if animation has finished
if (animation_step_ >= sequence.size()) {
animation_step_ = 0;
}
animation_timer_->start(100); // 100ms interval
start_btn_->setText("Pause Animation");
animation_running_ = true;
}
}
void updateAnimation() {
const ControlSequence& sequence = tracker_->getControlSequence();
if (animation_step_ >= sequence.size()) {
animation_timer_->stop();
start_btn_->setText("Restart Animation");
animation_running_ = false;
return;
}
visualization_->setCurrentStep(animation_step_);
// Highlight current row in table
table_->selectRow(animation_step_);
animation_step_++;
}
private:
void updateAGVModel() {
double wheelbase = wheelbase_spin_->value();
double max_vel = max_vel_spin_->value();
double max_steer = max_steer_spin_->value() * M_PI / 180.0;
model_ = std::make_unique<AGVModel>(wheelbase, max_vel, max_steer);
tracker_ = std::make_unique<PathTracker>(*model_);
}
QDoubleSpinBox* createParamRow(const QString& label, double min, double max,
double value, QVBoxLayout* layout) {
QHBoxLayout* row = new QHBoxLayout();
row->addWidget(new QLabel(label, this));
QDoubleSpinBox* spin = new QDoubleSpinBox(this);
spin->setRange(min, max);
spin->setValue(value);
spin->setSingleStep((max - min) / 100.0);
spin->setDecimals(2);
row->addWidget(spin);
layout->addLayout(row);
return spin;
}
void updateTable(const ControlSequence& sequence) {
table_->setRowCount(sequence.size());
for (size_t i = 0; i < sequence.size(); ++i) {
table_->setItem(i, 0, new QTableWidgetItem(QString::number(i)));
table_->setItem(i, 1, new QTableWidgetItem(
QString::number(sequence.timestamps[i], 'f', 2)));
table_->setItem(i, 2, new QTableWidgetItem(
QString::number(sequence.controls[i].v, 'f', 4)));
table_->setItem(i, 3, new QTableWidgetItem(
QString::number(sequence.controls[i].delta * 180.0 / M_PI, 'f', 2)));
}
}
void updateStatistics(const ControlSequence& sequence) {
if (sequence.size() == 0) {
stats_label_->setText("No data");
return;
}
double avg_vel = 0, max_vel = -1e9, min_vel = 1e9;
double avg_steer = 0, max_steer = -1e9, min_steer = 1e9;
for (const auto& ctrl : sequence.controls) {
avg_vel += ctrl.v;
max_vel = std::max(max_vel, ctrl.v);
min_vel = std::min(min_vel, ctrl.v);
double delta_deg = ctrl.delta * 180.0 / M_PI;
avg_steer += delta_deg;
max_steer = std::max(max_steer, delta_deg);
min_steer = std::min(min_steer, delta_deg);
}
avg_vel /= sequence.size();
avg_steer /= sequence.size();
QString stats = QString(
"Total Steps: %1\n\n"
"Velocity:\n"
" Avg: %2 m/s\n"
" Max: %3 m/s\n"
" Min: %4 m/s\n\n"
"Steering:\n"
" Avg: %5°\n"
" Max: %6°\n"
" Min: %7°"
).arg(sequence.size())
.arg(avg_vel, 0, 'f', 4)
.arg(max_vel, 0, 'f', 4)
.arg(min_vel, 0, 'f', 4)
.arg(avg_steer, 0, 'f', 2)
.arg(max_steer, 0, 'f', 2)
.arg(min_steer, 0, 'f', 2);
stats_label_->setText(stats);
}
// Widgets
PathVisualizationWidget* visualization_;
QTableWidget* table_;
QComboBox* algorithm_combo_;
QComboBox* path_combo_;
QDoubleSpinBox* wheelbase_spin_;
QDoubleSpinBox* max_vel_spin_;
QDoubleSpinBox* max_steer_spin_;
QDoubleSpinBox* dt_spin_;
QDoubleSpinBox* horizon_spin_;
QPushButton* start_btn_;
QGroupBox* stats_group_;
QLabel* stats_label_;
// Model and tracker
std::unique_ptr<AGVModel> model_;
std::unique_ptr<PathTracker> tracker_;
// Animation
QTimer* animation_timer_;
int animation_step_;
bool animation_running_ = false;
//自定义路径
PathCurve custom_path_;
bool custom_path_loaded_ = false;
};
int main(int argc, char* argv[]) {
QApplication app(argc, argv);
MainWindow window;
window.show();
return app.exec();
}
#include "qt_gui_demo.moc"