gsGeometryTransform.h

Go to the documentation of this file.

#pragma once

#include <gsCore/gsAffineFunction.h>

namespace gismo {

template <typename T>
class gsGeometryTransform : public gsFunction<T>
{
protected:
    const gsGeometry<T> * m_geo;
    gsAffineFunction<T> m_af;
public:
    typedef memory::shared_ptr< gsGeometryTransform > Ptr;

    typedef memory::unique_ptr< gsGeometryTransform > uPtr;

    GISMO_CLONE_FUNCTION(gsGeometryTransform)

    
    gsGeometryTransform(const gsGeometry<T> * geo,
                        const gsMatrix<T> mat, const gsVector<T> trans)
    : m_geo(geo), m_af(mat,trans) { }

    static uPtr make(const gsMatrix<T> mat, const gsVector<T> trans)
    { return uPtr(new gsGeometryTransform(mat, trans)); }

    virtual short_t domainDim() const { return m_geo->domainDim(); }
    virtual short_t targetDim() const { return m_geo->targetDim(); }

    gsMatrix<T> support() const { return m_geo->support(); }
    
    virtual void eval_into(const gsMatrix<T>& u, gsMatrix<T>& result) const;
    
    virtual void deriv_into(const gsMatrix<T>& u, gsMatrix<T>& result) const;
    
    virtual void deriv2_into( const gsMatrix<T>& u, gsMatrix<T>& result ) const
    { GISMO_NO_IMPLEMENTATION }
};

template<class T> void
gsGeometryTransform<T>::eval_into(const gsMatrix<T>& u, gsMatrix<T>& result) const
{
    m_geo->eval_into(u, result);
    result = ( m_af.matrix() * result ).colwise() + m_af.translation();
}

template<class T> void
gsGeometryTransform<T>::deriv_into(const gsMatrix<T>& u, gsMatrix<T>& result) const
{
    m_geo->deriv_into(u, result);
    auto Jt = result.reshaped( u.rows(), result.size()/u.rows() );
    gsMatrix<T> tmp;
    index_t n = m_geo->targetDim();
    for(index_t i = 0; i!=u.cols(); ++i)
    {
        tmp = Jt.middleCols(n*i,n);
        Jt.middleCols(n*i,n).noalias() = tmp * m_af.matrix().transpose();
    }
}

} // namespace gismo

//#ifndef GISMO_BUILD_LIB
//#include GISMO_HPP_HEADER(gsGeometryTransform.hpp)
//#endif