1 files changed, 61 insertions, 0 deletions
diff --git a/crop-tool/atlas.cpp b/crop-tool/atlas.cpp
new file mode 100644
index 00000000..ca05cd15
--- /dev/null
+++ b/crop-tool/atlas.cpp
@@ -0,0 +1,61 @@
+#include "atlas.hpp"
+#include "serialize.hpp"
+#include "../defs.hpp"
+
+#include <filesystem>
+#include <opencv2/imgcodecs.hpp>
+
+void anim_atlas_row::add_entry(const anim_atlas_entry& x)
+{
+    auto& frame = *x.frame;
+    const auto& mat = x.mat;
+    frame.offset = {xpos, ypos};
+    frame.size = {mat.cols, mat.rows};
+
+    ASSERT(mat.rows > 0 && mat.cols > 0);
+    data.push_back(x);
+    xpos += mat.cols;
+    max_height = std::max(mat.rows, max_height);
+}
+
+void anim_atlas::advance_row()
+{
+    auto& row = rows.back();
+    if (row.data.empty())
+        return;
+    ASSERT(row.xpos); ASSERT(row.max_height);
+    ypos += row.max_height;
+    maxx = std::max(row.xpos, maxx);
+    rows.push_back({{}, 0, 0, ypos});
+}
+
+Magnum::Vector2i anim_atlas::offset() const
+{
+    const auto& row = rows.back();
+    return {row.xpos, row.ypos};
+}
+
+Magnum::Vector2i anim_atlas::size() const
+{
+    const anim_atlas_row& row = rows.back();
+    // prevent accidentally writing out of bounds by forgetting to call
+    // anim_atlas::advance_row() one last time prior to anim_atlas::size()
+    return {std::max(maxx, row.xpos), ypos + row.max_height};
+}
+
+bool anim_atlas::dump(const std::filesystem::path& filename) const
+{
+    auto sz = size();
+    cv::Mat4b mat(sz[1], sz[0]);
+    mat.setTo(0);
+
+    for (const anim_atlas_row& row : rows)
+        for (const anim_atlas_entry& x : row.data)
+        {
+            auto offset = x.frame->offset, size = x.frame->size;
+            cv::Rect roi = {offset[0], offset[1], size[0], size[1]};
+            x.mat.copyTo(mat(roi));
+        }
+
+    return cv::imwrite(filename.string(), mat);
+}