gsHBSpline.h

Go to the documentation of this file.

#pragma once

#include <ostream>

#include <gsCore/gsLinearAlgebra.h>
#include <gsCore/gsGeometry.h>
#include <gsHSplines/gsHBSplineBasis.h>

namespace gismo
{

template<short_t d, class T>
class gsHBSpline : public gsGeoTraits<d,T>::GeometryBase
{
public:
    typedef typename gsGeoTraits<d,T>::GeometryBase Base;

    typedef gsHBSplineBasis<d,T> Basis;

    typedef memory::shared_ptr< gsHBSpline > Ptr;

    typedef memory::unique_ptr< gsHBSpline > uPtr;

    typedef typename
    util::conditional<d==1, gsConstantFunction<T>, gsHBSpline<static_cast<short_t>(d-1),T>
                      >::type BoundaryGeometryType;

    typedef typename gsHBSplineBasis<d,T>::BoundaryBasisType BoundaryBasisType;

public:

    gsHBSpline() { }

    gsHBSpline( const Basis * basis, const gsMatrix<T> * coefs ) :
    Base( basis, coefs ) { }

    gsHBSpline( const Basis & basis, const gsMatrix<T> & coefs ) :
    Base( basis, coefs ) { }

    explicit gsHBSpline( const gsTensorBSpline<d,T> & tbsp )
    {
        this->m_basis = new Basis( tbsp.basis() );
        this->m_coefs = tbsp.coefs();
    }

    GISMO_CLONE_FUNCTION(gsHBSpline)

    GISMO_BASIS_ACCESSORS

public:

    void slice(index_t dir_fixed,T par,BoundaryGeometryType & result) const
    {
        GISMO_ASSERT(d-1>=0,"d must be greater or equal than 1");
        GISMO_ASSERT(dir_fixed>=0 && static_cast<unsigned>(dir_fixed)<d,"cannot fix a dir greater than dim or smaller than 0");
        const BoundaryBasisType * bBasis = this->basis().basisSlice(dir_fixed,par);
        if(d==1)
        {
            gsMatrix<T> val(1,1),point;
            val(0,0)=par;
            this->eval_into(val,point);
            result=BoundaryGeometryType(*bBasis,point);
        }
        else
        {
            gsMatrix<T> vals,anchorsSlice,anchorsInGeom;
            bBasis->anchors_into(anchorsSlice);
            anchorsInGeom.resize(anchorsSlice.rows()+1,anchorsSlice.cols());
            anchorsInGeom.topRows(dir_fixed)=anchorsSlice.topRows(dir_fixed);
            anchorsInGeom.row(dir_fixed)=gsVector<T>::Constant(anchorsSlice.cols(),par);
            anchorsInGeom.bottomRows(anchorsSlice.rows()-dir_fixed)=anchorsSlice.bottomRows(anchorsSlice.rows()-dir_fixed);
            this->eval_into(anchorsInGeom,vals);
            BoundaryGeometryType* geom =
                    dynamic_cast<BoundaryGeometryType *>(bBasis->interpolateAtAnchors(vals).release()); //todo make it better
            GISMO_ASSERT(geom!=NULL,"bBasis should have BoundaryGeometryType.");
            result = *geom;
            delete geom;
        }
        delete bBasis;
    }
}; // class gsHBSpline

#ifdef GISMO_WITH_PYBIND11

  void pybind11_init_gsHBSpline2(pybind11::module &m);
  void pybind11_init_gsHBSpline3(pybind11::module &m);
  void pybind11_init_gsHBSpline4(pybind11::module &m);

#endif // GISMO_WITH_PYBIND11

} // namespace gismo

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