FlatteningError.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
 
#include <opencv2/core.hpp>
 
#include "vc/core/types/ITKMesh.hpp"
#include "vc/core/util/ColorMaps.hpp"
#include "vc/core/util/Iteration.hpp"
 
namespace volcart::texturing
{
 
template <typename ErrContainer>
cv::Mat PlotPerFaceError(
    const cv::Mat& cellMap,
    const ErrContainer& errorMap,
    float defaultValue = 0)
{
    cv::Mat output(cellMap.rows, cellMap.cols, CV_32FC1);
    output = cv::Scalar::all(defaultValue);
    for (const auto& [y, x] : range2D(cellMap.rows, cellMap.cols)) {
        auto cell = cellMap.at<std::int32_t>(y, x);
        if (cell >= 0) {
            auto error = errorMap.at(cell);
            output.at<float>(y, x) = static_cast<float>(error);
        }
    }
    return output;
}
 
struct LStretchMetrics {
    double l2{0};
    double lInf{0};
    std::vector<double> faceL2;
    std::vector<double> faceLInf;
};
 
LStretchMetrics LStretch(
    const ITKMesh::Pointer& mesh3D, const ITKMesh::Pointer& mesh2D);
 
LStretchMetrics InvertLStretchMetrics(const LStretchMetrics& metrics);
 
std::vector<cv::Mat> PlotLStretchError(
    const LStretchMetrics& metrics,
    const cv::Mat& cellMap,
    ColorMap cm = ColorMap::Plasma,
    bool drawLegend = false);
 
}  // namespace volcart::texturing