Filip Leonarski
8b356a7001
All checks were successful
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 7m47s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 7m20s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 8m13s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 7m10s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 7m53s
Build Packages / build:rpm (rocky8) (push) Successful in 7m57s
Build Packages / Generate python client (push) Successful in 13s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 7m39s
Build Packages / Create release (push) Has been skipped
Build Packages / Build documentation (push) Successful in 36s
Build Packages / build:rpm (rocky9) (push) Successful in 9m0s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 6m55s
Build Packages / Unit tests (push) Successful in 1h10m44s
This is an UNSTABLE release. * Fixes in CI pipeline * jfjoch_broker: Remove PNG preview, no dependency on libpng * jfjoch_writer: Fix UTC timestamp being generated wrong (mix between milli- and microseconds) * jfjoch_viewer: Show data collection time in dataset tooltip * jfjoch_viewer: Allow to choose the calibrant (presets for LaB6 and silver behenate) * jfjoch_viewer: Auto foreground value * Use external libjpeg-turbo and libtiff: simpler build stack, these are built and linked statically in automated Docker builds * Remove OpenBLAS dependency Reviewed-on: #1 Co-authored-by: Filip Leonarski <filip.leonarski@psi.ch> Co-committed-by: Filip Leonarski <filip.leonarski@psi.ch>
77 lines
2.8 KiB
C++
77 lines
2.8 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JUNGFRAUJOCH_PREVIEWIMAGE_H
|
|
#define JUNGFRAUJOCH_PREVIEWIMAGE_H
|
|
|
|
#include <vector>
|
|
#include <mutex>
|
|
|
|
#include "../common/SpotToSave.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/PixelMask.h"
|
|
#include "PreviewCounter.h"
|
|
#include "../common/ColorScale.h"
|
|
|
|
enum class PreviewImageFormat {JPEG, TIFF};
|
|
|
|
struct PreviewImageSettings {
|
|
float saturation_value = 10;
|
|
int64_t jpeg_quality = 70;
|
|
bool show_beam_center = true;
|
|
bool show_spots = true;
|
|
bool show_roi = false;
|
|
bool show_user_mask = true;
|
|
std::optional<float> resolution_ring;
|
|
ColorScaleEnum scale = ColorScaleEnum::Indigo;
|
|
PreviewImageFormat format = PreviewImageFormat::JPEG;
|
|
};
|
|
|
|
class PreviewImage {
|
|
mutable std::mutex m;
|
|
DiffractionExperiment experiment;
|
|
|
|
std::vector<uint8_t> mask;
|
|
|
|
std::vector<SpotToSave> spots;
|
|
size_t xpixel = 0;
|
|
size_t ypixel = 0;
|
|
size_t pixel_depth_bytes = 2;
|
|
bool pixel_is_signed = false;
|
|
float beam_x = 0;
|
|
float beam_y = 0;
|
|
|
|
constexpr static uint8_t MaskGap = 1;
|
|
constexpr static uint8_t MaskUsr = 2;
|
|
constexpr static uint8_t MaskDet = 3;
|
|
|
|
template<class T>
|
|
std::vector<rgb> GenerateRGB(const uint8_t* value,
|
|
int64_t special_value,
|
|
const ColorScale &scale,
|
|
const PreviewImageSettings& settings) const;
|
|
std::vector<rgb> GenerateRGB(const PreviewImageSettings& settings,
|
|
const CompressedImage& image,
|
|
const std::vector<SpotToSave>& spots) const;
|
|
|
|
void AddResolutionRing(std::vector<rgb> &rgb_image, float d) const;
|
|
void AddBeamCenter(std::vector<rgb> &rgb_image) const;
|
|
void AddSpots(std::vector<rgb> &rgb_image, const std::vector<SpotToSave>& spots) const;
|
|
void AddROI(std::vector<rgb> &rgb_image) const;
|
|
|
|
void color_pixel(std::vector<rgb>& ret, int64_t xpixel, int64_t ypixel, const rgb &color) const;
|
|
void spot(std::vector<rgb>& ret, int64_t xpixel, int64_t ypixel, const rgb &color) const;
|
|
void roi(std::vector<rgb>& ret, int64_t xpixel, int64_t ypixel, int64_t roi_number) const;
|
|
std::string GenerateImage(const PreviewImageSettings& settings,
|
|
const CompressedImage& image,
|
|
const std::vector<SpotToSave>& spots) const;
|
|
public:
|
|
void Configure(const DiffractionExperiment& experiment, const PixelMask& mask);
|
|
void Configure();
|
|
|
|
[[nodiscard]] std::string GenerateImage(const PreviewImageSettings& settings, const std::vector<uint8_t>& cbor_format);
|
|
[[nodiscard]] static std::string GenerateTIFF(const std::vector<uint8_t>& cbor_format) ;
|
|
};
|
|
|
|
#endif //JUNGFRAUJOCH_PREVIEWIMAGE_H
|