53 lines
1.8 KiB
C++
53 lines
1.8 KiB
C++
// Copyright (2019-2024) Paul Scherrer Institute
|
|
|
|
#ifndef JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|
|
#define JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|
|
|
|
#include <mutex>
|
|
|
|
#include "ImagePusher.h"
|
|
#include "../common/ZMQWrappers.h"
|
|
#include "../preview/PreviewCounter.h"
|
|
#include "ZMQWriterNotificationPuller.h"
|
|
|
|
class ZMQStream2Pusher : public ImagePusher {
|
|
std::vector<uint8_t> serialization_buffer;
|
|
CBORStream2Serializer serializer;
|
|
|
|
std::vector<std::unique_ptr<ZMQSocket>> socket;
|
|
|
|
std::unique_ptr<ZMQSocket> preview_socket;
|
|
PreviewCounter preview_counter;
|
|
|
|
std::unique_ptr<ZMQWriterNotificationPuller> writer_notification_socket;
|
|
|
|
int64_t images_per_file = 1;
|
|
uint64_t run_number = 0;
|
|
std::string run_name;
|
|
public:
|
|
explicit ZMQStream2Pusher(const std::vector<std::string>& addr,
|
|
int32_t send_buffer_high_watermark = -1,
|
|
int32_t send_buffer_size = -1);
|
|
|
|
ZMQStream2Pusher& PreviewSocket(const std::string& addr);
|
|
ZMQStream2Pusher& WriterNotificationSocket(const std::string& addr);
|
|
ZMQStream2Pusher& PreviewCounterPeriod(std::chrono::microseconds input);
|
|
std::string GetPreviewAddress();
|
|
|
|
std::vector<std::string> GetAddress();
|
|
|
|
// Strictly serial, as order of these is important
|
|
void StartDataCollection(StartMessage& message) override;
|
|
bool EndDataCollection(const EndMessage& message) override;
|
|
bool SendCalibration(const CompressedImage& message) override;
|
|
|
|
// Thread-safe
|
|
void SendImage(const uint8_t *image_data, size_t image_size, int64_t image_number, ZeroCopyReturnValue *z) override;
|
|
bool SendImage(const uint8_t *image_data, size_t image_size, int64_t image_number) override;
|
|
|
|
void Finalize() override;
|
|
std::string GetWriterNotificationSocketAddress() const override;
|
|
};
|
|
|
|
#endif //JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|