reveresed logic for enabling main file, and made tests link arch files

1 files changed, 31 insertions, 6 deletions

diff --git a/gui/mainwindow.cc b/gui/mainwindow.cc
index c436fd6c..bae02813 100644
--- a/gui/mainwindow.cc
+++ b/gui/mainwindow.cc
@@ -18,16 +18,41 @@ MainWindow::MainWindow(Design *_design, QWidget *parent)
     ui->setupUi(this);
     ui->treeWidget->setColumnCount(1);
     ui->treeWidget->setHeaderLabel(QString("Items"));
-    QTreeWidgetItem *belroot = new QTreeWidgetItem(ui->treeWidget);
-    belroot->setText(0, QString("Bels"));
-    ui->treeWidget->insertTopLevelItem(0, belroot);
-    QList<QTreeWidgetItem *> items;
+
+    QTreeWidgetItem *bel_root = new QTreeWidgetItem(ui->treeWidget);
+    bel_root->setText(0, QString("Bels"));
+    ui->treeWidget->insertTopLevelItem(0, bel_root);
+    QList<QTreeWidgetItem *> bel_items;
     for (auto bel : design->chip.getBels()) {
         auto name = design->chip.getBelName(bel);
-        items.append(new QTreeWidgetItem((QTreeWidget *)nullptr,
+        bel_items.append(new QTreeWidgetItem((QTreeWidget *)nullptr,
+                                         QStringList(QString(name.c_str()))));
+    }
+    bel_root->addChildren(bel_items);
+
+    QTreeWidgetItem *wire_root = new QTreeWidgetItem(ui->treeWidget);
+    QList<QTreeWidgetItem *> wire_items;
+    wire_root->setText(0, QString("Wires"));
+    ui->treeWidget->insertTopLevelItem(0, wire_root);
+    for (auto wire : design->chip.getWires()) {
+        auto name = design->chip.getWireName(wire);
+        wire_items.append(new QTreeWidgetItem((QTreeWidget *)nullptr,
                                          QStringList(QString(name.c_str()))));
     }
-    belroot->addChildren(items);
+    wire_root->addChildren(wire_items);
+
+    QTreeWidgetItem *pip_root = new QTreeWidgetItem(ui->treeWidget);
+    QList<QTreeWidgetItem *> pip_items;
+    pip_root->setText(0, QString("Pips"));
+    ui->treeWidget->insertTopLevelItem(0, pip_root);
+    for (auto pip : design->chip.getPips()) {
+        auto name = design->chip.getPipName(pip);
+        pip_items.append(new QTreeWidgetItem((QTreeWidget *)nullptr,
+                                         QStringList(QString(name.c_str()))));
+    }
+    pip_root->addChildren(pip_items);
+
+
     PyImport_ImportModule("emb");
 
     write = [this](std::string s) {