90 lines
2.9 KiB
C++
90 lines
2.9 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCHIMAGEREADINGWORKER_H
|
|
#define JFJOCHIMAGEREADINGWORKER_H
|
|
|
|
#include <QObject>
|
|
#include <QString>
|
|
#include <QVector>
|
|
#include <QRect>
|
|
#include <QMutex>
|
|
|
|
#include "JFJochViewerToolbar.h"
|
|
|
|
#include "../reader/JFJochHDF5Reader.h"
|
|
#include "../common/Logger.h"
|
|
#include "../reader/JFJochHttpReader.h"
|
|
#include "../image_analysis/MXAnalysisWithoutFPGA.h"
|
|
|
|
Q_DECLARE_METATYPE(std::shared_ptr<const JFJochReaderDataset>)
|
|
Q_DECLARE_METATYPE(std::shared_ptr<const JFJochReaderImage>)
|
|
Q_DECLARE_METATYPE(DiffractionExperiment)
|
|
Q_DECLARE_METATYPE(SpotFindingSettings)
|
|
Q_DECLARE_METATYPE(IndexingSettings)
|
|
Q_DECLARE_METATYPE(UnitCell)
|
|
|
|
class JFJochImageReadingWorker : public QObject {
|
|
Q_OBJECT
|
|
|
|
mutable QMutex m;
|
|
bool http_mode = false;
|
|
|
|
JFJochHDF5Reader file_reader;
|
|
JFJochHttpReader http_reader;
|
|
|
|
QString current_file;
|
|
|
|
DiffractionExperiment curr_experiment;
|
|
IndexingSettings indexing_settings;
|
|
std::unique_ptr<IndexerThreadPool> indexing;
|
|
|
|
std::shared_ptr<JFJochReaderImage> current_image_ptr;
|
|
std::unique_ptr<AzimuthalIntegration> azint_mapping;
|
|
std::unique_ptr<MXAnalysisWithoutFPGA> image_analysis;
|
|
|
|
std::unique_ptr<ROIElement> roi;
|
|
|
|
SpotFindingSettings spot_finding_settings;
|
|
|
|
std::optional<int64_t> current_image;
|
|
int64_t current_summation = 1;
|
|
int64_t total_images = 0;
|
|
Logger logger{"jfjoch_viewer"};
|
|
|
|
bool auto_reanalyze = false;
|
|
|
|
void LoadImage_i(int64_t image_number, int64_t summation);
|
|
void CalcROI_i();
|
|
void ReanalyzeImage_i();
|
|
void UpdateDataset_i(const DiffractionExperiment& experiment);
|
|
void UpdateAzint_i(const JFJochReaderDataset *dataset);
|
|
signals:
|
|
void datasetLoaded(std::shared_ptr<const JFJochReaderDataset>);
|
|
void imageLoaded(std::shared_ptr<const JFJochReaderImage>);
|
|
void imageStatsUpdated(std::shared_ptr<const JFJochReaderImage>);
|
|
void imageNumberChanged(int64_t total_images, int64_t current_image);
|
|
void setToolbarMode(JFJochViewerToolbar::ToolbarMode input);
|
|
void setRings(const QVector<float> &v);
|
|
public:
|
|
JFJochImageReadingWorker(const SpotFindingSettings &settings, const DiffractionExperiment& experiment, QObject *parent = nullptr);
|
|
~JFJochImageReadingWorker() override = default;
|
|
public slots:
|
|
void LoadFile(const QString &filename, qint64 image_number, qint64 summation);
|
|
void CloseFile();
|
|
void LoadImage(int64_t image_number, int64_t summation);
|
|
void SetROIBox(QRect box);
|
|
void SetROICircle(double x, double y, double radius);
|
|
|
|
void UpdateDataset(const DiffractionExperiment& experiment);
|
|
|
|
void FindCenter(const UnitCell& calibrant, bool guess);
|
|
|
|
void Analyze();
|
|
|
|
void UpdateSpotFindingSettings(const SpotFindingSettings &settings, const IndexingSettings &indexing, int64_t max_spots, bool reanalyze);
|
|
};
|
|
|
|
|
|
#endif //JFJOCHIMAGEREADINGWORKER_H
|