|
|
#include <DataSource/DataSourceWidget.h>
|
|
|
|
|
|
#include <ui_DataSourceWidget.h>
|
|
|
|
|
|
#include <DataSource/DataSourceItem.h>
|
|
|
#include <DataSource/DataSourceTreeWidgetItem.h>
|
|
|
|
|
|
#include <QMenu>
|
|
|
|
|
|
namespace {
|
|
|
|
|
|
/// Number of columns displayed in the tree
|
|
|
const auto TREE_NB_COLUMNS = 1;
|
|
|
|
|
|
/// Header labels for the tree
|
|
|
const auto TREE_HEADER_LABELS = QStringList{QObject::tr("Name")};
|
|
|
|
|
|
/**
|
|
|
* Creates the item associated to a data source
|
|
|
* @param dataSource the data source for which to create the item
|
|
|
* @return the new item
|
|
|
*/
|
|
|
DataSourceTreeWidgetItem *createTreeWidgetItem(DataSourceItem *dataSource)
|
|
|
{
|
|
|
// Creates item for the data source
|
|
|
auto item = new DataSourceTreeWidgetItem{dataSource};
|
|
|
|
|
|
// Generates items for the children of the data source
|
|
|
for (auto i = 0; i < dataSource->childCount(); ++i) {
|
|
|
item->addChild(createTreeWidgetItem(dataSource->child(i)));
|
|
|
}
|
|
|
|
|
|
return item;
|
|
|
}
|
|
|
|
|
|
} // namespace
|
|
|
|
|
|
DataSourceWidget::DataSourceWidget(QWidget *parent) : QWidget{parent}, ui{new Ui::DataSourceWidget}
|
|
|
{
|
|
|
ui->setupUi(this);
|
|
|
|
|
|
// Set tree properties
|
|
|
ui->treeWidget->setColumnCount(TREE_NB_COLUMNS);
|
|
|
ui->treeWidget->setHeaderLabels(TREE_HEADER_LABELS);
|
|
|
ui->treeWidget->setContextMenuPolicy(Qt::CustomContextMenu);
|
|
|
|
|
|
// Connection to show a menu when right clicking on the tree
|
|
|
connect(ui->treeWidget, &QTreeWidget::customContextMenuRequested, this,
|
|
|
&DataSourceWidget::onTreeMenuRequested);
|
|
|
}
|
|
|
|
|
|
void DataSourceWidget::addDataSource(DataSourceItem *dataSource) noexcept
|
|
|
{
|
|
|
// Creates the item associated to the source and adds it to the tree widget. The tree widget
|
|
|
// takes the ownership of the item
|
|
|
if (dataSource) {
|
|
|
ui->treeWidget->addTopLevelItem(createTreeWidgetItem(dataSource));
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void DataSourceWidget::onTreeMenuRequested(const QPoint &pos) noexcept
|
|
|
{
|
|
|
// Retrieves the selected item in the tree, and build the menu from its actions
|
|
|
if (auto selectedItem = dynamic_cast<DataSourceTreeWidgetItem *>(ui->treeWidget->itemAt(pos))) {
|
|
|
QMenu treeMenu{};
|
|
|
treeMenu.addActions(selectedItem->actions());
|
|
|
|
|
|
if (!treeMenu.isEmpty()) {
|
|
|
treeMenu.exec(mapToGlobal(pos));
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
