cvSelectionAnnotation.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
 
// clang-format off
// Qt - must be included before qPCL.h for MOC to work correctly
#include <QtGui/QColor>
#include <QtCore/QDateTime>
#include <QtGui/QFont>
#include <QtCore/QList>
#include <QtCore/QMap>
#include <QtCore/QObject>
#include <QtCore/QString>
// clang-format on
 
#include "qPCL.h"
 
// LOCAL
#include "cvSelectionData.h"
 
// Forward declarations
class vtkActor2D;
class vtkPolyData;
class vtkTextActor;
namespace PclUtils {
class PCLVis;
}
 
// Include for LabelProperties struct (nested type requires full definition)
#include "cvSelectionLabelPropertiesDialog.h"
 
struct QPCL_ENGINE_LIB_API cvAnnotation {
    QString text;                     
    QString id;                       
    double position[3];               
    QColor color;                     
    int fontSize;                     
    QString fontFamily;               
    bool bold;                        
    bool italic;                      
    bool shadow;                      
    double opacity;                   
    QString horizontalJustification;  
    QString verticalJustification;    
    bool visible;                     
    bool followSelection;             
    qint64 timestamp;                 
 
    cvAnnotation()
        : fontSize(12),
          fontFamily("Arial"),
          bold(false),
          italic(false),
          shadow(true),
          opacity(1.0),
          horizontalJustification("Left"),
          verticalJustification("Bottom"),
          visible(true),
          followSelection(false),
          timestamp(0) {
        position[0] = position[1] = position[2] = 0.0;
        color = Qt::yellow;
    }
 
    cvAnnotation(const QString& txt,
                 const double pos[3],
                 const QString& uid = QString())
        : text(txt),
          id(uid.isEmpty()
                     ? QString::number(QDateTime::currentMSecsSinceEpoch())
                     : uid),
          fontSize(12),
          fontFamily("Arial"),
          bold(false),
          italic(false),
          shadow(true),
          opacity(1.0),
          horizontalJustification("Left"),
          verticalJustification("Bottom"),
          visible(true),
          followSelection(false),
          timestamp(QDateTime::currentMSecsSinceEpoch()) {
        position[0] = pos[0];
        position[1] = pos[1];
        position[2] = pos[2];
        color = Qt::yellow;
    }
};
 
class QPCL_ENGINE_LIB_API cvSelectionAnnotationManager : public QObject {
    Q_OBJECT
 
public:
    explicit cvSelectionAnnotationManager(QObject* parent = nullptr);
    ~cvSelectionAnnotationManager() override;
 
    void setVisualizer(PclUtils::PCLVis* viewer);
 
    QString addAnnotation(const cvSelectionData& selection,
                          const QString& text,
                          bool autoPosition = true);
 
    QString addAnnotationAt(const QString& text,
                            const double position[3],
                            const QString& id = QString());
 
    bool removeAnnotation(const QString& id);
 
    bool updateAnnotationText(const QString& id, const QString& text);
 
    bool updateAnnotationPosition(const QString& id, const double position[3]);
 
    bool setAnnotationVisible(const QString& id, bool visible);
 
    bool setAnnotationColor(const QString& id, const QColor& color);
 
    bool setAnnotationFontSize(const QString& id, int fontSize);
 
    void applyLabelProperties(
            const cvSelectionLabelPropertiesDialog::LabelProperties& props,
            bool isCellLabel);
 
    void setDefaultLabelProperties(
            const cvSelectionLabelPropertiesDialog::LabelProperties& props,
            bool isCellLabel);
 
    cvAnnotation getAnnotation(const QString& id) const;
 
    QList<cvAnnotation> allAnnotations() const;
 
    QStringList annotationIds() const;
 
    int count() const { return m_annotations.size(); }
 
    void clearAll();
 
    void showAll();
 
    void hideAll();
 
    bool exportToFile(const QString& filename) const;
 
    bool importFromFile(const QString& filename, bool merge = true);
 
signals:
    void annotationsChanged();
 
    void annotationAdded(const QString& id);
 
    void annotationRemoved(const QString& id);
 
    void annotationUpdated(const QString& id);
 
private:
    void createTextActor(const cvAnnotation& annotation);
    void updateTextActor(const QString& id);
    void removeTextActor(const QString& id);
    double* computeSelectionCenter(const cvSelectionData& selection,
                                   vtkPolyData* polyData);
 
    bool worldToDisplay(const double worldPos[3], double displayPos[2]);
 
    PclUtils::PCLVis* m_viewer;
    QMap<QString, cvAnnotation> m_annotations;
    QMap<QString, vtkSmartPointer<vtkTextActor>> m_textActors;
 
    // Default label properties for new annotations (applied from dialog)
    cvSelectionLabelPropertiesDialog::LabelProperties m_defaultCellLabelProps;
    cvSelectionLabelPropertiesDialog::LabelProperties m_defaultPointLabelProps;
};