leonarski_f
6133da1377
Some checks failed
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 11m5s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 11m6s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 13m47s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 14m9s
Build Packages / Generate python client (push) Successful in 49s
Build Packages / Unit tests (push) Has been skipped
Build Packages / Create release (push) Has been skipped
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 12m38s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 12m35s
Build Packages / build:rpm (rocky8) (push) Successful in 9m16s
Build Packages / build:rpm (rocky9) (push) Successful in 10m22s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 9m20s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 7m52s
Build Packages / Build documentation (push) Failing after 9s
This is an UNSTABLE release. The release has significant modifications and bug fixes, if things go wrong, it is better to revert to 1.0.0-rc.132. * jfjoch_broker: Use httplib for HTTP server instead of Pistache * jfjoch_broker: Drop OpenSSL support * jfjoch_broker: Base work for multi-lattice support in the future * Update dependencies to more recent versions (spdlog, HDF5, Catch2, httplib) Reviewed-on: #41
50 lines
1.8 KiB
C++
50 lines
1.8 KiB
C++
|
|
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_HDF5FILEPUSHER_H
|
|
#define JFJOCH_HDF5FILEPUSHER_H
|
|
|
|
#include <future>
|
|
|
|
#include "ImagePusher.h"
|
|
#include "../writer/FileWriter.h"
|
|
#include "../common/ThreadSafeFIFO.h"
|
|
#include "../common/ZMQWrappers.h"
|
|
#include "../common/Logger.h"
|
|
|
|
class HDF5FilePusher : public ImagePusher {
|
|
std::unique_ptr<FileWriter> writer;
|
|
std::mutex m;
|
|
std::future<void> writer_future;
|
|
ThreadSafeFIFO<ImagePusherQueueElement> writer_queue;
|
|
void WriterThread();
|
|
|
|
std::atomic<uint64_t> images_written = 0;
|
|
|
|
static constexpr uint32_t default_repub_watermark = 220;
|
|
static constexpr auto RepubTimeout = std::chrono::milliseconds(100);
|
|
|
|
std::unique_ptr<ZMQSocket> repub_socket;
|
|
bool repub_active = false;
|
|
Logger logger{"HDF5FilePusher"};
|
|
public:
|
|
explicit HDF5FilePusher(const std::string &repub_address = "",
|
|
const std::optional<int32_t> &repub_watermark = {});
|
|
// Thread safety: StartDataCollection, EndDataCollection and SendCalibration must run poorly in serial context
|
|
// SendImage can be executed in parallel
|
|
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 PrintSetup() const override;
|
|
|
|
std::optional<uint64_t> GetImagesWritten() const override;
|
|
size_t GetConnectedWriters() const override;
|
|
ImagePusherType GetType() const override { return ImagePusherType::HDF5; }
|
|
};
|
|
|
|
#endif //JFJOCH_HDF5FILEPUSHER_H
|