RGBDImage.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 "open3d/geometry/RGBDImage.h"
#include "open3d/t/geometry/Image.h"
 
namespace open3d {
namespace t {
namespace geometry {

class RGBDImage : public Geometry {
public:
    RGBDImage() : Geometry(Geometry::GeometryType::RGBDImage, 2) {}

    RGBDImage(const Image &color, const Image &depth, bool aligned = true)
        : Geometry(Geometry::GeometryType::RGBDImage, 2),
          color_(color),
          depth_(depth),
          aligned_(aligned) {
        if (color.GetRows() != depth.GetRows() ||
            color.GetCols() != depth.GetCols()) {
            aligned_ = false;
            utility::LogWarning(
                    "Aligned image pair must have the same resolution.");
        }
    }
 
    core::Device GetDevice() const override {
        core::Device color_device = color_.GetDevice();
        core::Device depth_device = depth_.GetDevice();
        if (color_device != depth_device) {
            utility::LogError(
                    "Color {} and depth {} are not on the same device.",
                    color_device.ToString(), depth_device.ToString());
        }
        return color_device;
    }
 
    ~RGBDImage() override{};

    RGBDImage &Clear() override;

    bool IsEmpty() const override;

    bool AreAligned() const { return aligned_; }

    core::Tensor GetMinBound() const {
        return core::Tensor::Zeros({2}, core::Int64);
    }

    core::Tensor GetMaxBound() const {
        return core::Tensor(
                std::vector<int64_t>{color_.GetCols() + depth_.GetCols(),
                                     color_.GetRows()},
                {2}, core::Int64);
    }

    RGBDImage To(const core::Device &device, bool copy = false) const {
        return RGBDImage(color_.To(device, copy), depth_.To(device, copy),
                         aligned_);
    }

    RGBDImage Clone() const { return To(color_.GetDevice(), /*copy=*/true); }

    open3d::geometry::RGBDImage ToLegacy() const {
        return open3d::geometry::RGBDImage(color_.ToLegacy(),
                                           depth_.ToLegacy());
    }

    std::string ToString() const;
 
public:
    Image color_;
    Image depth_;
    bool aligned_ = true;
};
 
}  // namespace geometry
}  // namespace t
}  // namespace open3d