int main(int argc, char* argv[])
{
allquadrics::TriangleMesh inputMesh;
char *defaultInput = "../example_data/cylinder.obj";
char *meshfile = argc > 1 ? argv[1] : 0;
if (!meshfile) {
cerr << "No mesh file specified? Loading default mesh: " << defaultInput << endl;
meshfile = defaultInput;
}
if (!inputMesh.loadObj(meshfile)) {
cerr << "Couldn't load file " << meshfile << endl;
return 1;
}
const char *quadricTypeNames[] = { "general", "rotationally symmetric", "plane", "sphere",
"general cylinder", "circular cylinder", "general cone", "circular cone",
"ellipsoid (BIASED METHOD)", "hyperboloid (BIASED METHOD)",
"ellipsoid (IMPROVED)", "hyperboloid (IMPROVED)",
"hyperboloid (1 sheet)", "hyperboloid (2 sheet)", "paraboloid",
"paraboloid (elliptical)", "paraboloid (hyperbolic)",
"elliptical cylinder", "hyperbolic cylinder", "parabolic cylinder" };
// Always recenter and scale your data before fitting!
inputMesh.centerAndScale(1);
// EXAMPLE USAGE 1: fit all quadric types at once
cout << "fitting using all fits function ..." << endl;
vector<allquadrics::Quadric> qfits;
fitAllQuadricTypes(inputMesh, qfits);
for (size_t i = 0; i < qfits.size(); i++) {
cout << "direct fit for quadric type: " << quadricTypeNames[i] << ":" << endl;
outputQuadric(qfits[i]);
cout << endl << endl;
}
// EXAMPLE USAGE 2: fit specific quadric types
cout << "fitting using specific single-type functions ..." << endl;
allquadrics::Quadric general;
fitGeneralQuadric(inputMesh, general);
cout << "direct fit for general quadric (unconstrained type): " << endl;
outputQuadric(general);
cout << endl << endl;
allquadrics::Quadric sphere;
fitSphere(inputMesh, sphere);
cout << "direct fit for sphere: " << endl;
outputQuadric(sphere);
cout << endl << endl;
allquadrics::Quadric ellipsoid;
fitEllipsoid(inputMesh, ellipsoid);
cout << "direct fit for ellipsoid: " << endl;
outputQuadric(ellipsoid);
cout << endl << endl;
allquadrics::Quadric hyperboloid;
fitHyperboloid(inputMesh, hyperboloid);
cout << "direct fit for hyperboloid: " << endl;
outputQuadric(hyperboloid);
cout << endl << endl;
allquadrics::Quadric generalCone;
fitGeneralCone(inputMesh, generalCone, true);
cout << "direct fit for general cone: " << endl;
outputQuadric(generalCone);
cout << endl << endl;
allquadrics::Quadric circularCone;
fitCircularCone(inputMesh, circularCone);
cout << "direct fit for circular cone: " << endl;
outputQuadric(circularCone);
cout << endl << endl;
allquadrics::Quadric generalCylinder;
fitGeneralCylinder(inputMesh, generalCylinder);
cout << "direct fit for general cylinder: " << endl;
outputQuadric(generalCylinder);
cout << endl << endl;
allquadrics::Quadric circularCylinder;
fitCircularCylinder(inputMesh, circularCylinder);
cout << "direct fit for circular cylinder: " << endl;
outputQuadric(circularCylinder);
cout << endl << endl;
// EXAMPLE USAGE 3: filter & fit only chosen triangles
cout << "fitting using a selected subset of the input shape ..." << endl;
// set triangle tags on mesh to activate 'selection mode'
inputMesh.triangleTags.resize(inputMesh.triangles.size(), 0);
// choose triangles with tag == 1 as the active selection, to be used in fitting
inputMesh.activeTag = 1;
// randomly select some of those triangles
inputMesh.triangleTags[rand() % inputMesh.triangles.size()] = 1;
inputMesh.triangleTags[rand() % inputMesh.triangles.size()] = 1;
inputMesh.triangleTags[rand() % inputMesh.triangles.size()] = 1;
inputMesh.triangleTags[rand() % inputMesh.triangles.size()] = 1;
// fit a general quadric to that selection
allquadrics::Quadric generalSubset;
fitGeneralQuadric(inputMesh, generalSubset);
cout << "direct fit (on four random triangles of input) for general quadric (unconstrained type): " << endl;
outputQuadric(generalSubset);
cout << endl << endl;
return 0;
}
MainApplication::MainApplication(QWidget *parent) : QWidget(parent)
{
const int textSize = 12;
const int toolBoxWidth = 160;
const int toolBoxWidgetsWidth = 140;
const int toolBoxSubWidgetsWidth = 120;
QSize textEditSize = QSize(40, 30);
int windowHeight = 500;
int windowWidth = 800;
/*---- Buttons ----*/
QPushButton *rangeQueryButton = new QPushButton(tr("Range"));
rangeQueryButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *radiusQueryButton = new QPushButton(tr("Radius"));
radiusQueryButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *loadButton = new QPushButton(tr("Load"));
loadButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *nnQueryButton = new QPushButton(tr("NN-Query"));
nnQueryButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *smoothingButton = new QPushButton(tr("Smooth"));
smoothingButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *distanceColorMapButton = new QPushButton(tr("ColorbyDist"));
distanceColorMapButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *thinningButton = new QPushButton(tr("Thinning"));
thinningButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *lineFittingButton = new QPushButton(tr("fit Line"));
lineFittingButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *planeFittingButton = new QPushButton(tr("fit Plane"));
planeFittingButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
QPushButton *sphereFittingButton = new QPushButton(tr("fit Sphere"));
sphereFittingButton->setFont(QFont("Times", textSize, QFont::AnyStyle));
connect(loadButton, SIGNAL(clicked()), this, SLOT(loadPoints()));
connect(rangeQueryButton, SIGNAL(clicked()), this, SLOT(rangeQuery()));
connect(radiusQueryButton, SIGNAL(clicked()), this, SLOT(radiusQuery()));
connect(smoothingButton, SIGNAL(clicked()), this, SLOT(smoothPointCloud()));
connect(nnQueryButton, SIGNAL(clicked()), this, SLOT(nnQuery()));
connect(distanceColorMapButton, SIGNAL(clicked()), this, SLOT(colorPointsByDistance()));
connect(thinningButton, SIGNAL(clicked()), this, SLOT(applyThinning()));
connect(lineFittingButton, SIGNAL(clicked()), this, SLOT(fitLine()));
connect(planeFittingButton, SIGNAL(clicked()), this, SLOT(fitPlane()));
connect(sphereFittingButton, SIGNAL(clicked()), this, SLOT(fitSphere()));
/*---- Labels ----*/
labelCloudBounds = new QLabel("---", this);
labelCloudBounds->setMaximumHeight(60);
labelPoints = new QLabel("---", this);
labelPoints->setMaximumHeight(60);
labelTime = new QLabel("---", this);
labelTime->setMaximumHeight(60);
labelFitting = new QLabel("p: dir:", this);
labelFitting->setMaximumHeight(120);
/*---- Text Edits ----*/
QDoubleValidator *validDouble = new QDoubleValidator();
minXRange = new QLineEdit();
minXRange->setMaximumSize(textEditSize);
minXRange->setValidator(validDouble);
maxXRange = new QLineEdit();
maxXRange->setMaximumSize(textEditSize);
maxXRange->setValidator(validDouble);
minYRange = new QLineEdit();
minYRange->setMaximumSize(textEditSize);
minYRange->setValidator(validDouble);
maxYRange = new QLineEdit();
maxYRange->setMaximumSize(textEditSize);
maxYRange->setValidator(validDouble);
minZRange = new QLineEdit();
minZRange->setMaximumSize(textEditSize);
minZRange->setValidator(validDouble);
maxZRange = new QLineEdit();
maxZRange->setMaximumSize(textEditSize);
maxZRange->setValidator(validDouble);
xRadius = new QLineEdit();
xRadius->setMaximumSize(textEditSize);
xRadius->setValidator(validDouble);
yRadius = new QLineEdit();
yRadius->setMaximumSize(textEditSize);
yRadius->setValidator(validDouble);
zRadius = new QLineEdit();
zRadius->setMaximumSize(textEditSize);
zRadius->setValidator(validDouble);
rRadius = new QLineEdit();
rRadius->setMaximumSize(textEditSize);
rRadius->setValidator(validDouble);
xNeighbour = new QLineEdit();
xNeighbour->setMaximumSize(textEditSize);
xNeighbour->setValidator(validDouble);
yNeighbour = new QLineEdit();
yNeighbour->setMaximumSize(textEditSize);
yNeighbour->setValidator(validDouble);
zNeighbour = new QLineEdit();
zNeighbour->setMaximumSize(textEditSize);
zNeighbour->setValidator(validDouble);
rSmoothing = new QLineEdit();
rSmoothing->setMaximumSize(textEditSize);
rSmoothing->setMaximumWidth(toolBoxSubWidgetsWidth);
rSmoothing->setValidator(validDouble);
rThinning = new QLineEdit();
rThinning->setMaximumSize(textEditSize);
rThinning->setMaximumWidth(toolBoxSubWidgetsWidth);
rThinning->setValidator(validDouble);
/*---- Tool Box and Tool Box Widgets ----*/
QToolBox *toolBox = new QToolBox();
//Load
QVBoxLayout *layoutLoad = new QVBoxLayout();
layoutLoad->addWidget(loadButton);
QWidget* LoadWidget = new QWidget();
LoadWidget->setLayout(layoutLoad);
LoadWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(LoadWidget, "Load Data");
// Range Query
QGridLayout *layoutRangeTextEdits = new QGridLayout();
layoutRangeTextEdits->addWidget(minXRange,0,0,0);
layoutRangeTextEdits->addWidget(maxXRange,0,1,0);
layoutRangeTextEdits->addWidget(minYRange,1,0,0);
layoutRangeTextEdits->addWidget(maxYRange,1,1,0);
layoutRangeTextEdits->addWidget(minZRange,2,0,0);
layoutRangeTextEdits->addWidget(maxZRange,2,1,0);
QWidget* RangeTextEditsWidget = new QWidget();
RangeTextEditsWidget->setLayout(layoutRangeTextEdits);
RangeTextEditsWidget->setFixedWidth(toolBoxSubWidgetsWidth);
QVBoxLayout *layoutRange = new QVBoxLayout();
layoutRange->addWidget(RangeTextEditsWidget);
layoutRange->addWidget(rangeQueryButton);
QWidget* RangeWidget = new QWidget();
RangeWidget->setLayout(layoutRange);
RangeWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(RangeWidget, "Range Query");
// Radius Query
QGridLayout *layoutRadiusTextEdits = new QGridLayout();
layoutRadiusTextEdits->addWidget(xRadius, 0, 0, 0);
layoutRadiusTextEdits->addWidget(yRadius, 0, 1, 0);
layoutRadiusTextEdits->addWidget(zRadius, 0, 3, 0);
layoutRadiusTextEdits->addWidget(rRadius, 1, 1, 0);
QWidget* RadiusTextEditsWidget = new QWidget();
RadiusTextEditsWidget->setLayout(layoutRadiusTextEdits);
RadiusTextEditsWidget->setFixedWidth(toolBoxSubWidgetsWidth);
QVBoxLayout *layoutRadius = new QVBoxLayout();
layoutRadius->addWidget(RadiusTextEditsWidget);
layoutRadius->addWidget(radiusQueryButton);
QWidget* RadiusWidget = new QWidget();
RadiusWidget->setLayout(layoutRadius);
RadiusWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(RadiusWidget, "Radius Query");
// NN Query
QGridLayout *layoutNNTextEdits = new QGridLayout();
layoutNNTextEdits->addWidget(xNeighbour, 0, 0, 0);
layoutNNTextEdits->addWidget(yNeighbour, 0, 1, 0);
layoutNNTextEdits->addWidget(zNeighbour, 0, 3, 0);
QWidget* NNTextEditsWidget = new QWidget();
NNTextEditsWidget->setLayout(layoutNNTextEdits);
NNTextEditsWidget->setFixedWidth(toolBoxSubWidgetsWidth);
QVBoxLayout *layoutNN = new QVBoxLayout();
layoutNN->addWidget(NNTextEditsWidget);
layoutNN->addWidget(nnQueryButton);
QWidget* NNWidget = new QWidget();
NNWidget->setLayout(layoutNN);
NNWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(NNWidget, "Nearest Neighbour");
// Thinning
QVBoxLayout *layoutThinning = new QVBoxLayout();
layoutThinning->addWidget(rThinning);
layoutThinning->addWidget(thinningButton);
QWidget* ThinningWidget = new QWidget();
ThinningWidget->setLayout(layoutThinning);
ThinningWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(ThinningWidget, "Thinning");
// Smoothing
QVBoxLayout *layoutSmoothing = new QVBoxLayout();
layoutSmoothing->addWidget(rSmoothing);
layoutSmoothing->addWidget(smoothingButton);
QWidget* SmoothingWidget = new QWidget();
SmoothingWidget->setLayout(layoutSmoothing);
SmoothingWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(SmoothingWidget, "Smoothing");
// Fitting
QVBoxLayout *layoutFitting = new QVBoxLayout();
layoutFitting->addWidget(labelFitting);
layoutFitting->addWidget(planeFittingButton);
layoutFitting->addWidget(lineFittingButton);
layoutFitting->addWidget(sphereFittingButton);
QWidget* FittingWidget = new QWidget();
FittingWidget->setLayout(layoutFitting);
FittingWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(FittingWidget, "Fitting");
// Color
QVBoxLayout *layoutColorByDist = new QVBoxLayout();
layoutColorByDist->addWidget(distanceColorMapButton);
QWidget* ColorByDistWidget = new QWidget();
ColorByDistWidget->setLayout(layoutColorByDist);
ColorByDistWidget->setFixedWidth(toolBoxWidgetsWidth);
toolBox->addItem(ColorByDistWidget, "Color by Distance");
/*---- Data Group Box ----*/
QGroupBox *dataBox = new QGroupBox(tr("Data"));
QVBoxLayout *layoutDataBox = new QVBoxLayout;
layoutDataBox->addWidget(labelPoints);
layoutDataBox->addWidget(labelCloudBounds);
dataBox->setLayout(layoutDataBox);
/*---- Side Bar ----*/
QVBoxLayout *layoutSideBar = new QVBoxLayout();
layoutSideBar->addWidget(dataBox);
layoutSideBar->addWidget(toolBox);
layoutSideBar->addWidget(labelTime);
QWidget* sideBarWidget = new QWidget();
sideBarWidget->setLayout(layoutSideBar);
sideBarWidget->setFixedWidth(toolBoxWidth);
/*---- Main Widget ----*/
glWidget = new MainGLWidget();
glWidget->resize(windowWidth, windowHeight);
glWidget->setMinimumWidth(windowWidth);
glWidget->setMinimumHeight(windowHeight);
QHBoxLayout *layoutMain = new QHBoxLayout();
layoutMain->addWidget(glWidget);
layoutMain->addWidget(sideBarWidget);
setLayout(layoutMain);
}
