module_dialog.cpp

Go to the documentation of this file.

#include "gui/module_dialog/module_dialog.h"
#include "gui/module_dialog/module_select_model.h"
#include "gui/gui_globals.h"
#include "gui/module_model/module_model.h"
#include "gui/module_model/module_proxy_model.h"
#include "gui/module_model/module_item.h"
#include "gui/module_widget/module_widget.h"
#include "gui/graph_tab_widget/graph_tab_widget.h"
#include "gui/searchbar/searchbar.h"
#include "gui/content_manager/content_manager.h"
#include "gui/settings/settings_items/settings_item_keybind.h"
#include "gui/user_action/action_add_items_to_object.h"
#include "hal_core/utilities/log.h"
 
#include <QDialogButtonBox>
#include <QGridLayout>
#include <QPushButton>
#include <QTabWidget>
#include <QTableView>
#include <QTreeView>
#include <QMessageBox>
#include <QApplication>
 
namespace hal {
    ModuleDialog::ModuleDialog(const QSet<u32>& excludeIds, const QString &title, bool omitCreateNew, ModuleSelectReceiver* receiver, QWidget* parent)
        : QDialog(parent),
          mSelectedId(0),
          mExcludeIds(excludeIds),
          mLastUsed(nullptr),
          mSearchbar(new Searchbar(this)),
          mNewModule(false),
          mReceiver(receiver),
          mPickerModeActivated(false),
          mWindowTitle(title)
    {
        setWindowTitle(mWindowTitle + " …");
        QGridLayout* layout = new QGridLayout(this);
 
        mButtonPick = new QPushButton("Pick module from graph", this);
        if (mReceiver)
            connect(mButtonPick, &QPushButton::pressed, this, &ModuleDialog::handlePickFromGraph);
        else
            mButtonPick->setDisabled(true);
 
 
        QPushButton* butSearch = new QPushButton("Search", this);
        connect(butSearch, &QPushButton::pressed, this, &ModuleDialog::handleToggleSearchbar);
 
        if (omitCreateNew)
        {
            layout->addWidget(mButtonPick, 0, 0);
            layout->addWidget(butSearch, 0, 1);
        }
        else {
            QPushButton* butNew = new QPushButton("Create new module", this);
            connect(butNew, &QPushButton::pressed, this, &ModuleDialog::handleCreateNewModule);
            layout->addWidget(butNew, 0, 0);
            layout->addWidget(mButtonPick, 0, 1);
            layout->addWidget(butSearch, 0, 2);
        }
 
 
 
        layout->addWidget(mSearchbar, 1, 0, 1, 3);
        mTabWidget = new QTabWidget(this);
        mTreeView  = new QTreeView(mTabWidget);
        mTabWidget->addTab(mTreeView, "Module tree");
 
        mTableView = new ModuleSelectView(false, mSearchbar, &mExcludeIds, mTabWidget);
 
        connect(mTableView, &ModuleSelectView::moduleSelected, this, &ModuleDialog::handleTableSelection);
        mTabWidget->addTab(mTableView, "Module list");
 
        if (!ModuleSelectHistory::instance()->isEmpty())
        {
            mLastUsed = new ModuleSelectView(true, mSearchbar, &mExcludeIds, mTabWidget);
            if (mLastUsed->model()->rowCount())
            {
                connect(mLastUsed, &ModuleSelectView::moduleSelected, this, &ModuleDialog::handleTableSelection);
                mTabWidget->addTab(mLastUsed, "Recent selection");
            }
            else
            {
                delete mLastUsed;
                mLastUsed = nullptr;
            }
 
        }
 
        layout->addWidget(mTabWidget, 2, 0, 1, 3);
 
        mModuleTreeProxyModel = new ModuleProxyModel(this);
        mModuleTreeProxyModel->setFilterKeyColumn(-1);
        mModuleTreeProxyModel->setDynamicSortFilter(true);
        mModuleTreeProxyModel->setSourceModel(gContentManager->getModuleWidget()->getModuleModel());
        mTreeView->setModel(mModuleTreeProxyModel);
        mTreeView->expandAll();
 
        mButtonBox = new QDialogButtonBox(QDialogButtonBox::Ok|QDialogButtonBox::Cancel, this);
        layout->addWidget(mButtonBox, 3, 0, 1, 3, Qt::AlignHCenter);
 
        mToggleSearchbar = new QAction(this);
        mToggleSearchbar->setShortcut(QKeySequence(ContentManager::sSettingSearch->value().toString()));
        addAction(mToggleSearchbar);
 
        mTabWidget->setCurrentIndex(0);
        enableButtons();
        mSearchbar->hide();
 
        //get column names for searchbar
        if(mTabWidget->currentWidget() == mTreeView)
            mSearchbar->setColumnNames(mModuleTreeProxyModel->getColumnNames());
        else
            mSearchbar->setColumnNames(static_cast<ModuleSelectProxy*>(mTableView->model())->getColumnNames());
 
        connect(mTabWidget, &QTabWidget::currentChanged, this, &ModuleDialog::handleCurrentTabChanged);
        connect(mToggleSearchbar, &QAction::triggered, this, &ModuleDialog::handleToggleSearchbar);
// TODO textEdited        connect(mSearchbar, &Searchbar::textEdited, this, &ModuleDialog::filter);
        connect(ContentManager::sSettingSearch, &SettingsItemKeybind::keySequenceChanged, this, &ModuleDialog::keybindToggleSearchbar);
        connect(mButtonBox, &QDialogButtonBox::accepted, this, &QDialog::accept);
        connect(mButtonBox, &QDialogButtonBox::rejected, this, &QDialog::reject);
        connect(mTreeView, &QTreeView::doubleClicked, this, &ModuleDialog::handleTreeDoubleClick);
        connect(mTreeView->selectionModel(), &QItemSelectionModel::selectionChanged, this, &ModuleDialog::handleTreeSelectionChanged);
 
        connect(mSearchbar, &Searchbar::triggerNewSearch, mModuleTreeProxyModel, &ModuleProxyModel::startSearch);
        connect(mSearchbar, &Searchbar::triggerNewSearch, static_cast<ModuleSelectProxy*>(mTableView->model()), &ModuleSelectProxy::startSearch);
    }
 
    void ModuleDialog::enableButtons()
    {
        mButtonBox->button(QDialogButtonBox::Ok)->setEnabled(mSelectedId>0);
        mButtonPick->setEnabled(mTableView->model()->rowCount()>0);
        QString target = "…";
        if (mSelectedId>0)
        {
            Module* m = gNetlist->get_module_by_id(mSelectedId);
            if (m) target = QString("%1[%2]").arg(QString::fromStdString(m->get_name())).arg(mSelectedId);
        }
        setWindowTitle(mWindowTitle + " " + target);
    }
 
     u32 ModuleDialog::treeModuleId(const QModelIndex& index)
     {
        ModuleModel* treeModel = static_cast<ModuleModel*>(mModuleTreeProxyModel->sourceModel());
        Q_ASSERT(treeModel);
        QModelIndex sourceIndex = mModuleTreeProxyModel->mapToSource(index);
        if (!sourceIndex.isValid()) return 0;
        ModuleItem* item = treeModel->getItem(sourceIndex);
        if (!item) return 0;
        return item->id();
     }
 
     void ModuleDialog::handleTreeSelectionChanged(const QItemSelection &selected, const QItemSelection &deselected)
     {
        Q_UNUSED(deselected);
        auto sel = selected.indexes();
        if (sel.empty())
            handleTableSelection(0, false);
        else
            handleTableSelection(treeModuleId(sel.at(0)), false);
     }
 
     void ModuleDialog::handleTreeDoubleClick(const QModelIndex& index)
     {
        u32 moduleId = treeModuleId(index);
        handleTableSelection(moduleId, true);
     }
 
    void ModuleDialog::handleTableSelection(u32 id, bool doubleClick)
    {
        mSelectedId = mSelectExclude.isAccepted(id) ? id : 0;
        enableButtons();
        if (mSelectedId && doubleClick) accept();
    }
 
    void AddToModuleReceiver::handleModulesPicked(const QSet<u32>& mods)
    {
        if (!mods.empty())
        {
            u32 moduleId = *mods.constBegin();
            Module* firstAccepted = gNetlist->get_module_by_id(moduleId);
            Q_ASSERT(firstAccepted);
            if (QMessageBox::question(qApp->activeWindow(),
                                      "Confirm:",
                                      QString("Ok to move %1 into module '%2'[%3]").arg(mSelectExclude.selectionToString()).arg(QString::fromStdString(firstAccepted->get_name())).arg(moduleId),
                                      QMessageBox::Ok | QMessageBox::Cancel)
                == QMessageBox::Ok)
            {
                ActionAddItemsToObject* act = new ActionAddItemsToObject(mSelectExclude.modules(), mSelectExclude.gates());
                act->setObject(UserActionObject(moduleId, UserActionObjectType::Module));
                act->exec();
                gSelectionRelay->clear();
                gSelectionRelay->addModule(moduleId);
                gSelectionRelay->setFocus(SelectionRelay::ItemType::Module, moduleId);
                gSelectionRelay->relaySelectionChanged(this);
                gContentManager->getGraphTabWidget()->ensureSelectionVisible();
 
                ModuleSelectHistory::instance()->add(moduleId);
            }
        }
        deleteLater();
    }
 
    void ModuleDialog::handlePickFromGraph()
    {
        Q_ASSERT(mReceiver);
        new ModuleSelectPicker(mReceiver);
        mPickerModeActivated = true;
        reject(); // wait for picker, no selection done in dialog
    }
 
    void ModuleDialog::accept()
    {
        ModuleSelectHistory::instance()->add(mSelectedId);
        QDialog::accept();
    }
 
    void ModuleDialog::handleCreateNewModule()
    {
        mNewModule = true;
        QDialog::accept();
    }
 
    void ModuleDialog::handleToggleSearchbar()
    {
        if (mSearchbar->isHidden())
        {
            mSearchbar->show();
            mSearchbar->setFocus();
        }
        else
        {
            mSearchbar->hide();
            setFocus();
        }
    }
 
    void ModuleDialog::handleCurrentTabChanged(int index)
    {
        Q_UNUSED(index);
        //set columnNames for searchbar
        if(mTabWidget->currentWidget() == mTreeView)
            mSearchbar->setColumnNames(mModuleTreeProxyModel->getColumnNames());
        else
            mSearchbar->setColumnNames(static_cast<ModuleSelectProxy*>(mTableView->model())->getColumnNames());
        mTreeView->clearSelection();
        mTableView->clearSelection();
        mSearchbar->clear();
        if (mLastUsed) mLastUsed->clearSelection();
    }
 
    void ModuleDialog::keybindToggleSearchbar(const QKeySequence& seq)
    {
        mToggleSearchbar->setShortcut(seq);
    }
}