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 "./primitive.h"
 
namespace Ponca {
 
    template<class DataPoint, class _NFilter, typename T>
    class MeanPosition : public T {
    PONCA_FITTING_DECLARE_DEFAULT_TYPES
 
    protected:
        enum { PROVIDES_MEAN_POSITION };
        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::getNeighborFilter().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>
    class MeanNormal : public T {
    PONCA_FITTING_DECLARE_DEFAULT_TYPES
 
    protected:
        enum { PROVIDES_MEAN_NORMAL };
        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>
    class MeanPositionDer : public T {
    PONCA_FITTING_DECLARE_DEFAULT_TYPES
    PONCA_FITTING_DECLARE_DEFAULT_DER_TYPES
 
    protected:
        enum {
            Check = Base::PROVIDES_PRIMITIVE_DERIVATIVE &&
                    Base::PROVIDES_MEAN_POSITION,
            PROVIDES_MEAN_POSITION_DERIVATIVE,    
        };
 
        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>
    class MeanNormalDer : public T {
        PONCA_FITTING_DECLARE_DEFAULT_TYPES
        PONCA_FITTING_DECLARE_DEFAULT_DER_TYPES
 
    protected:
        enum {
            Check = Base::PROVIDES_PRIMITIVE_DERIVATIVE && 
                    Base::PROVIDES_MEAN_NORMAL,
            PROVIDES_MEAN_NORMAL_DERIVATIVE,  
        };
 
        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