imshow for double data type

subratappt · subratappt · commit 2b3e13daf5ef · 2022-01-28T08:23:23.000+05:30
diff --git a/matplotlibcpp.h b/matplotlibcpp.h
@@ -940,7 +940,7 @@ namespace detail {
 inline void imshow(void* ptr, const NPY_TYPES type, const int rows, const int columns,
                    const int colors, const std::map<std::string, std::string>& keywords,
                    PyObject** out) {
-    assert(type == NPY_UINT8 || type == NPY_FLOAT);
+    assert(type == NPY_UINT8 || type == NPY_FLOAT || type == NPY_DOUBLE);
     assert(colors == 1 || colors == 3 || colors == 4);
 
     detail::_interpreter::get();
@@ -982,6 +982,7 @@ inline void imshow(const float* ptr, const int rows, const int columns, const in
     detail::imshow((void*)ptr, NPY_FLOAT, rows, columns, colors, keywords, out);
 }
 
+inline void imshow(const double* ptr, const int rows, const int columns, const int colors,
 #ifdef WITH_OPENCV
 void imshow(const cv::Mat& image, const std::map<std::string, std::string>& keywords = {}) {
     // Convert underlying type of matrix, if needed
