WolmanCustomPlot.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 <qcustomplot.h>
 
// Qt5/Qt6 Compatibility
#include <QtCompat.h>
 
// Eigen
#include <Eigen/Geometry>
 
namespace Ui {
class WolmanCustomPlot;
}
 
class QCPBarsWithText : public QCPBars {
    Q_OBJECT
 
public:
    QCPBarsWithText(QCPAxis* keyAxis, QCPAxis* valueAxis)
        : QCPBars(keyAxis, valueAxis), m_textOnTheLeft(false) {}
 
    void setText(QString text) { m_text = QStringList(text); }
    void appendText(QString text) { m_text.append(text); }
    void setTextAlignment(bool left) { m_textOnTheLeft = left; }
 
protected:
    QStringList m_text;
    bool m_textOnTheLeft;
 
    // reimplemented virtual draw method
    virtual void draw(QCPPainter* painter) {
        if (!mKeyAxis || !mValueAxis) {
            qDebug() << Q_FUNC_INFO << "invalid key or value axis";
            return;
        }
 
        // switch to standard display
        QCPBars::draw(painter);
 
        int fontHeight = painter->fontMetrics().height();
 
        if (!data()->isEmpty()) {
            double& key = data()->begin()->key;
            double& value = data()->begin()->value;
            QPointF P = coordsToPixels(key, value);
            // apply a small shift
            int margin = 5;  // in pixels
            if (m_textOnTheLeft) margin = -margin;
            P.setX(P.x() + margin);
            // we draw at the 'base' line
            P.setY(P.y() + fontHeight);
 
            for (int i = 0; i < m_text.size(); ++i) {
                QPointF Pstart = P;
                if (m_textOnTheLeft)
                    Pstart.setX(P.x() -
                                QTCOMPAT_FONTMETRICS_WIDTH(
                                        painter->fontMetrics(), m_text[i]));
                painter->drawText(Pstart, m_text[i]);
                P.setY(P.y() + fontHeight);
            }
        }
    }
};
 
class WolmanCustomPlot : public QCustomPlot {
    Q_OBJECT
public:
    WolmanCustomPlot(const Eigen::ArrayXf& d_sample,
                     const Eigen::Array3d& dq_final,
                     const Eigen::Array3d& edq);
 
    QSharedPointer<QCPGraphDataContainer> dataContainer() {
        return m_graph->data();
    }
 
    Eigen::Array3d m_dq_final;
 
    Eigen::Array3d m_edq;
 
    QCPGraph* m_graph;
 
    void rescale();
 
    void emitCloseTab() { emit closeTab(); }
 
    void mouseDoubleClickEvent(QMouseEvent* event) override;
 
    void mousePressEvent(QMouseEvent* event) override;
 
signals:
    void closeTab();
 
private:
    Ui::WolmanCustomPlot* ui;
};