main_window.h

#ifndef MAIN_WINDOW_H
#define MAIN_WINDOW_H
 
#include <QMainWindow>
#include <QMediaPlayer>
#include <QVideoWidget>
#include <QGraphicsView>
#include <QGraphicsScene>
#include <QGraphicsRectItem>
#include <QMouseEvent>
#include <QVideoFrame>
#include <QVideoSink>
#include <QPainter>
 
#include "cvtrack/tracker.h"
 
class VideoWidget : public QWidget
{
    Q_OBJECT
 
public:
    VideoWidget(QWidget *parent = nullptr);
    void set_video_sink(QVideoSink *sink);
    void update_tracking_boxes(const std::vector<cvtrack::TrackedObject> &objects);
 
protected:
    void mousePressEvent(QMouseEvent *event) override;
    void mouseMoveEvent(QMouseEvent *event) override;
    void mouseReleaseEvent(QMouseEvent *event) override;
    void paintEvent(QPaintEvent *event) override;
 
signals:
    void new_bounding_box(const QRect &rect);
 
private slots:
    void on_video_frame(const QVideoFrame &frame);
 
private:
    QPoint start_point_;
    QPoint end_point_;
    bool drawing_;
    QRect current_selection_;
    QVideoSink *video_sink_;
    QImage current_frame_;
    std::vector<cvtrack::TrackedObject> tracked_objects_;
};
 
class MainWindow : public QMainWindow
{
    Q_OBJECT
 
public:
    MainWindow(QWidget *parent = nullptr);
    ~MainWindow();
 
    void set_video_file(const QString &file_path);
 
private slots:
    void on_new_bounding_box(const QRect &rect);
    void on_media_status_changed(QMediaPlayer::MediaStatus status);
    void on_video_frame_changed(const QVideoFrame &frame);
 
private:
    QMediaPlayer *player_;
    VideoWidget *video_widget_;
    QVideoSink *video_sink_;
    cvtrack::Tracker tracker_;
    bool tracking_started_;
};
 
#endif // MAIN_WINDOW_H