139 lines
4.6 KiB
C++
139 lines
4.6 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#include "JFJochReceiverPlots.h"
|
|
|
|
void JFJochReceiverPlots::Setup(const DiffractionExperiment &experiment, const AzimuthalIntegrationMapping &mapping) {
|
|
{
|
|
std::unique_lock ul(az_int_profile_mutex);
|
|
az_int_profile = std::make_unique<AzimuthalIntegrationProfile>(mapping);
|
|
az_int_profile->SetTitle("dataset");
|
|
}
|
|
|
|
default_binning = experiment.GetDefaultPlotBinning();
|
|
|
|
// Reset all status vectors
|
|
xfel_pulse_id.Clear();
|
|
xfel_event_code.Clear();
|
|
bkg_estimate.Clear();
|
|
spot_count.Clear();
|
|
indexing_solution.Clear();
|
|
error_pixels.Clear();
|
|
strong_pixels.Clear();
|
|
receiver_delay.Clear();
|
|
receiver_free_send_buf.Clear();
|
|
image_collection_efficiency.Clear();
|
|
roi_sum.Clear();
|
|
roi_max_count.Clear();
|
|
roi_pixels.Clear();
|
|
}
|
|
|
|
void JFJochReceiverPlots::Add(const DataMessage &msg, const AzimuthalIntegrationProfile &profile) {
|
|
bkg_estimate.AddElement(msg.number, msg.bkg_estimate);
|
|
|
|
spot_count.AddElement("Spots (crystal)", msg.number, msg.spots.size());
|
|
spot_count.AddElement("Spots (rings)", msg.number, msg.spot_count_in_rings);
|
|
|
|
error_pixels.AddElement("Error", msg.number, msg.error_pixel_count);
|
|
error_pixels.AddElement("Saturated", msg.number, msg.saturated_pixel_count);
|
|
|
|
strong_pixels.AddElement(msg.number, msg.strong_pixel_count);
|
|
|
|
image_collection_efficiency.AddElement(msg.number, msg.image_collection_efficiency);
|
|
receiver_delay.AddElement(msg.number, msg.receiver_aq_dev_delay);
|
|
receiver_free_send_buf.AddElement(msg.number, msg.receiver_free_send_buf);
|
|
|
|
indexing_solution.AddElement(msg.number, msg.indexing_result);
|
|
{
|
|
std::unique_lock ul(az_int_profile_mutex);
|
|
*az_int_profile += profile;
|
|
}
|
|
|
|
xfel_pulse_id.AddElement(msg.number, msg.xfel_pulse_id);
|
|
xfel_event_code.AddElement(msg.number, msg.xfel_event_code);
|
|
|
|
for (const auto &[key, value] : msg.roi) {
|
|
roi_sum.AddElement(key, msg.number, value.sum);
|
|
roi_max_count.AddElement(key, msg.number, value.max_count);
|
|
roi_pixels.AddElement(key, msg.number, value.pixels);
|
|
}
|
|
}
|
|
|
|
void JFJochReceiverPlots::AddEmptyImage(const DataMessage &msg) {
|
|
image_collection_efficiency.AddElement(msg.number, msg.image_collection_efficiency);
|
|
}
|
|
|
|
MultiLinePlot JFJochReceiverPlots::GetPlots(const PlotRequest &request) {
|
|
MultiLinePlot ret;
|
|
auto nbins = default_binning;
|
|
if (request.binning > 0)
|
|
nbins = request.binning;
|
|
MultiLinePlot tmp;
|
|
switch (request.type) {
|
|
case PlotType::SpotCount:
|
|
return spot_count.GetMeanPlot(nbins);
|
|
case PlotType::IndexingRate:
|
|
return indexing_solution.GetMeanPlot(nbins);
|
|
case PlotType::BkgEstimate:
|
|
return bkg_estimate.GetMeanPlot(nbins);
|
|
case PlotType::ErrorPixels:
|
|
return error_pixels.GetMeanPlot(nbins);
|
|
case PlotType::ImageCollectionEfficiency:
|
|
return image_collection_efficiency.GetMeanPlot(nbins);
|
|
case PlotType::ReceiverDelay:
|
|
return receiver_delay.GetMeanPlot(nbins);
|
|
case PlotType::ReceiverFreeSendBuf:
|
|
return receiver_free_send_buf.GetMeanPlot(nbins);
|
|
case PlotType::StrongPixels:
|
|
return strong_pixels.GetMeanPlot(nbins);
|
|
case PlotType::ROISum:
|
|
return roi_sum.GetMeanPlot(nbins);
|
|
case PlotType::ROIMaxCount:
|
|
return roi_max_count.GetMeanPlot(nbins);
|
|
case PlotType::ROIPixels:
|
|
return roi_pixels.GetMeanPlot(nbins);
|
|
case PlotType::AzInt:
|
|
return GetAzIntProfilePlot();
|
|
default:
|
|
// Do nothing
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
std::optional<float> JFJochReceiverPlots::GetIndexingRate() const {
|
|
auto tmp = indexing_solution.Mean();
|
|
if (std::isfinite(tmp))
|
|
return tmp;
|
|
else
|
|
return {};
|
|
}
|
|
|
|
std::optional<float> JFJochReceiverPlots::GetBkgEstimate() const {
|
|
auto tmp = bkg_estimate.Mean();
|
|
if (std::isfinite(tmp))
|
|
return tmp;
|
|
else
|
|
return {};
|
|
}
|
|
|
|
void JFJochReceiverPlots::GetXFELPulseID(std::vector<uint64_t> &v) const {
|
|
v = xfel_pulse_id.ExportArray(0);
|
|
}
|
|
|
|
void JFJochReceiverPlots::GetXFELEventCode(std::vector<uint64_t> &v) const {
|
|
v = xfel_event_code.ExportArray(0);
|
|
}
|
|
|
|
std::vector<float> JFJochReceiverPlots::GetAzIntProfile() const {
|
|
std::unique_lock ul(az_int_profile_mutex);
|
|
if (!az_int_profile)
|
|
return {};
|
|
return az_int_profile->GetResult();
|
|
}
|
|
|
|
MultiLinePlot JFJochReceiverPlots::GetAzIntProfilePlot() const {
|
|
std::unique_lock ul(az_int_profile_mutex);
|
|
if (!az_int_profile)
|
|
return {};
|
|
return az_int_profile->GetPlot();
|
|
}
|