scenariolistview.cpp

Go to the documentation of this file.

/*
 * This file is part of ArmarX.
 *
 * Copyright (C) 2011-2016, High Performance Humanoid Technologies (H2T), Karlsruhe Institute of Technology (KIT), all rights reserved.
 *
 * ArmarX is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * ArmarX is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * @package    ArmarXCore::core
 * @author     Cedric Seehausen (usdnr at kit dot edu)
 * @date       2016
 * @copyright  http://www.gnu.org/licenses/gpl-2.0.txt
 *             GNU General Public License
 */
 
 
#include "scenariolistview.h"
 
#include <QAction>
#include <QComboBox>
#include <QHeaderView>
 
#include <ArmarXCore/core/logging/Logging.h>
 
#include <ArmarXGui/gui-plugins/ScenarioManager/gui/ui_scenariolistview.h>
 
#include "scenarioitem.h"
#include "treeitem.h"
 
using namespace ScenarioManager;
using namespace Data_Structure;
using namespace armarx;
 
ScenarioListView::ScenarioListView(QWidget* parent) :
    QWidget(parent),
    ui(new Ui::ScenarioListView),
    startButtonDelegate(this),
    stopButtonDelegate(this),
    restartButtonDelegate(this),
    removeAction("Remove", &contextMenu),
    expandAction("Expand", &contextMenu),
    collapseAction("Collapse", &contextMenu)
{
    ui->setupUi(this);
 
    ui->treeView->setModel(model.get());
    ui->treeView->setSortingEnabled(true);
    ui->treeView->setItemDelegateForColumn(1, &startButtonDelegate);
    ui->treeView->setItemDelegateForColumn(2, &stopButtonDelegate);
    ui->treeView->setItemDelegateForColumn(3, &restartButtonDelegate);
 
 
    contextMenu.setParent(ui->treeView);
    ui->treeView->setContextMenuPolicy(Qt::CustomContextMenu);
    QObject::connect(ui->treeView,
                     SIGNAL(customContextMenuRequested(const QPoint&)),
                     this,
                     SLOT(onCustomContextMenu(const QPoint&)));
 
    ui->treeView->addAction(&removeAction);
 
    ui->treeView->setDragEnabled(true);
    ui->treeView->setDefaultDropAction(Qt::CopyAction);
    ui->treeView->setAcceptDrops(true);
    ui->treeView->setDropIndicatorShown(true);
 
    QObject::connect(&removeAction, SIGNAL(triggered()), this, SLOT(removeItemTriggered()));
    QObject::connect(&expandAction, SIGNAL(triggered()), this, SLOT(expandItemTriggered()));
    QObject::connect(&collapseAction, SIGNAL(triggered()), this, SLOT(collapseItemTriggered()));
 
    QObject::connect(&startButtonDelegate,
                     SIGNAL(buttonClicked(int, int, QModelIndex)),
                     this,
                     SLOT(startButtonClicked(int, int, QModelIndex)));
 
 
    QObject::connect(&stopButtonDelegate,
                     SIGNAL(buttonClicked(int, int, QModelIndex)),
                     this,
                     SLOT(stopButtonClicked(int, int, QModelIndex)));
 
    QObject::connect(&restartButtonDelegate,
                     SIGNAL(buttonClicked(int, int, QModelIndex)),
                     this,
                     SLOT(restartButtonClicked(int, int, QModelIndex)));
}
 
ScenarioListView::~ScenarioListView()
{
    delete ui;
}
 
void
ScenarioListView::setModel(FilterableTreeModelSortFilterProxyModelPtr treeModel)
{
    this->model = treeModel;
    ui->treeView->setModel(model.get());
 
    QHeaderView* header = ui->treeView->header();
    header->setStretchLastSection(false);
    header->setSectionResizeMode(0, QHeaderView::Stretch);
    header->setSectionResizeMode(1, QHeaderView::Fixed);
    header->setSectionResizeMode(2, QHeaderView::Fixed);
    header->setSectionResizeMode(3, QHeaderView::Fixed);
    header->setSectionResizeMode(4, QHeaderView::ResizeToContents);
    const int buttonWidth = 60;
    header->resizeSection(1, buttonWidth);
    header->resizeSection(2, buttonWidth);
    header->resizeSection(3, buttonWidth);
}
 
void
ScenarioListView::on_searchBar_textEdited(const QString& text)
{
    model->setFilterRegExp(QRegExp(text, Qt::CaseInsensitive, QRegExp::FixedString));
    ui->treeView->expandAll();
}
 
void
ScenarioListView::startButtonClicked(int row, int column, QModelIndex parent)
{
    emit startApplication(row, column, parent);
}
 
void
ScenarioListView::stopButtonClicked(int row, int column, QModelIndex parent)
{
    emit stopApplication(row, column, parent);
}
 
void
ScenarioListView::restartButtonClicked(int row, int column, QModelIndex parent)
{
    emit restartApplication(row, column, parent);
}
 
void
ScenarioListView::on_newButton_clicked()
{
    emit createScenario();
}
 
void
ScenarioListView::on_openButton_clicked()
{
    emit showOpenDialog();
}
 
void
ScenarioListView::on_expandAllButton_clicked()
{
    ui->treeView->expandAll();
}
 
void
ScenarioListView::on_collapseAllButton_clicked()
{
    ui->treeView->collapseAll();
}
 
void
ScenarioListView::removeItemTriggered()
{
    QModelIndex index = ui->treeView->currentIndex();
    emit removeItem(index);
}
 
void
ScenarioListView::on_treeView_clicked(const QModelIndex& index)
{
    emit itemClicked(index);
}
 
void
ScenarioListView::onCustomContextMenu(const QPoint& point)
{
    QModelIndex index = ui->treeView->currentIndex();
 
    QList<QAction*> actions;
    actions.append(&expandAction);
    actions.append(&collapseAction);
    QAction separator;
    separator.setSeparator(true);
    actions.append(&separator);
    actions.append(&removeAction);
 
    QMenu::exec(actions, ui->treeView->mapToGlobal(point));
}
 
// Generated by ChatGPT
inline void
expandRecursively(QTreeView* view, const QModelIndex& index)
{
    if (!index.isValid())
    {
        return;
    }
 
    view->expand(index);
 
    const QAbstractItemModel* model = index.model();
    int rows = model->rowCount(index);
    int cols = model->columnCount(index);
 
    for (int r = 0; r < rows; ++r)
    {
        for (int c = 0; c < cols; ++c)
        {
            QModelIndex child = model->index(r, c, index);
            if (child.isValid())
            {
                expandRecursively(view, child);
            }
        }
    }
}
 
void
ScenarioListView::expandItemTriggered()
{
    QModelIndex index = ui->treeView->currentIndex();
 
    // Support for Ubuntu 18 with older Qt version (>=5.9.5)
    #if QT_VERSION >= QT_VERSION_CHECK(5, 15, 0)
        ui->treeView->expandRecursively(index);
    #else
        expandRecursively(ui->treeView, index);
    #endif
}
 
void
ScenarioListView::collapseItemTriggered()
{
    QModelIndex index = ui->treeView->currentIndex();
    ui->treeView->collapse(index);
}