image_framework_ymj/include/open3d/visualization/shader/NormalShader.h

93 lines
3.3 KiB
C
Raw Permalink Normal View History

2024-12-06 16:25:16 +08:00
// ----------------------------------------------------------------------------
// - Open3D: www.open3d.org -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2023 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
#pragma once
#include <Eigen/Core>
#include <vector>
#include "open3d/visualization/shader/ShaderWrapper.h"
namespace open3d {
namespace visualization {
namespace glsl {
class NormalShader : public ShaderWrapper {
public:
~NormalShader() override { Release(); }
protected:
NormalShader(const std::string &name) : ShaderWrapper(name) { Compile(); }
protected:
bool Compile() final;
void Release() final;
bool BindGeometry(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view) final;
bool RenderGeometry(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view) final;
void UnbindGeometry() final;
protected:
virtual bool PrepareRendering(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view) = 0;
virtual bool PrepareBinding(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view,
std::vector<Eigen::Vector3f> &points,
std::vector<Eigen::Vector3f> &normals) = 0;
protected:
GLuint vertex_position_;
GLuint vertex_position_buffer_;
GLuint vertex_normal_;
GLuint vertex_normal_buffer_;
GLuint MVP_;
GLuint V_;
GLuint M_;
};
class NormalShaderForPointCloud : public NormalShader {
public:
NormalShaderForPointCloud() : NormalShader("NormalShaderForPointCloud") {}
protected:
bool PrepareRendering(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view) final;
bool PrepareBinding(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view,
std::vector<Eigen::Vector3f> &points,
std::vector<Eigen::Vector3f> &normals) final;
};
class NormalShaderForTriangleMesh : public NormalShader {
public:
NormalShaderForTriangleMesh()
: NormalShader("NormalShaderForTriangleMesh") {}
protected:
bool PrepareRendering(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view) final;
bool PrepareBinding(const geometry::Geometry &geometry,
const RenderOption &option,
const ViewControl &view,
std::vector<Eigen::Vector3f> &points,
std::vector<Eigen::Vector3f> &normals) final;
};
} // namespace glsl
} // namespace visualization
} // namespace open3d