diff vamp-client/PipedQProcessTransport.h @ 90:6429a99abcad

Split out classes

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/vamp-client/PipedQProcessTransport.h	Thu Oct 13 10:17:59 2016 +0100
@@ -0,0 +1,95 @@
+
+#ifndef PIPER_PIPED_QPROCESS_TRANSPORT_H
+#define PIPER_PIPED_QPROCESS_TRANSPORT_H
+
+#include "SynchronousTransport.h"
+
+#include <QProcess>
+#include <QString>
+
+#include <iostream>
+
+namespace piper { //!!! change
+
+class PipedQProcessTransport : public SynchronousTransport
+{
+public:
+    PipedQProcessTransport(QString processName,
+                           MessageCompletenessChecker *checker) : //!!! ownership
+        m_completenessChecker(checker) {
+        m_process = new QProcess();
+        m_process->setReadChannel(QProcess::StandardOutput);
+        m_process->setProcessChannelMode(QProcess::ForwardedErrorChannel);
+        m_process->start(processName);
+        if (!m_process->waitForStarted()) {
+            std::cerr << "server failed to start" << std::endl;
+            delete m_process;
+            m_process = nullptr;
+        }
+    }
+
+    ~PipedQProcessTransport() {
+        if (m_process) {
+            if (m_process->state() != QProcess::NotRunning) {
+		m_process->closeWriteChannel();
+                m_process->waitForFinished(200);
+                m_process->close();
+                m_process->waitForFinished();
+                std::cerr << "server exited" << std::endl;
+            }
+            delete m_process;
+        }
+    }
+
+    bool isOK() const override {
+        return m_process != nullptr;
+    }
+    
+    std::vector<char>
+    call(const char *ptr, size_t size) override {
+
+        m_process->write(ptr, size);
+        
+        std::vector<char> buffer;
+        size_t wordSize = sizeof(capnp::word);
+        bool complete = false;
+        
+        while (!complete) {
+
+            m_process->waitForReadyRead(1000);
+            qint64 byteCount = m_process->bytesAvailable();
+            qint64 wordCount = byteCount / wordSize;
+
+            if (!wordCount) {
+                if (m_process->state() == QProcess::NotRunning) {
+                    std::cerr << "ERROR: Subprocess exited: Load failed" << std::endl;
+                    throw std::runtime_error("Piper server exited unexpectedly");
+                }
+            } else {
+                // only read whole words
+                byteCount = wordCount * wordSize;
+                size_t formerSize = buffer.size();
+                buffer.resize(formerSize + byteCount);
+                m_process->read(buffer.data() + formerSize, byteCount);
+                complete = m_completenessChecker->isComplete(buffer);
+            }
+        }
+/*
+        cerr << "buffer = ";
+        for (int i = 0; i < buffer.size(); ++i) {
+            if (i % 16 == 0) cerr << "\n";
+            cerr << int(buffer[i]) << " ";
+        }
+        cerr << "\n";
+*/        
+        return buffer;
+    }
+    
+private:
+    MessageCompletenessChecker *m_completenessChecker; //!!! I don't own this (currently)
+    QProcess *m_process; // I own this
+};
+
+}
+
+#endif