initial base app

Author: shincurry

Animeme.pro

@@ -22,12 +22,18 @@ DEFINES += QT_DEPRECATED_WARNINGS
 # You can also select to disable deprecated APIs only up to a certain version of Qt.
 #DEFINES += QT_DISABLE_DEPRECATED_BEFORE=0x060000    # disables all the APIs deprecated before Qt 6.0.0
 
+INCLUDEPATH += /usr/include/opencv2
+
+LIBS += `pkg-config --cflags --libs opencv`
+
 SOURCES += \
-        main.cpp \
-        mainwindow.cpp
+    main.cpp \
+    mainwindow.cpp \
+    loopplayer.cpp
 
 HEADERS += \
-        mainwindow.h
+    mainwindow.h \
+    loopplayer.h
 
 FORMS += \
-        mainwindow.ui
+    mainwindow.ui

loopplayer.cpp

@@ -0,0 +1,110 @@
+#include "loopplayer.h"
+
+#include <iostream>
+
+LoopPlayer::LoopPlayer(QObject *parent) : QThread(parent)
+{
+    stop = true;
+}
+
+LoopPlayer::~LoopPlayer()
+{
+    mutex.lock();
+    stop = true;
+    condition.wakeOne();
+    mutex.unlock();
+    wait();
+}
+
+bool LoopPlayer::loadVideo(std::string filename)
+{
+    cv::VideoCapture capture;
+    capture.open(filename);
+    if (capture.isOpened())
+    {
+        frameRate = (int)capture.get(CV_CAP_PROP_FPS);
+
+        cv::Mat frame;
+        while (capture.read(frame))
+        {
+            if (frame.channels() == 3)
+            {
+                cv::cvtColor(frame, frame, CV_BGR2RGB);
+            }
+            frames.push_back(frame.clone());
+        }
+
+        capture.release();
+        return true;
+    }
+
+    return false;
+}
+
+void LoopPlayer::play()
+{
+    if (!isRunning())
+    {
+        if (isStopped())
+        {
+            stop = false;
+        }
+
+        start(LowPriority);
+    }
+}
+
+void LoopPlayer::run()
+{
+    int delay = 1000.0 / frameRate;
+
+    auto it = frames.begin();
+    while (!stop) {
+        if (it == frames.end())
+        {
+            it = frames.begin();
+        }
+
+        cv::Mat frame = *it;
+        if (frame.channels() == 3)
+        {
+            img = QImage((const unsigned char*)(frame.data),
+                         frame.cols,
+                         frame.rows,
+                         QImage::Format_RGB888);
+        }
+        else
+        {
+            img = QImage((const unsigned char*)(frame.data),
+                         frame.cols,
+                         frame.rows,
+                         QImage::Format_Indexed8);
+        }
+
+        emit processedImage(img);
+        this->msleep(delay);
+
+        it++;
+    }
+}
+
+void LoopPlayer::pause()
+{
+    stop = true;
+}
+
+void LoopPlayer::msleep(int ms)
+{
+    struct timespec ts =
+    {
+        ms / 1000, (ms % 1000) * 1000 * 1000
+    };
+
+    nanosleep(&ts, NULL);
+}
+
+bool LoopPlayer::isStopped() const
+{
+    return stop;
+}
+

loopplayer.h

@@ -0,0 +1,38 @@
+#ifndef LOOPPLAYER_H
+#define LOOPPLAYER_H
+
+#include <vector>
+#include <QMutex>
+#include <QThread>
+#include <QImage>
+#include <QWaitCondition>
+#include <opencv2/core/core.hpp>
+#include <opencv2/imgproc/imgproc.hpp>
+#include <opencv2/highgui/highgui.hpp>
+
+class LoopPlayer : public QThread
+{   Q_OBJECT
+private:
+    bool stop;
+    QMutex mutex;
+    QWaitCondition condition;
+    std::vector<cv::Mat> frames;
+    int frameRate;
+    QImage img;
+
+signals:
+    void processedImage(const QImage &image);
+
+protected:
+    void run();
+    void msleep(int ms);
+public:
+    LoopPlayer(QObject *parent = 0);
+    ~LoopPlayer();
+    bool loadVideo(std::string filename);
+    void play();
+    void pause();
+    bool isStopped() const;
+};
+
+#endif // LOOPPLAYER_H

main.cpp

@@ -5,6 +5,7 @@ int main(int argc, char *argv[])
 {
     QApplication a(argc, argv);
     MainWindow w;
+    w.setAttribute(Qt::WA_DeleteOnClose);
     w.show();
 
     return a.exec();

mainwindow.cpp

@@ -5,10 +5,72 @@ MainWindow::MainWindow(QWidget *parent) :
     QMainWindow(parent),
     ui(new Ui::MainWindow)
 {
+    player = new LoopPlayer();
+    QObject::connect(player, SIGNAL(processedImage(QImage)), this, SLOT(updateLoopPlayerUI(QImage)));
+
+    if (!player->loadVideo("/home/shin/Videos/59adf78ac4c0bc657943222d.mp4"))
+    {
+        QMessageBox msgBox;
+        msgBox.setText("The selected video could not be opened!");
+        msgBox.exec();
+    }
+
+    player->play();
+
     ui->setupUi(this);
 }
 
 MainWindow::~MainWindow()
 {
+    delete player;
     delete ui;
 }
+
+void MainWindow::updateLoopPlayerUI(QImage img)
+{
+    if (!img.isNull())
+    {
+        ui->playerScreen->setAlignment(Qt::AlignCenter);
+        ui->playerScreen->setPixmap(QPixmap::fromImage(img)
+                             .scaled(
+                                 ui->playerScreen->size(),
+                                 Qt::KeepAspectRatio,
+                                 Qt::FastTransformation)
+                             );
+    }
+}
+
+void MainWindow::on_openFileButton_clicked()
+{
+    QString filename = QFileDialog::getOpenFileName(this,
+                                                    tr("Open Video"),
+                                                    ".",
+                                                    tr("Video Files (*.avi *.mpg *.mp4")
+                                                    );
+
+    if (!filename.isEmpty())
+    {
+        if (!player->loadVideo(filename.toStdString().data()))
+        {
+            QMessageBox msgBox;
+            msgBox.setText("The selected video could not be opened!");
+            msgBox.exec();
+        }
+
+        player->play();
+    }
+}
+
+void MainWindow::on_pushButton_2_clicked()
+{
+    if (player->isStopped())
+    {
+        player->play();
+        ui->pushButton_2->setText(tr("Pause"));
+    }
+    else
+    {
+        player->pause();
+        ui->pushButton_2->setText(tr("Play"));
+    }
+}

mainwindow.h

@@ -2,6 +2,9 @@
 #define MAINWINDOW_H
 
 #include <QMainWindow>
+#include <QFileDialog>
+#include <QMessageBox>
+#include "loopplayer.h"
 
 namespace Ui {
 class MainWindow;
@@ -15,8 +18,14 @@ class MainWindow : public QMainWindow
     explicit MainWindow(QWidget *parent = 0);
     ~MainWindow();
 
+private slots:
+    void updateLoopPlayerUI(QImage img);
+    void on_openFileButton_clicked();
+    void on_pushButton_2_clicked();
+
 private:
     Ui::MainWindow *ui;
+    LoopPlayer *player;
 };
 
 #endif // MAINWINDOW_H