File: image_background.cpp

package info (click to toggle)
satdump 1.2.2%2Bgb79af48-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 81,648 kB
  • sloc: cpp: 276,768; ansic: 164,598; lisp: 1,219; sh: 283; xml: 106; makefile: 7
file content (48 lines) | stat: -rw-r--r-- 1,243 bytes parent folder | download | duplicates (2)
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
#include "image_background.h"
#include "logger.h"
#include "common/geodetic/geodetic_coordinates.h"
#include "common/projection/reprojector.h"
#include "common/projection/projs2/proj_json.h"

namespace image
{
	void remove_background(Image &img, nlohmann::json proj_cfg, float *progress)
	{
		size_t width = img.width();
		size_t height = img.height();

		satdump::reprojection::rescaleProjectionScalarsIfNeeded(proj_cfg, width, height);
		proj::projection_t product_proj;
		try
		{
			product_proj = proj_cfg;
		}
		catch (std::exception &)
		{
			logger->warn("Cannot remove background - Failed to get projection config");
			return;
		}
		if (proj::projection_setup(&product_proj))
		{
			// TODO: Support for warp?
			logger->warn("Cannot remove background - Failed to set up projection algorithm");
			return;
		}

		img.to_rgba();
		double dummy; // Hey! I resemble that remark!

		for (size_t y = 0; y < height; y++)
		{
			for (size_t x = 0; x < width; x++)
				if (proj::projection_perform_inv(&product_proj, x, y, &dummy, &dummy))
					for (int c = 0; c < img.channels(); c++)
						img.set(c, y * width + x, 0);

			if (progress != nullptr)
				*progress = (float)y / (float)height;
		}

		proj::projection_free(&product_proj);
	}
}