search for headers in other locations

Author: Calsign

Makefile

@@ -1,5 +1,10 @@
 BUILD=_build
 
+# By default (if left blank), the generator script searches for
+# the opencv4 headers in /usr/include/ and /usr/local/include/.
+# Set this variable if your opencv4 headers are located elsewhere.
+SYSTEM_INCLUDE_DIR=
+
 GEN=gen
 GENERATOR=generator.py
 GENERATOR_SRC=$(GENERATOR) hdr_parser.py type_manager.py incl/*
@@ -28,7 +33,7 @@ default: install
 $(GEN)/%: ${GENERATOR_SRC}
 	@echo "Running generator script"
 	mkdir -p $(GEN)
-	./$(GENERATOR) $(GEN)
+	./$(GENERATOR) $(GEN) $(SYSTEM_INCLUDE_DIR)
 
 $(BUILD)/$(SHARED_LIB): $(CPP) $(HEADERS)
 	@echo "Compiling shared library"

generator.py

@@ -11,12 +11,17 @@
 import type_manager
 
 src_files = [
-    '/usr/include/opencv4/opencv2/core/types.hpp',
-    '/usr/include/opencv4/opencv2/core/mat.hpp',
-    '/usr/include/opencv4/opencv2/core.hpp',
-    '/usr/include/opencv4/opencv2/imgproc.hpp',
-    '/usr/include/opencv4/opencv2/videoio.hpp',
-    '/usr/include/opencv4/opencv2/highgui.hpp',
+    'opencv4/opencv2/core/types.hpp',
+    'opencv4/opencv2/core/mat.hpp',
+    'opencv4/opencv2/core.hpp',
+    'opencv4/opencv2/imgproc.hpp',
+    'opencv4/opencv2/videoio.hpp',
+    'opencv4/opencv2/highgui.hpp',
+]
+
+system_include_dir_search = [
+    '/usr/include/',
+    '/usr/local/include',
 ]
 
 c_reserved = {
@@ -343,9 +348,24 @@ def param_sub(match):
 
 
 if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print('No output folder specified!')
+        exit(1)
+
     dest = sys.argv[1]
     print('Output directory: {}'.format(dest))
 
+    if len(sys.argv) > 2:
+        system_include_dir_search.insert(0, sys.argv[2])
+
+    system_include_dir = None
+    for include_dir in system_include_dir_search:
+        if os.path.exists(os.path.join(include_dir, src_files[0])):
+            system_include_dir = include_dir
+            break
+
+    print('Using include dir: {}'.format(system_include_dir))
+
     # TODO enable UMat support, and make a wrapper for handling both Mat and UMat identically
     generate_umat = False
 
@@ -475,7 +495,7 @@ def sanitize_param(param):
 
     decls = []
     for hname in src_files:
-        decls += parser.parse(hname, wmode=False)
+        decls += parser.parse(os.path.join(system_include_dir, hname), wmode=False)
 
     # first pass to collect classes and add types
     for decl in decls: