ViewMapIterators.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once

 
#include "ViewMap.h"
 
#include "../system/Iterator.h"  //soc
 
namespace Freestyle {
 
/**********************************/
/*                                */
/*                                */
/*             ViewMap            */
/*                                */
/*                                */
/**********************************/
 
/**********************************/
/*                                */
/*                                */
/*             ViewVertex         */
/*                                */
/*                                */
/**********************************/
 
namespace ViewVertexInternal {

class orientedViewEdgeIterator : public Iterator {
 public:
  friend class ViewVertex;
  friend class TVertex;
  friend class NonTVertex;
  friend class ViewEdge;
 
  // FIXME
  typedef TVertex::edge_pointers_container edge_pointers_container;
  typedef NonTVertex::edges_container edges_container;
 
 protected:
  Nature::VertexNature _Nature;  // the nature of the underlying vertex
  // T vertex attributes
  edge_pointers_container::iterator _tbegin;
  edge_pointers_container::iterator _tend;
  edge_pointers_container::iterator _tvertex_iter;
 
  // Non TVertex attributes
  edges_container::iterator _begin;
  edges_container::iterator _end;
  edges_container::iterator _nontvertex_iter;
 
 public:
  inline orientedViewEdgeIterator() {}
 
  inline orientedViewEdgeIterator(Nature::VertexNature iNature)
  {
    _Nature = iNature;
  }

  orientedViewEdgeIterator(const orientedViewEdgeIterator &iBrother)
  {
    _Nature = iBrother._Nature;
    if (_Nature & Nature::T_VERTEX) {
      _tbegin = iBrother._tbegin;
      _tend = iBrother._tend;
      _tvertex_iter = iBrother._tvertex_iter;
    }
    else {
      _begin = iBrother._begin;
      _end = iBrother._end;
      _nontvertex_iter = iBrother._nontvertex_iter;
    }
  }
 
  virtual ~orientedViewEdgeIterator() {}
 
 public:
  inline orientedViewEdgeIterator(edge_pointers_container::iterator begin,
                                  edge_pointers_container::iterator end,
                                  edge_pointers_container::iterator iter)
  {
    _Nature = Nature::T_VERTEX;
    _tbegin = begin;
    _tend = end;
    _tvertex_iter = iter;
  }
 
  inline orientedViewEdgeIterator(edges_container::iterator begin,
                                  edges_container::iterator end,
                                  edges_container::iterator iter)
  {
    _Nature = Nature::NON_T_VERTEX;
    _begin = begin;
    _end = end;
    _nontvertex_iter = iter;
  }
 
 public:
  virtual bool isBegin() const
  {
    if (_Nature & Nature::T_VERTEX) {
      return (_tvertex_iter == _tbegin);
    }
    else {
      return (_nontvertex_iter == _begin);
    }
  }

  virtual bool isEnd() const
  {
    if (_Nature & Nature::T_VERTEX) {
      return (_tvertex_iter == _tend);
    }
    else {
      return (_nontvertex_iter == _end);
    }
  }
 
  // operators
  // operator corresponding to ++i
  virtual orientedViewEdgeIterator &operator++()
  {
    increment();
    return *this;
  }
 
  // operator corresponding to i++, i.e. which returns the value *and then* increments.
  // That's why we store the value in a temp.
  virtual orientedViewEdgeIterator operator++(int)
  {
    orientedViewEdgeIterator tmp = *this;
    increment();
    return tmp;
  }
 
  // comparability
  virtual bool operator!=(const orientedViewEdgeIterator &b) const
  {
    if (_Nature & Nature::T_VERTEX) {
      return (_tvertex_iter != b._tvertex_iter);
    }
    else {
      return (_nontvertex_iter != b._nontvertex_iter);
    }
  }

  virtual bool operator==(const orientedViewEdgeIterator &b) const
  {
    return !(*this != b);
  }
 
  // dereferencing
  virtual ViewVertex::directedViewEdge &operator*() const
  {
    if (_Nature & Nature::T_VERTEX) {
      // return _tvertex_iter;
      return **_tvertex_iter;
    }
    else {
      return (*_nontvertex_iter);
    }
  }

  virtual ViewVertex::directedViewEdge *operator->() const
  {
    return &(operator*());
  }
 
 public:
  virtual inline int increment()
  {
    if (_Nature & Nature::T_VERTEX) {
      ViewVertex::directedViewEdge tmp = (**_tvertex_iter);
      ++_tvertex_iter;
      if (_tvertex_iter != _tend) {
        // FIXME : pquoi deja ?
        ViewVertex::directedViewEdge tmp2 = (**_tvertex_iter);
        if (tmp2.first == tmp.first) {
          ++_tvertex_iter;
        }
      }
    }
    else {
      ++_nontvertex_iter;
    }
    return 0;
  }
 
  MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:orientedViewEdgeIterator")
};
 
}  // namespace ViewVertexInternal
 
/**********************************/
/*                                */
/*                                */
/*             ViewEdge           */
/*                                */
/*                                */
/**********************************/
 
namespace ViewEdgeInternal {
 
//
// SVertexIterator
//
 
class SVertexIterator : public Interface0DIteratorNested {
 public:
  SVertexIterator()
  {
    _vertex = nullptr;
    _begin = nullptr;
    _previous_edge = nullptr;
    _next_edge = nullptr;
    _t = 0;
  }
 
  SVertexIterator(const SVertexIterator &vi)
  {
    _vertex = vi._vertex;
    _begin = vi._begin;
    _previous_edge = vi._previous_edge;
    _next_edge = vi._next_edge;
    _t = vi._t;
  }
 
  SVertexIterator(SVertex *v, SVertex *begin, FEdge *prev, FEdge *next, float t)
  {
    _vertex = v;
    _begin = begin;
    _previous_edge = prev;
    _next_edge = next;
    _t = t;
  }
 
  SVertexIterator &operator=(const SVertexIterator &vi)
  {
    _vertex = vi._vertex;
    _begin = vi._begin;
    _previous_edge = vi._previous_edge;
    _next_edge = vi._next_edge;
    _t = vi._t;
    return *this;
  }
 
  virtual ~SVertexIterator() {}
 
  virtual string getExactTypeName() const
  {
    return "SVertexIterator";
  }
 
  virtual SVertex &operator*()
  {
    return *_vertex;
  }
 
  virtual SVertex *operator->()
  {
    return &(operator*());
  }
 
  virtual SVertexIterator &operator++()
  {
    increment();
    return *this;
  }
 
  virtual SVertexIterator operator++(int)
  {
    SVertexIterator ret(*this);
    increment();
    return ret;
  }
 
  virtual SVertexIterator &operator--()
  {
    decrement();
    return *this;
  }
 
  virtual SVertexIterator operator--(int)
  {
    SVertexIterator ret(*this);
    decrement();
    return ret;
  }
 
  virtual int increment()
  {
    if (!_next_edge) {
      _vertex = nullptr;
      return 0;
    }
    _t += (float)_next_edge->getLength2D();
    _vertex = _next_edge->vertexB();
    _previous_edge = _next_edge;
    _next_edge = _next_edge->nextEdge();
    return 0;
  }
 
  virtual int decrement()
  {
    if (!_previous_edge) {
      _vertex = nullptr;
      return 0;
    }
    if ((!_next_edge) && (!_vertex)) {
      _vertex = _previous_edge->vertexB();
      return 0;
    }
    _t -= (float)_previous_edge->getLength2D();
    _vertex = _previous_edge->vertexA();
    _next_edge = _previous_edge;
    _previous_edge = _previous_edge->previousEdge();
    return 0;
  }
 
  virtual bool isBegin() const
  {
    return _vertex == _begin;
  }
 
  virtual bool isEnd() const
  {
    return (!_vertex) || (_vertex == _begin && _previous_edge);
  }
 
  virtual float t() const
  {
    return _t;
  }
 
  virtual float u() const
  {
    return _t / (float)_next_edge->viewedge()->getLength2D();
  }
 
  virtual bool operator==(const Interface0DIteratorNested &it) const
  {
    const SVertexIterator *it_exact = dynamic_cast<const SVertexIterator *>(&it);
    if (!it_exact) {
      return false;
    }
    return (_vertex == it_exact->_vertex);
  }
 
  virtual SVertexIterator *copy() const
  {
    return new SVertexIterator(*this);
  }
 
 private:
  SVertex *_vertex;
  SVertex *_begin;
  FEdge *_previous_edge;
  FEdge *_next_edge;
  float _t;  // curvilinear abscissa
 
  MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:SVertexIterator")
};
 
//
// ViewEdgeIterator (base class)
//

class ViewEdgeIterator : public Iterator {
 public:
  ViewEdgeIterator(ViewEdge *begin = nullptr, bool orientation = true)
  {
    _orientation = orientation;
    _edge = begin;
    _begin = begin;
  }

  ViewEdgeIterator(const ViewEdgeIterator &it)
  {
    _orientation = it._orientation;
    _edge = it._edge;
    _begin = it._begin;
  }
 
  virtual ~ViewEdgeIterator() {}

  virtual string getExactTypeName() const
  {
    return "ViewEdgeIterator";
  }

  ViewEdge *getCurrentEdge()
  {
    return _edge;
  }

  void setCurrentEdge(ViewEdge *edge)
  {
    _edge = edge;
  }

  ViewEdge *getBegin()
  {
    return _begin;
  }

  void setBegin(ViewEdge *begin)
  {
    _begin = begin;
  }

  bool getOrientation() const
  {
    return _orientation;
  }

  void setOrientation(bool orientation)
  {
    _orientation = orientation;
  }

  void changeOrientation()
  {
    _orientation = !_orientation;
  }

  virtual ViewEdge *operator*()
  {
    return _edge;
  }
 
  virtual ViewEdge *operator->()
  {
    return operator*();
  }

  virtual ViewEdgeIterator &operator++()
  {
    increment();
    return *this;
  }

  virtual ViewEdgeIterator operator++(int)
  {
    ViewEdgeIterator tmp(*this);
    increment();
    return tmp;
  }

  virtual int increment()
  {
    cerr << "Warning: method increment() not implemented" << endl;
    return 0;
  }

  virtual ViewEdgeIterator &operator--()
  {
    decrement();
    return *this;
  }

  virtual ViewEdgeIterator operator--(int)
  {
    ViewEdgeIterator tmp(*this);
    decrement();
    return tmp;
  }

  virtual int decrement()
  {
    cerr << "Warning: method decrement() not implemented" << endl;
    return 0;
  }

  virtual bool isBegin() const
  {
    return _edge == _begin;
  }

  virtual bool isEnd() const
  {
    return !_edge;
  }

  virtual bool operator==(ViewEdgeIterator &it) const
  {
    return _edge == it._edge;
  }

  virtual bool operator!=(ViewEdgeIterator &it) const
  {
    return !(*this == it);
  }
 
 protected:
  bool _orientation;
  ViewEdge *_edge;
  ViewEdge *_begin;
 
  MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:ViewEdgeIterator")
};
 
}  // end of namespace ViewEdgeInternal
 
} /* namespace Freestyle */