55 lines
1.7 KiB
C++
55 lines
1.7 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_IMAGEANALYSISCPU_H
|
|
#define JFJOCH_IMAGEANALYSISCPU_H
|
|
|
|
#include "../frame_serialize/JFJochMessages.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/AzimuthalIntegration.h"
|
|
#include "ImageSpotFinder.h"
|
|
#include "IndexerWrapper.h"
|
|
|
|
class ImageAnalysisCPU {
|
|
constexpr static const int64_t max_spot_count = 250;
|
|
|
|
DiffractionGeometry geom;
|
|
|
|
ROIMap rois;
|
|
AzimuthalIntegration integration;
|
|
std::vector<uint16_t> roi_map;
|
|
std::map<std::string, uint16_t> roi_names;
|
|
size_t roi_count;
|
|
|
|
size_t npixels;
|
|
size_t xpixels;
|
|
std::vector<uint8_t> mask_1byte;
|
|
|
|
ImageSpotFinder spotFinder;
|
|
std::unique_ptr<IndexerWrapper> indexer;
|
|
|
|
mutable std::mutex spot_finding_settings_mutex;
|
|
SpotFindingSettings settings;
|
|
SpotFindingSettings GetSpotFindingSettings() const;
|
|
|
|
template <class T>
|
|
void Analyze(DataMessage &output,
|
|
const uint8_t *image,
|
|
T err_pixel_val,
|
|
T sat_pixel_val);
|
|
public:
|
|
ImageAnalysisCPU(const DiffractionGeometry &geom,
|
|
const AzimuthalIntegrationSettings &azint_settings,
|
|
const SpotFindingSettings &spot_finding_settings,
|
|
const ROIMap &roi_map,
|
|
const std::optional<UnitCell> &uc,
|
|
const std::vector<uint32_t> &mask,
|
|
size_t width, size_t height);
|
|
void Analyze(DataMessage &output, std::vector<uint8_t> &buffer);
|
|
void FillStartMessage(StartMessage &msg);
|
|
void SetSpotFindingSettings(const SpotFindingSettings& settings);
|
|
};
|
|
|
|
|
|
#endif //JFJOCH_IMAGEANALYSISCPU_H
|