(temporary) Classify dialog added

2026-08-29 16:40:49 +08:00 · 2018-11-04 23:31:48 +01:00
parent 3c365089cf
commit 37eabb4cea
8 changed files with 1267 additions and 467 deletions
@@ -0,0 +1,190 @@
+//##########################################################################
+//#                                                                        #
+//#                     CLOUDCOMPARE PLUGIN: qCANUPO                       #
+//#                                                                        #
+//#  This program is free software; you can redistribute it and/or modify  #
+//#  it under the terms of the GNU General Public License as published by  #
+//#  the Free Software Foundation; version 2 or later of the License.      #
+//#                                                                        #
+//#  This program 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.                          #
+//#                                                                        #
+//#            COPYRIGHT: UNIVERSITE EUROPEENNE DE BRETAGNE                #
+//#                                                                        #
+//##########################################################################
+
+#include "qClassify3DMASCDialog.h"
+
+//qCC_plugins
+#include "../../ccMainAppInterface.h"
+
+//qCC_db
+#include <ccPointCloud.h>
+
+//Qt
+#include <QMainWindow>
+#include <QPushButton>
+#include <QComboBox>
+//#include <QApplication>
+
+//system
+#include <limits>
+
+static ccPointCloud* GetCloudFromCombo(QComboBox* comboBox, ccHObject* dbRoot)
+{
+	assert(comboBox && dbRoot);
+	if (!comboBox || !dbRoot)
+	{
+		assert(false);
+		return 0;
+	}
+
+	//return the cloud currently selected in the combox box
+	int index = comboBox->currentIndex();
+	if (index < 0)
+	{
+		assert(false);
+		return 0;
+	}
+	unsigned uniqueID = comboBox->itemData(index).toUInt();
+	ccHObject* item = dbRoot->find(uniqueID);
+	if (!item || !item->isA(CC_TYPES::POINT_CLOUD))
+	{
+		assert(false);
+		return 0;
+	}
+	return static_cast<ccPointCloud*>(item);
+}
+
+Classify3DMASCDialog::Classify3DMASCDialog(ccMainAppInterface* app)
+	: QDialog(app ? app->getMainWindow() : 0)
+	, Ui::Classify3DMASCDialog()
+	, m_app(app)
+{
+	setupUi(this);
+
+	if (m_app)
+	{
+		//add list of clouds to the combo-boxes
+		ccHObject::Container clouds;
+		if (m_app->dbRootObject())
+		{
+			m_app->dbRootObject()->filterChildren(clouds, true, CC_TYPES::POINT_CLOUD);
+		}
+
+		unsigned cloudCount = 0;
+		for (size_t i = 0; i < clouds.size(); ++i)
+		{
+			if (clouds[i]->isA(CC_TYPES::POINT_CLOUD)) //as filterChildren only test 'isKindOf'
+			{
+				QString name = clouds[i]->getName() + QString(" [%1]").arg(clouds[i]->getUniqueID());
+				QVariant uniqueID(clouds[i]->getUniqueID());
+				cloud1ComboBox->addItem(name, uniqueID);
+				cloud2ComboBox->addItem(name, uniqueID);
+				cloud3ComboBox->addItem(name, uniqueID);
+				++cloudCount;
+			}
+		}
+
+		//if 3 clouds are loaded, then there's chances that the first one is the global  cloud!
+		cloud1ComboBox->setCurrentIndex(cloudCount > 0 ? (cloudCount > 2 ? 1 : 0) : -1);
+		cloud2ComboBox->setCurrentIndex(cloudCount > 1 ? (cloudCount > 2 ? 2 : 1) : -1);
+		cloud3ComboBox->setCurrentIndex(cloudCount > 2 ? 0 : -1);
+
+		if (cloudCount == 0 && app)
+		{
+			app->dispToConsole("You need at least 1 loaded cloud to classify it...", ccMainAppInterface::ERR_CONSOLE_MESSAGE);
+		}
+	}
+
+	cloud1RadioButton->setEnabled(false);
+	cloud2RadioButton->setEnabled(false);
+	cloud3RadioButton->setEnabled(false);
+	cloud1ComboBox->setEnabled(false);
+	cloud2ComboBox->setEnabled(false);
+	cloud3ComboBox->setEnabled(false);
+
+	onCloudChanged(0);
+}
+
+void Classify3DMASCDialog::setCloudRoles(const QSet<QString>& roles)
+{
+	int index = 0;
+	for (const QString& role : roles)
+	{
+		switch (index)
+		{
+		case 0:
+			cloud1RadioButton->setText(role);
+
+			cloud1RadioButton->setEnabled(true);
+			cloud1ComboBox->setEnabled(false);
+
+			cloud1RadioButton->setChecked(true);
+			break;
+		case 1:
+			cloud2RadioButton->setText(role);
+
+			cloud2RadioButton->setEnabled(true);
+			cloud2ComboBox->setEnabled(false);
+			break;
+		case 2:
+			cloud3RadioButton->setText(role);
+
+			cloud3RadioButton->setEnabled(true);
+			cloud3ComboBox->setEnabled(false);
+			break;
+		default:
+			//this dialog can't handle more than 3 roles!
+			break;
+		}
+	}
+}
+
+void Classify3DMASCDialog::getClouds(QMap<QString, ccPointCloud*>& clouds, QString& mainCloud)
+{
+	if (!m_app)
+	{
+		assert(false);
+		return;
+	}
+	
+	if (cloud1RadioButton->isEnabled())
+	{
+		clouds.insert(cloud1RadioButton->text(), GetCloudFromCombo(cloud1ComboBox, m_app->dbRootObject()));
+		if (cloud1RadioButton->isChecked())
+		{
+			mainCloud = cloud1RadioButton->text();
+		}
+	}
+	if (cloud2RadioButton->isEnabled())
+	{
+		clouds.insert(cloud2RadioButton->text(), GetCloudFromCombo(cloud2ComboBox, m_app->dbRootObject()));
+		if (cloud2RadioButton->isChecked())
+		{
+			mainCloud = cloud2RadioButton->text();
+		}
+	}
+	if (cloud3RadioButton->isEnabled())
+	{
+		clouds.insert(cloud3RadioButton->text(), GetCloudFromCombo(cloud3ComboBox, m_app->dbRootObject()));
+		if (cloud3RadioButton->isChecked())
+		{
+			mainCloud = cloud3RadioButton->text();
+		}
+	}
+}
+
+void Classify3DMASCDialog::onCloudChanged(int dummy)
+{
+	if (!cloud1RadioButton->isEnabled())
+	{
+		//this means that no role has been defined yet
+		buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
+		return;
+	}
+
+	buttonBox->button(QDialogButtonBox::Ok)->setEnabled(cloud1ComboBox->currentIndex() >= 0);
+}