68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
#ifndef __DRIVER_H_
|
|
#define __DRIVER_H_
|
|
|
|
#ifdef __cplusplus
|
|
|
|
#include <vector>
|
|
#include <chrono>
|
|
#include <atomic>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include "icsneo/api/eventmanager.h"
|
|
#include "icsneo/third-party/concurrentqueue/blockingconcurrentqueue.h"
|
|
|
|
namespace icsneo {
|
|
|
|
class Driver {
|
|
public:
|
|
Driver(const device_eventhandler_t& handler) : report(handler) {}
|
|
virtual ~Driver() {}
|
|
virtual bool open() = 0;
|
|
virtual bool isOpen() = 0;
|
|
virtual void modeChangeIncoming() {}
|
|
virtual void awaitModeChangeComplete() {}
|
|
virtual bool isDisconnected() { return disconnected; };
|
|
virtual bool close() = 0;
|
|
bool read(std::vector<uint8_t>& bytes, size_t limit = 0);
|
|
bool readWait(std::vector<uint8_t>& bytes, std::chrono::milliseconds timeout = std::chrono::milliseconds(100), size_t limit = 0);
|
|
bool write(const std::vector<uint8_t>& bytes);
|
|
virtual bool isEthernet() const { return false; }
|
|
|
|
device_eventhandler_t report;
|
|
|
|
size_t writeQueueSize = 50;
|
|
bool writeBlocks = true; // Otherwise it just fails when the queue is full
|
|
|
|
protected:
|
|
class WriteOperation {
|
|
public:
|
|
WriteOperation() {}
|
|
WriteOperation(const std::vector<uint8_t>& b) : bytes(b) {}
|
|
std::vector<uint8_t> bytes;
|
|
};
|
|
enum IOTaskState {
|
|
LAUNCH,
|
|
WAIT
|
|
};
|
|
|
|
virtual void readTask() = 0;
|
|
virtual void writeTask() = 0;
|
|
|
|
// Overridable in case the driver doesn't want to use writeTask and writeQueue
|
|
virtual bool writeQueueFull() { return writeQueue.size_approx() > writeQueueSize; }
|
|
virtual bool writeQueueAlmostFull() { return writeQueue.size_approx() > (writeQueueSize * 3 / 4); }
|
|
virtual bool writeInternal(const std::vector<uint8_t>& b) { return writeQueue.enqueue(WriteOperation(b)); }
|
|
|
|
moodycamel::BlockingConcurrentQueue<uint8_t> readQueue;
|
|
moodycamel::BlockingConcurrentQueue<WriteOperation> writeQueue;
|
|
std::thread readThread, writeThread;
|
|
std::atomic<bool> closing{false};
|
|
std::atomic<bool> disconnected{false};
|
|
};
|
|
|
|
}
|
|
|
|
#endif // __cplusplus
|
|
|
|
#endif |