template to handle strings on x axis w/ legend

impressostudios · web-flow · commit 84148ab72145 · 2021-03-02T08:53:42.000-05:00
bool named_plot(const std::string&amp; name, const std::vector&lt;std::string&gt;&amp; x, const std::vector&lt;Numeric&gt;&amp; y, const std::string&amp; format = "") { .. }

added template to be able to have named legend plotting: ex: time(x) vs signal(y) with legend. 
Prior to this change, the current library implements simple plots with no legends and with this change the above should be possible.
diff --git a/matplotlibcpp.h b/matplotlibcpp.h
@@ -1394,6 +1394,33 @@ bool named_plot(const std::string& name, const std::vector<Numeric>& x, const st
     return res;
 }
 
+template<typename Numeric>
+bool named_plot(const std::string& name, const std::vector<std::string>& x, const std::vector<Numeric>& y, const std::string& format = "")
+{
+    detail::_interpreter::get();
+
+    PyObject* kwargs = PyDict_New();
+    PyDict_SetItemString(kwargs, "label", PyString_FromString(name.c_str()));
+
+    PyObject* xarray = detail::get_array(x);
+    PyObject* yarray = detail::get_array(y);
+
+    PyObject* pystring = PyString_FromString(format.c_str());
+
+    PyObject* plot_args = PyTuple_New(3);
+    PyTuple_SetItem(plot_args, 0, xarray);
+    PyTuple_SetItem(plot_args, 1, yarray);
+    PyTuple_SetItem(plot_args, 2, pystring);
+
+    PyObject* res = PyObject_Call(detail::_interpreter::get().s_python_function_plot, plot_args, kwargs);
+
+    Py_DECREF(kwargs);
+    Py_DECREF(plot_args);
+    if (res) Py_DECREF(res);
+
+    return res;
+}
+
 template<typename Numeric>
 bool named_semilogx(const std::string& name, const std::vector<Numeric>& x, const std::vector<Numeric>& y, const std::string& format = "")
 {
