121 lines
5.1 KiB
C++
121 lines
5.1 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#include "HDF5DataFilePluginMX.h"
|
|
#define RESERVE_IMAGES 1000
|
|
|
|
HDF5DataFilePluginMX::HDF5DataFilePluginMX(size_t in_max_spots) : max_spots(in_max_spots) {}
|
|
|
|
void HDF5DataFilePluginMX::OpenFile(HDF5File &data_file, const DataMessage &msg) {
|
|
bkg_estimate.reserve(RESERVE_IMAGES);
|
|
|
|
if (max_spots == 0)
|
|
return;
|
|
|
|
spot_x.reserve(max_spots * RESERVE_IMAGES);
|
|
spot_y.reserve(max_spots * RESERVE_IMAGES);
|
|
spot_int.reserve(max_spots * RESERVE_IMAGES);
|
|
spot_indexed.reserve(max_spots * RESERVE_IMAGES);
|
|
spot_ice_ring.reserve(max_spots * RESERVE_IMAGES);
|
|
angle_from_ewald_sphere.reserve(max_spots * RESERVE_IMAGES);
|
|
|
|
npeaks.reserve(RESERVE_IMAGES);
|
|
strong_pixel_count.reserve(RESERVE_IMAGES);
|
|
indexed.reserve(RESERVE_IMAGES);
|
|
mosaicity.reserve(RESERVE_IMAGES);
|
|
b_factor.reserve(RESERVE_IMAGES);
|
|
indexed_lattice.reserve(9 * RESERVE_IMAGES);
|
|
spot_count_rings.reserve(RESERVE_IMAGES);
|
|
resolution_estimate.reserve(RESERVE_IMAGES);
|
|
}
|
|
|
|
void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number) {
|
|
if (msg.bkg_estimate.has_value())
|
|
bkg_estimate[image_number] = msg.bkg_estimate.value();
|
|
|
|
if (max_spots == 0)
|
|
return;
|
|
|
|
if (image_number >= max_image_number) {
|
|
max_image_number = image_number;
|
|
|
|
spot_x.resize(max_spots * (max_image_number + 1));
|
|
spot_y.resize(max_spots * (max_image_number + 1));
|
|
spot_int.resize(max_spots * (max_image_number + 1));
|
|
spot_indexed.resize(max_spots * (max_image_number + 1));
|
|
spot_ice_ring.resize(max_spots * (max_image_number + 1));
|
|
|
|
indexed_lattice.resize((max_image_number + 1) * 9);
|
|
angle_from_ewald_sphere.resize((max_image_number + 1) * 9);
|
|
}
|
|
|
|
uint32_t spot_cnt = std::min(msg.spots.size(), max_spots);
|
|
|
|
for (int i = 0; i < spot_cnt; i++) {
|
|
spot_x[max_spots * image_number + i] = msg.spots[i].x;
|
|
spot_y[max_spots * image_number + i] = msg.spots[i].y;
|
|
spot_int[max_spots * image_number + i] = msg.spots[i].intensity;
|
|
spot_indexed[max_spots * image_number + i] = msg.spots[i].indexed;
|
|
spot_ice_ring[max_spots * image_number + i] = msg.spots[i].ice_ring;
|
|
angle_from_ewald_sphere[max_spots * image_number + i] = msg.spots[i].angle_from_ewald_sphere;
|
|
}
|
|
|
|
npeaks[image_number] = spot_cnt;
|
|
if (msg.strong_pixel_count.has_value())
|
|
strong_pixel_count[image_number] = msg.strong_pixel_count.value();
|
|
|
|
if (msg.indexing_result.has_value())
|
|
indexed[image_number] = msg.indexing_result.value();
|
|
|
|
if (msg.mosaicity.has_value())
|
|
mosaicity[image_number] = msg.mosaicity.value();
|
|
|
|
if (msg.b_factor.has_value())
|
|
b_factor[image_number] = msg.b_factor.value();
|
|
|
|
if (msg.resolution_estimate.has_value())
|
|
resolution_estimate[image_number] = msg.resolution_estimate.value();
|
|
|
|
if (msg.spot_count_ice_rings)
|
|
spot_count_rings[image_number] = msg.spot_count_ice_rings.value();
|
|
|
|
if (msg.indexing_lattice) {
|
|
auto tmp = msg.indexing_lattice->GetVector();
|
|
|
|
for (int i = 0; i < 9; i++)
|
|
indexed_lattice[image_number * 9 + i] = tmp[i];
|
|
}
|
|
}
|
|
|
|
void HDF5DataFilePluginMX::WriteFinal(HDF5File &data_file) {
|
|
HDF5Group(data_file, "/entry/MX").NXClass("NXcollection");
|
|
|
|
if (!spot_x.empty()) {
|
|
data_file.SaveVector("/entry/MX/peakXPosRaw", spot_x, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/peakYPosRaw", spot_y, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/peakTotalIntensity", spot_int, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/peakIndexed", spot_indexed, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/peakIceRingRes", spot_ice_ring, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/peakIndexedAngleFromEwald", angle_from_ewald_sphere, {(hsize_t) (max_image_number + 1), max_spots});
|
|
data_file.SaveVector("/entry/MX/nPeaks", npeaks.vec());
|
|
}
|
|
|
|
if (!strong_pixel_count.empty())
|
|
data_file.SaveVector("/entry/MX/strongPixels", strong_pixel_count.vec());
|
|
if (!spot_count_rings.empty())
|
|
data_file.SaveVector("/entry/MX/nPeaksRingFiltered", spot_count_rings.vec());
|
|
|
|
if (!indexed.empty())
|
|
data_file.SaveVector("/entry/MX/imageIndexed", indexed.vec());
|
|
if (!indexed_lattice.empty())
|
|
data_file.SaveVector("/entry/MX/latticeIndexed", indexed_lattice, {(hsize_t) (max_image_number + 1), 9});
|
|
if (!bkg_estimate.empty())
|
|
data_file.SaveVector("/entry/MX/bkgEstimate", bkg_estimate.vec());
|
|
if (!mosaicity.empty())
|
|
data_file.SaveVector("/entry/MX/mosaicity", mosaicity.vec());
|
|
if (!b_factor.empty())
|
|
data_file.SaveVector("/entry/MX/bFactor", b_factor.vec());
|
|
if (!resolution_estimate.empty())
|
|
data_file.SaveVector("/entry/MX/resolutionEstimate", resolution_estimate.vec());
|
|
}
|