mean.h

/*
 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"
 
#define MEAN_NORMAL_REQUIREMENTS ProvidesNormal<DataPoint>
#define MEAN_POSITION_DER_REQUIREMENTS ProvidesBasketDiffUnitBase<T>&& ProvidesMeanPosition<T>
#define MEAN_NORMAL_DER_REQUIREMENTS ProvidesBasketDiffUnitBase<T>&& ProvidesMeanNormal<T>
 
namespace Ponca
{
 
    template <class DataPoint, class _NFilter, typename T>
    class MeanPosition : public T
    {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
 
    protected:
        VectorType m_sumP{VectorType::Zero()}; 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS(MeanPosition, meanPosition)
        PONCA_FITTING_DECLARE_INIT
        PONCA_FITTING_DECLARE_ADDNEIGHBOR
 
        PONCA_MULTIARCH inline VectorType barycenter() const
        {
            return Base::getNeighborFrame().convertToGlobalBasis(barycenterLocal());
        }
 
        PONCA_MULTIARCH inline Scalar barycenterDistance() const { return barycenterLocal().norm(); }
 
    protected:
        PONCA_MULTIARCH inline VectorType barycenterLocal() const { return (m_sumP / Base::getWeightSum()); }
 
    }; // class MeanPosition
 
    template <class DataPoint, class _NFilter, typename T>
        requires MEAN_NORMAL_REQUIREMENTS
    class MeanNormal : public T
    {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
 
    protected:
        VectorType m_sumN{VectorType::Zero()}; 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS(MeanNormal, meanNormal)
        PONCA_FITTING_DECLARE_INIT
        PONCA_FITTING_DECLARE_ADDNEIGHBOR
 
        PONCA_MULTIARCH [[nodiscard]] inline VectorType meanNormalVector() const
        {
            return (m_sumN / Base::getWeightSum());
        }
 
    }; // class MeanNormal
 
    template <class DataPoint, class _NFilter, int DiffType, typename T>
        requires MEAN_POSITION_DER_REQUIREMENTS
    class MeanPositionDer : public T
    {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
        PONCA_FITTING_DECLARE_DEFAULT_DER_TYPES
 
    protected:
        VectorArray m_dSumP{VectorArray::Zero()};
 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS_DER(MeanPositionDer, meanPositionDer)
        PONCA_FITTING_DECLARE_INIT
        PONCA_FITTING_DECLARE_ADDNEIGHBOR_DER
 
        PONCA_MULTIARCH [[nodiscard]] VectorArray barycenterDerivatives() const
        {
            return (m_dSumP - Base::barycenterLocal() * Base::m_dSumW) / Base::getWeightSum();
        }
 
    }; // class MeanPositionDer
 
    template <class DataPoint, class _NFilter, int DiffType, typename T>
        requires MEAN_NORMAL_DER_REQUIREMENTS
    class MeanNormalDer : public T
    {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
        PONCA_FITTING_DECLARE_DEFAULT_DER_TYPES
 
    protected:
        VectorArray m_dSumN{VectorArray::Zero()};
 
    public:
        PONCA_EXPLICIT_CAST_OPERATORS_DER(MeanNormalDer, meanNormalDer)
        PONCA_FITTING_DECLARE_INIT
        PONCA_FITTING_DECLARE_ADDNEIGHBOR_DER
 
 
        PONCA_MULTIARCH [[nodiscard]] VectorArray dMeanNormal() const
        {
            return (m_dSumN - Base::meanNormalVector() * Base::m_dSumW) / Base::getWeightSum();
        }
 
    }; // class MeanNormalDer
 
#include "mean.hpp"
 
} // namespace Ponca