132 lines
4.3 KiB
C++
132 lines
4.3 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#pragma once
|
|
|
|
#include <map>
|
|
#include <shared_mutex>
|
|
#include <string>
|
|
|
|
#include "StatusVector.h"
|
|
#include "Histogram.h"
|
|
#include "ADUHistogram.h"
|
|
#include "AutoIncrVector.h"
|
|
#include "DiffractionExperiment.h"
|
|
#include "AzimuthalIntegrationProfile.h"
|
|
#include "JFJochMessages.h"
|
|
#include "Plot.h"
|
|
#include "ScanResult.h"
|
|
|
|
struct MeanProcessingTime {
|
|
float compression;
|
|
float preprocessing;
|
|
float azint;
|
|
float spot_finding;
|
|
float indexing;
|
|
float refinement;
|
|
float integration;
|
|
float bragg_prediction;
|
|
float processing;
|
|
float indexing_analysis;
|
|
float image_scale;
|
|
};
|
|
|
|
class JFJochReceiverPlots {
|
|
mutable std::mutex m; // protects xfel_pulse_id, xfel_event_code and az_int_profile
|
|
|
|
std::optional<GoniometerAxis> goniometer;
|
|
std::optional<GridScanSettings> grid_scan;
|
|
int64_t default_binning = 1;
|
|
std::unique_ptr<AzimuthalIntegrationProfile> az_int_profile;
|
|
AutoIncrVector<uint64_t> xfel_pulse_id;
|
|
AutoIncrVector<uint64_t> xfel_event_code;
|
|
|
|
StatusVector bkg_estimate;
|
|
StatusVector spot_count;
|
|
StatusVector spot_count_low_res;
|
|
StatusVector spot_count_indexed;
|
|
StatusVector spot_count_ice;
|
|
|
|
StatusVector indexing_solution;
|
|
StatusVector indexing_lattice_count;
|
|
StatusVector indexing_uc_a;
|
|
StatusVector indexing_uc_b;
|
|
StatusVector indexing_uc_c;
|
|
StatusVector indexing_uc_alpha;
|
|
StatusVector indexing_uc_beta;
|
|
StatusVector indexing_uc_gamma;
|
|
StatusVector error_pixels;
|
|
StatusVector saturated_pixels;
|
|
StatusVector strong_pixels;
|
|
StatusVector receiver_delay;
|
|
StatusVector receiver_buf_available;
|
|
StatusVector receiver_buf_in_sending;
|
|
StatusVector receiver_buf_in_preparation;
|
|
StatusVector image_collection_efficiency;
|
|
StatusVector packets_received;
|
|
StatusVector max_value;
|
|
StatusVector resolution_estimate;
|
|
StatusVector integrated_reflections;
|
|
StatusVector image_scale_factor;
|
|
StatusVector image_scale_cc;
|
|
StatusVector image_scale_b;
|
|
StatusVector compression_ratio;
|
|
|
|
// StatusVector objects are fully thread-safe (protected by internal mutex)
|
|
// It is OK to have concurrent access to StatusVector
|
|
// roi_m lock is needed to make sure that std::map<std::string, ROIStatus> is not mutable within critical section
|
|
// so no new elements added outside of a unique lock, but it is OK to modify ROIStatus under shared lock
|
|
struct ROIStatus {
|
|
StatusVector sum;
|
|
StatusVector max_count;
|
|
StatusVector pixels;
|
|
StatusVector x;
|
|
StatusVector y;
|
|
StatusVector mean;
|
|
};
|
|
mutable std::shared_mutex roi_m;
|
|
std::map<std::string, ROIStatus> roi_status;
|
|
|
|
StatusVector profile_radius;
|
|
StatusVector mosaicity_deg;
|
|
StatusVector b_factor;
|
|
StatusVector pixel_sum;
|
|
|
|
StatusVector beam_center_x;
|
|
StatusVector beam_center_y;
|
|
|
|
StatusVector spot_finding_time;
|
|
StatusVector indexing_time;
|
|
StatusVector refinement_time;
|
|
StatusVector integration_time;
|
|
StatusVector bragg_prediction_time;
|
|
StatusVector total_processing_time;
|
|
StatusVector preprocessing_time;
|
|
StatusVector compression_time;
|
|
StatusVector azint_time;
|
|
StatusVector indexing_analysis_time;
|
|
StatusVector image_scale_time;
|
|
|
|
MultiLinePlot GetROIPlot(PlotType type, int64_t nbins, float start, float incr,
|
|
const std::optional<float> &fill_value) const;
|
|
public:
|
|
void Setup(const DiffractionExperiment& experiment, const AzimuthalIntegrationMapping& mapping);
|
|
|
|
void Add(const DataMessage& msg, const AzimuthalIntegrationProfile &profile);
|
|
void AddEmptyImage(const DataMessage& msg);
|
|
MultiLinePlot GetPlots(const PlotRequest& request);
|
|
|
|
void GetXFELPulseID(std::vector<uint64_t>& v) const;
|
|
void GetXFELEventCode(std::vector<uint64_t>& v) const;
|
|
|
|
std::optional<float> GetIndexingRate() const;
|
|
std::optional<float> GetBkgEstimate() const;
|
|
|
|
std::vector<float> GetAzIntProfile() const;
|
|
MultiLinePlot GetAzIntProfilePlot(bool force_1d = false, PlotAzintUnit azint_unit = PlotAzintUnit::Q_recipA) const;
|
|
|
|
MeanProcessingTime GetMeanProcessingTime() const;
|
|
|
|
void GetPlotRaw(std::vector<float> &v, PlotType type, const std::string &roi);
|
|
};
|