70 lines
2.4 KiB
C++
70 lines
2.4 KiB
C++
#include "component_preview_grid_widget.h"
|
|
#include "component_list_model.h"
|
|
#include "document.h"
|
|
#include <memory>
|
|
|
|
ComponentPreviewGridWidget::ComponentPreviewGridWidget(Document* document, QWidget* parent)
|
|
: PreviewGridView(parent)
|
|
, m_document(document)
|
|
{
|
|
m_componentListModel = std::make_unique<ComponentListModel>(m_document);
|
|
setModel(m_componentListModel.get());
|
|
|
|
connect(this, &ComponentPreviewGridWidget::doubleClicked, [this](const QModelIndex& index) {
|
|
const SkeletonComponent* component = this->componentListModel()->modelIndexToComponent(index);
|
|
if (nullptr == component)
|
|
return;
|
|
if (component->childrenIds.empty()) {
|
|
std::vector<dust3d::Uuid> partIds;
|
|
this->m_document->collectComponentDescendantParts(component->id, partIds);
|
|
for (const auto& partId : partIds)
|
|
emit this->selectPartOnCanvas(partId);
|
|
} else {
|
|
this->componentListModel()->setListingComponentId(component->id);
|
|
}
|
|
});
|
|
}
|
|
|
|
ComponentListModel* ComponentPreviewGridWidget::componentListModel()
|
|
{
|
|
return m_componentListModel.get();
|
|
}
|
|
|
|
std::vector<const SkeletonComponent*> ComponentPreviewGridWidget::getSelectedComponents() const
|
|
{
|
|
std::vector<const SkeletonComponent*> components;
|
|
QModelIndexList selected = selectionModel()->selectedIndexes();
|
|
for (const auto& it : selected) {
|
|
const auto& component = m_componentListModel->modelIndexToComponent(it);
|
|
if (nullptr == component)
|
|
continue;
|
|
components.push_back(component);
|
|
}
|
|
return components;
|
|
}
|
|
|
|
std::vector<dust3d::Uuid> ComponentPreviewGridWidget::getSelectedComponentIds() const
|
|
{
|
|
std::vector<dust3d::Uuid> componentIds;
|
|
QModelIndexList selected = selectionModel()->selectedIndexes();
|
|
for (const auto& it : selected) {
|
|
const auto& componentId = m_componentListModel->modelIndexToComponentId(it);
|
|
if (componentId.isNull())
|
|
continue;
|
|
componentIds.push_back(componentId);
|
|
}
|
|
return componentIds;
|
|
}
|
|
|
|
std::vector<dust3d::Uuid> ComponentPreviewGridWidget::getSelectedPartIds() const
|
|
{
|
|
auto selectedComponents = getSelectedComponents();
|
|
std::vector<dust3d::Uuid> partIds;
|
|
for (const auto& component : selectedComponents) {
|
|
if (component->linkToPartId.isNull())
|
|
continue;
|
|
partIds.push_back(component->linkToPartId);
|
|
}
|
|
return partIds;
|
|
}
|