32 lines
1.3 KiB
C++
32 lines
1.3 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#pragma once
|
|
|
|
#include "TCPStreamPusherSocket.h"
|
|
#include "ZMQWriterNotificationPuller.h"
|
|
|
|
class TCPStreamPusher : public ImagePusher {
|
|
std::vector<uint8_t> serialization_buffer;
|
|
CBORStream2Serializer serializer;
|
|
std::vector<std::unique_ptr<TCPStreamPusherSocket>> socket;
|
|
|
|
int64_t images_per_file = 1;
|
|
uint64_t run_number = 0;
|
|
std::string run_name;
|
|
std::atomic<bool> transmission_error = false;
|
|
public:
|
|
explicit TCPStreamPusher(const std::vector<std::string>& addr,
|
|
std::optional<int32_t> send_buffer_size = {},
|
|
std::optional<size_t> zerocopy_threshold = {},
|
|
size_t send_queue_size = 4096);
|
|
|
|
void StartDataCollection(StartMessage& message) override;
|
|
bool EndDataCollection(const EndMessage& message) override;
|
|
bool SendImage(const uint8_t *image_data, size_t image_size, int64_t image_number) override;
|
|
void SendImage(ZeroCopyReturnValue &z) override;
|
|
bool SendCalibration(const CompressedImage& message) override;
|
|
|
|
std::string Finalize() override;
|
|
std::string PrintSetup() const override;
|
|
}; |