ImageIO.hpp

#pragma once
 
#include <optional>
 
#include <opencv2/core.hpp>
 
#include "vc/core/filesystem.hpp"
#include "vc/core/util/Iteration.hpp"
#include "vc/core/util/Logging.hpp"
#include "vc/core/util/String.hpp"
 
namespace volcart
{
 
auto ReadImage(const filesystem::path& path) -> cv::Mat;
 
struct WriteImageOpts {
    std::optional<int> compression;
 
    std::optional<int> padding;
};
 
void WriteImage(
    const filesystem::path& path, const cv::Mat& img, WriteImageOpts = {});
 
template <class Iterable>
void WriteImageSequence(
    const filesystem::path& path,
    const Iterable& iterable,
    const WriteImageOpts& opts = {})
{
    namespace fs = volcart::filesystem;
 
    // components
    fs::path parent;
    std::string prefix;
    std::string suffix;
    fs::path ext;
 
    // If directory, default to dir/###.tif
    if (fs::is_directory(path)) {
        parent = path;
        ext = ".tif";
    }
 
    // If path, decompose to replace {} with a number
    else {
        parent = path.parent_path();
        ext = path.extension();
 
        // Split into a prefix and suffix
        auto stem = path.stem().string();
        std::tie(prefix, std::ignore, suffix) = partition(stem, "{}");
 
        // Log when separator not found
        if (suffix.empty()) {
            Logger()->debug(
                "Index placement separator \\{\\} not found in stem: {}", stem);
        }
    }
 
    // Setup padding (default or user-provided)
    auto pad = std::to_string(std::size(iterable)).size();
    pad = opts.padding.value_or(pad);
 
    // Write images
    for (const auto [i, image] : enumerate(iterable)) {
        const auto name = prefix + to_padded_string(i, pad) + suffix;
        auto filepath = (parent / name).replace_extension(ext);
        WriteImage(filepath, image, opts);
    }
}
 
}  // namespace volcart