CurveEditor: Use std::array where applicable

Allows simplifying how redo/undo data gets moved around and also
eliminates a few hardcoded magic values for sizes.
This commit is contained in:
Lioncash 2019-08-25 05:42:24 -04:00
parent bc8218e183
commit b1ab67f8c2
2 changed files with 82 additions and 57 deletions

View File

@ -1,51 +1,64 @@
#include "CurveEditor.hpp" #include "CurveEditor.hpp"
#include <algorithm>
#include <array>
#include "MainWindow.hpp" #include "MainWindow.hpp"
#include <QHBoxLayout>
#include <QGridLayout> #include <QGridLayout>
#include <QPainter> #include <QHBoxLayout>
#include <QMouseEvent>
#include <QJSValueIterator> #include <QJSValueIterator>
#include <QMouseEvent>
#include <QPainter>
class CurveEditUndoCommand : public EditorUndoCommand { class CurveEditUndoCommand : public EditorUndoCommand {
uint8_t m_redoData[128];
uint8_t m_undoData[128];
bool m_usedExpr;
bool m_undid = false;
public: public:
CurveEditUndoCommand(const uint8_t* redoData, bool usedExpr, amuse::ObjToken<ProjectModel::CurveNode> node) using RedoData = std::array<uint8_t, 128>;
: EditorUndoCommand(node.get(), CurveControls::tr("Edit Curve")), m_usedExpr(usedExpr) { using UndoData = std::array<uint8_t, 128>;
std::memcpy(m_redoData, redoData, 128);
} CurveEditUndoCommand(const RedoData& redoData, bool usedExpr, amuse::ObjToken<ProjectModel::CurveNode> node)
: EditorUndoCommand(node.get(), CurveControls::tr("Edit Curve")), m_redoData{redoData}, m_usedExpr(usedExpr) {}
void undo() override { void undo() override {
m_undid = true; m_undid = true;
amuse::ITable& table = **m_node.cast<ProjectModel::CurveNode>()->m_obj; amuse::ITable& table = **m_node.cast<ProjectModel::CurveNode>()->m_obj;
if (table.Isa() == amuse::ITable::Type::Curve) { if (table.Isa() == amuse::ITable::Type::Curve) {
amuse::Curve& curve = static_cast<amuse::Curve&>(table); auto& curve = static_cast<amuse::Curve&>(table);
curve.data.resize(128); curve.data.assign(m_undoData.cbegin(), m_undoData.cend());
std::memcpy(curve.data.data(), m_undoData, 128);
} }
EditorUndoCommand::undo(); EditorUndoCommand::undo();
} }
void redo() override { void redo() override {
amuse::ITable& table = **m_node.cast<ProjectModel::CurveNode>()->m_obj; amuse::ITable& table = **m_node.cast<ProjectModel::CurveNode>()->m_obj;
if (table.Isa() == amuse::ITable::Type::Curve) { if (table.Isa() == amuse::ITable::Type::Curve) {
amuse::Curve& curve = static_cast<amuse::Curve&>(table); auto& curve = static_cast<amuse::Curve&>(table);
curve.data.resize(128); curve.data.resize(128);
std::memcpy(m_undoData, curve.data.data(), 128); std::copy(curve.data.cbegin(), curve.data.cend(), m_undoData.begin());
std::memcpy(curve.data.data(), m_redoData, 128); curve.data.assign(m_redoData.cbegin(), m_redoData.cend());
} }
if (m_undid) if (m_undid) {
EditorUndoCommand::redo(); EditorUndoCommand::redo();
}
} }
bool mergeWith(const QUndoCommand* other) override { bool mergeWith(const QUndoCommand* other) override {
if (other->id() == id() && !m_usedExpr && !static_cast<const CurveEditUndoCommand*>(other)->m_usedExpr) { const auto* const command = static_cast<const CurveEditUndoCommand*>(other);
std::memcpy(m_redoData, static_cast<const CurveEditUndoCommand*>(other)->m_redoData, 128);
if (other->id() == id() && !m_usedExpr && !command->m_usedExpr) {
m_redoData = command->m_redoData;
return true; return true;
} }
return false; return false;
} }
int id() const override { return int(Id::CurveEdit); } int id() const override { return int(Id::CurveEdit); }
private:
RedoData m_redoData;
UndoData m_undoData;
bool m_usedExpr;
bool m_undid = false;
}; };
CurveEditor* CurveView::getEditor() const { return qobject_cast<CurveEditor*>(parentWidget()); } CurveEditor* CurveView::getEditor() const { return qobject_cast<CurveEditor*>(parentWidget()); }
@ -57,51 +70,56 @@ void CurveView::unloadData() { m_node.reset(); }
ProjectModel::INode* CurveView::currentNode() const { return m_node.get(); } ProjectModel::INode* CurveView::currentNode() const { return m_node.get(); }
void CurveView::paintEvent(QPaintEvent* ev) { void CurveView::paintEvent(QPaintEvent* ev) {
if (!m_node) if (!m_node) {
return; return;
}
amuse::ITable& table = **m_node->m_obj; amuse::ITable& table = **m_node->m_obj;
if (table.Isa() != amuse::ITable::Type::Curve) if (table.Isa() != amuse::ITable::Type::Curve) {
return; return;
amuse::Curve& curve = static_cast<amuse::Curve&>(table); }
auto& curve = static_cast<amuse::Curve&>(table);
QPainter painter(this); QPainter painter(this);
painter.setRenderHint(QPainter::Antialiasing); painter.setRenderHint(QPainter::Antialiasing);
qreal deviceRatio = devicePixelRatioF(); const qreal deviceRatio = devicePixelRatioF();
qreal penWidth = std::max(std::floor(deviceRatio), 1.0) / deviceRatio; const qreal penWidth = std::max(std::floor(deviceRatio), 1.0) / deviceRatio;
painter.setPen(QPen(QColor(127, 127, 127), penWidth)); painter.setPen(QPen(QColor(127, 127, 127), penWidth));
painter.setFont(m_gridFont); painter.setFont(m_gridFont);
qreal yIncrement = (height() - 16.0) / 10.0; const qreal yIncrement = (height() - 16.0) / 10.0;
for (int i = 0; i < 11; ++i) { for (size_t i = 0; i < m_percentTexts.size(); ++i) {
qreal thisY = i * yIncrement; const qreal thisY = i * yIncrement;
qreal textY = thisY - (i == 0 ? 2.0 : (i == 10 ? 16.0 : 8.0)); const qreal textY = thisY - (i == 0 ? 2.0 : (i == 10 ? 16.0 : 8.0));
painter.drawStaticText(QPointF(0.0, textY), m_percentTexts[i]); painter.drawStaticText(QPointF(0.0, textY), m_percentTexts[i]);
painter.drawLine(QPointF(30.0, thisY), QPointF(width(), thisY)); painter.drawLine(QPointF(30.0, thisY), QPointF(width(), thisY));
} }
qreal xIncrement = (width() - 30.0) / 10.0; qreal xIncrement = (width() - 30.0) / 10.0;
for (int i = 0; i < 11; ++i) { for (size_t i = 0; i < m_percentTextsCenter.size(); ++i) {
qreal thisX = i * xIncrement + 30.0; const qreal thisX = i * xIncrement + 30.0;
qreal textX = thisX - (i == 10 ? 30.0 : 15.0); const qreal textX = thisX - (i == 10 ? 30.0 : 15.0);
painter.drawStaticText(QPointF(textX, height() - 16.0), m_percentTextsCenter[i]); painter.drawStaticText(QPointF(textX, height() - 16.0), m_percentTextsCenter[i]);
painter.drawLine(QPointF(thisX, 0.0), QPointF(thisX, height() - 16.0)); painter.drawLine(QPointF(thisX, 0.0), QPointF(thisX, height() - 16.0));
} }
int i; size_t i;
xIncrement = (width() - 30.0) / 127.0; xIncrement = (width() - 30.0) / 127.0;
QPointF lastPt; QPointF lastPt;
painter.setPen(QPen(Qt::white, penWidth * 3.0)); painter.setPen(QPen(Qt::white, penWidth * 3.0));
for (i = 0; i < curve.data.size(); ++i) { for (i = 0; i < curve.data.size(); ++i) {
QPointF thisPt(i * xIncrement + 30.0, (height() - 16.0) - (height() - 16.0) * (curve.data[i] / 127.0)); const QPointF thisPt(i * xIncrement + 30.0, (height() - 16.0) - (height() - 16.0) * (curve.data[i] / 127.0));
if (i) if (i != 0) {
painter.drawLine(lastPt, thisPt); painter.drawLine(lastPt, thisPt);
}
lastPt = thisPt; lastPt = thisPt;
} }
for (; i < 128; ++i) { for (; i < 128; ++i) {
QPointF thisPt(i * xIncrement + 30.0, height() - 16.0); const QPointF thisPt(i * xIncrement + 30.0, height() - 16.0);
if (i) if (i != 0) {
painter.drawLine(lastPt, thisPt); painter.drawLine(lastPt, thisPt);
}
lastPt = thisPt; lastPt = thisPt;
} }
} }
@ -111,19 +129,22 @@ void CurveView::mousePressEvent(QMouseEvent* ev) { mouseMoveEvent(ev); }
void CurveView::mouseMoveEvent(QMouseEvent* ev) { void CurveView::mouseMoveEvent(QMouseEvent* ev) {
CurveView* view = getEditor()->m_curveView; CurveView* view = getEditor()->m_curveView;
amuse::ITable& table = **view->m_node->m_obj; amuse::ITable& table = **view->m_node->m_obj;
if (table.Isa() != amuse::ITable::Type::Curve) if (table.Isa() != amuse::ITable::Type::Curve) {
return; return;
amuse::Curve& curve = static_cast<amuse::Curve&>(table); }
qreal xIncrement = (width() - 30.0) / 127.0; const qreal xIncrement = (width() - 30.0) / 127.0;
int idx = int(std::round((ev->localPos().x() - 30.0) / xIncrement)); const int idx = int(std::round((ev->localPos().x() - 30.0) / xIncrement));
if (idx < 0 || idx > 127) if (idx < 0 || idx > 127) {
return; return;
int val = 127 - amuse::clamp(0, int(std::round(ev->localPos().y() / (height() - 16.0) * 127.0)), 127); }
const int val = 127 - amuse::clamp(0, int(std::round(ev->localPos().y() / (height() - 16.0) * 127.0)), 127);
curve.data.resize(128); CurveEditUndoCommand::RedoData newData;
uint8_t newData[128]; auto& curve = static_cast<amuse::Curve&>(table);
std::memcpy(newData, curve.data.data(), 128); curve.data.resize(newData.size());
std::copy(curve.data.cbegin(), curve.data.cend(), newData.begin());
newData[idx] = uint8_t(val); newData[idx] = uint8_t(val);
g_MainWindow->pushUndoCommand(new CurveEditUndoCommand(newData, false, m_node)); g_MainWindow->pushUndoCommand(new CurveEditUndoCommand(newData, false, m_node));
@ -131,7 +152,7 @@ void CurveView::mouseMoveEvent(QMouseEvent* ev) {
} }
CurveView::CurveView(QWidget* parent) : QWidget(parent) { CurveView::CurveView(QWidget* parent) : QWidget(parent) {
for (int i = 0; i < 11; ++i) { for (size_t i = 0; i < m_percentTexts.size(); ++i) {
m_percentTexts[i].setText(QStringLiteral("%1%").arg(100 - i * 10)); m_percentTexts[i].setText(QStringLiteral("%1%").arg(100 - i * 10));
m_percentTexts[i].setTextOption(QTextOption(Qt::AlignVCenter | Qt::AlignRight)); m_percentTexts[i].setTextOption(QTextOption(Qt::AlignVCenter | Qt::AlignRight));
m_percentTexts[i].setTextWidth(28.0); m_percentTexts[i].setTextWidth(28.0);
@ -157,16 +178,18 @@ void CurveControls::unloadData() {
void CurveControls::exprCommit() { void CurveControls::exprCommit() {
CurveView* view = getEditor()->m_curveView; CurveView* view = getEditor()->m_curveView;
amuse::ITable& table = **view->m_node->m_obj; amuse::ITable& table = **view->m_node->m_obj;
if (table.Isa() != amuse::ITable::Type::Curve) if (table.Isa() != amuse::ITable::Type::Curve) {
return; return;
amuse::Curve& curve = static_cast<amuse::Curve&>(table); }
const QString progText = m_lineEdit->text();
CurveEditUndoCommand::RedoData newData;
auto& curve = static_cast<amuse::Curve&>(table);
curve.data.resize(newData.size());
std::copy(curve.data.cbegin(), curve.data.cend(), newData.begin());
QString progText = m_lineEdit->text();
curve.data.resize(128);
uint8_t newData[128];
std::memcpy(newData, curve.data.data(), 128);
bool notANumber = false; bool notANumber = false;
for (int i = 0; i < 128; ++i) { for (size_t i = 0; i < newData.size(); ++i) {
m_engine.globalObject().setProperty(QStringLiteral("x"), i / 127.0); m_engine.globalObject().setProperty(QStringLiteral("x"), i / 127.0);
QJSValue val = m_engine.evaluate(progText); QJSValue val = m_engine.evaluate(progText);
if (val.isError()) { if (val.isError()) {

View File

@ -1,5 +1,7 @@
#pragma once #pragma once
#include <array>
#include "EditorWidget.hpp" #include "EditorWidget.hpp"
#include <QFrame> #include <QFrame>
#include <QLabel> #include <QLabel>
@ -15,8 +17,8 @@ class CurveView : public QWidget {
friend class CurveControls; friend class CurveControls;
amuse::ObjToken<ProjectModel::CurveNode> m_node; amuse::ObjToken<ProjectModel::CurveNode> m_node;
QFont m_gridFont; QFont m_gridFont;
QStaticText m_percentTexts[11]; std::array<QStaticText, 11> m_percentTexts;
QStaticText m_percentTextsCenter[11]; std::array<QStaticText, 11> m_percentTextsCenter;
CurveEditor* getEditor() const; CurveEditor* getEditor() const;
public: public: