aboutsummaryrefslogtreecommitdiffstats
path: root/gui
diff options
context:
space:
mode:
Diffstat (limited to 'gui')
-rw-r--r--gui/designwidget.cc356
-rw-r--r--gui/designwidget.h33
-rw-r--r--gui/treemodel.cc329
-rw-r--r--gui/treemodel.h93
4 files changed, 509 insertions, 302 deletions
diff --git a/gui/designwidget.cc b/gui/designwidget.cc
index d55c84e9..0f01b3c5 100644
--- a/gui/designwidget.cc
+++ b/gui/designwidget.cc
@@ -30,48 +30,14 @@
NEXTPNR_NAMESPACE_BEGIN
-class ElementTreeItem : public QTreeWidgetItem
+DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), selectionModel(nullptr)
{
- public:
- ElementTreeItem(ElementType t, QString str, QTreeWidgetItem *parent)
- : QTreeWidgetItem(parent, QStringList(str)), type(t)
- {
- this->setFlags(this->flags() & ~Qt::ItemIsSelectable);
- }
- virtual ~ElementTreeItem(){};
-
- ElementType getType() { return type; };
-
- private:
- ElementType type;
-};
-
-class IdStringTreeItem : public ElementTreeItem
-{
- public:
- IdStringTreeItem(IdString d, ElementType t, QString str, QTreeWidgetItem *parent) : ElementTreeItem(t, str, parent)
- {
- this->setFlags(this->flags() | Qt::ItemIsSelectable);
- this->data = d;
- }
- virtual ~IdStringTreeItem(){};
-
- IdString getData() { return this->data; };
-
- private:
- IdString data;
-};
-
-DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), nets_root(nullptr), cells_root(nullptr)
-{
-
- treeWidget = new QTreeWidget();
-
// Add tree view
- treeWidget->setColumnCount(1);
- treeWidget->setHeaderLabel("Items");
- treeWidget->setContextMenuPolicy(Qt::CustomContextMenu);
- treeWidget->setSelectionMode(QAbstractItemView::ExtendedSelection);
+ treeView = new QTreeView();
+ treeModel = new ContextTreeModel();
+ treeView->setModel(treeModel);
+ treeView->setContextMenuPolicy(Qt::CustomContextMenu);
+ treeView->setSelectionMode(QAbstractItemView::ExtendedSelection);
// Add property view
variantManager = new QtVariantPropertyManager(this);
@@ -96,7 +62,7 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
connect(actionFirst, &QAction::triggered, this, [this] {
history_ignore = true;
history_index = 0;
- treeWidget->setCurrentItem(history.at(history_index));
+ selectionModel->setCurrentIndex(history.at(history_index), QItemSelectionModel::ClearAndSelect);
updateButtons();
});
@@ -106,7 +72,7 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
connect(actionPrev, &QAction::triggered, this, [this] {
history_ignore = true;
history_index--;
- treeWidget->setCurrentItem(history.at(history_index));
+ selectionModel->setCurrentIndex(history.at(history_index), QItemSelectionModel::ClearAndSelect);
updateButtons();
});
@@ -116,7 +82,7 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
connect(actionNext, &QAction::triggered, this, [this] {
history_ignore = true;
history_index++;
- treeWidget->setCurrentItem(history.at(history_index));
+ selectionModel->setCurrentIndex(history.at(history_index), QItemSelectionModel::ClearAndSelect);
updateButtons();
});
@@ -126,7 +92,7 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
connect(actionLast, &QAction::triggered, this, [this] {
history_ignore = true;
history_index = int(history.size() - 1);
- treeWidget->setCurrentItem(history.at(history_index));
+ selectionModel->setCurrentIndex(history.at(history_index), QItemSelectionModel::ClearAndSelect);
updateButtons();
});
@@ -136,11 +102,11 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
connect(actionClear, &QAction::triggered, this, [this] {
history_index = -1;
history.clear();
- QTreeWidgetItem *clickItem = treeWidget->selectedItems().at(0);
- if (clickItem->parent()) {
- ElementType type = static_cast<ElementTreeItem *>(clickItem)->getType();
+ QModelIndex index = selectionModel->selectedIndexes().at(0);
+ if (index.isValid()) {
+ ElementType type = treeModel->nodeFromIndex(index)->type();
if (type != ElementType::NONE)
- addToHistory(treeWidget->selectedItems().at(0));
+ addToHistory(index);
}
updateButtons();
});
@@ -158,7 +124,7 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
vbox1->setSpacing(5);
vbox1->setContentsMargins(0, 0, 0, 0);
vbox1->addWidget(lineEdit);
- vbox1->addWidget(treeWidget);
+ vbox1->addWidget(treeView);
QWidget *toolbarWidget = new QWidget();
QHBoxLayout *hbox = new QHBoxLayout;
@@ -192,8 +158,11 @@ DesignWidget::DesignWidget(QWidget *parent) : QWidget(parent), ctx(nullptr), net
&DesignWidget::prepareMenuProperty);
connect(propertyEditor->treeWidget(), &QTreeWidget::itemDoubleClicked, this, &DesignWidget::onItemDoubleClicked);
- connect(treeWidget, SIGNAL(itemSelectionChanged()), SLOT(onItemSelectionChanged()));
- connect(treeWidget, &QTreeWidget::customContextMenuRequested, this, &DesignWidget::prepareMenuTree);
+ connect(treeView, &QTreeWidget::customContextMenuRequested, this, &DesignWidget::prepareMenuTree);
+
+ selectionModel = treeView->selectionModel();
+ connect(selectionModel, SIGNAL(selectionChanged(const QItemSelection &, const QItemSelection &)),
+ SLOT(onSelectionChanged(const QItemSelection &, const QItemSelection &)));
history_index = -1;
history_ignore = false;
@@ -219,7 +188,7 @@ void DesignWidget::updateButtons()
actionLast->setEnabled(history_index < (count - 1));
}
-void DesignWidget::addToHistory(QTreeWidgetItem *item)
+void DesignWidget::addToHistory(QModelIndex item)
{
if (!history_ignore) {
int count = int(history.size());
@@ -234,194 +203,38 @@ void DesignWidget::addToHistory(QTreeWidgetItem *item)
void DesignWidget::newContext(Context *ctx)
{
+ if (!ctx)
+ return;
+
highlightSelected.clear();
- treeWidget->clear();
- // reset pointers since they are not valid after clear
- nets_root = nullptr;
- cells_root = nullptr;
history_ignore = false;
history_index = -1;
history.clear();
updateButtons();
- for (int i = 0; i < 6; i++)
- nameToItem[i].clear();
-
+ highlightSelected.clear();
this->ctx = ctx;
-
- // Add bels to tree
- QTreeWidgetItem *bel_root = new QTreeWidgetItem(treeWidget);
- QMap<QString, QTreeWidgetItem *> bel_items;
- bel_root->setText(0, "Bels");
- bel_root->setFlags(bel_root->flags() & ~Qt::ItemIsSelectable);
- treeWidget->insertTopLevelItem(0, bel_root);
- if (ctx) {
- for (auto bel : ctx->getBels()) {
- auto id = ctx->getBelName(bel);
- QStringList items = QString(id.c_str(ctx)).split("/");
- QString name;
- QTreeWidgetItem *parent = nullptr;
- for (int i = 0; i < items.size(); i++) {
- if (!name.isEmpty())
- name += "/";
- name += items.at(i);
- if (!bel_items.contains(name)) {
- if (i == items.size() - 1)
- nameToItem[0].insert(name, new IdStringTreeItem(id, ElementType::BEL, items.at(i), parent));
- else
- bel_items.insert(name, new ElementTreeItem(ElementType::NONE, items.at(i), parent));
- }
- parent = bel_items[name];
- }
- }
- }
- for (auto bel : bel_items.toStdMap()) {
- bel_root->addChild(bel.second);
- }
- for (auto bel : nameToItem[0].toStdMap()) {
- bel_root->addChild(bel.second);
- }
-
- // Add wires to tree
- QTreeWidgetItem *wire_root = new QTreeWidgetItem(treeWidget);
- QMap<QString, QTreeWidgetItem *> wire_items;
- wire_root->setText(0, "Wires");
- wire_root->setFlags(wire_root->flags() & ~Qt::ItemIsSelectable);
- treeWidget->insertTopLevelItem(0, wire_root);
- if (ctx) {
- for (auto wire : ctx->getWires()) {
- auto id = ctx->getWireName(wire);
- QStringList items = QString(id.c_str(ctx)).split("/");
- QString name;
- QTreeWidgetItem *parent = nullptr;
- for (int i = 0; i < items.size(); i++) {
- if (!name.isEmpty())
- name += "/";
- name += items.at(i);
- if (!wire_items.contains(name)) {
- if (i == items.size() - 1)
- nameToItem[1].insert(name, new IdStringTreeItem(id, ElementType::WIRE, items.at(i), parent));
- else
- wire_items.insert(name, new ElementTreeItem(ElementType::NONE, items.at(i), parent));
- }
- parent = wire_items[name];
- }
- }
- }
- for (auto wire : wire_items.toStdMap()) {
- wire_root->addChild(wire.second);
- }
- for (auto wire : nameToItem[1].toStdMap()) {
- wire_root->addChild(wire.second);
- }
- // Add pips to tree
- QTreeWidgetItem *pip_root = new QTreeWidgetItem(treeWidget);
- QMap<QString, QTreeWidgetItem *> pip_items;
- pip_root->setText(0, "Pips");
- pip_root->setFlags(pip_root->flags() & ~Qt::ItemIsSelectable);
- treeWidget->insertTopLevelItem(0, pip_root);
-#ifndef ARCH_ECP5
- if (ctx) {
- for (auto pip : ctx->getPips()) {
- auto id = ctx->getPipName(pip);
- QStringList items = QString(id.c_str(ctx)).split("/");
- QString name;
- QTreeWidgetItem *parent = nullptr;
- for (int i = 0; i < items.size(); i++) {
- if (!name.isEmpty())
- name += "/";
- name += items.at(i);
- if (!pip_items.contains(name)) {
- if (i == items.size() - 1)
- nameToItem[2].insert(name, new IdStringTreeItem(id, ElementType::PIP, items.at(i), parent));
- else
- pip_items.insert(name, new ElementTreeItem(ElementType::NONE, items.at(i), parent));
- }
- parent = pip_items[name];
- }
- }
- }
- for (auto pip : pip_items.toStdMap()) {
- pip_root->addChild(pip.second);
- }
- for (auto pip : nameToItem[2].toStdMap()) {
- pip_root->addChild(pip.second);
- }
-#endif
-
- nets_root = new QTreeWidgetItem(treeWidget);
- nets_root->setText(0, "Nets");
- nets_root->setFlags(nets_root->flags() & ~Qt::ItemIsSelectable);
- treeWidget->insertTopLevelItem(0, nets_root);
-
- cells_root = new QTreeWidgetItem(treeWidget);
- cells_root->setText(0, "Cells");
- cells_root->setFlags(cells_root->flags() & ~Qt::ItemIsSelectable);
- treeWidget->insertTopLevelItem(0, cells_root);
-
+ treeModel->loadData(ctx);
updateTree();
}
void DesignWidget::updateTree()
{
- if (!ctx)
- return;
-
clearProperties();
- // treeWidget->setSortingEnabled(false);
-
- // Remove nets not existing any more
- QMap<QString, QTreeWidgetItem *>::iterator i = nameToItem[3].begin();
- while (i != nameToItem[3].end()) {
- QMap<QString, QTreeWidgetItem *>::iterator prev = i;
+ QMap<ContextTreeItem *, int>::iterator i = highlightSelected.begin();
+ while (i != highlightSelected.end()) {
+ QMap<ContextTreeItem *, int>::iterator prev = i;
++i;
- if (ctx->nets.find(ctx->id(prev.key().toStdString())) == ctx->nets.end()) {
- if (treeWidget->currentItem() == prev.value())
- treeWidget->setCurrentItem(nets_root);
- if (highlightSelected.contains(prev.value()))
- highlightSelected.remove(prev.value());
- delete prev.value();
- nameToItem[3].erase(prev);
+ if (prev.key()->type() == ElementType::NET && ctx->nets.find(prev.key()->id()) == ctx->nets.end()) {
+ highlightSelected.erase(prev);
}
- }
- // Add nets to tree
- for (auto &item : ctx->nets) {
- auto id = item.first;
- QString name = QString(id.c_str(ctx));
- if (!nameToItem[3].contains(name)) {
- IdStringTreeItem *newItem = new IdStringTreeItem(id, ElementType::NET, name, nullptr);
- nets_root->addChild(newItem);
- nameToItem[3].insert(name, newItem);
+ if (prev.key()->type() == ElementType::CELL && ctx->cells.find(prev.key()->id()) == ctx->cells.end()) {
+ highlightSelected.erase(prev);
}
}
- // Remove cells not existing any more
- i = nameToItem[4].begin();
- while (i != nameToItem[4].end()) {
- QMap<QString, QTreeWidgetItem *>::iterator prev = i;
- ++i;
- if (ctx->cells.find(ctx->id(prev.key().toStdString())) == ctx->cells.end()) {
- if (treeWidget->currentItem() == prev.value())
- treeWidget->setCurrentItem(cells_root);
- if (highlightSelected.contains(prev.value()))
- highlightSelected.remove(prev.value());
- delete prev.value();
- nameToItem[4].erase(prev);
- }
- }
- // Add cells to tree
- for (auto &item : ctx->cells) {
- auto id = item.first;
- QString name = QString(id.c_str(ctx));
- if (!nameToItem[4].contains(name)) {
- IdStringTreeItem *newItem = new IdStringTreeItem(id, ElementType::CELL, name, nullptr);
- cells_root->addChild(newItem);
- nameToItem[4].insert(name, newItem);
- }
- }
- // treeWidget->sortByColumn(0, Qt::AscendingOrder);
- // treeWidget->setSortingEnabled(true);
+ treeModel->updateData(ctx);
}
QtProperty *DesignWidget::addTopLevelProperty(const QString &id)
{
@@ -460,21 +273,6 @@ QString DesignWidget::getElementTypeName(ElementType type)
return "CELL";
return "";
}
-int DesignWidget::getElementIndex(ElementType type)
-{
- if (type == ElementType::BEL)
- return 0;
- if (type == ElementType::WIRE)
- return 1;
- if (type == ElementType::PIP)
- return 2;
- if (type == ElementType::NET)
- return 3;
- if (type == ElementType::CELL)
- return 4;
- return -1;
-}
-
ElementType DesignWidget::getElementTypeByName(QString type)
{
if (type == "BEL")
@@ -510,59 +308,58 @@ QtProperty *DesignWidget::addSubGroup(QtProperty *topItem, const QString &name)
void DesignWidget::onClickedBel(BelId bel, bool keep)
{
- QTreeWidgetItem *item = nameToItem[getElementIndex(ElementType::BEL)].value(ctx->getBelName(bel).c_str(ctx));
- treeWidget->setCurrentItem(item);
+ ContextTreeItem *item = treeModel->nodeForIdType(ElementType::BEL, ctx->getBelName(bel).c_str(ctx));
+ selectionModel->setCurrentIndex(treeModel->indexFromNode(item),
+ keep ? QItemSelectionModel::Select : QItemSelectionModel::ClearAndSelect);
Q_EMIT selected(getDecals(ElementType::BEL, ctx->getBelName(bel)), keep);
}
void DesignWidget::onClickedWire(WireId wire, bool keep)
{
- QTreeWidgetItem *item = nameToItem[getElementIndex(ElementType::WIRE)].value(ctx->getWireName(wire).c_str(ctx));
- treeWidget->setCurrentItem(item);
+ ContextTreeItem *item = treeModel->nodeForIdType(ElementType::WIRE, ctx->getWireName(wire).c_str(ctx));
+ selectionModel->setCurrentIndex(treeModel->indexFromNode(item),
+ keep ? QItemSelectionModel::Select : QItemSelectionModel::ClearAndSelect);
Q_EMIT selected(getDecals(ElementType::WIRE, ctx->getWireName(wire)), keep);
}
void DesignWidget::onClickedPip(PipId pip, bool keep)
{
- QTreeWidgetItem *item = nameToItem[getElementIndex(ElementType::PIP)].value(ctx->getPipName(pip).c_str(ctx));
- treeWidget->setCurrentItem(item);
+ ContextTreeItem *item = treeModel->nodeForIdType(ElementType::PIP, ctx->getPipName(pip).c_str(ctx));
+ selectionModel->setCurrentIndex(treeModel->indexFromNode(item),
+ keep ? QItemSelectionModel::Select : QItemSelectionModel::ClearAndSelect);
Q_EMIT selected(getDecals(ElementType::PIP, ctx->getPipName(pip)), keep);
}
-void DesignWidget::onItemSelectionChanged()
+void DesignWidget::onSelectionChanged(const QItemSelection &, const QItemSelection &)
{
- if (treeWidget->selectedItems().size() == 0)
+ if (selectionModel->selectedIndexes().size() == 0)
return;
- if (treeWidget->selectedItems().size() > 1) {
+ if (selectionModel->selectedIndexes().size() > 1) {
std::vector<DecalXY> decals;
- for (auto clickItem : treeWidget->selectedItems()) {
- IdString value = static_cast<IdStringTreeItem *>(clickItem)->getData();
- ElementType type = static_cast<ElementTreeItem *>(clickItem)->getType();
- std::vector<DecalXY> d = getDecals(type, value);
+ for (auto index : selectionModel->selectedIndexes()) {
+ ContextTreeItem *item = treeModel->nodeFromIndex(index);
+ std::vector<DecalXY> d = getDecals(item->type(), item->id());
std::move(d.begin(), d.end(), std::back_inserter(decals));
}
Q_EMIT selected(decals, false);
return;
}
-
- QTreeWidgetItem *clickItem = treeWidget->selectedItems().at(0);
-
- if (!clickItem->parent())
+ QModelIndex index = selectionModel->selectedIndexes().at(0);
+ if (!index.isValid())
return;
+ ContextTreeItem *clickItem = treeModel->nodeFromIndex(index);
- ElementType type = static_cast<ElementTreeItem *>(clickItem)->getType();
- if (type == ElementType::NONE) {
+ ElementType type = clickItem->type();
+ if (type == ElementType::NONE)
return;
- }
-
std::vector<DecalXY> decals;
- addToHistory(clickItem);
+ addToHistory(index);
clearProperties();
- IdString c = static_cast<IdStringTreeItem *>(clickItem)->getData();
+ IdString c = clickItem->id();
Q_EMIT selected(getDecals(type, c), false);
if (type == ElementType::BEL) {
@@ -799,7 +596,7 @@ std::vector<DecalXY> DesignWidget::getDecals(ElementType type, IdString value)
return decals;
}
-void DesignWidget::updateHighlightGroup(QList<QTreeWidgetItem *> items, int group)
+void DesignWidget::updateHighlightGroup(QList<ContextTreeItem *> items, int group)
{
const bool shouldClear = items.size() == 1;
for (auto item : items) {
@@ -814,9 +611,7 @@ void DesignWidget::updateHighlightGroup(QList<QTreeWidgetItem *> items, int grou
std::vector<DecalXY> decals[8];
for (auto it : highlightSelected.toStdMap()) {
- ElementType type = static_cast<ElementTreeItem *>(it.first)->getType();
- IdString value = static_cast<IdStringTreeItem *>(it.first)->getData();
- std::vector<DecalXY> d = getDecals(type, value);
+ std::vector<DecalXY> d = getDecals(it.first->type(), it.first->id());
std::move(d.begin(), d.end(), std::back_inserter(decals[it.second]));
}
for (int i = 0; i < 8; i++)
@@ -826,7 +621,7 @@ void DesignWidget::updateHighlightGroup(QList<QTreeWidgetItem *> items, int grou
void DesignWidget::prepareMenuProperty(const QPoint &pos)
{
QTreeWidget *tree = propertyEditor->treeWidget();
- QList<QTreeWidgetItem *> items;
+ QList<ContextTreeItem *> items;
for (auto itemContextMenu : tree->selectedItems()) {
QtBrowserItem *browserItem = propertyEditor->itemToBrowserItem(itemContextMenu);
if (!browserItem)
@@ -836,11 +631,11 @@ void DesignWidget::prepareMenuProperty(const QPoint &pos)
if (type == ElementType::NONE)
continue;
IdString value = ctx->id(selectedProperty->valueText().toStdString());
- items.append(nameToItem[getElementIndex(type)].value(value.c_str(ctx)));
+ items.append(treeModel->nodeForIdType(type, value.c_str(ctx)));
}
int selectedIndex = -1;
if (items.size() == 1) {
- QTreeWidgetItem *item = items.at(0);
+ ContextTreeItem *item = items.at(0);
if (highlightSelected.contains(item))
selectedIndex = highlightSelected[item];
}
@@ -850,9 +645,7 @@ void DesignWidget::prepareMenuProperty(const QPoint &pos)
connect(selectAction, &QAction::triggered, this, [this, items] {
std::vector<DecalXY> decals;
for (auto clickItem : items) {
- IdString value = static_cast<IdStringTreeItem *>(clickItem)->getData();
- ElementType type = static_cast<ElementTreeItem *>(clickItem)->getType();
- std::vector<DecalXY> d = getDecals(type, value);
+ std::vector<DecalXY> d = getDecals(clickItem->type(), clickItem->id());
std::move(d.begin(), d.end(), std::back_inserter(decals));
}
Q_EMIT selected(decals, false);
@@ -878,12 +671,18 @@ void DesignWidget::prepareMenuProperty(const QPoint &pos)
void DesignWidget::prepareMenuTree(const QPoint &pos)
{
- if (treeWidget->selectedItems().size() == 0)
- return;
int selectedIndex = -1;
- QList<QTreeWidgetItem *> items = treeWidget->selectedItems();
- if (treeWidget->selectedItems().size() == 1) {
- QTreeWidgetItem *item = treeWidget->selectedItems().at(0);
+
+ if (selectionModel->selectedIndexes().size() == 0)
+ return;
+
+ QList<ContextTreeItem *> items;
+ for (auto index : selectionModel->selectedIndexes()) {
+ ContextTreeItem *item = treeModel->nodeFromIndex(index);
+ items.append(item);
+ }
+ if (items.size() == 1) {
+ ContextTreeItem *item = items.at(0);
if (highlightSelected.contains(item))
selectedIndex = highlightSelected[item];
}
@@ -902,17 +701,16 @@ void DesignWidget::prepareMenuTree(const QPoint &pos)
action->setChecked(true);
connect(action, &QAction::triggered, this, [this, i, items] { updateHighlightGroup(items, i); });
}
- menu.exec(treeWidget->mapToGlobal(pos));
+ menu.exec(treeView->mapToGlobal(pos));
}
void DesignWidget::onItemDoubleClicked(QTreeWidgetItem *item, int column)
{
QtProperty *selectedProperty = propertyEditor->itemToBrowserItem(item)->property();
ElementType type = getElementTypeByName(selectedProperty->propertyId());
- QString value = selectedProperty->valueText();
- int index = getElementIndex(type);
- if (type != ElementType::NONE && nameToItem[index].contains(value))
- treeWidget->setCurrentItem(nameToItem[index].value(value));
+ ContextTreeItem *it = treeModel->nodeForIdType(type, selectedProperty->valueText());
+ if (it)
+ selectionModel->setCurrentIndex(treeModel->indexFromNode(it), QItemSelectionModel::ClearAndSelect);
}
NEXTPNR_NAMESPACE_END
diff --git a/gui/designwidget.h b/gui/designwidget.h
index 60291cf3..bbd188cd 100644
--- a/gui/designwidget.h
+++ b/gui/designwidget.h
@@ -20,27 +20,17 @@
#ifndef DESIGNWIDGET_H
#define DESIGNWIDGET_H
-#include <QTreeWidget>
+#include <QTreeView>
#include <QVariant>
#include "nextpnr.h"
#include "qtgroupboxpropertybrowser.h"
#include "qtpropertymanager.h"
#include "qttreepropertybrowser.h"
#include "qtvariantproperty.h"
+#include "treemodel.h"
NEXTPNR_NAMESPACE_BEGIN
-enum class ElementType
-{
- NONE,
- BEL,
- WIRE,
- PIP,
- NET,
- CELL,
- GROUP
-};
-
class DesignWidget : public QWidget
{
Q_OBJECT
@@ -59,9 +49,9 @@ class DesignWidget : public QWidget
ElementType getElementTypeByName(QString type);
int getElementIndex(ElementType type);
void updateButtons();
- void addToHistory(QTreeWidgetItem *item);
+ void addToHistory(QModelIndex item);
std::vector<DecalXY> getDecals(ElementType type, IdString value);
- void updateHighlightGroup(QList<QTreeWidgetItem *> item, int group);
+ void updateHighlightGroup(QList<ContextTreeItem *> item, int group);
Q_SIGNALS:
void info(std::string text);
void selected(std::vector<DecalXY> decal, bool keep);
@@ -70,7 +60,7 @@ class DesignWidget : public QWidget
private Q_SLOTS:
void prepareMenuProperty(const QPoint &pos);
void prepareMenuTree(const QPoint &pos);
- void onItemSelectionChanged();
+ void onSelectionChanged(const QItemSelection &selected, const QItemSelection &deselected);
void onItemDoubleClicked(QTreeWidgetItem *item, int column);
public Q_SLOTS:
void newContext(Context *ctx);
@@ -82,8 +72,9 @@ class DesignWidget : public QWidget
private:
Context *ctx;
- QTreeWidget *treeWidget;
-
+ QTreeView *treeView;
+ QItemSelectionModel *selectionModel;
+ ContextTreeModel *treeModel;
QtVariantPropertyManager *variantManager;
QtVariantPropertyManager *readOnlyManager;
QtGroupPropertyManager *groupManager;
@@ -93,14 +84,10 @@ class DesignWidget : public QWidget
QMap<QtProperty *, QString> propertyToId;
QMap<QString, QtProperty *> idToProperty;
- QMap<QString, QTreeWidgetItem *> nameToItem[6];
- std::vector<QTreeWidgetItem *> history;
+ std::vector<QModelIndex> history;
int history_index;
bool history_ignore;
- QTreeWidgetItem *nets_root;
- QTreeWidgetItem *cells_root;
-
QAction *actionFirst;
QAction *actionPrev;
QAction *actionNext;
@@ -108,7 +95,7 @@ class DesignWidget : public QWidget
QAction *actionClear;
QColor highlightColors[8];
- QMap<QTreeWidgetItem *, int> highlightSelected;
+ QMap<ContextTreeItem *, int> highlightSelected;
};
NEXTPNR_NAMESPACE_END
diff --git a/gui/treemodel.cc b/gui/treemodel.cc
new file mode 100644
index 00000000..59391f02
--- /dev/null
+++ b/gui/treemodel.cc
@@ -0,0 +1,329 @@
+/*
+ * nextpnr -- Next Generation Place and Route
+ *
+ * Copyright (C) 2018 Miodrag Milanovic <miodrag@symbioticeda.com>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ */
+
+#include "treemodel.h"
+
+NEXTPNR_NAMESPACE_BEGIN
+
+static bool contextTreeItemLessThan(const ContextTreeItem *v1, const ContextTreeItem *v2)
+ {
+ return v1->name() < v2->name();
+ }
+
+ContextTreeItem::ContextTreeItem() { parentNode = nullptr; }
+
+ContextTreeItem::ContextTreeItem(QString name)
+ : parentNode(nullptr), itemId(IdString()), itemType(ElementType::NONE), itemName(name)
+{
+}
+
+ContextTreeItem::ContextTreeItem(IdString id, ElementType type, QString name)
+ : parentNode(nullptr), itemId(id), itemType(type), itemName(name)
+{
+}
+
+ContextTreeItem::~ContextTreeItem()
+{
+ if (parentNode)
+ parentNode->children.removeOne(this);
+ qDeleteAll(children);
+}
+void ContextTreeItem::addChild(ContextTreeItem *item)
+{
+ item->parentNode = this;
+ children.append(item);
+}
+
+void ContextTreeItem::sort()
+{
+ for (auto item : children)
+ if (item->count()>1) item->sort();
+ qSort(children.begin(), children.end(), contextTreeItemLessThan);
+}
+
+ContextTreeModel::ContextTreeModel(QObject *parent) : QAbstractItemModel(parent) { root = new ContextTreeItem(); }
+
+ContextTreeModel::~ContextTreeModel() { delete root; }
+
+void ContextTreeModel::loadData(Context *ctx)
+{
+ if (!ctx)
+ return;
+
+ beginResetModel();
+
+ delete root;
+ root = new ContextTreeItem();
+
+ for (int i = 0; i < 6; i++)
+ nameToItem[i].clear();
+
+ IdString none;
+
+ ContextTreeItem *bels_root = new ContextTreeItem("Bels");
+ root->addChild(bels_root);
+ QMap<QString, ContextTreeItem *> bel_items;
+
+ // Add bels to tree
+ for (auto bel : ctx->getBels()) {
+ IdString id = ctx->getBelName(bel);
+ QStringList items = QString(id.c_str(ctx)).split("/");
+ QString name;
+ ContextTreeItem *parent = bels_root;
+ for (int i = 0; i < items.size(); i++) {
+ if (!name.isEmpty())
+ name += "/";
+ name += items.at(i);
+ if (!bel_items.contains(name)) {
+ if (i == items.size() - 1) {
+ ContextTreeItem *item = new ContextTreeItem(id, ElementType::BEL, items.at(i));
+ parent->addChild(item);
+ nameToItem[0].insert(name, item);
+ } else {
+ ContextTreeItem *item = new ContextTreeItem(none, ElementType::NONE, items.at(i));
+ parent->addChild(item);
+ bel_items.insert(name, item);
+ }
+ }
+ parent = bel_items[name];
+ }
+ }
+ bels_root->sort();
+
+ ContextTreeItem *wire_root = new ContextTreeItem("Wires");
+ root->addChild(wire_root);
+ QMap<QString, ContextTreeItem *> wire_items;
+
+ // Add wires to tree
+ for (auto wire : ctx->getWires()) {
+ auto id = ctx->getWireName(wire);
+ QStringList items = QString(id.c_str(ctx)).split("/");
+ QString name;
+ ContextTreeItem *parent = wire_root;
+ for (int i = 0; i < items.size(); i++) {
+ if (!name.isEmpty())
+ name += "/";
+ name += items.at(i);
+ if (!wire_items.contains(name)) {
+ if (i == items.size() - 1) {
+ ContextTreeItem *item = new ContextTreeItem(id, ElementType::WIRE, items.at(i));
+ parent->addChild(item);
+ nameToItem[1].insert(name, item);
+ } else {
+ ContextTreeItem *item = new ContextTreeItem(none, ElementType::NONE, items.at(i));
+ parent->addChild(item);
+ wire_items.insert(name, item);
+ }
+ }
+ parent = wire_items[name];
+ }
+ }
+ wire_root->sort();
+
+ ContextTreeItem *pip_root = new ContextTreeItem("Pips");
+ root->addChild(pip_root);
+ QMap<QString, ContextTreeItem *> pip_items;
+
+ // Add pips to tree
+ for (auto pip : ctx->getPips()) {
+ auto id = ctx->getPipName(pip);
+ QStringList items = QString(id.c_str(ctx)).split("/");
+ QString name;
+ ContextTreeItem *parent = pip_root;
+ for (int i = 0; i < items.size(); i++) {
+ if (!name.isEmpty())
+ name += "/";
+ name += items.at(i);
+ if (!pip_items.contains(name)) {
+ if (i == items.size() - 1) {
+ ContextTreeItem *item = new ContextTreeItem(id, ElementType::PIP, items.at(i));
+ parent->addChild(item);
+ nameToItem[2].insert(name, item);
+ } else {
+ ContextTreeItem *item = new ContextTreeItem(none, ElementType::NONE, items.at(i));
+ parent->addChild(item);
+ pip_items.insert(name, item);
+ }
+ }
+ parent = pip_items[name];
+ }
+ }
+ pip_root->sort();
+
+ nets_root = new ContextTreeItem("Nets");
+ root->addChild(nets_root);
+
+ cells_root = new ContextTreeItem("Cells");
+ root->addChild(cells_root);
+
+ endResetModel();
+}
+
+void ContextTreeModel::updateData(Context *ctx)
+{
+ if (!ctx)
+ return;
+
+ beginResetModel();
+
+ //QModelIndex nets_index = indexFromNode(nets_root);
+ // Remove nets not existing any more
+ QMap<QString, ContextTreeItem *>::iterator i = nameToItem[3].begin();
+ while (i != nameToItem[3].end()) {
+ QMap<QString, ContextTreeItem *>::iterator prev = i;
+ ++i;
+ if (ctx->nets.find(ctx->id(prev.key().toStdString())) == ctx->nets.end()) {
+ //int pos = prev.value()->parent()->indexOf(prev.value());
+ //beginRemoveRows(nets_index, pos, pos);
+ delete prev.value();
+ nameToItem[3].erase(prev);
+ //endRemoveRows();
+ }
+ }
+ // Add nets to tree
+ for (auto &item : ctx->nets) {
+ auto id = item.first;
+ QString name = QString(id.c_str(ctx));
+ if (!nameToItem[3].contains(name)) {
+ //beginInsertRows(nets_index, nets_root->count() + 1, nets_root->count() + 1);
+ ContextTreeItem *newItem = new ContextTreeItem(id, ElementType::NET, name);
+ nets_root->addChild(newItem);
+ nameToItem[3].insert(name, newItem);
+ //endInsertRows();
+ }
+ }
+
+ nets_root->sort();
+
+ //QModelIndex cell_index = indexFromNode(cells_root);
+ // Remove cells not existing any more
+ i = nameToItem[4].begin();
+ while (i != nameToItem[4].end()) {
+ QMap<QString, ContextTreeItem *>::iterator prev = i;
+ ++i;
+ if (ctx->cells.find(ctx->id(prev.key().toStdString())) == ctx->cells.end()) {
+ //int pos = prev.value()->parent()->indexOf(prev.value());
+ //beginRemoveRows(cell_index, pos, pos);
+ delete prev.value();
+ nameToItem[4].erase(prev);
+ //endRemoveRows();
+ }
+ }
+ // Add cells to tree
+ for (auto &item : ctx->cells) {
+ auto id = item.first;
+ QString name = QString(id.c_str(ctx));
+ if (!nameToItem[4].contains(name)) {
+ //beginInsertRows(cell_index, cells_root->count() + 1, cells_root->count() + 1);
+ ContextTreeItem *newItem = new ContextTreeItem(id, ElementType::CELL, name);
+ cells_root->addChild(newItem);
+ nameToItem[4].insert(name, newItem);
+ //endInsertRows();
+ }
+ }
+
+ cells_root->sort();
+
+ endResetModel();
+}
+
+int ContextTreeModel::rowCount(const QModelIndex &parent) const { return nodeFromIndex(parent)->count(); }
+
+int ContextTreeModel::columnCount(const QModelIndex &parent) const { return 1; }
+
+QModelIndex ContextTreeModel::index(int row, int column, const QModelIndex &parent) const
+{
+ ContextTreeItem *node = nodeFromIndex(parent);
+ if (row >= node->count())
+ return QModelIndex();
+ return createIndex(row, column, node->at(row));
+}
+
+QModelIndex ContextTreeModel::parent(const QModelIndex &child) const
+{
+ ContextTreeItem *parent = nodeFromIndex(child)->parent();
+ if (parent == root)
+ return QModelIndex();
+ ContextTreeItem *node = parent->parent();
+ return createIndex(node->indexOf(parent), 0, parent);
+}
+
+QVariant ContextTreeModel::data(const QModelIndex &index, int role) const
+{
+ if (index.column() != 0)
+ return QVariant();
+ if (role != Qt::DisplayRole)
+ return QVariant();
+ ContextTreeItem *node = nodeFromIndex(index);
+ return node->name();
+}
+
+QVariant ContextTreeModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+ Q_UNUSED(section);
+ if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
+ return QString("Items");
+
+ return QVariant();
+}
+
+ContextTreeItem *ContextTreeModel::nodeFromIndex(const QModelIndex &idx) const
+{
+ if (idx.isValid())
+ return (ContextTreeItem *)idx.internalPointer();
+ return root;
+}
+
+static int getElementIndex(ElementType type)
+{
+ if (type == ElementType::BEL)
+ return 0;
+ if (type == ElementType::WIRE)
+ return 1;
+ if (type == ElementType::PIP)
+ return 2;
+ if (type == ElementType::NET)
+ return 3;
+ if (type == ElementType::CELL)
+ return 4;
+ return -1;
+}
+
+ContextTreeItem *ContextTreeModel::nodeForIdType(const ElementType type, const QString name) const
+{
+ int index = getElementIndex(type);
+ if (type != ElementType::NONE && nameToItem[index].contains(name))
+ return nameToItem[index].value(name);
+ return nullptr;
+}
+
+QModelIndex ContextTreeModel::indexFromNode(ContextTreeItem *node)
+{
+ ContextTreeItem *parent = node->parent();
+ if (parent == root)
+ return QModelIndex();
+ return createIndex(parent->indexOf(node), 0, node);
+}
+
+Qt::ItemFlags ContextTreeModel::flags(const QModelIndex &index) const
+{
+ ContextTreeItem *node = nodeFromIndex(index);
+ return Qt::ItemIsEnabled | (node->type() != ElementType::NONE ? Qt::ItemIsSelectable : Qt::NoItemFlags);
+}
+NEXTPNR_NAMESPACE_END \ No newline at end of file
diff --git a/gui/treemodel.h b/gui/treemodel.h
new file mode 100644
index 00000000..a85c290a
--- /dev/null
+++ b/gui/treemodel.h
@@ -0,0 +1,93 @@
+/*
+ * nextpnr -- Next Generation Place and Route
+ *
+ * Copyright (C) 2018 Miodrag Milanovic <miodrag@symbioticeda.com>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ */
+
+#ifndef TREEMODEL_H
+#define TREEMODEL_H
+
+#include <QAbstractItemModel>
+#include "nextpnr.h"
+
+NEXTPNR_NAMESPACE_BEGIN
+
+enum class ElementType
+{
+ NONE,
+ BEL,
+ WIRE,
+ PIP,
+ NET,
+ CELL,
+ GROUP
+};
+
+class ContextTreeItem
+{
+ public:
+ ContextTreeItem();
+ ContextTreeItem(QString name);
+ ContextTreeItem(IdString id, ElementType type, QString name);
+ ~ContextTreeItem();
+
+ void addChild(ContextTreeItem *item);
+ int indexOf(ContextTreeItem *n) const { return children.indexOf(n); }
+ ContextTreeItem *at(int idx) const { return children.at(idx); }
+ int count() const { return children.count(); }
+ ContextTreeItem *parent() const { return parentNode; }
+ IdString id() const { return itemId; }
+ ElementType type() const { return itemType; }
+ QString name() const { return itemName; }
+ void sort();
+ private:
+ ContextTreeItem *parentNode;
+ QList<ContextTreeItem *> children;
+ IdString itemId;
+ ElementType itemType;
+ QString itemName;
+};
+
+class ContextTreeModel : public QAbstractItemModel
+{
+ public:
+ ContextTreeModel(QObject *parent = nullptr);
+ ~ContextTreeModel();
+
+ void loadData(Context *ctx);
+ void updateData(Context *ctx);
+ ContextTreeItem *nodeFromIndex(const QModelIndex &idx) const;
+ QModelIndex indexFromNode(ContextTreeItem *node);
+ ContextTreeItem *nodeForIdType(const ElementType type, const QString name) const;
+ // Override QAbstractItemModel methods
+ int rowCount(const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+ int columnCount(const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+ QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const Q_DECL_OVERRIDE;
+ QModelIndex parent(const QModelIndex &child) const Q_DECL_OVERRIDE;
+ QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const Q_DECL_OVERRIDE;
+ QVariant headerData(int section, Qt::Orientation orientation, int role) const Q_DECL_OVERRIDE;
+ Qt::ItemFlags flags(const QModelIndex &index) const Q_DECL_OVERRIDE;
+
+ private:
+ ContextTreeItem *root;
+ QMap<QString, ContextTreeItem *> nameToItem[6];
+ ContextTreeItem *nets_root;
+ ContextTreeItem *cells_root;
+};
+
+NEXTPNR_NAMESPACE_END
+
+#endif // TREEMODEL_H