NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed numpy_image not compiling when C++14 is enabled.

Browse Source
This commit is contained in:
Davis King 2018-06-02 18:52:42 -04:00
parent 85fc0e0b91
commit 44c06150a0
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
dlib/python/numpy_image.h
View File

@ -13,6 +13,7 @@
#include <pybind11/pybind11.h> #include <pybind11/pybind11.h>
#include <dlib/image_transforms/assign_image.h> #include <dlib/image_transforms/assign_image.h>
#include <stdint.h> #include <stdint.h>
#include <type_traits>
namespace py = pybind11; namespace py = pybind11;
@ -356,18 +357,28 @@ namespace pybind11
{ {
using basic_pixel_type = typename dlib::pixel_traits<pixel_type>::basic_pixel_type; using basic_pixel_type = typename dlib::pixel_traits<pixel_type>::basic_pixel_type;
static PYBIND11_DESCR name() { template <size_t channels>
constexpr size_t channels = dlib::pixel_traits<pixel_type>::num; static PYBIND11_DESCR getname(typename std::enable_if<channels==1,int>::type) {
if (channels == 1) return _("numpy.ndarray[(rows,cols),") + npy_format_descriptor<basic_pixel_type>::name() + _("]");
return _("numpy.ndarray[(rows,cols),") + npy_format_descriptor<basic_pixel_type>::name() + _("]"); };
else if (channels == 2) template <size_t channels>
static PYBIND11_DESCR getname(typename std::enable_if<channels!=1,int>::type) {
if (channels == 2)
return _("numpy.ndarray[(rows,cols,2),") + npy_format_descriptor<basic_pixel_type>::name() + _("]"); return _("numpy.ndarray[(rows,cols,2),") + npy_format_descriptor<basic_pixel_type>::name() + _("]");
else if (channels == 3) else if (channels == 3)
return _("numpy.ndarray[(rows,cols,3),") + npy_format_descriptor<basic_pixel_type>::name() + _("]"); return _("numpy.ndarray[(rows,cols,3),") + npy_format_descriptor<basic_pixel_type>::name() + _("]");
else if (channels == 4) else if (channels == 4)
return _("numpy.ndarray[(rows,cols,4),") + npy_format_descriptor<basic_pixel_type>::name() + _("]"); return _("numpy.ndarray[(rows,cols,4),") + npy_format_descriptor<basic_pixel_type>::name() + _("]");
else };
DLIB_CASSERT(false,"unsupported pixel type");
static PYBIND11_DESCR name() {
constexpr size_t channels = dlib::pixel_traits<pixel_type>::num;
// The reason we have to call getname() in this wonky way is because
// pybind11 uses a type that records the length of the returned string in
// the type. So we have to do this overloading to make the return type
// from name() consistent. In C++17 this would be a lot cleaner with
// constexpr if, but can't use C++17 yet because of lack of wide support :(
return getname<channels>(0);
} }
}; };