cvPointPickingHelper.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <vtkSmartPointer.h>
 
#include <QKeySequence>
#include <QObject>
#include <QPointer>
 
// CV_DB_LIB
#include <Shortcuts/ecvModalShortcut.h>
 
class QWidget;
class vtkCell;
class vtkCellPicker;
class vtkDataSet;
class vtkPointPicker;
class vtkPropPicker;
class vtkHardwareSelector;
class vtkRenderWindowInteractor;
class vtkRenderer;
 
class cvPointPickingHelper : public QObject {
    Q_OBJECT
 
public:
    enum PickOption {
        Coordinates,          
        Normal,               
        CoordinatesAndNormal  
    };
 
    cvPointPickingHelper(const QKeySequence& keySequence,
                         bool pickOnPoint,
                         QWidget* parent = nullptr,
                         PickOption pickOpt = Coordinates);
    ~cvPointPickingHelper() override;
 
    bool pickOnPoint() const { return m_pickOnPoint; }
 
    PickOption getPickOption() const { return m_pickOption; }
 
    void setInteractor(vtkRenderWindowInteractor* interactor);
 
    void setRenderer(vtkRenderer* renderer);
 
    void setContextWidget(QWidget* widget);
 
    void setEnabled(bool enabled, bool setFocus = false);
 
    bool isEnabled() const;
 
    void clearSelectionCache();
 
Q_SIGNALS:
    void pick(double x, double y, double z);
 
    void pickNormal(
            double px, double py, double pz, double nx, double ny, double nz);
 
private Q_SLOTS:
    void pickPoint();
 
private:
    Q_DISABLE_COPY(cvPointPickingHelper)
 
    
    void getCellNormal(vtkDataSet* dataset,
                       vtkIdType cellId,
                       vtkCell* cell,
                       double normal[3]);
 
    QPointer<ecvModalShortcut> m_shortcut;
    QPointer<QWidget> m_contextWidget;
    vtkRenderWindowInteractor* m_interactor = nullptr;
    vtkRenderer* m_renderer = nullptr;
    bool m_pickOnPoint;
    PickOption m_pickOption;
 
    // Reusable pickers to avoid creating new objects on each pick
    vtkSmartPointer<vtkPointPicker> m_pointPicker;
    vtkSmartPointer<vtkCellPicker> m_cellPicker;
    vtkSmartPointer<vtkPropPicker>
            m_propPicker;  // Fast hardware-accelerated picker
    vtkSmartPointer<vtkHardwareSelector>
            m_hardwareSelector;  // Ultra-fast GPU-based picking
 
    // Selection cache (ParaView-style optimization)
    struct SelectionCache {
        int displayX = -1;
        int displayY = -1;
        bool pickOnPoint = false;
        double position[3] = {0, 0, 0};
        double normal[3] = {0, 0, 1};
        bool valid = false;
        vtkIdType id = -1;
    } m_selectionCache;
};