56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JUNGFRAUJOCH_ZMQIMAGEPULLER_H
|
|
#define JUNGFRAUJOCH_ZMQIMAGEPULLER_H
|
|
|
|
#include <future>
|
|
|
|
#include "../common/ZMQWrappers.h"
|
|
#include "../common/Logger.h"
|
|
#include "../common/SpotToSave.h"
|
|
#include "../frame_serialize/CBORStream2Deserializer.h"
|
|
#include "../common/ThreadSafeFIFO.h"
|
|
#include "ImagePuller.h"
|
|
|
|
|
|
|
|
class ZMQImagePuller : public ImagePuller {
|
|
const uint32_t default_receive_watermark = 100;
|
|
const uint32_t default_repub_watermark = 220;
|
|
|
|
// ZeroMQ receive timeout allows to check for abort value from time to time
|
|
constexpr static auto ReceiveTimeout = std::chrono::milliseconds(100);
|
|
constexpr static auto RepubTimeout = std::chrono::milliseconds(100);
|
|
|
|
ZMQSocket socket;
|
|
std::string addr;
|
|
|
|
volatile int disconnect = 0;
|
|
|
|
std::unique_ptr<ZMQSocket> repub_socket;
|
|
|
|
ThreadSafeFIFO<ImagePullerOutput> cbor_fifo{200};
|
|
ThreadSafeFIFO<ImagePullerOutput> repub_fifo{200};
|
|
|
|
std::thread puller_thread;
|
|
std::thread cbor_thread;
|
|
std::thread repub_thread;
|
|
|
|
void PullerThread();
|
|
void CBORThread();
|
|
void RepubThread();
|
|
Logger logger{"ZMQImagePuller"};
|
|
public:
|
|
explicit ZMQImagePuller(
|
|
const std::string &zmq_addr,
|
|
const std::string &repub_address = "",
|
|
const std::optional<int32_t> &rcv_watermark = {},
|
|
const std::optional<int32_t> &repub_watermark = {});
|
|
~ZMQImagePuller();
|
|
|
|
void Disconnect() override;
|
|
};
|
|
|
|
#endif //JUNGFRAUJOCH_ZMQIMAGEPULLER_H
|