theo-vt
136 lines
4.6 KiB
C++
136 lines
4.6 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2026 Théo Veilleux-Trinh <theo.veilleux.trinh@proton.me>*
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library 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 Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
#include "TaskCommandLink.h"
|
|
|
|
#include "ui_TaskCommandLink.h"
|
|
|
|
#include "Application.h"
|
|
#include "Document.h"
|
|
#include "ViewProvider.h"
|
|
|
|
#include <App/Application.h>
|
|
#include <App/DocumentObject.h>
|
|
#include <App/Document.h>
|
|
|
|
namespace Gui
|
|
{
|
|
TaskCommandLink::TaskCommandLink()
|
|
: ui(new Ui_TaskCommandLinkDialog())
|
|
{
|
|
proxy = new QWidget(this);
|
|
ui->setupUi(proxy);
|
|
ui->objectsList->header()->hide();
|
|
ui->objectsList->setSelectionMode(QAbstractItemView::SelectionMode::ExtendedSelection);
|
|
|
|
this->groupLayout()->addWidget(proxy);
|
|
|
|
buildObjectsList();
|
|
}
|
|
TaskCommandLink::~TaskCommandLink()
|
|
{
|
|
delete proxy;
|
|
delete ui;
|
|
}
|
|
std::vector<App::DocumentObject*> TaskCommandLink::selectedObjects()
|
|
{
|
|
auto selected = ui->objectsList->selectedItems();
|
|
std::vector<App::DocumentObject*> dst;
|
|
dst.reserve(selected.size());
|
|
|
|
for (auto sel : selected) {
|
|
dst.push_back(sel->data(0, Qt::UserRole).value<App::DocumentObject*>());
|
|
}
|
|
return dst;
|
|
}
|
|
void processObjectsHelper(std::vector<App::DocumentObject*> objs, QTreeWidgetItem* item)
|
|
{
|
|
for (auto obj : objs) {
|
|
auto objItem = new QTreeWidgetItem(item);
|
|
|
|
objItem->setText(0, obj->Label.getValue());
|
|
objItem->setData(0, Qt::UserRole, QVariant::fromValue(obj));
|
|
|
|
Gui::ViewProvider* vp = nullptr;
|
|
if (auto doc = Application::Instance->getDocument(obj->getDocument())) {
|
|
vp = doc->getViewProvider(obj);
|
|
}
|
|
if (vp) {
|
|
objItem->setIcon(0, vp->getIcon());
|
|
processObjectsHelper(vp->claimChildren(), objItem);
|
|
}
|
|
else {
|
|
objItem->setIcon(0, QIcon());
|
|
}
|
|
}
|
|
}
|
|
void TaskCommandLink::buildObjectsList()
|
|
{
|
|
ui->objectsList->clear();
|
|
|
|
auto allDocuments = App::GetApplication().getDocuments();
|
|
bool collapse = true;
|
|
std::map<QTreeWidgetItem*, App::Document*> docItemMap;
|
|
|
|
for (auto doc : allDocuments) {
|
|
auto docItem = new QTreeWidgetItem();
|
|
std::string itemName = doc->Label.getValue();
|
|
|
|
docItem->setText(0, QString::fromStdString(itemName));
|
|
docItem->setIcon(0, QIcon(QStringLiteral(":/icons/Document.svg")));
|
|
docItem->setFlags(docItem->flags() & ~Qt::ItemIsSelectable); // Can't link a whole document
|
|
|
|
docItemMap[docItem] = doc;
|
|
|
|
ui->objectsList->addTopLevelItem(docItem);
|
|
|
|
processObjectsHelper(Application::Instance->getDocument(doc)->getTreeRootObjects(), docItem);
|
|
|
|
if (collapse) {
|
|
ui->objectsList->collapseAll();
|
|
}
|
|
else {
|
|
ui->objectsList->expandToDepth(0);
|
|
}
|
|
}
|
|
ui->objectsList->selectedItems();
|
|
}
|
|
|
|
// dialog
|
|
|
|
TaskCommandLinkDialog::TaskCommandLinkDialog(
|
|
std::function<void(std::vector<App::DocumentObject*>)> executor_
|
|
)
|
|
: executor(executor_)
|
|
{
|
|
commandLink = new TaskCommandLink();
|
|
Content.push_back(commandLink);
|
|
}
|
|
void TaskCommandLinkDialog::open()
|
|
{}
|
|
bool TaskCommandLinkDialog::accept()
|
|
{
|
|
executor(commandLink->selectedObjects());
|
|
return true;
|
|
}
|
|
} // namespace Gui
|