src/tools/lvr2_mesh_reducer/Main.cpp

Go to the documentation of this file.

#include <iostream>
#include <memory>
#include <tuple>
#include <stdlib.h>

#include <boost/optional.hpp>
#include "Options.hpp"



using std::unique_ptr;
using std::make_unique;

#include "lvr2/io/Progress.hpp"
#include "lvr2/io/Model.hpp"
#include "lvr2/geometry/HalfEdgeMesh.hpp"
#include "lvr2/algorithm/FinalizeAlgorithms.hpp"
#include "lvr2/algorithm/NormalAlgorithms.hpp"
#include "lvr2/algorithm/ReductionAlgorithms.hpp"

using Vec = lvr2::BaseVector<float>;

int main(int argc, char** argv)
{
    // =======================================================================
    // Parse and print command line parameters
    // =======================================================================
    // Parse command line arguments
    meshreduce::Options options(argc, argv);
  
    // Exit if options had to generate a usage message
    // (this means required parameters are missing)
    if (options.printUsage())
    {
        return EXIT_SUCCESS;
    }
    std::cout << options << std::endl;
    cout << "LOAD" << endl;
    lvr2::ModelPtr model = lvr2::ModelFactory::readModel(options.getInputFileName());
    cout << "MODEL" << endl;
    cout << model << endl;
    lvr2::MeshBufferPtr meshBuffer = model->m_mesh;
    cout << meshBuffer << endl;
    lvr2::HalfEdgeMesh<Vec> mesh(meshBuffer);

    std::cout << lvr2::timestamp << "Computing face normals..." << std::endl;

    // Calculate initial face normals
    auto faceNormals = calcFaceNormals(mesh);

    // Reduce mesh complexity
    const auto reductionRatio = options.getEdgeCollapseReductionRatio();
    std::cout << lvr2::timestamp << "Collapsing faces..." << std::endl;

    if (reductionRatio > 0.0)
    {
        if (reductionRatio > 1.0)
        {
            throw "The reduction ratio needs to be between 0 and 1!";
        }

        // Each edge collapse removes two faces in the general case.
        // TODO: maybe we should calculate this differently...
        const auto count = static_cast<size_t>((mesh.numFaces() / 2) * reductionRatio);
        auto collapsedCount = simpleMeshReduction(mesh, count, faceNormals);
    }

    // =======================================================================
    // Finalize mesh
    // =======================================================================
    lvr2::SimpleFinalizer<Vec> finalizer;
    // Run finalize algorithm
    auto buffer = finalizer.apply(mesh);

    // =======================================================================
    // Write all results (including the mesh) to file
    // =======================================================================
    // Create output model and save to file
    auto m = lvr2::ModelPtr( new lvr2::Model(buffer));

    lvr2::ModelFactory::saveModel(m, "reduced_mesh.ply");
   
    cout << lvr2::timestamp << "Program end." << endl;

    return 0;
}