CrystFEL geometry: read the pixel masks and bad regions instead of ignoring them
The .geom parser now takes the masking information the format defines: - mask arrays: mask/mask_file/mask_good/mask_bad (mask 0) and maskN_data/maskN_file/maskN_goodbits/ maskN_badbits for N = 0..7; global keys hold for the panels after them, a panel may override. A mask that is the same on every panel is one whole-frame mask, otherwise one per panel, applied on that panel's pixels only. - a relative mask_file is looked for in the working directory (as CrystFEL does), then beside the .geom. - bad regions (bad*): a lab-plane box (min_x..max_y) or a fs/ss range on a panel (or, old syntax, on the whole frame); mask_edge_pixels and no_index. JFJochXFELReader already had the static pixel mask and the per-frame mask path for these; the masks go through them unchanged. A mask whose path holds % (one per shot group) is applied frame by frame like a 3D one. Where the geometry names mask arrays, the data file's own Cheetah masks (CXI mask_shared and per-shot mask with Cheetah's generic bad bits, /metadata/pixelmask) are not added: the geometry says which bits matter (e.g. an AGIPD mask_bad that includes bit 32). --mask-good / --mask-bad still add their masks to these. Tests: [CrystFELGeometry] (new keys, per-panel override, relative mask file), JFJochXFELReader_* (new GeometryMasks case on a synthetic CXI + .geom), [JFJochMultiReader], JFJochXTCReader*. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_015CeEGgpKdfkCTJb3YNMJcu
This commit is contained in:
1 parent
30ea607686
commit
7c23b4befe
10 files changed
+380
-49
No files matched your search
@@ -3,6 +3,9 @@
|
||||
|
||||
#include "CrystFELGeometry.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <array>
|
||||
#include <filesystem>
|
||||
#include <fstream>
|
||||
#include <map>
|
||||
#include <regex>
|
||||
@@ -98,8 +101,8 @@ bool Bookkeeping(const std::string &key) {
|
||||
// Keywords that change where a pixel lands or what it is worth, which this reader does not do.
|
||||
const std::set<std::string> REFUSED = {"detector_shift_x", "detector_shift_y"};
|
||||
|
||||
// Sets one panel property; false if the key is not one. Only where the panel is: its pixel range, axes,
|
||||
// corner, pixel size, depth offset and camera length.
|
||||
// Sets one panel property; false if the key is not one. Where the panel is - its pixel range, axes,
|
||||
// corner, pixel size, depth offset and camera length - and which of its pixels are left out.
|
||||
bool SetPanel(const std::string &filename, Panel &p, const std::string &key, const std::string &value) {
|
||||
if (key == "min_fs") p.min_fs = std::stoll(value);
|
||||
else if (key == "max_fs") p.max_fs = std::stoll(value);
|
||||
@@ -112,11 +115,58 @@ bool SetPanel(const std::string &filename, Panel &p, const std::string &key, con
|
||||
else if (key == "res") p.res = std::stod(value);
|
||||
else if (key == "clen") p.clen = ParseValue(filename, key, value, LENGTH_UNITS, 1e-3);
|
||||
else if (key == "coffset") p.coffset = std::stod(value);
|
||||
else if (key == "mask_edge_pixels") p.mask_edge_pixels = std::stoll(value);
|
||||
else if (key == "no_index") p.no_index = value == "1" || value == "true";
|
||||
else
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
// The up to eight masks a panel has (maskN_data, maskN_file, maskN_goodbits, maskN_badbits, N = 0..7;
|
||||
// mask, mask_file, mask_good and mask_bad are mask 0's). Sets the one the key names; false if the key
|
||||
// is not a mask key.
|
||||
using Masks = std::array<xfel::Mask, 8>;
|
||||
bool SetMask(Masks &masks, const std::string &key, const std::string &value) {
|
||||
static const std::map<std::string, std::string> OLD = {
|
||||
{"mask", "data"}, {"mask_file", "file"}, {"mask_good", "goodbits"}, {"mask_bad", "badbits"}};
|
||||
static const std::regex CURRENT(R"(mask([0-7])_(data|file|goodbits|badbits))");
|
||||
size_t n = 0;
|
||||
std::string field;
|
||||
std::smatch match;
|
||||
if (OLD.contains(key))
|
||||
field = OLD.at(key);
|
||||
else if (std::regex_match(key, match, CURRENT)) {
|
||||
n = std::stoul(match[1].str());
|
||||
field = match[2].str();
|
||||
} else
|
||||
return false;
|
||||
if (field == "data") masks[n].data = value;
|
||||
else if (field == "file") masks[n].file = value;
|
||||
else if (field == "goodbits") masks[n].good = static_cast<uint32_t>(std::stoul(value, nullptr, 0));
|
||||
else masks[n].bad = static_cast<uint32_t>(std::stoul(value, nullptr, 0));
|
||||
return true;
|
||||
}
|
||||
|
||||
// Sets one property of a bad region: a box in the laboratory plane (min_x .. max_y, in pixels) or a
|
||||
// pixel range in the stored frame (min_fs .. max_ss, inclusive) on one panel - or, in the old syntax,
|
||||
// with no panel, on the whole frame. False if the key is not one.
|
||||
bool SetBad(xfel::BadRegion &b, const std::string &key, const std::string &value) {
|
||||
if (key == "min_x") b.min_x = std::stod(value);
|
||||
else if (key == "max_x") b.max_x = std::stod(value);
|
||||
else if (key == "min_y") b.min_y = std::stod(value);
|
||||
else if (key == "max_y") b.max_y = std::stod(value);
|
||||
else if (key == "min_fs") b.min_fs = std::stoll(value);
|
||||
else if (key == "max_fs") b.max_fs = std::stoll(value);
|
||||
else if (key == "min_ss") b.min_ss = std::stoll(value);
|
||||
else if (key == "max_ss") b.max_ss = std::stoll(value);
|
||||
else if (key == "panel") b.panel = value;
|
||||
else
|
||||
return false;
|
||||
if (key.ends_with("_x") || key.ends_with("_y"))
|
||||
b.in_lab = true;
|
||||
return true;
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored) {
|
||||
@@ -128,7 +178,9 @@ xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored
|
||||
// Global panel keys hold for every panel that appears after them, as in the old syntax; a file in
|
||||
// the current syntax states them first, so this reads both the same.
|
||||
Panel defaults;
|
||||
std::map<std::string, size_t> panel_index;
|
||||
Masks default_masks;
|
||||
std::vector<Masks> panel_masks; // each panel's, beside geom.panels
|
||||
std::map<std::string, size_t> panel_index, bad_index;
|
||||
std::map<std::string, std::string> dims;
|
||||
|
||||
std::string line;
|
||||
@@ -147,7 +199,8 @@ xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored
|
||||
if (slash == std::string::npos) {
|
||||
if (key == "data") geom.data = value;
|
||||
else if (key.starts_with("dim")) dims[key] = value;
|
||||
else if (!SetPanel(filename, defaults, key, value) && !Bookkeeping(key))
|
||||
else if (!SetPanel(filename, defaults, key, value) && !SetMask(default_masks, key, value)
|
||||
&& !Bookkeeping(key))
|
||||
ignored.push_back(key);
|
||||
continue;
|
||||
}
|
||||
@@ -156,7 +209,12 @@ xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored
|
||||
const std::string prop = key.substr(slash + 1);
|
||||
|
||||
if (name.starts_with("bad")) {
|
||||
ignored.push_back(key);
|
||||
if (!bad_index.contains(name)) {
|
||||
bad_index[name] = geom.bad.size();
|
||||
geom.bad.emplace_back();
|
||||
}
|
||||
if (!SetBad(geom.bad[bad_index[name]], prop, value))
|
||||
ignored.push_back(key);
|
||||
continue;
|
||||
}
|
||||
|
||||
@@ -173,8 +231,11 @@ xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored
|
||||
panel_index[name] = geom.panels.size();
|
||||
geom.panels.push_back(defaults);
|
||||
geom.panels.back().name = name;
|
||||
panel_masks.push_back(default_masks);
|
||||
}
|
||||
if (!SetPanel(filename, geom.panels[panel_index[name]], prop, value) && !Bookkeeping(prop))
|
||||
const size_t n = panel_index[name];
|
||||
if (!SetPanel(filename, geom.panels[n], prop, value) && !SetMask(panel_masks[n], prop, value)
|
||||
&& !Bookkeeping(prop))
|
||||
ignored.push_back(key);
|
||||
}
|
||||
|
||||
@@ -195,6 +256,33 @@ xfel::Layout Read(const std::string &filename, std::vector<std::string> &ignored
|
||||
throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
|
||||
filename + ": panel " + p.name + " has no pixel size (res)");
|
||||
}
|
||||
|
||||
// Each mask once for the whole frame where every panel has it alike (the usual case: stated once,
|
||||
// before the panels), otherwise once for each panel that has it, on that panel's pixels only.
|
||||
for (size_t n = 0; n < default_masks.size(); n++) {
|
||||
const auto &first = panel_masks[0][n];
|
||||
const bool alike = std::all_of(panel_masks.begin(), panel_masks.end(),
|
||||
[&](const Masks &m) { return m[n] == first; });
|
||||
if (alike) {
|
||||
if (!first.data.empty())
|
||||
geom.masks.push_back(first);
|
||||
continue;
|
||||
}
|
||||
for (size_t i = 0; i < geom.panels.size(); i++) {
|
||||
if (panel_masks[i][n].data.empty())
|
||||
continue;
|
||||
geom.masks.push_back(panel_masks[i][n]);
|
||||
geom.masks.back().panel = geom.panels[i].name;
|
||||
}
|
||||
}
|
||||
// A mask file is named as it is, or relative to the working directory, as CrystFEL reads it; a
|
||||
// relative one not found there is looked for beside the geometry file.
|
||||
for (auto &m: geom.masks) {
|
||||
const std::filesystem::path file(m.file);
|
||||
const auto beside = std::filesystem::path(filename).parent_path() / file;
|
||||
if (!m.file.empty() && file.is_relative() && !std::filesystem::exists(file) && std::filesystem::exists(beside))
|
||||
m.file = beside.string();
|
||||
}
|
||||
return geom;
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user