AngleBasedFlattening.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <memory>
 
#include <opencv2/core.hpp>
 
#include "vc/core/types/ITKMesh.hpp"
#include "vc/core/types/UVMap.hpp"
#include "vc/texturing/FlatteningAlgorithm.hpp"
 
namespace volcart::texturing
{
class AngleBasedFlattening : public FlatteningAlgorithm
{
public:
    enum class Solver { SparseLU = 0, ConjugateGradient = 1 };
 
    static constexpr std::size_t DEFAULT_ITERATIONS{10};
 
    using Pointer = std::shared_ptr<AngleBasedFlattening>;
 
    AngleBasedFlattening() = default;
 
    explicit AngleBasedFlattening(const ITKMesh::Pointer& m);
 
    template <typename... Args>
    static auto New(Args... args) -> Pointer
    {
        return std::make_shared<AngleBasedFlattening>(
            std::forward<Args>(args)...);
    }
 
    ~AngleBasedFlattening() override = default;
    void setUseABF(bool a);
 
    [[nodiscard]] auto useABF() const -> bool;
 
    void setABFMaxIterations(std::size_t i);
 
    [[nodiscard]] auto abfMaxIterations() const -> std::size_t;
    void setSolver(Solver solver);
 
    [[nodiscard]] auto solver() const -> Solver;
    auto compute() -> ITKMesh::Pointer override;
private:
    bool useABF_{true};
    Solver solver_{Solver::SparseLU};
    std::size_t maxABFIterations_{DEFAULT_ITERATIONS};
};
}  // namespace volcart::texturing