ecvAnnotationsTool.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>
#include <ui_annotationsDlg.h>
 
// CV_DB_LIB
#include <ecvBBox.h>
#include <ecvDisplayTools.h>
 
// SYSTEM
#include <vector>
 
class ccGenericPointCloud;
class ecvProgressDialog;
class ccHObject;
class ccBBox;
class ecvGenericAnnotationTool;
 
class ecvAnnotationsTool : public ccOverlayDialog, public Ui::AnnotationsDlg {
    Q_OBJECT
 
public:
    explicit ecvAnnotationsTool(QWidget* parent);
    virtual ~ecvAnnotationsTool() override;
 
    // inherited from ccOverlayDialog
    virtual bool start() override;
    virtual void stop(bool state) override;
    virtual bool linkWith(QWidget* win) override;
 
    bool setAnnotationsTool(ecvGenericAnnotationTool* annotationTool);
 
 
    bool addAssociatedEntity(ccHObject* anObject);
 
    void releaseAssociatedEntities();
 
    unsigned getNumberOfAssociatedEntity() const;
 
public slots:
    void toggleInteractors(bool state);
    void toggleEditMode(bool state);
 
protected slots:
    void onShortcutTriggered(int);
 
    void importClassesFromFile();
    void toggleBox(bool);
    void toggleOrigin(bool dummy);
 
    void onNewMode();
    void onUnionMode();
    void onTrimMode();
    void onIntersectMode();
    void onLabelSelected();
    void onLabelChanged(int index);
    void onItemPicked(bool isPicked);
 
    void saveAnnotations();
    void reset();
    void closeDialog();
    void exportAnnotationToSF();
    void updateLabelsCombox(const std::vector<std::string>& labels);
 
    inline void shiftXMinus() { shiftBox(0, true); }
    inline void shiftXPlus() { shiftBox(0, false); }
    inline void shiftYMinus() { shiftBox(1, true); }
    inline void shiftYPlus() { shiftBox(1, false); }
    inline void shiftZMinus() { shiftBox(2, true); }
    inline void shiftZPlus() { shiftBox(2, false); }
 
    void setFrontView();
    void setBottomView();
    void setTopView();
    void setBackView();
    void setLeftView();
    void setRightView();
 
protected:
    void shiftBox(unsigned char dim, bool minus);
 
    void setView(CC_VIEW_ORIENTATION orientation);
 
    ecvGenericAnnotationTool* m_annotationTool;
 
    ccBBox getSelectedEntityBbox();
 
    ccBBox m_box;
 
    ccHObject m_entityContainer;
 
    bool m_disabledCombEvent;
 
    bool m_editMode;
};