59 lines
2.0 KiB
C++
59 lines
2.0 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "ZMQPreviewSocket.h"
|
|
#include "../common/Definitions.h"
|
|
|
|
ZMQPreviewSocket::ZMQPreviewSocket(const std::string &addr, const std::optional<int32_t> &watermark)
|
|
: socket(ZMQSocketType::Pub),
|
|
counter(std::chrono::seconds(1)) {
|
|
if (!watermark || watermark <= 0)
|
|
socket.Conflate(true);
|
|
else
|
|
socket.SendWaterMark(watermark.value());
|
|
|
|
socket.SendTimeout(std::chrono::milliseconds(100));
|
|
socket.Bind(addr);
|
|
}
|
|
|
|
ZMQPreviewSocket &ZMQPreviewSocket::Period(const std::optional<std::chrono::microseconds> &period) {
|
|
counter.Period(period);
|
|
return *this;
|
|
}
|
|
|
|
|
|
void ZMQPreviewSocket::SendImage(const uint8_t *image_data, size_t image_size) {
|
|
if (counter.GeneratePreview())
|
|
socket.Send(image_data, image_size, false);
|
|
}
|
|
|
|
void ZMQPreviewSocket::StartDataCollection(const StartMessage &message) {
|
|
std::vector<uint8_t> serialization_buffer(MESSAGE_SIZE_FOR_START_END);
|
|
CBORStream2Serializer serializer(serialization_buffer.data(), serialization_buffer.size());
|
|
serializer.SerializeSequenceStart(message);
|
|
socket.Send(serialization_buffer.data(), serializer.GetBufferSize(), true);
|
|
}
|
|
|
|
void ZMQPreviewSocket::EndDataCollection(const EndMessage &message) {
|
|
std::vector<uint8_t> serialization_buffer(MESSAGE_SIZE_FOR_START_END);
|
|
CBORStream2Serializer serializer(serialization_buffer.data(), serialization_buffer.size());
|
|
serializer.SerializeSequenceEnd(message);
|
|
socket.Send(serialization_buffer.data(), serializer.GetBufferSize(), true);
|
|
}
|
|
|
|
std::string ZMQPreviewSocket::GetAddress() {
|
|
return socket.GetEndpointName();
|
|
}
|
|
|
|
ZMQPreviewSettings ZMQPreviewSocket::GetSettings() {
|
|
return {
|
|
.period = counter.GetPeriod(),
|
|
.address = socket.GetEndpointName()
|
|
};
|
|
}
|
|
|
|
ZMQPreviewSocket &ZMQPreviewSocket::ImportSettings(const ZMQPreviewSettings &settings) {
|
|
Period(settings.period);
|
|
return *this;
|
|
}
|