52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#include "ZMQPreviewPublisher.h"
|
|
|
|
#include "CBORStream2Serializer.h"
|
|
|
|
ZMQPreviewPublisher::ZMQPreviewPublisher(ZMQContext& context, const std::string& addr) :
|
|
socket(context, ZMQSocketType::Pub) {
|
|
socket.SendWaterMark(2).NoLinger();
|
|
socket.Bind(addr);
|
|
}
|
|
|
|
void ZMQPreviewPublisher::SetPreviewStride(int64_t preview_stride) {
|
|
std::unique_lock<std::mutex> ul(m);
|
|
stride = preview_stride;
|
|
}
|
|
|
|
void ZMQPreviewPublisher::StartDataCollection(const StartMessage& message) {
|
|
std::unique_lock<std::mutex> ul(m);
|
|
current_part = -1;
|
|
|
|
size_t approx_size = 1024*1024;
|
|
for (const auto &x : message.pixel_mask)
|
|
approx_size += x.size;
|
|
|
|
for (const auto &x : message.calibration)
|
|
approx_size += x.size;
|
|
|
|
std::vector<uint8_t> serialization_buffer(approx_size);
|
|
CBORStream2Serializer serializer(serialization_buffer.data(), serialization_buffer.size());
|
|
serializer.SerializeSequenceStart(message);
|
|
socket.Send(serialization_buffer.data(), serializer.GetBufferSize());
|
|
}
|
|
|
|
void ZMQPreviewPublisher::SendImage(const DataMessage& message) {
|
|
{
|
|
std::unique_lock<std::mutex> ul(m);
|
|
int64_t part = message.number / stride;
|
|
if (current_part >= part)
|
|
return;
|
|
else
|
|
current_part = part;
|
|
}
|
|
size_t approx_size = message.image.size + 2*1024*1024;
|
|
std::vector<uint8_t> serialization_buffer(approx_size);
|
|
|
|
CBORStream2Serializer serializer(serialization_buffer.data(), serialization_buffer.size());
|
|
serializer.SerializeImage(message);
|
|
|
|
socket.Send(serialization_buffer.data(), serializer.GetBufferSize());
|
|
}
|