leonarski_f
f1472a5fe1
* Writer can send a PUB message with information on closed file * Writer uses getopt to format input parameters * DiffractionExperiment: Add frame number limit
57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#ifndef JUNGFRAUJOCH_STREAMWRITER_H
|
|
#define JUNGFRAUJOCH_STREAMWRITER_H
|
|
|
|
#include <future>
|
|
|
|
#include "ZMQImagePuller.h"
|
|
#include "HDF5DataFile.h"
|
|
|
|
enum class StreamWriterState {Idle, Started, Receiving};
|
|
|
|
struct StreamWriterStatistics {
|
|
uint64_t processed_images;
|
|
float performance_MBs;
|
|
float performance_Hz;
|
|
std::string file_prefix;
|
|
StreamWriterState state;
|
|
};
|
|
|
|
struct StreamWriterOutput {
|
|
StreamWriterStatistics image_puller_stats;
|
|
std::vector<HDF5DataFileStatistics> data_file_stats;
|
|
};
|
|
|
|
class StreamWriter {
|
|
ZMQContext &zmq_context;
|
|
std::string file_done_address;
|
|
|
|
StreamWriterState state = StreamWriterState::Idle;
|
|
|
|
ZMQImagePullerOutput image_puller_output;
|
|
|
|
std::atomic<uint64_t> processed_images;
|
|
std::atomic<uint64_t> processed_image_size;
|
|
std::chrono::time_point<std::chrono::system_clock> start_time;
|
|
std::chrono::time_point<std::chrono::system_clock> end_time;
|
|
std::string file_prefix;
|
|
|
|
ZMQImagePuller image_puller;
|
|
Logger &logger;
|
|
void CollectImages(std::vector<HDF5DataFileStatistics> &v);
|
|
bool WaitForImage();
|
|
public:
|
|
StreamWriter(ZMQContext& context,
|
|
Logger &logger,
|
|
const std::string& zmq_addr,
|
|
const std::string& repub_address = "",
|
|
const std::string& file_done_address = "");
|
|
StreamWriterOutput Run();
|
|
void Cancel();
|
|
StreamWriterStatistics GetStatistics() const;
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_STREAMWRITER_H
|