Add set_xscale and set_yscale

jacketsj · jacketsj · commit a179e3b57a07 · 2023-07-31T22:25:18.000-07:00
diff --git a/matplotlibcpp.h b/matplotlibcpp.h
@@ -70,6 +70,8 @@ struct _interpreter {
     PyObject *s_python_function_subplot2grid;
     PyObject *s_python_function_legend;
     PyObject *s_python_function_xlim;
+    PyObject *s_python_function_set_xscale;
+    PyObject *s_python_function_set_yscale;
     PyObject *s_python_function_ion;
     PyObject *s_python_function_ginput;
     PyObject *s_python_function_ylim;
@@ -246,6 +248,8 @@ struct _interpreter {
         s_python_function_subplot2grid = safe_import(pymod, "subplot2grid");
         s_python_function_legend = safe_import(pymod, "legend");
         s_python_function_xlim = safe_import(pymod, "xlim");
+        s_python_function_set_xscale = safe_import(pymod, "set_xscale");
+        s_python_function_set_yscale = safe_import(pymod, "set_yscale");
         s_python_function_ylim = safe_import(pymod, "ylim");
         s_python_function_title = safe_import(pymod, "title");
         s_python_function_axis = safe_import(pymod, "axis");
@@ -883,6 +887,52 @@ bool arrow(Numeric x, Numeric y, Numeric end_x, Numeric end_y, const std::string
     return res;
 }
 
+void set_xscale(std::string value,
+ const std::map<std::string, std::string> &keywords = {})
+{
+    detail::_interpreter::get();
+
+    PyObject* pystr = PyString_FromString(value.c_str());
+    PyObject* args = PyTuple_New(1);
+    PyTuple_SetItem(args, 0, pystr);
+
+    PyObject* kwargs = PyDict_New();
+    for (auto it = keywords.begin(); it != keywords.end(); ++it) {
+        PyDict_SetItemString(kwargs, it->first.c_str(), PyUnicode_FromString(it->second.c_str()));
+    }
+
+    PyObject* res = PyObject_Call(detail::_interpreter::get().s_python_function_set_xscale, args, kwargs);
+
+    if(!res) throw std::runtime_error("Call to set_xscale() failed.");
+
+    Py_DECREF(args);
+    Py_DECREF(kwargs);
+    Py_DECREF(res);
+}
+
+void set_yscale(std::string value,
+ const std::map<std::string, std::string> &keywords = {})
+{
+    detail::_interpreter::get();
+
+    PyObject* pystr = PyString_FromString(value.c_str());
+    PyObject* args = PyTuple_New(1);
+    PyTuple_SetItem(args, 0, pystr);
+
+    PyObject* kwargs = PyDict_New();
+    for (auto it = keywords.begin(); it != keywords.end(); ++it) {
+        PyDict_SetItemString(kwargs, it->first.c_str(), PyUnicode_FromString(it->second.c_str()));
+    }
+
+    PyObject* res = PyObject_Call(detail::_interpreter::get().s_python_function_set_yscale, args, kwargs);
+
+    if(!res) throw std::runtime_error("Call to set_yscale() failed.");
+
+    Py_DECREF(args);
+    Py_DECREF(kwargs);
+    Py_DECREF(res);
+}
+
 template< typename Numeric>
 bool hist(const std::vector<Numeric>& y, long bins=10,std::string color="b",
           double alpha=1.0, bool cumulative=false)
