ElementIterator.h

Go to the documentation of this file.

// Copyright (C) 2004 Johan Hoffman and Anders Logg.
// Licensed under the GNU GPL Version 2.

#ifndef __ELEMENT_ITERATOR_H
#define __ELEMENT_ITERATOR_H

#include <dolfin/NewArray.h>

namespace dolfin
{

  class Element;
  class ElementGroup;
  class ElementGroupList;

  typedef Element* ElementPointer;


  class ElementIterator
  {
  public:

    ElementIterator(ElementGroupList& groups);

    ElementIterator(ElementGroup& group);
    
    ~ElementIterator();
    
    operator ElementPointer() const;
    
    ElementIterator& operator++();
    
    Element& operator*() const;

    Element* operator->() const;

    bool end();
    
    class GenericElementIterator
    {
    public:
      virtual void operator++() = 0;
      virtual Element& operator*() const = 0;
      virtual Element* operator->() const = 0;
      virtual Element* pointer() const = 0;
      virtual bool end() = 0;
    };

    class ElementGroupListElementIterator : public GenericElementIterator
    {
    public:
      ElementGroupListElementIterator(ElementGroupList& groups);
      void operator++();
      Element& operator*() const;
      Element* operator->() const;
      Element* pointer() const;
      bool end();
    private:
      Element* element;
      NewArray<Element*>::iterator element_it;
      NewArray<Element*>::iterator element_at_end;
      NewArray<ElementGroup*>::iterator group_it;
      NewArray<ElementGroup*>::iterator group_at_end;
    };

    class ElementGroupElementIterator : public GenericElementIterator
    {
    public:
      ElementGroupElementIterator(ElementGroup& group);
      void operator++();
      Element& operator*() const;
      Element* operator->() const;
      Element* pointer() const;
      bool end();
    private:
      NewArray<Element*>::iterator it;
      NewArray<Element*>::iterator at_end;
    };

  private:

    GenericElementIterator* it;

  };

}

#endif

Documentation automatically generated with Doxygen on 10 Sep 2004.