PickPointsInteractor.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2023 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <queue>
 
#include "open3d/visualization/gui/SceneWidget.h"
#include "open3d/visualization/rendering/MatrixInteractorLogic.h"
 
namespace open3d {
 
namespace geometry {
class Geometry3D;
class Image;
}  // namespace geometry
 
namespace visualization {
 
namespace rendering {
class Camera;
struct MaterialRecord;
class MatrixInteractorLogic;
class Open3DScene;
}  // namespace rendering
 
namespace gui {
 
class SelectionIndexLookup;
 
// This is an internal class used by SceneWidget
class PickPointsInteractor : public SceneWidget::MouseInteractor {
public:
    PickPointsInteractor(rendering::Open3DScene* scene,
                         rendering::Camera* camera);
    virtual ~PickPointsInteractor();
 
    void SetPointSize(int px);
 
    void SetPickableGeometry(
            const std::vector<SceneWidget::PickableGeometry>& geometry);
 
    void SetNeedsRedraw();
 
    void SetOnPointsPicked(
            std::function<void(
                    const std::map<
                            std::string,
                            std::vector<std::pair<size_t, Eigen::Vector3d>>>&,
                    int)> f);
 
    void SetOnUIChanged(
            std::function<void(const std::vector<Eigen::Vector2i>&)>);
 
    void SetOnStartedPolygonPicking(std::function<void()> on_poly_pick);
 
    void DoPick();
    void ClearPick();
 
    rendering::MatrixInteractorLogic& GetMatrixInteractor() override;
    void Mouse(const MouseEvent& e) override;
    void Key(const KeyEvent& e) override;
 
protected:
    void OnPickImageDone(std::shared_ptr<geometry::Image> img);
 
    rendering::MaterialRecord MakeMaterial();
 
private:
    rendering::Open3DScene* scene_;
    rendering::Camera* camera_;
 
    std::function<void(
            const std::map<std::string,
                           std::vector<std::pair<size_t, Eigen::Vector3d>>>&,
            int)>
            on_picked_;
    std::function<void(const std::vector<Eigen::Vector2i>&)> on_ui_changed_;
    std::function<void()> on_started_poly_pick_;
    int point_size_ = 3;
    rendering::MatrixInteractorLogic matrix_logic_;
    std::shared_ptr<rendering::Open3DScene> picking_scene_;
    std::vector<Eigen::Vector3d> points_;
    // This is a pointer rather than unique_ptr so that we don't have
    // to define this (internal) class in the header file.
    SelectionIndexLookup* lookup_ = nullptr;
    std::shared_ptr<geometry::Image> pick_image_;
    bool dirty_ = true;
    struct PickInfo {
        std::vector<gui::Point> polygon;  // or point, if only one item
        int keymods;
    };
    std::queue<PickInfo> pending_;
};
 
}  // namespace gui
}  // namespace visualization
}  // namespace open3d