1 files changed, 118 insertions, 0 deletions
diff --git a/plugins/warp/ui/shared/constraint.cpp b/plugins/warp/ui/shared/constraint.cpp
new file mode 100644
index 00000000..1d369bcd
--- /dev/null
+++ b/plugins/warp/ui/shared/constraint.cpp
@@ -0,0 +1,118 @@
+#include "constraint.h"
+
+#include <QGridLayout>
+#include <QHeaderView>
+
+WarpConstraintItem::WarpConstraintItem(const Warp::Constraint &constraint) : m_constraint(constraint)
+{
+    QString guidStr = QString::fromStdString(constraint.guid.ToString());
+    if (const auto offset = constraint.offset; offset)
+        guidStr += QString(" @ %1").arg(*offset, 0, 16);
+    setText(guidStr);
+}
+
+WarpConstraintItemModel::WarpConstraintItemModel(const QStringList &labels, QObject *parent)
+{
+    this->setHorizontalHeaderLabels(labels);
+}
+
+void WarpConstraintItemModel::AddItem(WarpConstraintItem *item)
+{
+    QList<QStandardItem *> row = {};
+    row.insert(COL_CONSTRAINT_ITEM, item);
+    appendRow(row);
+}
+
+WarpConstraintItem *WarpConstraintItemModel::GetItem(const QModelIndex &index) const
+{
+    if (!index.isValid())
+        return nullptr;
+    return dynamic_cast<WarpConstraintItem *>(item(index.row(), COL_CONSTRAINT_ITEM));
+}
+
+QVariant WarpConstraintItemModel::data(const QModelIndex &index, int role) const
+{
+    // Highlight constraints that are found in analysis.
+    if (role == Qt::BackgroundRole)
+    {
+        if (const auto item = GetItem(index); item)
+        {
+            auto itemConstraint = item->GetConstraint();
+            // TODO: We really should store the guid in a hashmap or something instead of looping over it for every item.
+            // TODO: A less intense green?
+            // TODO: Take into account the constraint offset.
+            for (const auto &constraint: m_matchedConstraints)
+                if (constraint.guid == itemConstraint.guid)
+                    return QBrush(Qt::green);
+        }
+    }
+
+    return QStandardItemModel::data(index, role);
+}
+
+WarpConstraintTableWidget::WarpConstraintTableWidget(QWidget *parent)
+{
+    QGridLayout *layout = new QGridLayout(this);
+    layout->setContentsMargins(2, 2, 2, 2);
+    layout->setVerticalSpacing(4);
+
+    m_table = new QTableView(this);
+    m_model = new WarpConstraintItemModel({"Constraint"}, this);
+    m_proxyModel = new GenericTextFilterModel(this);
+    m_proxyModel->setSourceModel(m_model);
+    m_table->setModel(m_proxyModel);
+
+    m_filterEdit = new FilterEdit(this);
+    m_filterView = new FilteredView(this, m_table, this, m_filterEdit);
+    m_filterView->setFilterPlaceholderText("Search constraints");
+
+    layout->addWidget(m_filterEdit, 0, 0, 1, 5);
+    layout->addWidget(m_table, 1, 0, 1, 5);
+
+    // Make the table look nice.
+    m_table->horizontalHeader()->setStretchLastSection(true);
+    m_table->verticalHeader()->hide();
+    m_table->setSelectionBehavior(QAbstractItemView::SelectRows);
+    m_table->setSelectionMode(QAbstractItemView::SingleSelection);
+    m_table->setEditTriggers(QAbstractItemView::NoEditTriggers);
+    m_table->setFocusPolicy(Qt::NoFocus);
+    m_table->setShowGrid(false);
+    m_table->setAlternatingRowColors(false);
+    m_table->setSortingEnabled(true);
+    // NOTE: We only have a single column right now, so disable the header.
+    m_table->horizontalHeader()->hide();
+    // Decrease row height to make it look nice.
+    m_table->verticalHeader()->setDefaultSectionSize(30);
+}
+
+void WarpConstraintTableWidget::SetConstraints(QVector<WarpConstraintItem *> constraints)
+{
+    // Clear matches as they are no longer valid.
+    m_model->clear();
+    m_model->setRowCount(0);
+
+    // Temporarily disable sorting so we can add rows faster
+    m_table->setModel(m_model);
+    m_table->setSortingEnabled(false);
+    m_table->setEnabled(false);
+
+    for (const auto &constraint: constraints)
+        m_model->AddItem(constraint);
+
+    // We are done, re-enable table.
+    m_table->setEnabled(true);
+    m_table->setModel(m_proxyModel);
+    m_table->setSortingEnabled(true);
+}
+
+void WarpConstraintTableWidget::SetMatchedConstraints(
+    const std::vector<Warp::Constraint> &analysisConstraints)
+{
+    m_model->SetMatchedConstraints(analysisConstraints);
+}
+
+void WarpConstraintTableWidget::setFilter(const std::string &filter)
+{
+    m_proxyModel->setFilterFixedString(QString::fromStdString(filter));
+    m_filterView->showFilter(QString::fromStdString(filter));
+}