54 lines
1.7 KiB
C++
54 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 "../common/JFJochMessages.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/AzimuthalIntegration.h"
|
|
#include "ImageSpotFinder.h"
|
|
#include "indexing/Indexer.h"
|
|
#include "../common/PixelMask.h"
|
|
#include "../common/AzimuthalIntegrationProfile.h"
|
|
#include "NeuralNetInferenceClient.h"
|
|
#include "bragg_integration/BraggIntegrate2D.h"
|
|
|
|
class ImageAnalysisCPU {
|
|
constexpr static const int64_t max_spot_count = 250;
|
|
|
|
const DiffractionExperiment &experiment;
|
|
const 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;
|
|
|
|
NeuralNetInferenceClient *inference_client = nullptr;
|
|
ImageSpotFinder spotFinder;
|
|
std::unique_ptr<Indexer> indexer;
|
|
|
|
const int64_t saturation_limit;
|
|
|
|
int nquads = 2;
|
|
|
|
template <class T>
|
|
void Analyze(DataMessage &output, const uint8_t *image, T err_pixel_val, T sat_pixel_val, AzimuthalIntegrationProfile &profile, const SpotFindingSettings &spot_finding_settings);
|
|
public:
|
|
ImageAnalysisCPU(const DiffractionExperiment& experiment,
|
|
const AzimuthalIntegration& integration,
|
|
const PixelMask& mask);
|
|
|
|
void UpdateROI();
|
|
|
|
ImageAnalysisCPU& NeuralNetInference(NeuralNetInferenceClient *client);
|
|
void Analyze(DataMessage &output, std::vector<uint8_t> &buffer, AzimuthalIntegrationProfile &profile, const SpotFindingSettings &spot_finding_settings);
|
|
};
|
|
|
|
|
|
#endif //JFJOCH_IMAGEANALYSISCPU_H
|