added cm for mplot3d/subplot3d.cpp

Author: soblin

gallery/mplot3d/subplot3d.cpp

@@ -5,6 +5,7 @@
 #include <pybind11/numpy.h>
 
 #include <matplotlibcpp17/pyplot.h>
+#include <matplotlibcpp17/cm.h>
 
 #include <xtensor/xbuilder.hpp>
 #include <xtensor/xmath.hpp>
@@ -79,8 +80,9 @@ int main() {
     auto Y_ = py::array(py::cast(std::move(Y)));
     auto Z_ = py::array(py::cast(std::move(Z)));
     auto surf = ax.plot_surface(
-        Args(X_, Y_, Z_), Kwargs("rstride"_a = 1, "cstride"_a = 1,
-                                 "linewidth"_a = 0, "antialiased"_a = false));
+        Args(X_, Y_, Z_),
+        Kwargs("rstride"_a = 1, "cstride"_a = 1, "linewidth"_a = 0,
+               "antialiased"_a = false, "cmap"_a = cm::coolwarm()));
     ax.set_zlim(Args(-1.01, 1.01));
     fig.colorbar(Args(surf), Kwargs("shrink"_a = 0.5, "aspect"_a = 10));
   }

include/matplotlibcpp17/cm.h

@@ -0,0 +1,23 @@
+/**
+ * @file cm.h
+ * @brief corresponding header for matplotlib.animation
+ **/
+
+#ifndef MATPLOTLIBCPP17_CM_H
+#define MATPLOTLIBCPP17_CM_H
+
+#include <pybind11/pybind11.h>
+
+#include <matplotlibcpp17/common.h>
+
+namespace matplotlibcpp17::cm {
+
+pybind11::object coolwarm() {
+  pybind11::object ret =
+      pybind11::module::import("matplotlib.cm").attr("coolwarm");
+  return ret;
+}
+
+} // namespace matplotlibcpp17::cm
+
+#endif /* MATPLOTLIBCPP17_CM_H */