acquired_points.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef UTILS_ACQUIRED_POINTS_H
#define UTILS_ACQUIRED_POINTS_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../utils_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QMap>
#include <QObject>
#include <QString>
#include <QVector>
#include <QVector3D>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
enum class PointKind {
    Fiducial = 0,    
    Eeg,             
    HeadShape        
};
 
//=============================================================================================================
enum class FiducialId {
    LPA = 1,
    NAS = 2,
    RPA = 3
};
 
//=============================================================================================================
struct UTILSSHARED_EXPORT DigitizedPoint
{
    PointKind   kind = PointKind::HeadShape;
    QString     label;            
    int         identNumber = 0;  
    QVector3D   position;         
};
 
//=============================================================================================================
class UTILSSHARED_EXPORT AcquiredPoints : public QObject
{
    Q_OBJECT
public:
    explicit AcquiredPoints(QObject* parent = nullptr) : QObject(parent) {}
 
    const QVector<DigitizedPoint>& points() const { return m_points; }

    void append(const DigitizedPoint& p) {
        m_points.append(p);
        emit pointsChanged();
    }

    void undoLast(PointKind kind) {
        for (int i = m_points.size() - 1; i >= 0; --i) {
            if (m_points[i].kind == kind) {
                m_points.removeAt(i);
                emit pointsChanged();
                return;
            }
        }
    }

    void clear() {
        if (m_points.isEmpty()) return;
        m_points.clear();
        emit pointsChanged();
    }

    QVector3D fiducial(FiducialId id) const {
        for (const auto& p : m_points) {
            if (p.kind == PointKind::Fiducial && p.identNumber == static_cast<int>(id)) {
                return p.position;
            }
        }
        return {};
    }

    bool hasFiducial(FiducialId id) const {
        for (const auto& p : m_points) {
            if (p.kind == PointKind::Fiducial && p.identNumber == static_cast<int>(id)) {
                return true;
            }
        }
        return false;
    }

    void removeFiducial(FiducialId id) {
        for (int i = m_points.size() - 1; i >= 0; --i) {
            if (m_points[i].kind == PointKind::Fiducial
                && m_points[i].identNumber == static_cast<int>(id)) {
                m_points.removeAt(i);
            }
        }
        emit pointsChanged();
    }

    bool hasAllFiducials() const {
        return hasFiducial(FiducialId::NAS)
            && hasFiducial(FiducialId::LPA)
            && hasFiducial(FiducialId::RPA);
    }
 
    int countOf(PointKind kind) const {
        int n = 0;
        for (const auto& p : m_points) if (p.kind == kind) ++n;
        return n;
    }
 
    void setPosition(int index, const QVector3D& pos) {
        if (index >= 0 && index < m_points.size())
            m_points[index].position = pos;
    }
 
    void emitChanged() { emit pointsChanged(); }
 
    // ── Twin (BEM-space) fiducials — clicked on digital twin surface ──
 
    void setTwinFiducial(FiducialId id, const QVector3D& pos) {
        m_twinFiducials[static_cast<int>(id)] = pos;
        emit pointsChanged();
    }
 
    void clearTwinFiducial(FiducialId id) {
        m_twinFiducials.remove(static_cast<int>(id));
        emit pointsChanged();
    }
 
    bool hasTwinFiducial(FiducialId id) const {
        return m_twinFiducials.contains(static_cast<int>(id));
    }
 
    QVector3D twinFiducial(FiducialId id) const {
        return m_twinFiducials.value(static_cast<int>(id));
    }
 
    bool hasAllTwinFiducials() const {
        return hasTwinFiducial(FiducialId::NAS)
            && hasTwinFiducial(FiducialId::LPA)
            && hasTwinFiducial(FiducialId::RPA);
    }
 
signals:
    void pointsChanged();
 
private:
    QVector<DigitizedPoint> m_points;
    QMap<int, QVector3D> m_twinFiducials; 
};
 
} // namespace UTILSLIB
 
#endif // UTILS_ACQUIRED_POINTS_H