120 lines
3.6 KiB
C++
120 lines
3.6 KiB
C++
#ifndef LIBREVNAUSBDRIVER_H
|
|
#define LIBREVNAUSBDRIVER_H
|
|
|
|
#include "librevnadriver.h"
|
|
|
|
#include <libusb-1.0/libusb.h>
|
|
#include <condition_variable>
|
|
#include <thread>
|
|
|
|
#include <QQueue>
|
|
#include <QTimer>
|
|
|
|
Q_DECLARE_METATYPE(Protocol::PacketInfo)
|
|
Q_DECLARE_METATYPE(LibreVNADriver::TransmissionResult)
|
|
|
|
class USBInBuffer : public QObject {
|
|
Q_OBJECT
|
|
public:
|
|
USBInBuffer(libusb_device_handle *handle, unsigned char endpoint, int buffer_size);
|
|
~USBInBuffer();
|
|
|
|
void removeBytes(int handled_bytes);
|
|
int getReceived() const;
|
|
uint8_t *getBuffer() const;
|
|
|
|
signals:
|
|
void DataReceived();
|
|
void TransferError();
|
|
|
|
private:
|
|
void Callback(libusb_transfer *transfer);
|
|
static void LIBUSB_CALL CallbackTrampoline(libusb_transfer *transfer);
|
|
libusb_transfer *transfer;
|
|
unsigned char *buffer;
|
|
int buffer_size;
|
|
int received_size;
|
|
bool inCallback;
|
|
bool cancelling;
|
|
std::condition_variable cv;
|
|
};
|
|
|
|
class LibreVNAUSBDriver : public LibreVNADriver
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
LibreVNAUSBDriver();
|
|
|
|
/**
|
|
* @brief Returns the driver name. It must be unique across all implemented drivers and is used to identify the driver
|
|
* @return driver name
|
|
*/
|
|
virtual QString getDriverName() override;
|
|
/**
|
|
* @brief Lists all available devices by their serial numbers
|
|
* @return Serial numbers of detected devices
|
|
*/
|
|
virtual std::set<QString> GetAvailableDevices() override;
|
|
/**
|
|
* @brief Connects to a device, given by its serial number
|
|
* @param serial Serial number of device that should be connected to
|
|
* @return true if connection successful, otherwise false
|
|
*/
|
|
virtual bool connectTo(QString serial) override;
|
|
/**
|
|
* @brief Disconnects from device. Has no effect if no device was connected
|
|
*/
|
|
virtual void disconnect() override;
|
|
|
|
/**
|
|
* @brief Registers metatypes within the Qt Framework.
|
|
*
|
|
* If the device driver uses a queued signal/slot connection with custom data types, these types must be registered before emitting the signal.
|
|
* Register them within this function with qRegisterMetaType<Type>("Name");
|
|
*/
|
|
virtual void registerTypes();
|
|
|
|
private slots:
|
|
void ReceivedData();
|
|
void ReceivedLog();
|
|
void transmissionTimeout() {
|
|
transmissionFinished(TransmissionResult::Timeout);
|
|
}
|
|
void transmissionFinished(TransmissionResult result);
|
|
private:
|
|
static constexpr int EP_Data_Out_Addr = 0x01;
|
|
static constexpr int EP_Data_In_Addr = 0x81;
|
|
static constexpr int EP_Log_In_Addr = 0x82;
|
|
|
|
virtual bool SendPacket(const Protocol::PacketInfo& packet, std::function<void(TransmissionResult)> cb = nullptr, unsigned int timeout = 500) override;
|
|
|
|
void USBHandleThread();
|
|
// foundCallback is called for every device that is found. If it returns true the search continues, otherwise it is aborted.
|
|
// When the search is aborted the last found device is still opened
|
|
static void SearchDevices(std::function<bool(libusb_device_handle *handle, QString getSerial)> foundCallback, libusb_context *context, bool ignoreOpenError);
|
|
|
|
libusb_device_handle *m_handle;
|
|
libusb_context *m_context;
|
|
USBInBuffer *dataBuffer;
|
|
USBInBuffer *logBuffer;
|
|
|
|
class Transmission {
|
|
public:
|
|
Protocol::PacketInfo packet;
|
|
unsigned int timeout;
|
|
std::function<void(TransmissionResult)> callback;
|
|
};
|
|
|
|
std::mutex transmissionMutex;
|
|
QQueue<Transmission> transmissionQueue;
|
|
bool startNextTransmission();
|
|
QTimer transmissionTimer;
|
|
bool transmissionActive;
|
|
|
|
std::thread *m_receiveThread;
|
|
|
|
std::mutex accessMutex;
|
|
};
|
|
|
|
#endif // LIBREVNAUSBDRIVER_H
|