Files
Jungfraujoch/image_analysis/bragg_prediction/BraggPredictionGPU.h
Filip Leonarski 07fe4dd3bb
All checks were successful
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 11m23s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 10m32s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 9m15s
Build Packages / Generate python client (push) Successful in 19s
Build Packages / Build documentation (push) Successful in 49s
Build Packages / Create release (push) Has been skipped
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 9m13s
Build Packages / build:rpm (rocky8) (push) Successful in 9m10s
Build Packages / build:rpm (rocky9) (push) Successful in 9m58s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 8m52s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 8m42s
Build Packages / Unit tests (push) Successful in 1h12m44s
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 11m30s
v1.0.0-rc.124 (#31)
This is an UNSTABLE release. This version significantly rewrites code to predict reflection position and integrate them,
especially in case of rotation crystallography. If things go wrong with analysis, it is better to revert to 1.0.0-rc.123.

* jfjoch_broker: Improve refection position prediction and Bragg integration code.
* jfjoch_broker: Align with XDS way of calculating Lorentz correction and general notation.
* jfjoch_writer: Fix saving mosaicity properly in HDF5 file.
* jfjoch_viewer: Introduce high-dynamic range mode for images
* jfjoch_viewer: Ctrl+mouse wheel has exponential change in foreground (+/-15%)
* jfjoch_viewer: Zoom-in numbers have better readability

Reviewed-on: #31
Co-authored-by: Filip Leonarski <filip.leonarski@psi.ch>
Co-committed-by: Filip Leonarski <filip.leonarski@psi.ch>
2026-02-01 13:29:33 +01:00

50 lines
1.2 KiB
C++

// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only
#ifndef JFJOCH_BRAGGPREDICTIONGPU_H
#define JFJOCH_BRAGGPREDICTIONGPU_H
#include <vector>
#include "BraggPrediction.h"
#include "../indexing/CUDAMemHelpers.h"
struct KernelConsts {
float det_width_pxl;
float det_height_pxl;
float beam_x;
float beam_y;
float coeff_const;
float one_over_wavelength;
float one_over_dmax_sq;
float ewald_cutoff;
Coord Astar, Bstar, Cstar, S0;
float rot[9];
char centering;
};
class BraggPredictionGPU : public BraggPrediction {
CudaRegisteredVector<Reflection> reg_out; // requires stable storage
CudaDevicePtr<Reflection> d_out;
// Dedicated stream
CudaStream stream;
// Device allocations via helpers
CudaDevicePtr<KernelConsts> dK;
CudaDevicePtr<int> d_count;
// Host pinned buffer for async download (optional but faster)
CudaHostPtr<int> h_count;
public:
explicit BraggPredictionGPU(int max_reflections = 10000);
int Calc(const DiffractionExperiment &experiment,
const CrystalLattice &lattice,
const BraggPredictionSettings &settings) override;
};
#endif //JFJOCH_BRAGGPREDICTIONGPU_H