recording_connection.hpp

Go to the documentation of this file.

#pragma once
 
#include "../connection.hpp"
 
#include <memory>
#include <ostream>
#include <iostream>
 
namespace microstrain
{
namespace connections
{
 
 
class RecordingConnection : public Connection
{
public:
    static constexpr auto TYPE = "Recording";
 
    RecordingConnection(Connection *connection, std::ostream *recvStream = nullptr, std::ostream *sendStream = nullptr);
    ~RecordingConnection() = default;
 
    bool sendToDevice(const uint8_t* data, size_t length) override;
    bool recvFromDevice(uint8_t* buffer, size_t max_length, unsigned int wait_time_ms, size_t* length_out, EmbeddedTimestamp* timestamp_out) override;
 
    bool isConnected() const override
    {
        if(mConnection)
            return mConnection->isConnected();
 
        return false;
    };
 
    bool connect() override
    {
        if (mConnection) return mConnection->connect();
 
        return false;
    };
    bool disconnect() override
    {
        if (mConnection) return mConnection->disconnect();
 
        return false;
    };
 
    const char* interfaceName() const override { return mConnection->interfaceName(); }
    uint32_t parameter() const override { return mConnection->parameter(); }
 
    uint64_t recvFileBytesWritten() const
    {
        return mRecvFileWritten;
    }
 
    uint64_t sendFileBytesWritten() const
    {
        return mSendFileWritten;
    }
 
protected:
    Connection* mConnection;
 
    // Files may be NULL to not record one direction or the other
    std::ostream* mRecvFile;
    std::ostream* mSendFile;
 
    uint64_t mRecvFileWritten = 0;
    uint64_t mSendFileWritten = 0;
};
 
template<typename ConnectionType>
class RecordingConnectionWrapper : public RecordingConnection
{
public:
    template<class... Args>
    RecordingConnectionWrapper(std::ostream* recvStream, std::ostream* sendStream, Args&&... args) : RecordingConnection(new ConnectionType(std::forward<Args>(args)...), recvStream, sendStream) {}
 
    ~RecordingConnectionWrapper() { delete mConnection; }
};
 
 
}  // namespace connections
}  // namespace microstrain