From 4ec761fc7a8e95367fb66740b75880453634495c Mon Sep 17 00:00:00 2001
From: Naoto Kondo <cgigcp3yqt@gmail.com>
Date: Sat, 25 Apr 2020 05:06:37 +0900
Subject: [PATCH] get current source list

---
 src/main.cpp | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index 5d0d69c..6b18dab 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -269,8 +269,17 @@ PYBIND11_MODULE(NDIlib, m) {
 
   m.def("find_destroy", &NDIlib_find_destroy, py::arg("instance"));
 
-  m.def("find_get_current_sources", &NDIlib_find_get_current_sources,
-        py::arg("instance"), py::arg("no_sources"));
+  m.def(
+      "find_get_current_sources",
+      [](NDIlib_find_instance_t p_instance) {
+        uint32_t count = 0;
+        auto sources = NDIlib_find_get_current_sources(p_instance, &count);
+        py::list out;
+        for (uint32_t i = 0; i < count; ++i)
+          out.append(sources + i);
+        return out;
+      },
+      py::arg("instance"));
 
   m.def("find_wait_for_sources", &NDIlib_find_wait_for_sources,
         py::arg("instance"), py::arg("timeout_in_ms"));