mirror of
https://github.com/dgirardeau/q3DMASC.git
synced 2026-08-31 01:20:53 +08:00
Context-based features are now working
This commit is contained in:
Daniel Girardeau-Montaut
+229
-3
@@ -17,14 +17,207 @@
|
||||
|
||||
#include "ContextBasedFeature.h"
|
||||
|
||||
//Qt
|
||||
#include <QMutex>
|
||||
|
||||
using namespace masc;
|
||||
|
||||
bool ContextBasedFeature::prepare( const CorePoints& corePoints,
|
||||
QString& error,
|
||||
CCLib::GenericProgressCallback* progressCb/*=nullptr*/)
|
||||
{
|
||||
//TODO
|
||||
return false;
|
||||
if (!cloud1 || !corePoints.cloud)
|
||||
{
|
||||
//invalid input
|
||||
assert(false);
|
||||
error = "internal error (no input core points)";
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!cloud2)
|
||||
{
|
||||
//invalid input
|
||||
assert(false);
|
||||
error = "internal error (no contextual cloud)";
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!checkValidity(error))
|
||||
{
|
||||
assert(false);
|
||||
return false;
|
||||
}
|
||||
|
||||
//build the final SF name
|
||||
QString typeStr = ToString(type);
|
||||
QString resultSFName = typeStr + "_" + cloud1Label + "_" + cloud2Label + "_" + QString::number(ctxClassLabel);
|
||||
if (scaled())
|
||||
{
|
||||
resultSFName += "@" + QString::number(scale);
|
||||
}
|
||||
else
|
||||
{
|
||||
resultSFName += "@kNN=" + QString::number(kNN);
|
||||
}
|
||||
|
||||
//and the scalar field
|
||||
assert(!sf);
|
||||
sf = PrepareSF(corePoints.cloud, qPrintable(resultSFName));
|
||||
if (!sf)
|
||||
{
|
||||
error = QString("Failed to prepare scalar %1 @ scale %2").arg(resultSFName).arg(scale);
|
||||
return false;
|
||||
}
|
||||
sourceName = sf->getName();
|
||||
|
||||
if (!scaled()) //with 'kNN' neighbors, we can compute the values right away
|
||||
{
|
||||
//get the octree
|
||||
ccOctree::Shared octree = cloud2->getOctree();
|
||||
if (!octree)
|
||||
{
|
||||
ccLog::Print(QString("Computing octree of cloud %1 (%2 points)").arg(cloud2->getName()).arg(cloud2->size()));
|
||||
octree = cloud2->computeOctree(progressCb);
|
||||
if (!octree)
|
||||
{
|
||||
error = "Failed to compute octree (not enough memory?)";
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
//now extract the neighborhoods
|
||||
unsigned char octreeLevel = octree->findBestLevelForAGivenPopulationPerCell(static_cast<unsigned>(std::min(3, kNN)));
|
||||
|
||||
unsigned pointCount = corePoints.size();
|
||||
QString logMessage = QString("Computing %1 on cloud %2 with context cloud %3\n(core points: %4)").arg(typeStr).arg(corePoints.cloud->getName()).arg(cloud2Label).arg(pointCount);
|
||||
if (progressCb)
|
||||
{
|
||||
progressCb->setMethodTitle(qPrintable("Compute " + typeStr));
|
||||
progressCb->setInfo(qPrintable(logMessage));
|
||||
}
|
||||
ccLog::Print(logMessage);
|
||||
CCLib::NormalizedProgress nProgress(progressCb, pointCount);
|
||||
|
||||
QMutex mutex;
|
||||
bool error = false;
|
||||
#ifndef _DEBUG
|
||||
#if defined(_OPENMP)
|
||||
#pragma omp parallel for
|
||||
#endif
|
||||
#endif
|
||||
for (int i = 0; i < static_cast<int>(pointCount); ++i)
|
||||
{
|
||||
const CCVector3* P = corePoints.cloud->getPoint(i);
|
||||
CCLib::ReferenceCloud Yk(cloud2);
|
||||
double maxSquareDist = 0;
|
||||
|
||||
ScalarType s = NAN_VALUE;
|
||||
|
||||
if (octree->findPointNeighbourhood(P, &Yk, static_cast<unsigned>(kNN), octreeLevel, maxSquareDist) >= kNN)
|
||||
{
|
||||
CCVector3d sumQ(0, 0, 0);
|
||||
for (int k = 0; k < kNN; ++k)
|
||||
{
|
||||
sumQ += CCVector3d::fromArray(Yk.getPoint(k)->u);
|
||||
}
|
||||
|
||||
switch (type)
|
||||
{
|
||||
case DZ:
|
||||
s = static_cast<ScalarType>(P->z - sumQ.z / kNN);
|
||||
break;
|
||||
case DH:
|
||||
s = static_cast<ScalarType>(sqrt(pow(P->x - sumQ.x / kNN, 2.0) + pow(P->y - sumQ.y / kNN, 2.0)));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
sf->setValue(i, s);
|
||||
|
||||
if (progressCb)
|
||||
{
|
||||
mutex.lock();
|
||||
bool cancelled = !nProgress.oneStep();
|
||||
mutex.unlock();
|
||||
if (cancelled)
|
||||
{
|
||||
//process cancelled by the user
|
||||
ccLog::Warning("Process cancelled");
|
||||
error = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
sf->computeMinAndMax();
|
||||
|
||||
if (progressCb)
|
||||
{
|
||||
progressCb->stop();
|
||||
}
|
||||
|
||||
if (error)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool ContextBasedFeature::computeValue(CCLib::DgmOctree::NeighboursSet& pointsInNeighbourhood, const CCVector3& queryPoint, ScalarType& outputValue) const
|
||||
{
|
||||
CCVector3d sumQ(0, 0, 0);
|
||||
for (CCLib::DgmOctree::PointDescriptor& Pd : pointsInNeighbourhood)
|
||||
{
|
||||
sumQ += CCVector3d::fromArray(Pd.point->u);
|
||||
}
|
||||
|
||||
switch (type)
|
||||
{
|
||||
case DZ:
|
||||
outputValue = static_cast<ScalarType>(queryPoint.z - sumQ.z / kNN);
|
||||
break;
|
||||
case DH:
|
||||
outputValue = static_cast<ScalarType>(sqrt(pow(queryPoint.x - sumQ.x / kNN, 2.0) + pow(queryPoint.y - sumQ.y / kNN, 2.0)));
|
||||
break;
|
||||
default:
|
||||
assert(false);
|
||||
outputValue = NAN_VALUE;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool ContextBasedFeature::finish(const CorePoints& corePoints, QString& error)
|
||||
{
|
||||
if (!corePoints.cloud)
|
||||
{
|
||||
//invalid input
|
||||
assert(false);
|
||||
error = "internal error (no input core points)";
|
||||
return false;
|
||||
}
|
||||
|
||||
bool success = true;
|
||||
|
||||
if (sf)
|
||||
{
|
||||
sf->computeMinAndMax();
|
||||
|
||||
//update display
|
||||
//if (corePoints.cloud->getDisplay())
|
||||
{
|
||||
int sfIndex = corePoints.cloud->getScalarFieldIndexByName(sf->getName());
|
||||
corePoints.cloud->setCurrentDisplayedScalarField(sfIndex);
|
||||
//corePoints.cloud->getDisplay()->redraw();
|
||||
//QCoreApplication::processEvents();
|
||||
}
|
||||
}
|
||||
|
||||
return success;
|
||||
}
|
||||
|
||||
bool ContextBasedFeature::checkValidity(QString &error) const
|
||||
@@ -34,6 +227,13 @@ bool ContextBasedFeature::checkValidity(QString &error) const
|
||||
return false;
|
||||
}
|
||||
|
||||
if (type == Invalid)
|
||||
{
|
||||
assert(false);
|
||||
error = "invalid feature type";
|
||||
return false;
|
||||
}
|
||||
|
||||
unsigned char cloudCount = (cloud1 ? (cloud2 ? 2 : 1) : 0);
|
||||
if (cloudCount < 2)
|
||||
{
|
||||
@@ -41,5 +241,31 @@ bool ContextBasedFeature::checkValidity(QString &error) const
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!scaled() && kNN < 1)
|
||||
{
|
||||
error = QString("invalid kNN value for a scale-less context-based feature (%1)").arg(kNN);
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
QString ContextBasedFeature::toString() const
|
||||
{
|
||||
//use the default keyword + number of neighbors + the scale + the context class
|
||||
QString str = ToString(type);
|
||||
if (!scaled())
|
||||
{
|
||||
if (kNN != 1)
|
||||
str += QString::number(kNN);
|
||||
str += "_SC0";
|
||||
}
|
||||
else
|
||||
{
|
||||
str += "_SC" + QString::number(scale);
|
||||
}
|
||||
|
||||
str += "_" + cloud1Label + "_" + cloud2Label + "_" + QString::number(ctxClassLabel);
|
||||
|
||||
return str;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user