gsCurvatureSmoothing.h

Go to the documentation of this file.

#pragma once

#include <iostream>
#include <stdexcept>

#include <gsCore/gsGeometry.h>
#include <gsNurbs/gsBSpline.h>
#include <gsCore/gsLinearAlgebra.h>

namespace gismo
{
template<class T>
class gsCurvatureSmoothing
{
public:
    gsCurvatureSmoothing()
    {
        m_curve_original = NULL;
        m_curve_smooth = NULL;
    }

    gsCurvatureSmoothing(const gsBSpline<T> & init_curve, const gsMatrix<T>& param_values, const gsMatrix<T>& points)
    {
        m_curve_original = &init_curve;
        m_curve_smooth = init_curve.clone().release();
        m_param_values = param_values;
        m_points = points;
    }

    ~gsCurvatureSmoothing()
    {
        delete m_curve_smooth; // deleting the B-spline curve
    }


    const gsBSpline<T> & curveOriginal() const { return *m_curve_original; }
    const gsBSpline<T> & curveSmooth() const { return *m_curve_smooth; }


    void smoothTotalVariation(const T omega1, const T omega2, const T lamda, const T tau, const unsigned iter=50);

    void smoothTotalVariationSelectLamda(const T omega1, const T omega2, const gsMatrix<T> listlamdas, const unsigned iter=50);

    void smoothTotalVariationSelectLamda(const T omega1, const T omega2, const T lamda, const unsigned iter=50);

    void smoothHadenfeld(const unsigned smooth_degree, const T delta, const index_t iter_step, const index_t iter_total, gsVector<index_t> &iterated, const bool original=true);


    void smoothAllHadenfeld(const unsigned smooth_degree=4, const unsigned iter=500);

    void write(std::ostream &os);

    void computeApproxError(T & error);

    void computeApproxErrorL2(T & error);

    void computeApproxErrorLMax(T & error);

    void computeApproxErrorCoef(T & error);

    void computeCurvatureError(T & error);

private:

    const gsBSpline<T> * m_curve_original;
    gsBSpline<T> * m_curve_smooth;
    gsMatrix<T> m_param_values;
    gsMatrix<T> m_points;

    void compute_AllValues(gsBSplineBasis<T> * basis, gsMatrix<T> u, gsMatrix<T> *coefs, gsMatrix<T> & values0, gsMatrix<T> & values1, gsMatrix<T> & values2, gsMatrix<T> & values3);

    void compute_ObjectiveFunction(gsBSplineBasis<T> * basis, gsMatrix<T> *coefs, const T omega1, const T omega2, T &value);

    // TODO: What is exactly the purpose of this function; why do we want to change output?
    void reset(gsBSpline<T> * newCurve)
    {
        if (m_curve_smooth)
            delete m_curve_smooth;
        m_curve_smooth = newCurve;
    }

}; // class gsCurvatureSmoothing

} // namespace gismo

#ifndef GISMO_BUILD_LIB
#include GISMO_HPP_HEADER(gsCurvatureSmoothing.hpp)
#endif