mx/Jungfraujoch
0
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases 37 Wiki Activity

v1.0.0-rc.31

Browse Source
This commit is contained in:
leonarski_f
2025-03-02 13:15:28 +01:00
parent aeabc81a4c
commit ddf4c75645
309 changed files with 8705 additions and 1421 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
writer/HDF5DataFilePluginMX.cpp
View File

@@ -13,11 +13,13 @@ void HDF5DataFilePluginMX::OpenFile(HDF5File &data_file, const DataMessage &msg)
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);
npeaks.reserve(RESERVE_IMAGES);
strong_pixel_count.reserve(RESERVE_IMAGES);
indexed.reserve(RESERVE_IMAGES);
indexed_lattice.reserve(9 * RESERVE_IMAGES);
spot_count_rings.reserve(RESERVE_IMAGES);
bkg_estimate.reserve(RESERVE_IMAGES);
}
void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number) {
@@ -30,14 +32,8 @@ void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t 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));
npeaks.resize(max_image_number + 1);
strong_pixel_count.resize(max_image_number + 1);
indexed.resize(max_image_number + 1);
spot_indexed.resize(max_spots * (max_image_number + 1));
indexed_lattice.resize((max_image_number + 1) * 9);
if (msg.spot_count_in_rings)
spot_count_rings.resize(max_image_number + 1);
}
uint32_t spot_cnt = std::min(msg.spots.size(), max_spots);
@@ -46,11 +42,14 @@ void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number)
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;
}
npeaks[image_number] = spot_cnt;
strong_pixel_count[image_number] = msg.strong_pixel_count;
indexed[image_number] = msg.indexing_result;
bkg_estimate[image_number] = msg.bkg_estimate.value_or(0.0f);
if (msg.spot_count_in_rings)
spot_count_rings[image_number] = msg.spot_count_in_rings.value();
@@ -67,16 +66,19 @@ void HDF5DataFilePluginMX::WriteFinal(HDF5File &data_file) {
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/nPeaks", npeaks);
data_file.SaveVector("/entry/MX/strongPixels", strong_pixel_count);
data_file.SaveVector("/entry/MX/nPeaks", npeaks.vec());
data_file.SaveVector("/entry/MX/strongPixels", strong_pixel_count.vec());
}
if (!spot_count_rings.empty())
data_file.SaveVector("/entry/MX/nPeaksRingFiltered", spot_count_rings);
data_file.SaveVector("/entry/MX/nPeaksRingFiltered", spot_count_rings.vec());
if (!indexed.empty())
data_file.SaveVector("/entry/MX/imageIndexed", indexed);
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());
}