summaryrefslogtreecommitdiff
path: root/py/pykms
diff options
context:
space:
mode:
Diffstat (limited to 'py/pykms')
-rw-r--r--py/pykms/CMakeLists.txt3
-rw-r--r--py/pykms/pykms.cpp6
2 files changed, 6 insertions, 3 deletions
diff --git a/py/pykms/CMakeLists.txt b/py/pykms/CMakeLists.txt
index 505c0c3..56a9c35 100644
--- a/py/pykms/CMakeLists.txt
+++ b/py/pykms/CMakeLists.txt
@@ -1,3 +1,4 @@
+include_directories(${PROJECT_BINARY_DIR}) # for config.h
include_directories(${LIBDRM_INCLUDE_DIRS})
link_directories(${LIBDRM_LIBRARY_DIRS})
@@ -12,7 +13,7 @@ include_directories(${PROJECT_SOURCE_DIR}/ext/pybind11/include)
set(SRCS pykms.cpp pykmsbase.cpp pykmsutil.cpp pyvid.cpp)
-if(LIBDRM_OMAP_FOUND)
+if(HAVE_OMAP_DRM)
set(SRCS ${SRCS} pykmsomap.cpp)
endif()
diff --git a/py/pykms/pykms.cpp b/py/pykms/pykms.cpp
index 7752f19..819093e 100644
--- a/py/pykms/pykms.cpp
+++ b/py/pykms/pykms.cpp
@@ -2,6 +2,8 @@
#include <pybind11/stl.h>
#include <kms++/kms++.h>
+#include "config.h"
+
namespace py = pybind11;
using namespace kms;
@@ -11,7 +13,7 @@ void init_pykmstest(py::module &m);
void init_pykmsbase(py::module &m);
void init_pyvid(py::module &m);
-#if HAS_LIBDRM_OMAP
+#if HAVE_OMAP_DRM
void init_pykmsomap(py::module &m);
#endif
@@ -29,7 +31,7 @@ PYBIND11_PLUGIN(pykms) {
init_pyvid(m);
-#if HAS_LIBDRM_OMAP
+#if HAVE_OMAP_DRM
init_pykmsomap(m);
#endif
return m.ptr();