avfilter/dnn: Don't show backends which are not supported by a filter

1 files changed, 8 insertions, 3 deletions

diff --git a/libavfilter/dnn/dnn_interface.c b/libavfilter/dnn/dnn_interface.c
index e9c3619abd..8dec3b671a 100644
--- a/libavfilter/dnn/dnn_interface.c
+++ b/libavfilter/dnn/dnn_interface.c
@@ -120,11 +120,16 @@ void *ff_dnn_child_next(DnnContext *obj, void *prev) {
     return NULL;
 }
 
-const AVClass *ff_dnn_child_class_iterate(void **iter)
+const AVClass *ff_dnn_child_class_iterate_with_mask(void **iter, uint32_t backend_mask)
 {
-    uintptr_t i = (uintptr_t) *iter;
+    for (uintptr_t i = (uintptr_t)*iter; i < FF_ARRAY_ELEMS(dnn_backend_info_list); i++) {
+        if (i > 0) {
+            const DNNModule *module = dnn_backend_info_list[i].module;
+
+            if (!(module->type & backend_mask))
+                continue;
+        }
 
-    if (i < FF_ARRAY_ELEMS(dnn_backend_info_list)) {
         *iter = (void *)(i + 1);
         return dnn_backend_info_list[i].class;
     }