weightFunc.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 PONCA_MULTIARCH_INCLUDE_CU_STD(utility)
 
namespace Ponca
{
template <class DataPoint, class WeightKernel>
class DistWeightFunc
{
public:
    using Scalar =  typename DataPoint::Scalar;
    using VectorType =  typename DataPoint::VectorType;
    using MatrixType = typename DataPoint::MatrixType;
    using WeightReturnType = PONCA_MULTIARCH_CU_STD_NAMESPACE(pair)<Scalar, VectorType>;
 
    PONCA_MULTIARCH inline DistWeightFunc(const VectorType & _evalPos = VectorType::Zero(),
                                          const Scalar& _t = Scalar(1.))
    : m_t(_t), m_p(_evalPos)
    {
        //\todo manage that assert on __host__ and __device__
        //assert(_t > Scalar(0));
    }
 
    PONCA_MULTIARCH inline const VectorType& basisCenter() const { return m_p; }
 
    PONCA_MULTIARCH inline VectorType convertToGlobalBasis(const VectorType& _q) const;
 
    PONCA_MULTIARCH inline VectorType convertToLocalBasis(const VectorType& _q) const;
 
    PONCA_MULTIARCH inline WeightReturnType w(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
 
    PONCA_MULTIARCH inline VectorType spacedw(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
 
    PONCA_MULTIARCH inline MatrixType spaced2w(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
    PONCA_MULTIARCH inline Scalar scaledw(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
    PONCA_MULTIARCH inline Scalar scaled2w(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
    PONCA_MULTIARCH inline VectorType scaleSpaced2w(const VectorType& _q,
        const DataPoint&  /*attributes*/) const;
 
    PONCA_MULTIARCH inline Scalar evalScale() const { return m_t; }
 
    PONCA_MULTIARCH inline const VectorType & evalPos() const { return m_p; }
 
protected:
    Scalar       m_t;  
    WeightKernel m_wk; 
    VectorType   m_p;  
};// class DistWeightFunc
 
 
template <class DataPoint>
class NoWeightFunc
{
public:
    using Scalar =  typename DataPoint::Scalar;
    using VectorType =  typename DataPoint::VectorType;
    using MatrixType = typename DataPoint::MatrixType;
    using WeightReturnType = PONCA_MULTIARCH_CU_STD_NAMESPACE(pair)<Scalar, VectorType>;
 
    PONCA_MULTIARCH inline NoWeightFunc(const VectorType & _evalPos = VectorType::Zero()) : m_p(_evalPos){ }
 
    PONCA_MULTIARCH inline void init( const VectorType& _evalPos = VectorType::Zero() ) { m_p = _evalPos; }
 
    PONCA_MULTIARCH inline const VectorType& basisCenter() const
    { return m_p; }
 
    PONCA_MULTIARCH inline VectorType convertToLocalBasis(const VectorType& _q) const
    { return _q - m_p; }
 
    PONCA_MULTIARCH inline WeightReturnType w(const VectorType& _q,
                                              const DataPoint&  /*attributes*/) const
    {
        VectorType q = convertToLocalBasis(_q);
        return {Scalar(1), q};
    }
 
 
    PONCA_MULTIARCH inline VectorType spacedw(const VectorType& /*_q*/,
                                              const DataPoint&  /*attributes*/) const
    { return VectorType::Zeros(); }
 
 
    PONCA_MULTIARCH inline MatrixType spaced2w(const VectorType& /*_q*/,
                                               const DataPoint&  /*attributes*/) const
    { return MatrixType::Zeros(); }
 
    PONCA_MULTIARCH inline Scalar scaledw(const VectorType& /*_q*/,
                                          const DataPoint&  /*attributes*/) const
    { return Scalar(0); }
 
    PONCA_MULTIARCH inline Scalar scaled2w(const VectorType& /*_q*/,
                                           const DataPoint&  /*attributes*/) const
    { return Scalar(0); }
 
    PONCA_MULTIARCH inline VectorType scaleSpaced2w(const VectorType& /*_q*/,
                                                    const DataPoint&  /*attributes*/) const
    { return VectorType::Zeros(); }
 
private:
    VectorType   m_p;  
};// class DistWeightFunc
 
#include "weightFunc.hpp"
 
}// namespace Ponca