127 lines
4.5 KiB
C++
127 lines
4.5 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JUNGFRAUJOCH_JFJOCHRECEIVER_H
|
|
#define JUNGFRAUJOCH_JFJOCHRECEIVER_H
|
|
|
|
#include <condition_variable>
|
|
#include <atomic>
|
|
#include <future>
|
|
#include <latch>
|
|
|
|
#include "JFJochReceiver.h"
|
|
|
|
#include "../acquisition_device/AcquisitionDeviceGroup.h"
|
|
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/JFJochException.h"
|
|
#include "FrameTransformation.h"
|
|
#include "../image_pusher/ImagePusher.h"
|
|
#include "../common/Logger.h"
|
|
#include "../common/ThreadSafeFIFO.h"
|
|
#include "../common/NUMAHWPolicy.h"
|
|
#include "../common/ImageBuffer.h"
|
|
#include "../common/PixelMask.h"
|
|
|
|
#include "../image_analysis/spot_finding/StrongPixelSet.h"
|
|
#include "../common/AzimuthalIntegration.h"
|
|
#include "../common/AzimuthalIntegrationProfile.h"
|
|
|
|
#include "../jungfrau/JFCalibration.h"
|
|
#include "../common/ADUHistogram.h"
|
|
#include "../common/Plot.h"
|
|
#include "JFJochReceiverPlots.h"
|
|
#include "../preview/PreviewImage.h"
|
|
#include "LossyFilter.h"
|
|
#include "JFJochReceiverCurrentStatus.h"
|
|
#include "../jungfrau/JFPedestalCalc.h"
|
|
#include "../preview/ZMQPreviewSocket.h"
|
|
#include "../preview/ZMQMetadataSocket.h"
|
|
#include "ImageMetadata.h"
|
|
#include "../common/ModuleSummation.h"
|
|
#include "../image_analysis/MXAnalysisAfterFPGA.h"
|
|
#include "../common/MovingAverage.h"
|
|
#include "JFJochReceiverOutput.h"
|
|
|
|
class JFJochReceiverFPGA : public JFJochReceiver {
|
|
const JFCalibration *calibration;
|
|
|
|
std::vector<std::unique_ptr<JFPedestalCalc>> pedestal;
|
|
|
|
std::vector<uint16_t> roi_map;
|
|
|
|
std::vector<std::future<void>> frame_transformation_futures;
|
|
std::vector<std::future<void>> data_acquisition_futures;
|
|
|
|
std::future<void> measurement;
|
|
|
|
ThreadSafeFIFO<int64_t> images_to_go;
|
|
|
|
AcquisitionDeviceGroup &acquisition_device;
|
|
uint16_t ndatastreams{0};
|
|
|
|
const int64_t frame_transformation_nthreads;
|
|
const int64_t pedestal_nthreads;
|
|
const int64_t summation_nthreads;
|
|
|
|
int64_t expected_packets_per_image;
|
|
bool only_2nd_sc_pedestal = false; // special mode used for pedestal G1/G2 with storage cells
|
|
|
|
std::latch frame_transformation_ready;
|
|
std::latch data_acquisition_ready;
|
|
|
|
void AcquireThread(uint16_t data_stream);
|
|
void LoadCalibrationToFPGA(uint16_t data_stream);
|
|
|
|
void FrameTransformationThread(uint32_t threadid);
|
|
void MeasurePedestalThread(uint16_t data_stream,
|
|
uint16_t module_number,
|
|
uint16_t storage_cell,
|
|
uint32_t threadid,
|
|
bool ignore);
|
|
|
|
int64_t SummationThread(uint16_t data_stream,
|
|
int64_t image_number,
|
|
uint16_t module_number,
|
|
uint32_t threadid,
|
|
ModuleSummation &summation);
|
|
void Cancel(const JFJochException &e) override;
|
|
void FinalizeMeasurement();
|
|
|
|
void SendCalibration();
|
|
void SendPedestal(const std::string &prefix, const std::vector<uint8_t> &v, int gain, int sc);
|
|
void RetrievePedestal(std::vector<JFModulePedestal> &output) const;
|
|
public:
|
|
JFJochReceiverFPGA(const DiffractionExperiment& experiment,
|
|
const PixelMask &pixel_mask,
|
|
const JFCalibration *calibration,
|
|
AcquisitionDeviceGroup &acquisition_devices,
|
|
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,
|
|
ZMQMetadataSocket *zmq_metadata_socket,
|
|
IndexerThreadPool *indexing_thread_pool);
|
|
~JFJochReceiverFPGA();
|
|
JFJochReceiverFPGA(const JFJochReceiverFPGA &other) = delete;
|
|
JFJochReceiverFPGA& operator=(const JFJochReceiverFPGA &other) = delete;
|
|
|
|
void StopReceiver() override;
|
|
|
|
float GetEfficiency() const override;
|
|
void Cancel(bool silent) override;
|
|
float GetProgress() const override;
|
|
|
|
void SetSpotFindingSettings(const SpotFindingSettings &spot_finding_settings) override;
|
|
|
|
JFJochReceiverOutput GetFinalStatistics() const override;
|
|
};
|
|
|
|
#endif //JUNGFRAUJOCH_JFJOCHRECEIVER_H
|
|
|