unorientedSphereFit.h

/*
 Copyright (C) 2013 Gael Guennebaud <gael.guennebaud@inria.fr>
 
 This Source Code Form is subject to the terms of the Mozilla Public
 License, v. 2.0. If a copy of the MPL was not distributed with this
 file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/
 
#pragma once
#include "../../defines.h"
#include "../concepts.h"
 
#include "../primitive/algebraicSphere.h" // used to define UnorientedSphereFit
#include "../tools/mean.h"                // used to define UnorientedSphereFit
 
#include <Eigen/Dense>
 
#define UNORIENTED_SPHERE_FIT_REQUIREMENTS \
    ProvidesAlgebraicSphere<T>&& ProvidesMeanPosition<T>&& ProvidesNormal<DataPoint>
#define UNORIENTED_SPHERE_DER_REQUIREMENTS \
    ProvidesBasketDiffUnitBase<T>&& ProvidesAlgebraicSphere<T>&& ProvidesMeanPositionDerivative<T>
 
namespace Ponca
{
    template <class DataPoint, class _NFilter, typename T>
        requires UNORIENTED_SPHERE_FIT_REQUIREMENTS
    class UnorientedSphereFitImpl : public T
    {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
 
    protected:
        using VectorB  = Eigen::Matrix<Scalar, DataPoint::Dim + 1, 1>;
        using MatrixBB = Eigen::Matrix<Scalar, DataPoint::Dim + 1, DataPoint::Dim + 1>;
 
    public:
        using Solver = Eigen::EigenSolver<MatrixBB>;
 
        MatrixBB m_matA{MatrixBB::Zero()}; 
        MatrixBB m_matQ{MatrixBB::Zero()}; 
        Scalar m_sumDotPP{0};              
        Solver m_solver;
 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS(UnorientedSphereFitImpl, unorientedSphereFit)
        PONCA_FITTING_DECLARE_INIT_ADD_FINALIZE
        PONCA_FITTING_IS_SIGNED(false)
 
    }; // class UnorientedSphereFitImpl
 
    template <class DataPoint, class _NFilter, typename T>
    using UnorientedSphereFit =
        UnorientedSphereFitImpl<DataPoint, _NFilter,
                                MeanPosition<DataPoint, _NFilter, AlgebraicSphere<DataPoint, _NFilter, T>>>;
 
    template <class DataPoint, class _NFilter, int DiffType, typename T>
        requires ProvidesBasketDiffUnitBase<T> && ProvidesAlgebraicSphere<T> && ProvidesMeanPositionDerivative<T>
    class UnorientedSphereDerImpl : public T
    {
    protected:
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
        PONCA_FITTING_DECLARE_DEFAULT_DER_TYPES
 
        using VectorB  = typename Base::VectorB;
        using MatrixBB = typename Base::MatrixBB;
 
    protected:
        // computation data
        MatrixBB m_dmatA[Base::NbDerivatives];
        ScalarArray m_dSumDotPP{ScalarArray::Zero()};
 
    public:
        // results
        ScalarArray m_dUc{ScalarArray::Zero()};
        VectorArray m_dUl{VectorArray::Zero()};
        ScalarArray m_dUq{ScalarArray::Zero()};
 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS_DER(UnorientedSphereDerImpl, unorientedSphereDer)
        PONCA_EXPLICIT_CAST_OPERATORS_DER(UnorientedSphereDerImpl, algebraicSphereDer)
        PONCA_EXPLICIT_CAST_OPERATORS_DER(UnorientedSphereDerImpl, implicitPrimitiveDer)
 
        PONCA_FITTING_DECLARE_INIT_ADDDER_FINALIZE
 
        PONCA_MULTIARCH inline ScalarArray dPotential() const;
        PONCA_MULTIARCH inline VectorArray dNormal() const;
 
        PONCA_MULTIARCH [[nodiscard]] inline ScalarArray dprattNorm2() const
        {
            return Scalar(2.) * Base::m_ul.transpose() * m_dUl - Scalar(4.) * Base::m_uq * m_dUc -
                   Scalar(4.) * Base::m_uc * m_dUq;
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline Scalar dprattNorm2(unsigned int _d) const
        {
            return Scalar(2.) * m_dUl.col(_d).dot(Base::m_ul) - Scalar(4.) * m_dUc.col(_d)[0] * Base::m_uq -
                   Scalar(4.) * m_dUq.col(_d)[0] * Base::m_uc;
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline Scalar dprattNorm(unsigned int _d) const
        {
            PONCA_MULTIARCH_STD_MATH(sqrt);
            return sqrt(dprattNorm2(_d));
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline ScalarArray dprattNorm() const
        {
            PONCA_MULTIARCH_STD_MATH(sqrt);
            return dprattNorm2().array().sqrt();
        }
 
        PONCA_MULTIARCH [[nodiscard]] inline bool applyPrattNorm();
 
    }; // class UnorientedSphereDerImpl
 
    template <class DataPoint, class _NFilter, int DiffType, typename T>
    using UnorientedSphereDer =
        UnorientedSphereDerImpl<DataPoint, _NFilter, DiffType, MeanPositionDer<DataPoint, _NFilter, DiffType, T>>;
 
} // namespace Ponca
 
#include "unorientedSphereFit.hpp"