Upsilon/python/port/mod/matplotlib/plot_store.h

#ifndef PYTHON_MATPLOTLIB_PLOT_STORE_H
#define PYTHON_MATPLOTLIB_PLOT_STORE_H

//#include <apps/shared/curve_view_range.h>
#include <apps/shared/interactive_curve_view_range.h>
extern "C" {
#include <py/runtime.h>
}

namespace Matplotlib {

class PlotStore : public Shared::InteractiveCurveViewRange {
public:
  PlotStore();
  void flush();

  // Iterators

  template <class T>
  class ListIterator {
  public:
    static ListIterator Begin(mp_obj_t list);
    static ListIterator End(mp_obj_t list);
    T operator*();
    ListIterator & operator++();
    bool operator!=(const ListIterator & it) const;
  private:
    ListIterator() : m_tupleIndex(0) {}
    mp_obj_t * m_tuples;
    size_t m_numberOfTuples;
    size_t m_tupleIndex;
  };

  template <class T>
  class Iterable {
  public:
    Iterable(mp_obj_t list) : m_list(list) {}
    T begin() const { return T::Begin(m_list); }
    T end() const { return T::End(m_list); }
  private:
    mp_obj_t m_list;
  };

  // Dots

  class Dot {
  public:
    Dot(mp_obj_t tuple);
    float x() const { return m_x; }
    float y() const { return m_y; }
    KDColor color() const { return m_color; }
  private:
    float m_x;
    float m_y;
    KDColor m_color;
  };

  void addDot(mp_obj_t x, mp_obj_t y, KDColor c);
  Iterable<ListIterator<Dot>> dots() { return Iterable<ListIterator<Dot>>(m_dots); }

  // Texts

  class Text {
  public:
    Text(mp_obj_t tuple);
    float x() const { return m_x; }
    float y() const { return m_y; }
    const char * string() const { return m_string; }
  private:
    float m_x;
    float m_y;
    const char * m_string;
  };

  void addText(mp_obj_t x, mp_obj_t y, mp_obj_t string);
  Iterable<ListIterator<Text>> texts() { return Iterable<ListIterator<Text>>(m_texts); }

  void setGrid(bool grid) { m_grid = grid; }
  bool grid() { return m_grid; }
private:
  mp_obj_t m_dots; // List of (x, y, color)
  mp_obj_t m_texts; // List of (x, y, string)
  mp_obj_t m_rects; // List of (x, y, w, h, color)
  mp_obj_t m_segments; // List of (x, y, dx, dy, style, color)

  bool m_grid;
};

}

#endif