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

76 lines
2.6 KiB
C
Raw 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 PickingShader : public ShaderWrapper {
public:
~PickingShader() override { Release(); }
protected:
PickingShader(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<float> &indices) = 0;
protected:
GLuint vertex_position_;
GLuint vertex_position_buffer_;
GLuint vertex_index_;
GLuint vertex_index_buffer_;
GLuint MVP_;
};
class PickingShaderForPointCloud : public PickingShader {
public:
PickingShaderForPointCloud()
: PickingShader("PickingShaderForPointCloud") {}
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<float> &indices) final;
};
} // namespace glsl
} // namespace visualization
} // namespace open3d