ImageIO.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <cstddef>
#include <cstdint>
#include <string>
#include <vector>
 
#include "open3d/t/geometry/Image.h"
 
namespace open3d {
namespace t {
namespace io {
 
std::shared_ptr<geometry::Image> CreateImageFromFile(
        const std::string &filename);
 
bool ReadImage(const std::string &filename, geometry::Image &image);
 
constexpr int kOpen3DImageIODefaultQuality = -1;
 
bool WriteImage(const std::string &filename,
                const geometry::Image &image,
                int quality = kOpen3DImageIODefaultQuality);
 
bool ReadImageFromPNG(const std::string &filename, geometry::Image &image);
 
bool ReadImageFromPNGInMemory(const uint8_t *data,
                              size_t size,
                              geometry::Image &image);
 
bool WriteImageToPNG(const std::string &filename,
                     const geometry::Image &image,
                     int quality = kOpen3DImageIODefaultQuality);
 
bool WriteImageToPNGInMemory(std::vector<uint8_t> &output_buffer,
                             const geometry::Image &image,
                             int quality = kOpen3DImageIODefaultQuality);
 
bool ReadImageFromJPG(const std::string &filename, geometry::Image &image);
 
bool ReadImageFromJPGInMemory(const uint8_t *data,
                              size_t size,
                              geometry::Image &image);
 
bool ReadImageFromMemory(const uint8_t *data,
                         size_t size,
                         geometry::Image &image);
 
bool WriteImageToJPG(const std::string &filename,
                     const geometry::Image &image,
                     int quality = kOpen3DImageIODefaultQuality);
 
class DepthNoiseSimulator {
public:
    explicit DepthNoiseSimulator(const std::string &noise_model_path);
 
    geometry::Image Simulate(const geometry::Image &im_src,
                             float depth_scale = 1000.0);
 
    core::Tensor GetNoiseModel() const { return model_; }
 
    void EnableDeterministicDebugMode() { deterministic_debug_mode_ = true; }
 
private:
    core::Tensor model_;  // ndims==3
    bool deterministic_debug_mode_ = false;
};
 
}  // namespace io
}  // namespace t
}  // namespace open3d