ecvEntitySelectionDlg.cpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#include "ecvEntitySelectionDlg.h"
 
// ui
#include <ui_entitySelectionDlg.h>
 
// Qt
#include <QDialog>
#include <QListWidgetItem>
 
ecvEntitySelectionDialog::ecvEntitySelectionDialog(
        const ccHObject::Container& entities,
        bool multiSelectionEnabled,
        int defaultSelectedIndex /*=0*/,
        QWidget* parent /*=0*/,
        QString labelStr /*=QString()*/)
    : QDialog(parent, Qt::Tool), m_ui(new Ui_EntitySelectionDialog) {
    m_ui->setupUi(this);
 
    // multi-selection mode
    if (multiSelectionEnabled) {
        m_ui->label->setText(tr("Select one or several entities:"));
        m_ui->listWidget->setSelectionMode(
                QAbstractItemView::ExtendedSelection);
        connect(m_ui->selectAllPushButton, &QPushButton::clicked, this,
                &ecvEntitySelectionDialog::selectAll);
        connect(m_ui->selectNonePushButton, &QPushButton::clicked, this,
                &ecvEntitySelectionDialog::selectNone);
    } else {
        m_ui->selectAllPushButton->setVisible(false);
        m_ui->selectNonePushButton->setVisible(false);
    }
 
    for (size_t i = 0; i < entities.size(); ++i) {
        // add one line per entity in the combo-box
        m_ui->listWidget->insertItem(
                static_cast<int>(i),
                new QListWidgetItem(QString("[ID %1] %2")
                                            .arg(entities[i]->getUniqueID())
                                            .arg(entities[i]->getName())));
    }
 
    // default selection
    if (defaultSelectedIndex >= 0 &&
        static_cast<size_t>(defaultSelectedIndex) < entities.size()) {
        m_ui->listWidget->item(defaultSelectedIndex)->setSelected(true);
    }
 
    // custom lalel
    if (!labelStr.isNull()) {
        m_ui->label->setText(labelStr);
    }
}
 
ecvEntitySelectionDialog::~ecvEntitySelectionDialog() {
    if (m_ui) {
        delete m_ui;
        m_ui = nullptr;
    }
}
 
void ecvEntitySelectionDialog::selectAll() { m_ui->listWidget->selectAll(); }
 
void ecvEntitySelectionDialog::selectNone() {
    m_ui->listWidget->clearSelection();
}
 
int ecvEntitySelectionDialog::getSelectedIndex() const {
    // get selected items
    QList<QListWidgetItem*> list = m_ui->listWidget->selectedItems();
    return list.empty() ? -1 : m_ui->listWidget->row(list.front());
}
 
void ecvEntitySelectionDialog::getSelectedIndexes(
        std::vector<int>& indexes) const {
    // get selected items
    QList<QListWidgetItem*> list = m_ui->listWidget->selectedItems();
 
    try {
        indexes.resize(static_cast<size_t>(list.size()));
    } catch (const std::bad_alloc&) {
        // not enough memory?!
        return;
    }
 
    for (int i = 0; i < list.size(); ++i) {
        indexes[i] = m_ui->listWidget->row(list[i]);
    }
}
 
int ecvEntitySelectionDialog::SelectEntity(const ccHObject::Container& entities,
                                           int selectedIndex /*=0*/,
                                           QWidget* parent /*=0*/,
                                           QString label /*=QString()*/) {
    ecvEntitySelectionDialog epDlg(entities, false, selectedIndex, parent,
                                   label);
    if (!epDlg.exec()) {
        return -1;
    }
 
    return epDlg.getSelectedIndex();
}
 
bool ecvEntitySelectionDialog::SelectEntities(
        const ccHObject::Container& entities,
        std::vector<int>& selectedIndexes,
        QWidget* parent /*=0*/,
        QString label /*=QString()*/) {
    selectedIndexes.clear();
 
    ecvEntitySelectionDialog epDlg(entities, true, -1, parent, label);
    if (!epDlg.exec()) {
        return false;
    }
 
    epDlg.getSelectedIndexes(selectedIndexes);
    return true;
}