Paul Leroy
99 lines
2.1 KiB
C++
99 lines
2.1 KiB
C++
#ifndef WOLMANCUSTOMPLOT_H
|
|
#define WOLMANCUSTOMPLOT_H
|
|
|
|
#include <qcustomplot.h>
|
|
|
|
// Eigen
|
|
#include <Eigen/Geometry>
|
|
|
|
namespace Ui {
|
|
class WolmanCustomPlot;
|
|
}
|
|
|
|
//! QCustomPlot: vertical bar with text along side
|
|
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() - painter->fontMetrics().boundingRect(m_text[i]).width());
|
|
}
|
|
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;
|
|
};
|
|
|
|
#endif // WOLMANCUSTOMPLOT_H
|