video/ps3eye: prototype open driver impl

1 files changed, 150 insertions, 6 deletions

diff --git a/video-ps3eye/module.cpp b/video-ps3eye/module.cpp
index ac6d8807..26b771e2 100644
--- a/video-ps3eye/module.cpp
+++ b/video-ps3eye/module.cpp
@@ -1,7 +1,19 @@
 #include "module.hpp"
-#if 0
+#include "compat/library-path.hpp"
+#include "compat/sleep.hpp"
+#include "compat/run-in-thread.hpp"
+
+#include <cstddef>
+
+#include <QCoreApplication>
+#include <QMessageBox>
+
 #include <libusb.h>
 
+#ifdef __GNUG__
+#   pragma clang diagnostic ignored "-Wcast-qual"
+#endif
+
 int device_count()
 {
     libusb_context * ctx = nullptr;
@@ -50,8 +62,6 @@ bool check_device_exists()
 
 static const QString camera_name = QStringLiteral("PS3 Eye open driver");
 
-namespace video::impl {
-
 std::vector<QString> ps3eye_camera_::camera_names() const
 {
     if (check_device_exists())
@@ -66,16 +76,150 @@ std::unique_ptr<camera> ps3eye_camera_::make_camera(const QString& name)
     else
         return {};
 }
-bool ps3eye_camera_::show_dialog(const QString& camera_name)
+bool ps3eye_camera_::show_dialog(const QString&)
 {
     // TODO
     return false;
 }
-bool ps3eye_camera_::can_show_dialog(const QString& camera_name)
+bool ps3eye_camera_::can_show_dialog(const QString&)
 {
     return false;
 }
 
-} // ns video::impl
+ps3eye_camera::ps3eye_camera()
+{
+    if (!shm.success())
+        return;
+
+    static const QString library_path(OPENTRACK_BASE_PATH + OPENTRACK_LIBRARY_PATH);
 
+    wrapper.setWorkingDirectory(library_path);
+#ifdef _WIN32
+    wrapper.setProgram("\"ps3eye-subprocess.exe\"");
+#else
+    wrapper.setProgram("ps3eye-subprocess");
 #endif
+}
+
+ps3eye_camera::~ps3eye_camera()
+{
+    stop();
+}
+
+void ps3eye_camera::stop()
+{
+    open = false;
+
+    if (wrapper.state() != QProcess::NotRunning)
+    {
+        if (shm.success())
+        {
+            volatile auto& ptr = *(ps3eye::shm*)shm.ptr();
+            ptr.in.do_exit = true;
+            if (!wrapper.waitForFinished(500))
+                wrapper.kill();
+            wrapper.waitForFinished(-1);
+        }
+        else
+        {
+            wrapper.kill();
+            wrapper.waitForFinished(-1);
+        }
+    }
+}
+
+bool ps3eye_camera::start(info& args)
+{
+    if (!shm.success())
+        return false;
+
+    volatile auto& ptr = *(ps3eye::shm*)shm.ptr();
+
+    using mode = ps3eye::shm_in::mode;
+
+    open = false;
+    fr = {};
+    fr.channels = 3;
+    fr.channel_size = 1;
+
+    if (!args.width || args.width > 320)
+    {
+        ptr.in.resolution = mode::vga;
+        fr.width = 640; fr.height = 480;
+    }
+    else
+    {
+        ptr.in.resolution = mode::qvga;
+        fr.width = 320; fr.height = 240;
+    }
+
+    ptr.in.framerate = (uint8_t)std::clamp(args.fps, 30, 187);
+    ptr.in.gain = (uint8_t)s.gain;
+    ptr.in.exposure = (uint8_t)s.exposure;
+
+    wrapper.start();
+
+    constexpr int sleep_ms = 10, max_sleeps = 5000/sleep_ms;
+
+    for (int i = 0; i < max_sleeps; i++)
+    {
+        if (ptr.out.timecode > 0)
+            goto ok;
+        portable::sleep(sleep_ms);
+    }
+
+    run_in_thread_async(qApp, [&]() {
+        QString error;
+        if (ptr.out.error_string[0] == '\0')
+            error = "Unknown error";
+        else
+            error = QString::fromLatin1((const char*)ptr.out.error_string,
+                                        strnlen((const char*)ptr.out.error_string, sizeof(ptr.out.error_string)));
+
+        QMessageBox::critical(nullptr, "Can't open camera", "PS3 Eye driver error: " + error, QMessageBox::Close);
+    });
+
+    return false;
+
+ok:
+    open = true;
+    return true;
+}
+
+std::tuple<const frame&, bool> ps3eye_camera::get_frame()
+{
+    volatile auto& ptr = *(ps3eye::shm*)shm.ptr();
+    constexpr int sleep_ms = 10;
+    constexpr int max_sleeps = 2000/sleep_ms;
+
+    if (!open)
+        goto fail;
+
+    for (int i = 0; i < max_sleeps; i++)
+    {
+        unsigned new_timecode = ptr.out.timecode;
+        if (timecode != new_timecode)
+        {
+            timecode = new_timecode;
+            goto ok;
+        }
+        portable::sleep(sleep_ms);
+    }
+
+fail:
+    stop();
+    return { fr, false };
+
+    static_assert(offsetof(decltype(ptr.out), data_640x480) == offsetof(decltype(ptr.out), data_320x240));
+
+ok:
+    fr.data = (unsigned char*)ptr.out.data_640x480;
+    return { fr, true};
+}
+
+bool ps3eye_camera::show_dialog()
+{
+    return false;
+}
+
+OTR_REGISTER_CAMERA(ps3eye_camera_)