gsDomainIterator.h

Go to the documentation of this file.

#pragma once

//#include <gsCore/gsBasis.h> // todo: remove
#include <gsCore/gsDomain.h>
#include <gsCore/gsBoundary.h>

namespace gismo
{

template <class T>
class gsDomainIterator
{
public:
    typedef memory::shared_ptr< gsDomainIterator > Ptr;
    typedef memory::unique_ptr< gsDomainIterator > uPtr;

public:

    gsDomainIterator( ) : m_basis(NULL), m_isGood( true ), m_id(0) { }

    gsDomainIterator( const gsBasis<T>& basisParam, const boxSide & s = boundary::none)
        : center( gsVector<T>::Zero(basisParam.dim()) ), m_basis( &basisParam ),
          m_isGood( true ), m_side(s), m_id(0)
    { }

    virtual ~gsDomainIterator() { }

public:

    virtual bool next() = 0;

    virtual bool next(index_t increment) = 0;

    virtual void reset()
    {
        GISMO_NO_IMPLEMENTATION
    }

public:

    size_t id() const   { return m_id; }

    bool good() const   { return m_isGood; }

    short_t dim() const   { return center.size(); }

    virtual void adjacent( const gsVector<bool> & ,
                           gsDomainIterator & )
    {
        GISMO_NO_IMPLEMENTATION
    }

    const gsVector<T>& centerPoint () const
    { return center; }

    virtual const gsVector<T>& lowerCorner() const
    {
        GISMO_NO_IMPLEMENTATION
    }

    virtual const gsVector<T>& upperCorner() const
    {
        GISMO_NO_IMPLEMENTATION
    }

    virtual const T getPerpendicularCellSize() const
    {
        GISMO_NO_IMPLEMENTATION
    }

    virtual bool isBoundaryElement() const
    {
        GISMO_NO_IMPLEMENTATION
    }

    T getCellSize() const
    {
        return (upperCorner() - lowerCorner()).norm();
    }

    T getMinCellLength() const
    {
        return (upperCorner() - lowerCorner()).minCoeff();
    }

    T getMaxCellLength() const
    {
        return (upperCorner() - lowerCorner()).maxCoeff();
    }

    T volume() const
    { return (upperCorner() - lowerCorner()).prod(); }

    virtual size_t numElements() const
    {
        //\todo Remove this implementation. Probably using a shallow
        //copy, "reset" and "next" would do this better.

        // Buggy, and probably a terrible implementation,
        // but needed and therefore can be useful
        // sometimes.
        typename gsBasis<T>::domainIter domIter = m_basis->makeDomainIterator(m_side);

        size_t numEl = 0;
        for (; domIter->good(); domIter->next(), numEl++){}

        return numEl;
    }


    inline boxSide side() const {return m_side;}

public:

    gsVector<T> center;

protected:
    const gsBasis<T> * m_basis;

    bool m_isGood;

    boxSide m_side;

    size_t m_id;

private:
    // disable copying
    gsDomainIterator( const gsDomainIterator& );
    gsDomainIterator& operator= ( const gsDomainIterator& );
}; // class gsDomainIterator


//template<class T>
//std::ostream &operator<<(std::ostream &os, const gsDomainIterator<T>& b)
//{return b.print(os); }


} // namespace gismo