1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#pragma once
#include "viewer.h"
#include "logger.h"
#include "products/image_products.h"
#include "common/widgets/image_view.h"
#include "imgui/imgui_stdlib.h"
#include "core/style.h"
#include "libs/ctpl/ctpl_stl.h"
#include "common/widgets/markdown_helper.h"
#include "common/widgets/timed_message.h"
namespace satdump
{
class ImageViewerHandler : public ViewerHandler
{
public:
~ImageViewerHandler();
// Products
ImageProducts *products;
// Image handling
int active_channel_id = 0, select_image_id = 1;
bool active_channel_calibrated = false;
std::string select_image_str;
image::Image rgb_image, current_image;
ImageViewWidget image_view;
// Map projection stuff
std::function<std::pair<int, int>(float, float, int, int)> proj_func;
nlohmann::json last_proj_cfg;
bool last_correct_image = false;
bool last_rotate_image = false;
size_t last_width = 0;
size_t last_height = 0;
// Other controls
bool remove_background = false;
bool median_blur = false;
bool despeckle = false;
bool rotate_image = false;
bool equalize_image = false;
bool individual_equalize_image = false;
bool invert_image = false;
bool correct_image = false;
bool normalize_image = false;
bool white_balance_image = false;
bool manual_brightness_contrast = false;
float manual_brightness_contrast_brightness = 0.0;
float manual_brightness_contrast_contrast = 0.0;
// GUI
bool range_window = false;
std::vector<std::pair<double, double>> temp_ranges, radiance_ranges;
bool update_needed;
bool is_updating = false;
OverlayHandler overlay_handler;
// Calibration
bool is_temp = false;
bool show_scale = false;
image::Image scale_image; // 512x25
unsigned int scale_texture_id = 0;
uint32_t *scale_buffer = nullptr;
bool scale_has_update = false;
// LUT
bool using_lut = false;
image::Image lut_image;
// Geo-Correction
std::vector<int> correction_factors;
// RGB Handling
ImageCompositeCfg rgb_compo_cfg;
std::vector<std::string> channel_numbers;
float rgb_progress = 0;
bool rgb_processing = false;
std::vector<std::pair<std::string, ImageCompositeCfg>> rgb_presets;
std::string preset_search_str;
int select_rgb_presets = -1;
bool show_markdown_description = false;
widgets::MarkdownHelper markdown_composite_info;
std::vector<double> current_timestamps;
nlohmann::json current_proj_metadata;
bool projection_use_old_algo = false;
// Utils
void updateScaleImage();
void updateCorrectionFactors(bool first = false);
// The Rest
void init();
void updateImage();
ctpl::thread_pool handler_thread_pool = ctpl::thread_pool(1);
std::mutex async_image_mutex;
void asyncUpdate();
void updateRGB();
void drawMenu();
void drawContents(ImVec2 win_size);
float drawTreeMenu();
bool canBeProjected();
void addCurrentToProjections();
widgets::TimedMessage proj_notif;
static std::string getID() { return "image_handler"; }
static std::shared_ptr<ViewerHandler> getInstance() { return std::make_shared<ImageViewerHandler>(); }
};
}
|
