Added NDI code

gst-plugin-ndi/src/ndilib.rs

@@ -148,7 +148,9 @@ impl Default for NDIlib_recv_create_v3_t {
 pub type NDIlib_recv_instance_t = *mut ::std::os::raw::c_void;
 
 //Rust wrapper around *mut ::std::os::raw::c_void
-pub struct NdiInstance (NDIlib_recv_instance_t);
+pub struct NdiInstance {
+    pub recv: NDIlib_recv_instance_t,
+}
 
 unsafe impl ::std::marker::Send for NdiInstance {}
 

gst-plugin-ndi/src/ndisrc.rs

@@ -438,11 +438,13 @@ impl BaseSrcImpl<BaseSrc> for NdiSrc {
 
         gst_debug!(self.cat, obj: element, "Configuring for caps {}", caps);
 
+        // TODO Puede que falle si no creamos la estructura de cero, pero si lo hacemos no podemos poner recv a none
         let mut state = self.state.lock().unwrap();
-            *state = State {
-              info: Some(info),
-              recv: None
-            };
+        state.info = Some(info);
+            // *state = State {
+            //   info: Some(info),
+            //   recv: Some(NdiInstance{recv: pNDI_recv}),
+            // };
 
         // element.set_blocksize(info.bpf() * (*self.settings.lock().unwrap()).samples_per_buffer);
         //
@@ -491,6 +493,89 @@ impl BaseSrcImpl<BaseSrc> for NdiSrc {
         self.unlock_stop(element);
 
         gst_warning!(self.cat, obj: element, "Starting");
+        let mut state = self.state.lock().unwrap();
+        //let mut pNDI_recv = state.recv;
+        unsafe {
+            //TODO Al inicializar el plugin
+            if !NDIlib_initialize() {
+                //TODO delete exits
+                println!("Cannot run NDI: NDIlib_initialize error.");
+                ::std::process::exit(1);
+            }
+
+            //TODO valores por defecto
+            let NDI_find_create_desc: NDIlib_find_create_t = Default::default();
+            let pNDI_find = NDIlib_find_create_v2(&NDI_find_create_desc);
+            if pNDI_find.is_null() {
+                println!("Cannot run NDI: NDIlib_find_create_v2 error.");
+                ::std::process::exit(1);
+            }
+
+            let mut no_sources: u32 = 0;
+            let mut p_sources = ptr::null();
+            //TODO Evitar bloqueo
+            while no_sources == 0 {
+                p_sources = NDIlib_find_get_current_sources(pNDI_find, &mut no_sources as *mut u32);
+            }
+
+            // We need at least one source
+            if p_sources.is_null() {
+                println!("Error getting NDIlib_find_get_current_sources.");
+                ::std::process::exit(1);
+            }
+
+            println!(
+                "no_source {}: Name '{}' Address '{}'",
+                no_sources,
+                CStr::from_ptr((*p_sources).p_ndi_name)
+                    .to_string_lossy()
+                    .into_owned(),
+                CStr::from_ptr((*p_sources).p_ip_address)
+                    .to_string_lossy()
+                    .into_owned()
+            );
+
+            // We now have at least one source, so we create a receiver to look at it.
+            // We tell it that we prefer YCbCr video since it is more efficient for us. If the source has an alpha channel
+            // it will still be provided in BGRA
+            let p_ndi_name = CString::new("Galicaster NDI Receiver").unwrap();
+            let NDI_recv_create_desc = NDIlib_recv_create_v3_t {
+                source_to_connect_to: *p_sources,
+                p_ndi_name: p_ndi_name.as_ptr(),
+                ..Default::default()
+            };
+
+            let pNDI_recv = NDIlib_recv_create_v3(&NDI_recv_create_desc);
+            if pNDI_recv.is_null() {
+                println!("Cannot run NDI: NDIlib_recv_create_v3 error.");
+                ::std::process::exit(1);
+            }
+
+            // Destroy the NDI finder. We needed to have access to the pointers to p_sources[0]
+            NDIlib_find_destroy(pNDI_find);
+
+            // We are now going to mark this source as being on program output for tally purposes (but not on preview)
+            let tally_state: NDIlib_tally_t = Default::default();
+            NDIlib_recv_set_tally(pNDI_recv, &tally_state);
+
+            // Enable Hardwqre Decompression support if this support has it. Please read the caveats in the documentation
+            // regarding this. There are times in which it might reduce the performance although on small stream numbers
+            // it almost always yields the same or better performance.
+            let data = CString::new("<ndi_hwaccel enabled=\"true\"/>").unwrap();
+            let enable_hw_accel = NDIlib_metadata_frame_t {
+                length: data.to_bytes().len() as i32,
+                timecode: 0,
+                p_data: data.as_ptr(),
+            };
+
+            NDIlib_recv_send_metadata(pNDI_recv, &enable_hw_accel);
+            state.recv = Some(NdiInstance{recv: pNDI_recv});
+            //TODO Otra opcion para guardar pNDI_recv es esta:
+            // *state = State{
+            //     info: state.info.clone(),
+            //     recv: Some(NdiInstance{recv: pNDI_recv}),
+            // };
+        }
 
         true
     }
@@ -561,6 +646,64 @@ impl BaseSrcImpl<BaseSrc> for NdiSrc {
             }
             Some(ref info) => info.clone(),
         };
+        let recv = match state.recv{
+            None => {
+                //TODO Cambiar gst_element_error por uno mas descriptivo
+                println!("pNDI_recv no encontrado");
+                gst_element_error!(element, gst::CoreError::Negotiation, ["No encontramos ndi recv"]);
+                return Err(gst::FlowReturn::NotNegotiated);
+               }
+            Some(ref recv) => recv.clone(),
+        };
+        let pNDI_recv = recv.recv;
+
+        unsafe{
+        // loop {
+            let video_frame: NDIlib_video_frame_v2_t = Default::default();
+            let audio_frame: NDIlib_audio_frame_v2_t = Default::default();
+            let metadata_frame: NDIlib_metadata_frame_t = Default::default();
+
+            let frame_type = NDIlib_recv_capture_v2(
+                pNDI_recv,
+                &video_frame,
+                &audio_frame,
+                &metadata_frame,
+                1000,
+            );
+
+            match frame_type {
+                NDIlib_frame_type_e::NDIlib_frame_type_video => {
+                    println!("Tengo video {:?}", video_frame);
+
+                }
+                NDIlib_frame_type_e::NDIlib_frame_type_audio => {
+                    println!("Tengo audio {:?}", audio_frame);
+                }
+                NDIlib_frame_type_e::NDIlib_frame_type_metadata => {
+                    println!(
+                        "Tengo metadata {} '{}'",
+                        metadata_frame.length,
+                        CStr::from_ptr(metadata_frame.p_data)
+                            .to_string_lossy()
+                            .into_owned(),
+                    );
+                }
+                NDIlib_frame_type_e::NDIlib_frame_type_error => {
+                    println!(
+                        "Tengo error {} '{}'",
+                        metadata_frame.length,
+                        CStr::from_ptr(metadata_frame.p_data)
+                            .to_string_lossy()
+                            .into_owned(),
+                    );
+                    // break;
+                }
+                _ => println!("Tengo {:?}", frame_type),
+            }
+        // }
+
+
+
 
         // // If a stop position is set (from a seek), only produce samples up to that
         // // point but at most samples_per_buffer samples per buffer
@@ -698,6 +841,7 @@ impl BaseSrcImpl<BaseSrc> for NdiSrc {
 
         Ok(buffer)
     }
+    }
 
     fn fixate(&self, element: &BaseSrc, caps: gst::Caps) -> gst::Caps {
         // Fixate the caps. BaseSrc will do some fixation for us, but