41 lines
1.2 KiB
C++
41 lines
1.2 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_IMAGEPULLER_H
|
|
#define JFJOCH_IMAGEPULLER_H
|
|
|
|
#include <string>
|
|
#include <memory>
|
|
|
|
#include "../common/ZMQWrappers.h"
|
|
#include "../frame_serialize/CBORStream2Deserializer.h"
|
|
#include "../common/ThreadSafeFIFO.h"
|
|
|
|
struct ImagePullerOutput {
|
|
std::shared_ptr<ZMQMessage> msg;
|
|
std::shared_ptr<CBORStream2DeserializerOutput> cbor;
|
|
};
|
|
|
|
struct ImagePullerStatistics {
|
|
size_t outside_fifo_max_util = 0;
|
|
};
|
|
|
|
class ImagePuller {
|
|
protected:
|
|
ThreadSafeFIFO<ImagePullerOutput> outside_fifo; // assuming 1 file is <1000 files, this should be enough to buffer
|
|
public:
|
|
constexpr static size_t DefaultQueueSize = 1000;
|
|
constexpr static auto PollTimeout = std::chrono::milliseconds(100);
|
|
|
|
explicit ImagePuller(size_t fifo_size = DefaultQueueSize);
|
|
virtual ~ImagePuller() = default;
|
|
|
|
[[nodiscard]] std::optional<ImagePullerOutput> PollImage(std::chrono::microseconds timeout = PollTimeout);
|
|
[[nodiscard]] size_t GetMaxFifoUtilization() const;
|
|
[[nodiscard]] size_t GetCurrentFifoUtilization() const;
|
|
|
|
virtual void Disconnect() = 0;
|
|
};
|
|
|
|
#endif //JFJOCH_IMAGEPULLER_H
|