looks like i forgot about commits

2026-06-25 12:43:53 +03:00
parent 49ead70b10
commit 51a1b1b2c8
8 changed files with 445 additions and 3 deletions
@@ -0,0 +1,72 @@
+#include <stdexcept>
+#include <vector>
+#include <string>
+#include <chrono>
+#include <Processing.NDI.Lib.h>
+
+namespace dmf {
+class NDIHelper {
+    public:
+        NDIHelper() {
+            if (!NDIlib_initialize()) {
+                throw std::runtime_error("NDI lib init failed");
+                if (!NDIlib_is_supported_CPU()) {
+                    throw std::runtime_error("CPU is not sufficient for NDI");
+                }
+            }
+        }
+
+        ~NDIHelper() {
+            NDIlib_destroy();
+            NDIlib_recv_destroy(pNDI_recv);
+        }
+
+        void find_sources(std::vector<std::string>* sources, u_int32_t timeout_ms) {
+            pNDI_find = NDIlib_find_create_v2();
+            if (!pNDI_find) {
+                throw std::runtime_error("Cannot create NDI finder");
+            }
+
+            while (!sources_amount) {
+                NDIlib_find_wait_for_sources(pNDI_find, timeout_ms);
+                p_sources = NDIlib_find_get_current_sources(pNDI_find, &sources_amount);
+            }
+
+            for (int i = 0; i < sources_amount; ++i){
+                sources->push_back(p_sources[i].p_ndi_name);
+            }
+            NDIlib_find_destroy(pNDI_find);
+        }
+
+        void select_source(uint32_t source_num) {
+            if (sources_amount == 0) {
+                throw std::runtime_error("0 sources found");
+            } else if (source_num > sources_amount) {
+                throw std::runtime_error("Source_num bigger that sources amount");
+            }
+
+            pNDI_recv = NDIlib_recv_create_v3();
+            if (!pNDI_recv) {
+                NDIlib_recv_destroy(pNDI_recv);
+                throw std::runtime_error("Cannot create NDI recieve instance");
+            }
+            NDIlib_recv_connect(pNDI_recv, p_sources + source_num);
+        }
+
+    private:
+        // receive
+        NDIlib_find_instance_t pNDI_find = nullptr;
+        uint32_t sources_amount = 0;
+        const NDIlib_source_t* p_sources = NULL;
+        NDIlib_recv_instance_t pNDI_recv = nullptr;
+
+        void get_source_info(uint32_t source_num) {
+            NDIlib_video_frame_v2_t video_frame;
+            uint8_t frames = 0;
+            while (frames < 2) {
+                NDIlib_recv_capture_v2(pNDI_recv, &video_frame, nullptr, nullptr, 1000);
+                NDIlib_recv_free_video_v2(pNDI_recv, &video_frame);
+            }
+        }
+};
+}