ecvGraphicalSegmentationTool.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
 
// LOCAL
#include <ecvOverlayDialog.h>
 
// CV_DB_LIB
#include <ecvHObject.h>
 
// QT
#include <QSet>
 
// GUI
#include <ui_graphicalSegmentationDlg.h>
 
class ccPolyline;
class ccPointCloud;
class ecvMainAppInterface;
 
class ccGraphicalSegmentationTool : public ccOverlayDialog,
                                    public Ui::GraphicalSegmentationDlg {
    Q_OBJECT
 
public:
    explicit ccGraphicalSegmentationTool(QWidget* parent);
    virtual ~ccGraphicalSegmentationTool();
 
 
    bool addEntity(ccHObject* anObject, bool silent = false);
 
    unsigned getNumberOfValidEntities() const;
 
    ccPolyline* getPolyLine() { return m_segmentationPoly; }
 
    QSet<ccHObject*>& entities() { return m_toSegment; }
    const QSet<ccHObject*>& entities() const { return m_toSegment; }
 
    // inherited from ccOverlayDialog
    virtual bool linkWith(QWidget* win) override;
    virtual bool start() override;
    virtual void stop(bool accepted) override;
 
    bool deleteHiddenParts() const { return m_deleteHiddenParts; }
 
 
    void removeAllEntities();
 
    bool applySegmentation(ecvMainAppInterface* app,
                           ccHObject::Container& newEntities);
 
protected slots:
 
    void segmentIn();
    void segmentOut();
    void segment(bool);
    void reset();
    void apply();
    void applyAndDelete();
    void cancel();
    void addPointToPolyline(int x, int y);
    void closePolyLine(int x = 0,
                       int y = 0);  // arguments for compatibility with
                                    // ccGlWindow::rightButtonClicked signal
    void updateSegmentation();
    void closeRectangle();
    void updatePolyLine(int x, int y, Qt::MouseButtons buttons);
    void pauseSegmentationMode(bool state, bool only2D = true);
    inline void pauseSegmentation(bool state) { pauseSegmentationMode(state); }
    void resetSegmentation();
    void doSetPolylineSelection();
    void doSetRectangularSelection();
    void doActionUseExistingPolyline();
    void doExportSegmentationPolyline();
 
    void onShortcutTriggered(int);
 
    void prepareEntityForRemoval(ccHObject* entity,
                                 bool unallocateVisibilityArrays);
 
    void allowPolylineExport(bool state);
 
protected:
    void setDrawFlag(bool state = true);
 
    QSet<ccHObject*> m_toSegment;
 
    bool m_somethingHasChanged;
 
    enum ProcessStates {
        POLYLINE = 1,
        RECTANGLE = 2,
        //...           = 4,
        //...           = 8,
        //...           = 16,
        PAUSED = 32,
        STARTED = 64,
        RUNNING = 128,
    };
 
    unsigned m_state;
 
    ccPolyline* m_segmentationPoly;
    ccPointCloud* m_polyVertices;
 
    bool m_rectangularSelection;
 
    bool m_deleteHiddenParts;
};