69 lines
2.0 KiB
C++
69 lines
2.0 KiB
C++
#include <filesystem>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#include "a2.h"
|
|
#include "bounding_box.h"
|
|
#include "mapping.h"
|
|
#include "mask.h"
|
|
#include "pack.h"
|
|
#include "png.h"
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
int main(int argc, const char* argv[]) {
|
|
// We take an image path and a list of points (grouped by 4 to define
|
|
// bounding boxes).
|
|
if (argc < 6 or (argc - 2) % 4 != 0) {
|
|
std::cerr
|
|
<< "Usage: " << argv[0] << " path/to/image"
|
|
<< " [x y width height ...]\n"
|
|
<< "x/y/w/h points must be grouped by 4 to define bounding boxes\n";
|
|
return 1;
|
|
}
|
|
|
|
const fs::path input(argv[1]);
|
|
if (!fs::exists(input)) {
|
|
std::cerr << "Input file " << input << " does not exist.\n";
|
|
return 1;
|
|
}
|
|
|
|
const std::optional<freling::Frame> in_frame = freling::load_png(input);
|
|
if (!in_frame) {
|
|
std::cerr << "Cannot load file " << input << "\n";
|
|
return 1;
|
|
}
|
|
|
|
int i = 2;
|
|
std::vector<freling::BoundingBox> bboxes;
|
|
bboxes.reserve((argc - 2) / 4);
|
|
while (i < argc) {
|
|
const int32_t x = atoi(argv[i]);
|
|
const int32_t y = atoi(argv[i + 1]);
|
|
const uint32_t w = atoi(argv[i + 2]);
|
|
const uint32_t h = atoi(argv[i + 3]);
|
|
bboxes.push_back(freling::BoundingBox({x, y, w, h}));
|
|
i += 4;
|
|
}
|
|
freling::Mask in_mask(in_frame->width, in_frame->height);
|
|
|
|
std::vector<freling::BoundingBox> packed_bboxes;
|
|
std::optional<freling::Frame> regions =
|
|
pack(*in_frame, bboxes, packed_bboxes);
|
|
|
|
in_mask.fill(false);
|
|
for (const auto& b : bboxes) {
|
|
for (int j = b.y; j < b.y + b.height; ++j) {
|
|
for (int i = b.x; i < b.x + b.width; ++i) {
|
|
in_mask.set(i, j, true);
|
|
}
|
|
}
|
|
};
|
|
|
|
std::vector<freling::BoundingBox> a2_bboxes = freling::A2(packed_bboxes);
|
|
std::vector<freling::BoundingBox> a2_bboxes_origin =
|
|
freling::map_to_origin(bboxes, packed_bboxes, a2_bboxes);
|
|
|
|
return 0;
|
|
}
|