v1.0.0-rc.68
This commit is contained in:
@@ -7,6 +7,8 @@
|
||||
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;
|
||||
|
||||
@@ -14,6 +16,7 @@ void HDF5DataFilePluginMX::OpenFile(HDF5File &data_file, const DataMessage &msg)
|
||||
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);
|
||||
@@ -23,11 +26,13 @@ void HDF5DataFilePluginMX::OpenFile(HDF5File &data_file, const DataMessage &msg)
|
||||
b_factor.reserve(RESERVE_IMAGES);
|
||||
indexed_lattice.reserve(9 * RESERVE_IMAGES);
|
||||
spot_count_rings.reserve(RESERVE_IMAGES);
|
||||
bkg_estimate.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;
|
||||
|
||||
@@ -38,6 +43,8 @@ void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number)
|
||||
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);
|
||||
}
|
||||
@@ -49,6 +56,7 @@ void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number)
|
||||
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;
|
||||
}
|
||||
|
||||
@@ -65,9 +73,6 @@ void HDF5DataFilePluginMX::Write(const DataMessage &msg, uint64_t image_number)
|
||||
if (msg.b_factor.has_value())
|
||||
b_factor[image_number] = msg.b_factor.value();
|
||||
|
||||
if (msg.bkg_estimate.has_value())
|
||||
bkg_estimate[image_number] = msg.bkg_estimate.value();
|
||||
|
||||
if (msg.resolution_estimate.has_value())
|
||||
resolution_estimate[image_number] = msg.resolution_estimate.value();
|
||||
|
||||
@@ -90,6 +95,7 @@ void HDF5DataFilePluginMX::WriteFinal(HDF5File &data_file) {
|
||||
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());
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user