73 lines
2.6 KiB
C++
73 lines
2.6 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_JFJOCHRECEIVERLITE_H
|
|
#define JFJOCH_JFJOCHRECEIVERLITE_H
|
|
|
|
#include <future>
|
|
#include <latch>
|
|
|
|
#include "JFJochReceiver.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/PixelMask.h"
|
|
#include "../common/ImageBuffer.h"
|
|
#include "../common/Logger.h"
|
|
#include "../image_pusher/ImagePusher.h"
|
|
#include "../common/NUMAHWPolicy.h"
|
|
#include "../preview/PreviewImage.h"
|
|
#include "../preview/ZMQPreviewSocket.h"
|
|
#include "../preview/ZMQMetadataSocket.h"
|
|
#include "LossyFilter.h"
|
|
#include "../common/MovingAverage.h"
|
|
#include "JFJochReceiverCurrentStatus.h"
|
|
#include "JFJochReceiverPlots.h"
|
|
#include "JFJochReceiverOutput.h"
|
|
#include "../image_puller/ZMQImagePuller.h"
|
|
#include "../image_analysis/ImageAnalysisCPU.h"
|
|
|
|
class JFJochReceiverLite : public JFJochReceiver {
|
|
ImagePuller &image_puller;
|
|
|
|
std::atomic<bool> end_message_received{false};
|
|
|
|
const int64_t data_analysis_nthreads;
|
|
|
|
std::vector<std::future<void>> data_analysis_futures;
|
|
std::future<void> measurement;
|
|
|
|
std::latch measurement_started;
|
|
std::latch data_analysis_started;
|
|
|
|
void MeasurementThread();
|
|
void DataAnalysisThread(uint32_t id);
|
|
void Configure(const StartMessage &message);
|
|
public:
|
|
|
|
static int64_t NumberOfDataAnalysisThreads(int64_t requested_thread_number, const DiffractionExperiment& experiment);
|
|
|
|
JFJochReceiverLite(const DiffractionExperiment& experiment,
|
|
const PixelMask &pixel_mask,
|
|
ImagePuller &image_puller,
|
|
ImagePusher &image_pusher,
|
|
Logger &logger, int64_t forward_and_sum_nthreads,
|
|
const NUMAHWPolicy &numa_policy,
|
|
const SpotFindingSettings &spot_finding_settings,
|
|
PreviewImage &preview_image,
|
|
JFJochReceiverCurrentStatus ¤t_status,
|
|
JFJochReceiverPlots &plots,
|
|
ImageBuffer &image_buffer,
|
|
ZMQPreviewSocket *zmq_preview_socket = nullptr,
|
|
ZMQMetadataSocket *zmq_metadata_socket = nullptr,
|
|
IndexerThreadPool *indexing_thread_pool = nullptr);
|
|
~JFJochReceiverLite() override;
|
|
|
|
void StopReceiver() override;
|
|
|
|
float GetEfficiency() const override;
|
|
float GetProgress() const override;
|
|
|
|
void SetSpotFindingSettings(const SpotFindingSettings &spot_finding_settings) override;
|
|
};
|
|
|
|
#endif //JFJOCH_JFJOCHRECEIVERLITE_H
|