netatmo-algo/src/pack.cpp

44 lines
1.4 KiB
C++
Raw Normal View History

2022-02-12 00:28:54 +01:00
#include "pack.h"
2022-02-14 00:13:09 +01:00
#include <algorithm>
#include <cmath>
#include <iostream>
2022-02-12 00:28:54 +01:00
Frame pack(const Frame& in_frame, const std::vector<BoundingBox>& bboxes) {
2022-02-14 00:13:09 +01:00
// We sort the bounding boxes by maximum area
std::vector<BoundingBox> sorted_bboxes = bboxes;
std::sort(sorted_bboxes.begin(), sorted_bboxes.end(),
[](const auto& a, const auto& b) { return a.area() > b.area(); });
// We keep a mapping between the sorted bounding boxes and the original
// order
std::vector<int> mapping(bboxes.size());
for (int i = 0; i < sorted_bboxes.size(); ++i) {
const auto& s_box = sorted_bboxes[i];
for (int j = 0; j < bboxes.size(); ++j) {
const auto& box = bboxes[j];
if (box == s_box) {
mapping[i] = j;
continue;
}
}
}
int max_area = 0;
for (const auto& box : sorted_bboxes) {
int area = box.area();
std::cout << "bounding box area: " << area << "\n";
max_area += area;
}
std::cout << "max area: " << max_area << "\n";
int min_dim = int(ceil(std::sqrt(max_area)));
std::cout << "optimal image dimention: " << min_dim << " x " << min_dim
<< "\n";
const int in_min_dim = std::min(in_frame.width, in_frame.height);
std::cout << "maximum image dimention: " << in_min_dim << " x "
<< in_min_dim << "\n";
2022-02-12 00:28:54 +01:00
// TODO
2022-02-13 23:41:08 +01:00
return Frame(in_frame);
2022-02-12 00:28:54 +01:00
}