Jungfraujoch/image_pusher/TCPStreamPusher.h

// 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;

    std::unique_ptr<ZMQWriterNotificationPuller> writer_notification_socket;

    int64_t images_per_file = 1;
    uint64_t run_number = 0;
    std::string run_name;

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);

    TCPStreamPusher& WriterNotificationSocket(const std::string& addr);
    std::string GetWriterNotificationSocketAddress() const override;

    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;
};