image_io.cc

Go to the documentation of this file.

#include "image_io.h"
#include <stdexcept>

namespace tlz {

cv::Mat_<cv::Vec3b> load_texture(const std::string& filename) {
    cv::Mat mat = cv::imread(filename, CV_LOAD_IMAGE_COLOR);
    if(mat.empty()) throw std::runtime_error("could not load texture " + filename);
    cv::Mat_<cv::Vec3b> mat_ = mat;
    return mat_;
}

void save_texture(const std::string& filename, const cv::Mat_<cv::Vec3b>& texture) {
    std::vector<int> params = { CV_IMWRITE_PNG_COMPRESSION, 0 };
    cv::imwrite(filename, texture, params);
}


cv::Mat_<ushort> load_ir(const std::string& filename) {
    cv::Mat mat = cv::imread(filename, CV_LOAD_IMAGE_ANYDEPTH);
    if(mat.empty()) throw std::runtime_error("could not load ir image " + filename);
    if(mat.depth() != CV_16U) throw std::runtime_error("input ir image " + filename + " is not must be 16 bit");
    cv::Mat_<ushort> mat_ = mat;
    return mat_;
}

void save_ir(const std::string& filename, const cv::Mat_<ushort>& ir) {
    std::vector<int> params = { CV_IMWRITE_PNG_COMPRESSION, 0 };
    cv::imwrite(filename, ir, params);
}


cv::Mat_<ushort> load_depth(const std::string& filename) {
    cv::Mat mat = cv::imread(filename, CV_LOAD_IMAGE_ANYDEPTH);
    if(mat.empty()) throw std::runtime_error("could not load depth map " + filename);
    if(mat.depth() != CV_16U) throw std::runtime_error("input depth map " + filename + " is not must be 16 bit");
    cv::Mat_<ushort> mat_ = mat;
    return mat_;
}

void save_depth(const std::string& filename, const cv::Mat_<ushort>& depth) {
    std::vector<int> params = { CV_IMWRITE_PNG_COMPRESSION, 0 };
    cv::imwrite(filename, depth, params);
}

}