leonarski_f
928789a67c
All checks were successful
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 10m14s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 10m4s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 12m0s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 10m31s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 10m0s
Build Packages / Generate python client (push) Successful in 45s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 11m17s
Build Packages / Create release (push) Has been skipped
Build Packages / Build documentation (push) Successful in 42s
Build Packages / build:rpm (rocky8) (push) Successful in 10m56s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 10m8s
Build Packages / build:rpm (rocky9) (push) Successful in 11m27s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 10m10s
Build Packages / Unit tests (push) Successful in 1h14m45s
This is an UNSTABLE release. The release has significant modifications and bug fixes, if things go wrong, it is better to revert to 1.0.0-rc.124. * jfjoch_broker: Rotation indexer has two retries if failes * jfjoch_broker: Rotation indexer handles small number of rotation images (like test shot) * jfjoch_broker: Integration calculates background mask based on R2 radius * jfjoch_process: HDF5 files are not saved by default Reviewed-on: #37
64 lines
2.2 KiB
C++
64 lines
2.2 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_ROTATIONINDEXER_H
|
|
#define JFJOCH_ROTATIONINDEXER_H
|
|
|
|
#include <vector>
|
|
#include <mutex>
|
|
|
|
#include "../common/DiffractionSpot.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "indexing/IndexerThreadPool.h"
|
|
#include "lattice_search/LatticeSearch.h"
|
|
|
|
// RotationIndexer works as following:
|
|
// 1. First accumulates spot results from rotation images (only images within a certain stride are included)
|
|
// 2. When a minimum number of images is reached (at least 10 images AND at least 10 deg), indexing is attempted
|
|
// 3. If indexing is successful - lattice is provided that is used by subsequent images
|
|
// 4. If indexing is not-successful - accumulation procedure is continued
|
|
|
|
struct RotationIndexerResult {
|
|
CrystalLattice lattice;
|
|
LatticeSearchResult search_result;
|
|
DiffractionGeometry geom;
|
|
std::optional<GoniometerAxis> axis;
|
|
};
|
|
|
|
class RotationIndexer {
|
|
mutable std::mutex m;
|
|
const DiffractionExperiment& experiment;
|
|
|
|
constexpr static int64_t max_spots = 10000;
|
|
constexpr static int64_t min_images_for_indexing = 10;
|
|
constexpr static int64_t max_indexing_range_multiplier = 3;
|
|
|
|
const bool index_ice_rings;
|
|
float angle_norm_deg = 1.0f;
|
|
|
|
std::vector<SpotToSave> v_;
|
|
std::vector<Coord> coords_;
|
|
std::optional<GoniometerAxis> axis_;
|
|
const DiffractionGeometry geom_;
|
|
DiffractionGeometry updated_geom_;
|
|
LatticeSearchResult search_result_;
|
|
|
|
IndexerThreadPool &indexer_;
|
|
|
|
int64_t last_accumulated_image = -1;
|
|
int64_t accumulated_images = 0;
|
|
|
|
int64_t image_stride = 1;
|
|
int64_t first_image_to_try_indexing = INT64_MAX;
|
|
int64_t next_image_to_try_indexing = INT64_MAX;
|
|
int64_t indexing_range_multiplier = 1;
|
|
|
|
std::optional<CrystalLattice> indexed_lattice;
|
|
void TryIndex();
|
|
public:
|
|
RotationIndexer(const DiffractionExperiment& x, IndexerThreadPool& indexer);
|
|
std::optional<RotationIndexerResult> ProcessImage(int64_t image, const std::vector<SpotToSave>& spots);
|
|
std::optional<RotationIndexerResult> GetLattice();
|
|
};
|
|
|
|
#endif //JFJOCH_ROTATIONINDEXER_H
|