58 lines
1.5 KiB
C++
58 lines
1.5 KiB
C++
#ifndef __DRIVER_H_
|
|
#define __DRIVER_H_
|
|
|
|
#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 bool close() = 0;
|
|
virtual bool read(std::vector<uint8_t>& bytes, size_t limit = 0);
|
|
virtual bool readWait(std::vector<uint8_t>& bytes, std::chrono::milliseconds timeout = std::chrono::milliseconds(100), size_t limit = 0);
|
|
virtual bool write(const std::vector<uint8_t>& bytes);
|
|
inline void onWrite() {
|
|
if(writeQueue.size_approx() < (writeQueueSize * 3/4))
|
|
writeCV.notify_one();
|
|
}
|
|
|
|
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;
|
|
moodycamel::BlockingConcurrentQueue<uint8_t> readQueue;
|
|
moodycamel::BlockingConcurrentQueue<WriteOperation> writeQueue;
|
|
std::mutex writeMutex;
|
|
std::condition_variable writeCV;
|
|
std::thread readThread, writeThread;
|
|
std::atomic<bool> closing{false};
|
|
};
|
|
|
|
}
|
|
|
|
#endif |