gsFlowFieldCreator.h

Go to the documentation of this file.

 
#pragma once
 
namespace gismo
{
 
template<class T>
class gsScaledOuterNormalField : public gsFunction<T> 
{
 
public: // *** Smart pointers ***
 
    typedef memory::shared_ptr< gsScaledOuterNormalField > Ptr;
    typedef memory::unique_ptr< gsScaledOuterNormalField > uPtr;
 
 
protected: // *** Class members ***
 
    index_t m_patch;
    boxSide m_side;
    const gsField<T>& m_scalarField;
 
 
public: // *** Constructor/destructor ***
 
    gsScaledOuterNormalField( index_t patch, boxSide side, const gsField<T>& scalarField):
    m_patch(patch), m_side(side), m_scalarField(scalarField)
    { }
 
    GISMO_CLONE_FUNCTION(gsScaledOuterNormalField)
 
 
public: // *** Member functions ***
 
    virtual void eval_into(const gsMatrix<T>& u, gsMatrix<T>& result) const
    {
        result.resize(this->targetDim(), u.cols());
 
        gsMapData<T> mapData(NEED_OUTER_NORMAL);
        mapData.side = m_side;
        mapData.points = u;
        m_scalarField.patches().patch(m_patch).computeMap(mapData);
 
        gsMatrix<T> scalarVals = getScalarValues(u);
 
        gsVector<> normal;
        for (index_t k = 0; k < u.cols(); k++)
        {
            outerNormal(mapData, k, m_side, normal);
            result.col(k) = scalarVals(0, k) * (1. / normal.norm()) * normal;
        }
    }
 
    short_t domainDim() const { return m_scalarField.patches().patch(m_patch).parDim(); }
    short_t targetDim() const { return m_scalarField.patches().patch(m_patch).geoDim(); }
 
    virtual std::ostream &print(std::ostream &os) const
    { os << "gsScaledOuterNormalField"; return os; };
 
 
protected: // *** Member functions ***
 
    virtual gsMatrix<T> getScalarValues(const gsMatrix<T>& u) const
    { return m_scalarField.function(m_patch).eval(u); }
 
 
}; // gsScaledOuterNormalField
 
// ----------------------------------------------------------------------------
 
template<class T>
class gsDiffScaledOuterNormalField : public gsScaledOuterNormalField<T> 
{
 
public:
    typedef gsScaledOuterNormalField<T> Base;
 
public: // *** Smart pointers ***
 
    typedef memory::shared_ptr< gsDiffScaledOuterNormalField > Ptr;
    typedef memory::unique_ptr< gsDiffScaledOuterNormalField > uPtr;
 
 
protected: // *** Class members ***
 
    T m_targetValue;
    
 
public: // *** Constructor/destructor ***
 
    gsDiffScaledOuterNormalField( index_t patch, boxSide side, const gsField<T>& scalarField, T targetValue):
    Base(patch, side, scalarField), m_targetValue(targetValue)
    { }
 
    GISMO_CLONE_FUNCTION(gsDiffScaledOuterNormalField)
 
 
public: // *** Member functions ***
 
    virtual std::ostream &print(std::ostream &os) const
    { os << "gsDiffScaledOuterNormalField"; return os; };
 
 
protected: // *** Member functions ***
 
    virtual gsMatrix<T> getScalarValues(const gsMatrix<T>& u) const
    { 
        gsMatrix<T> fieldVals = Base::getScalarValues(u);
 
        gsMatrix<T> targetVals(fieldVals.rows(), fieldVals.cols());
        targetVals.setOnes();
        targetVals *= m_targetValue;
 
        return (targetVals - fieldVals); 
    }
 
 
}; // gsDiffScaledOuterNormalField
 
} // namespace gismo