TriangleMesh.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// The MIT License (MIT)
//
// Copyright (c) 2018 www.open3d.org
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.
// ----------------------------------------------------------------------------

#pragma once

#include "open3d/core/Tensor.h"
#include "open3d/geometry/TriangleMesh.h"
#include "open3d/t/geometry/Geometry.h"
#include "open3d/t/geometry/TensorMap.h"

namespace open3d {
namespace t {
namespace geometry {

class TriangleMesh : public Geometry {
public:
    TriangleMesh(const core::Device &device = core::Device("CPU:0"));

    TriangleMesh(const core::Tensor &vertices, const core::Tensor &triangles);

    virtual ~TriangleMesh() override {}

public:
    TriangleMesh To(const core::Device &device, bool copy = false) const;

    TriangleMesh Clone() const { return To(GetDevice()); }

    TriangleMesh CPU() const { return To(core::Device("CPU:0")); };

    TriangleMesh CUDA(int device_id = 0) const {
        return To(core::Device(core::Device::DeviceType::CUDA, device_id));
    };

    const TensorMap &GetVertexAttr() const { return vertex_attr_; }

    core::Tensor &GetVertexAttr(const std::string &key) {
        return vertex_attr_.at(key);
    }

    core::Tensor &GetVertices() { return GetVertexAttr("vertices"); }

    core::Tensor &GetVertexColors() { return GetVertexAttr("colors"); }

    core::Tensor &GetVertexNormals() { return GetVertexAttr("normals"); }

    const TensorMap &GetTriangleAttr() const { return triangle_attr_; }

    core::Tensor &GetTriangleAttr(const std::string &key) {
        return triangle_attr_.at(key);
    }

    core::Tensor &GetTriangles() { return GetTriangleAttr("triangles"); }

    core::Tensor &GetTriangleNormals() { return GetTriangleAttr("normals"); }

    core::Tensor &GetTriangleColors() { return GetTriangleAttr("colors"); }

    const core::Tensor &GetVertexAttr(const std::string &key) const {
        return vertex_attr_.at(key);
    }

    void RemoveVertexAttr(const std::string &key) { vertex_attr_.Erase(key); }

    const core::Tensor &GetVertices() const {
        return GetVertexAttr("vertices");
    }

    const core::Tensor &GetVertexColors() const {
        return GetVertexAttr("colors");
    }

    const core::Tensor &GetVertexNormals() const {
        return GetVertexAttr("normals");
    }

    const core::Tensor &GetTriangleAttr(const std::string &key) const {
        return triangle_attr_.at(key);
    }

    void RemoveTriangleAttr(const std::string &key) {
        triangle_attr_.Erase(key);
    }

    const core::Tensor &GetTriangles() const {
        return GetTriangleAttr("triangles");
    }

    const core::Tensor &GetTriangleNormals() const {
        return GetTriangleAttr("normals");
    }

    const core::Tensor &GetTriangleColors() const {
        return GetTriangleAttr("colors");
    }

    void SetVertexAttr(const std::string &key, const core::Tensor &value) {
        value.AssertDevice(device_);
        vertex_attr_[key] = value;
    }

    void SetVertices(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetVertexAttr("vertices", value);
    }

    void SetVertexColors(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetVertexAttr("colors", value);
    }

    void SetVertexNormals(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetVertexAttr("normals", value);
    }

    void SetTriangleAttr(const std::string &key, const core::Tensor &value) {
        value.AssertDevice(device_);
        triangle_attr_[key] = value;
    }

    void SetTriangles(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetTriangleAttr("triangles", value);
    }

    void SetTriangleNormals(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetTriangleAttr("normals", value);
    }

    void SetTriangleColors(const core::Tensor &value) {
        value.AssertShapeCompatible({utility::nullopt, 3});
        SetTriangleAttr("colors", value);
    }

    bool HasVertexAttr(const std::string &key) const {
        return vertex_attr_.Contains(key) &&
               GetVertexAttr(key).GetLength() > 0 &&
               GetVertexAttr(key).GetLength() == GetVertices().GetLength();
    }

    bool HasVertices() const { return HasVertexAttr("vertices"); }

    bool HasVertexColors() const { return HasVertexAttr("colors"); }

    bool HasVertexNormals() const { return HasVertexAttr("normals"); }

    bool HasTriangleAttr(const std::string &key) const {
        return triangle_attr_.Contains(key) &&
               GetTriangleAttr(key).GetLength() > 0 &&
               GetTriangleAttr(key).GetLength() == GetTriangles().GetLength();
    }

    bool HasTriangles() const { return HasTriangleAttr("triangles"); }

    bool HasTriangleNormals() const { return HasTriangleAttr("normals"); }

    bool HasTriangleColors() const { return HasTriangleAttr("colors"); }

public:
    TriangleMesh &Clear() override {
        vertex_attr_.clear();
        triangle_attr_.clear();
        return *this;
    }

    bool IsEmpty() const override { return !HasVertices(); }

    core::Tensor GetMinBound() const { utility::LogError("Unimplemented"); }

    core::Tensor GetMaxBound() const { utility::LogError("Unimplemented"); }

    core::Tensor GetCenter() const { utility::LogError("Unimplemented"); }

    TriangleMesh &Transform(const core::Tensor &transformation) {
        utility::LogError("Unimplemented");
    }

    TriangleMesh &Translate(const core::Tensor &translation,
                            bool relative = true) {
        utility::LogError("Unimplemented");
    }

    TriangleMesh &Scale(double scale, const core::Tensor &center) {
        utility::LogError("Unimplemented");
    }

    TriangleMesh &Rotate(const core::Tensor &R, const core::Tensor &center) {
        utility::LogError("Unimplemented");
    }

    core::Device GetDevice() const { return device_; }

    static geometry::TriangleMesh FromLegacyTriangleMesh(
            const open3d::geometry::TriangleMesh &mesh_legacy,
            core::Dtype float_dtype = core::Dtype::Float32,
            core::Dtype int_dtype = core::Dtype::Int64,
            const core::Device &device = core::Device("CPU:0"));

    open3d::geometry::TriangleMesh ToLegacyTriangleMesh() const;

protected:
    core::Device device_ = core::Device("CPU:0");
    TensorMap vertex_attr_;
    TensorMap triangle_attr_;
};

}  // namespace geometry
}  // namespace t
}  // namespace open3d