TextureRasterizer.h

/*
 * Copyright 2025 Adobe. All rights reserved.
 * This file is licensed to you under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License. You may obtain a copy
 * of the License at http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under
 * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
 * OF ANY KIND, either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
#pragma once
 
#include <lagrange/SurfaceMesh.h>
#include <lagrange/image/Array3D.h>
#include <lagrange/image/View3D.h>
 
#include <Eigen/Geometry>
 
namespace lagrange::texproc {


struct CameraOptions
{
    Eigen::Affine3f view_transform = Eigen::Affine3f::Identity();

    Eigen::Projective3f projection_transform = Eigen::Projective3f::Identity();
};

struct TextureRasterizerOptions
{
    unsigned int depth_discontinuity_erosion_radius = 20;

    double depth_discontinuity_threshold = 1;

    double depth_precision = 1e-3;

    size_t width = 1024;

    size_t height = 1024;
};

template <typename Scalar, typename Index>
class TextureRasterizer
{
public:
    using Array3Df = image::experimental::Array3D<float>;
 
public:
    TextureRasterizer(
        const SurfaceMesh<Scalar, Index>& mesh,
        const TextureRasterizerOptions& options);

    ~TextureRasterizer();

    std::pair<Array3Df, Array3Df> weighted_texture_from_render(
        image::experimental::View3D<const float> image,
        const CameraOptions& options) const;
 
private:
 
    struct Impl;
    value_ptr<Impl> m_impl;

};

void filter_low_confidences(
    span<std::pair<image::experimental::Array3D<float>, image::experimental::Array3D<float>>>
        textures_and_weights,
    float low_ratio_threshold);

 
} // namespace lagrange::texproc