ipa: rpi: controller: awb: Add Neural Network Awb

Add an Awb algorithm which uses neural networks.

Signed-off-by: Peter Bailey <[email protected]>

Author: PineappleBeech

meson_options.txt

@@ -91,3 +91,8 @@ option('v4l2',
         value : 'auto',
         description : 'Compile the V4L2 compatibility layer',
         deprecated : {'true': 'enabled', 'false': 'disabled'})
+
+option('rpi_nn_awb',
+        type : 'feature',
+        value : 'disabled',
+        description : 'Enable Raspberry Pi neural network AWB')

src/ipa/rpi/controller/meson.build

@@ -32,6 +32,22 @@ rpi_ipa_controller_deps = [
     libcamera_private,
 ]
 
+if not get_option('rpi_nn_awb').disabled()
+    tflite_dep = dependency('tensorflow-lite', required : false)
+    if not tflite_dep.found()
+        # This may not have a pkg-config file, so try and find it directly
+        cc = meson.get_compiler('cpp')
+        tflite_dep = cc.find_library('tensorflow-lite', required : get_option('rpi_nn_awb'))
+    endif
+
+    if tflite_dep.found()
+        rpi_ipa_controller_sources += files([
+            'rpi/awb_nn.cpp',
+        ])
+        rpi_ipa_controller_deps += tflite_dep
+    endif
+endif
+
 rpi_ipa_controller_lib = static_library('rpi_ipa_controller', rpi_ipa_controller_sources,
                                         include_directories : libipa_includes,
                                         dependencies : rpi_ipa_controller_deps)