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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
#pragma once
#include "products/image_products.h"
#include "resources.h"
#define DEFINE_COMPOSITE_UTILS 1
#include "common/image/composite.h"
#include "common/projection/sat_proj/sat_proj.h"
#include "common/projection/projs/equirectangular.h"
#include "common/image/io.h"
#include "common/image/processing.h"
namespace cpp_compos
{
image::Image underlay_with_clouds(satdump::ImageProducts *img_pro,
std::vector<image::Image> &inputChannels,
std::vector<std::string> channelNumbers,
std::string /* cpp_id */,
nlohmann::json vars,
nlohmann::json offsets_cfg,
std::vector<double> *final_timestamps = nullptr,
float *progress = nullptr)
{
if (!img_pro->has_proj_cfg())
{
logger->error("This composite requires projection info");
return image::Image();
}
image::compo_cfg_t f = image::get_compo_cfg(inputChannels, channelNumbers, offsets_cfg);
image::Image img_background;
image::load_img(img_background, resources::getResourcePath("maps/nasa_hd.jpg"));
geodetic::projection::EquirectangularProjection equp;
equp.init(img_background.width(), img_background.height(), -180, 90, 180, -90);
int map_x, map_y;
// return 3 channels, RGB
image::Image rgb_output(f.img_depth, f.maxWidth, f.maxHeight, 3);
geodetic::geodetic_coords_t coords;
auto proj_cfg = img_pro->get_proj_cfg();
auto projFunc = satdump::get_sat_proj(proj_cfg, img_pro->get_tle(), *final_timestamps, true);
float cfg_offset = vars["minoffset"];
float cfg_scalar = vars["scalar"];
float cfg_thresold = vars["thresold"];
bool cfg_blend = vars["blend"];
bool cfg_invert = vars["invert"];
size_t background_size = img_background.width() * img_background.height();
auto &ch_equal = inputChannels[0];
image::equalize(ch_equal);
float val = 0;
float proj_scale = 1.0f;
if (proj_cfg.contains("image_width"))
proj_scale = proj_cfg["image_width"].get<double>() / double(rgb_output.width());
for (size_t x = 0; x < rgb_output.width(); x++)
{
for (size_t y = 0; y < rgb_output.height(); y++)
{
if (!projFunc->get_position(x * proj_scale, y * proj_scale, coords))
{
equp.forward(coords.lon, coords.lat, map_x, map_y);
if (cfg_invert)
val = 1.0 - (ch_equal.getf((y * rgb_output.width()) + x) - cfg_offset) * cfg_scalar;
else
val = (ch_equal.getf((y * rgb_output.width()) + x) - cfg_offset) * cfg_scalar;
if (val > 1)
val = 1;
if (val < 0)
val = 0;
int mappos = map_y * img_background.width() + map_x;
if (mappos >= (int)background_size)
mappos = background_size - 1;
if (mappos < 0)
mappos = 0;
if (cfg_blend == 1)
{
for (int c = 0; c < 3; c++)
{
float mval = img_background.getf(background_size * c + mappos);
float fval = mval * (1.0 - val) + val * val;
rgb_output.setf(c, y * rgb_output.width() + x, rgb_output.clampf(fval));
}
}
else
{
if (cfg_thresold == 0)
{
for (int c = 0; c < 3; c++)
{
float mval = img_background.getf(background_size * c + mappos);
float fval = mval * 0.4 + val * 0.6;
rgb_output.setf(c, y * rgb_output.width() + x, rgb_output.clampf(fval));
}
}
else
{
if (val < cfg_thresold)
{
for (int c = 0; c < 3; c++)
{
float fval = img_background.getf(background_size * c + mappos);
rgb_output.setf(c, y * rgb_output.width() + x, rgb_output.clampf(fval));
}
}
else
{
for (int c = 0; c < 3; c++)
{
float mval = img_background.getf(background_size * c + mappos);
float fval = mval * 0.4 + val * 0.6;
rgb_output.setf(c, y * rgb_output.width() + x, rgb_output.clampf(fval));
}
}
}
}
}
else
{
rgb_output.setf(0, y * rgb_output.width() + x, 0);
rgb_output.setf(1, y * rgb_output.width() + x, 0);
rgb_output.setf(2, y * rgb_output.width() + x, 0);
}
}
// set the progress bar accordingly
if (progress != nullptr)
*progress = double(x) / double(rgb_output.width());
}
return rgb_output;
}
}
|
