Files
Jungfraujoch/writer/HDF5DataFilePluginMX.h
T
leonarski_fandClaude Opus 5 14e2bf3c0f Merge rc167 into the detection-score branch
Both lanes added a per-image scalar to the same eleven files, so every conflict was two
additions competing for one line. All were resolved by keeping both, with three that needed
more than that:

- ScanResultGenerator: rc167 changed the per-image float vectors to resize(n, NAN) so a frame
  that never arrived does not read back as a real 0. v_protein_score and v_ice_score are exactly
  that case - 0 is a real answer ("nothing detected here") - so they take the NAN default too.
- HDF5MetadataSource: rc167 established that NaN in a stored per-image array means "no value" and
  the optional must come back absent. The two detection scores now follow it, which they did not
  before the merge; without the guard a missing score would come back as a NaN that a threshold
  would silently compare against.
- CBORTest: designated initialisers must follow member declaration order, so spindle_blind_fraction
  precedes the two scores in the DataMessage aggregate.

Verified after the merge that every CBOR key that is encoded is also decoded (198 encoded keys,
one intentional exception: the "type" discriminator), that both lanes' fields survive in the
writer, the reader, the plots and the API, and that a stored file still round-trips.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01EFEJG6WBQv8th4UJFNe53N
2026-09-08 00:13:51 +02:00

75 lines
2.6 KiB
C++

// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only
#pragma once
#include "HDF5DataFilePlugin.h"
#include "../common/AutoIncrVector.h"
class HDF5DataFilePluginMX : public HDF5DataFilePlugin {
const size_t max_spots;
const size_t max_extra_lattices;
const bool indexing;
// spots
std::vector<float> spot_x;
std::vector<float> spot_y;
std::vector<float> spot_int;
std::vector<int8_t> spot_indexed;
std::vector<int8_t> spot_ice_ring;
std::vector<int8_t> spot_lattice;
std::vector<int32_t> spot_h;
std::vector<int32_t> spot_k;
std::vector<int32_t> spot_l;
std::vector<float> spot_dist_ewald;
AutoIncrVector<float> beam_corr_x{NAN};
AutoIncrVector<float> beam_corr_y{NAN};
AutoIncrVector<uint32_t> npeaks;
AutoIncrVector<uint32_t> strong_pixel_count;
AutoIncrVector<uint32_t> spot_count_total;
AutoIncrVector<uint32_t> spot_count_ice;
AutoIncrVector<float> spot_count_ice_control{NAN};
AutoIncrVector<uint32_t> spot_count_indexed;
AutoIncrVector<uint32_t> spot_count_low_res;
// indexing
AutoIncrVector<uint8_t> indexed;
AutoIncrVector<uint32_t> indexing_lattice_count;
std::vector<float> indexed_lattice;
std::vector<float> extra_lattices; // [nimages, max_extra_lattices, 9], NaN-filled
// crystal
AutoIncrVector<float> profile_radius{NAN};
AutoIncrVector<float> mosaicity_deg{NAN};
AutoIncrVector<float> b_factor{NAN};
// bkg_estimate
AutoIncrVector<float> bkg_estimate{NAN};
AutoIncrVector<float> ice_ring_score{NAN};
AutoIncrVector<float> protein_score{NAN};
AutoIncrVector<float> ice_score{NAN};
// NaN = the frame had no value (CANNOT SAY), which is not a zero - see SpindleTrigger.
AutoIncrVector<float> spindle_blind_fraction{NAN};
// resolution_estimation
AutoIncrVector<float> resolution_estimate{NAN};
AutoIncrVector<uint32_t> niggli_class;
AutoIncrVector<std::string> bravais_lattice;
AutoIncrVector<int64_t> integrated_reflections;
// scaling
bool image_scale_present = false;
AutoIncrVector<float> image_scale_factor{NAN};
AutoIncrVector<float> image_scale_cc{NAN};
AutoIncrVector<float> image_scale_mosaicity{NAN};
public:
explicit HDF5DataFilePluginMX(const StartMessage& msg);
void OpenFile(HDF5File &data_file, const DataMessage& msg, size_t images_per_file) override;
void Write(const DataMessage& msg, uint64_t image_number) override;
void WriteFinal(HDF5File &data_file) override;
};