mirror of
https://github.com/UpsilonNumworks/Upsilon.git
synced 2026-01-19 00:37:25 +01:00
113 lines
2.9 KiB
C++
113 lines
2.9 KiB
C++
#include "plot_store.h"
|
|
|
|
namespace Matplotlib {
|
|
|
|
PlotStore::PlotStore() : Shared::InteractiveCurveViewRange(),
|
|
m_grid(false)
|
|
{
|
|
flush();
|
|
}
|
|
|
|
void PlotStore::flush() {
|
|
m_dots = mp_obj_new_list(0, nullptr);
|
|
m_segments = mp_obj_new_list(0, nullptr);
|
|
m_labels = mp_obj_new_list(0, nullptr);
|
|
}
|
|
|
|
// Iterators
|
|
|
|
template <class T>
|
|
PlotStore::ListIterator<T> PlotStore::ListIterator<T>::Begin(mp_obj_t list) {
|
|
ListIterator<T> it;
|
|
mp_obj_list_get(list, &(it.m_numberOfTuples), &(it.m_tuples));
|
|
return it;
|
|
}
|
|
|
|
template <class T>
|
|
PlotStore::ListIterator<T> PlotStore::ListIterator<T>::End(mp_obj_t list) {
|
|
ListIterator<T> it;
|
|
mp_obj_list_get(list, &(it.m_numberOfTuples), &(it.m_tuples));
|
|
if (it.m_numberOfTuples > 0) {
|
|
it.m_tupleIndex = it.m_numberOfTuples;
|
|
}
|
|
return it;
|
|
}
|
|
|
|
template <class T>
|
|
PlotStore::ListIterator<T> & PlotStore::ListIterator<T>::operator++() {
|
|
if (m_tupleIndex < m_numberOfTuples) {
|
|
m_tupleIndex++;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template <class T>
|
|
bool PlotStore::ListIterator<T>::operator!=(const PlotStore::ListIterator<T> & it) const {
|
|
return m_tupleIndex != it.m_tupleIndex;
|
|
};
|
|
|
|
template <class T>
|
|
T PlotStore::ListIterator<T>::operator*() {
|
|
return T(m_tuples[m_tupleIndex]);
|
|
};
|
|
|
|
// Dot
|
|
|
|
template class PlotStore::ListIterator<PlotStore::Dot>;
|
|
|
|
PlotStore::Dot::Dot(mp_obj_t tuple) {
|
|
mp_obj_t * elements;
|
|
mp_obj_get_array_fixed_n(tuple, 3, &elements);
|
|
m_x = mp_obj_get_float(elements[0]);
|
|
m_y = mp_obj_get_float(elements[1]);
|
|
m_color = KDColor::RGB16(mp_obj_get_int(elements[2]));
|
|
}
|
|
|
|
void PlotStore::addDot(mp_obj_t x, mp_obj_t y, KDColor c) {
|
|
mp_obj_t color = mp_obj_new_int(c);
|
|
mp_obj_t items[3] = {x, y, color};
|
|
mp_obj_t tuple = mp_obj_new_tuple(3, items);
|
|
mp_obj_list_append(m_dots, tuple);
|
|
}
|
|
|
|
// Segment
|
|
|
|
template class PlotStore::ListIterator<PlotStore::Segment>;
|
|
|
|
PlotStore::Segment::Segment(mp_obj_t tuple) {
|
|
mp_obj_t * elements;
|
|
mp_obj_get_array_fixed_n(tuple, 5, &elements);
|
|
m_xStart = mp_obj_get_float(elements[0]);
|
|
m_yStart = mp_obj_get_float(elements[1]);
|
|
m_xEnd = mp_obj_get_float(elements[2]);
|
|
m_yEnd = mp_obj_get_float(elements[3]);
|
|
m_color = KDColor::RGB16(mp_obj_get_int(elements[4]));
|
|
}
|
|
|
|
void PlotStore::addSegment(mp_obj_t xStart, mp_obj_t yStart, mp_obj_t xEnd, mp_obj_t yEnd, KDColor c) {
|
|
mp_obj_t color = mp_obj_new_int(c);
|
|
mp_obj_t items[5] = {xStart, yStart, xEnd, yEnd, color};
|
|
mp_obj_t tuple = mp_obj_new_tuple(5, items);
|
|
mp_obj_list_append(m_segments, tuple);
|
|
}
|
|
|
|
// Label
|
|
|
|
template class PlotStore::ListIterator<PlotStore::Label>;
|
|
|
|
PlotStore::Label::Label(mp_obj_t tuple) {
|
|
mp_obj_t * elements;
|
|
mp_obj_get_array_fixed_n(tuple, 3, &elements);
|
|
m_x = mp_obj_get_float(elements[0]);
|
|
m_y = mp_obj_get_float(elements[1]);
|
|
m_string = mp_obj_str_get_str(elements[2]);
|
|
}
|
|
|
|
void PlotStore::addLabel(mp_obj_t x, mp_obj_t y, mp_obj_t string) {
|
|
mp_obj_t items[3] = {x, y, string};
|
|
mp_obj_t tuple = mp_obj_new_tuple(3, items);
|
|
mp_obj_list_append(m_labels, tuple);
|
|
}
|
|
|
|
}
|