feat(interop): add NdiInteropPInvoke production INdiInterop implementation
Some checks failed
CI / build-and-test (push) Failing after 26s
Some checks failed
CI / build-and-test (push) Failing after 26s
This commit is contained in:
parent
90b0951a42
commit
da5818b690
1 changed files with 235 additions and 0 deletions
235
src/TeamsISO.Engine.NdiInterop/NdiInteropPInvoke.cs
Normal file
235
src/TeamsISO.Engine.NdiInterop/NdiInteropPInvoke.cs
Normal file
|
|
@ -0,0 +1,235 @@
|
||||||
|
using System.Runtime.InteropServices;
|
||||||
|
using System.Runtime.Versioning;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using TeamsISO.Engine.Interop;
|
||||||
|
using TeamsISO.Engine.Pipeline;
|
||||||
|
|
||||||
|
namespace TeamsISO.Engine.NdiInterop;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Production <see cref="INdiInterop"/> implementation backed by the NDI 6 SDK.
|
||||||
|
/// Initializes the NDI runtime on construction and tears it down on dispose.
|
||||||
|
/// All frame buffers crossing the managed boundary are copied so the engine never holds
|
||||||
|
/// a reference to NDI-owned memory after the call returns.
|
||||||
|
/// </summary>
|
||||||
|
[SupportedOSPlatform("windows")]
|
||||||
|
public sealed class NdiInteropPInvoke : INdiInterop, IDisposable
|
||||||
|
{
|
||||||
|
private readonly ILogger<NdiInteropPInvoke> _logger;
|
||||||
|
private bool _initialized;
|
||||||
|
private bool _disposed;
|
||||||
|
|
||||||
|
public NdiInteropPInvoke(ILogger<NdiInteropPInvoke> logger)
|
||||||
|
{
|
||||||
|
_logger = logger;
|
||||||
|
if (!NdiNative.Initialize())
|
||||||
|
throw new InvalidOperationException(
|
||||||
|
"NDI runtime failed to initialize. Ensure the NDI Runtime is installed and " +
|
||||||
|
"Processing.NDI.Lib.x64.dll is on the application's DLL search path.");
|
||||||
|
_initialized = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
public string GetRuntimeVersion()
|
||||||
|
{
|
||||||
|
var ptr = NdiNative.Version();
|
||||||
|
return Marshal.PtrToStringAnsi(ptr) ?? string.Empty;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---- Discovery ----
|
||||||
|
|
||||||
|
public NdiFindHandle CreateFinder()
|
||||||
|
{
|
||||||
|
var native = NdiNative.FindCreateV2(IntPtr.Zero);
|
||||||
|
if (native == IntPtr.Zero)
|
||||||
|
throw new InvalidOperationException("NDIlib_find_create_v2 returned null.");
|
||||||
|
return new NdiPInvokeFindHandle(native);
|
||||||
|
}
|
||||||
|
|
||||||
|
public IReadOnlyList<string> GetCurrentSources(NdiFindHandle finder)
|
||||||
|
{
|
||||||
|
var pInvokeFinder = (NdiPInvokeFindHandle)finder;
|
||||||
|
var ptr = NdiNative.FindGetCurrentSources(pInvokeFinder.Native, out var count);
|
||||||
|
if (ptr == IntPtr.Zero || count == 0) return Array.Empty<string>();
|
||||||
|
|
||||||
|
// The native call returns a pointer to an array of NDIlib_source_t.
|
||||||
|
// Each entry is sizeof(IntPtr)*2 bytes (p_ndi_name + p_url_address).
|
||||||
|
var sourceSize = Marshal.SizeOf<NdiNative.Source>();
|
||||||
|
var result = new string[count];
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
var sourcePtr = IntPtr.Add(ptr, i * sourceSize);
|
||||||
|
var src = Marshal.PtrToStructure<NdiNative.Source>(sourcePtr);
|
||||||
|
result[i] = Marshal.PtrToStringAnsi(src.p_ndi_name) ?? string.Empty;
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---- Receive ----
|
||||||
|
|
||||||
|
public NdiReceiverHandle CreateReceiver(string sourceFullName)
|
||||||
|
{
|
||||||
|
var nameUtf8 = Marshal.StringToHGlobalAnsi(sourceFullName);
|
||||||
|
var recvNameUtf8 = Marshal.StringToHGlobalAnsi("TeamsISO");
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var settings = new NdiNative.RecvCreateV3Settings
|
||||||
|
{
|
||||||
|
source_to_connect_to = new NdiNative.Source
|
||||||
|
{
|
||||||
|
p_ndi_name = nameUtf8,
|
||||||
|
p_url_address = IntPtr.Zero
|
||||||
|
},
|
||||||
|
color_format = NdiNative.RecvColorFormat.BgrxBgra,
|
||||||
|
bandwidth = NdiNative.RecvBandwidth.Highest,
|
||||||
|
allow_video_fields = false,
|
||||||
|
p_ndi_recv_name = recvNameUtf8
|
||||||
|
};
|
||||||
|
var native = NdiNative.RecvCreateV3(ref settings);
|
||||||
|
if (native == IntPtr.Zero)
|
||||||
|
throw new InvalidOperationException($"NDIlib_recv_create_v3 returned null for source '{sourceFullName}'.");
|
||||||
|
return new NdiPInvokeReceiverHandle(native, sourceFullName);
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
Marshal.FreeHGlobal(nameUtf8);
|
||||||
|
Marshal.FreeHGlobal(recvNameUtf8);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public RawFrame? CaptureFrame(NdiReceiverHandle receiver, int timeoutMs)
|
||||||
|
{
|
||||||
|
var pInvokeReceiver = (NdiPInvokeReceiverHandle)receiver;
|
||||||
|
var frameType = NdiNative.RecvCaptureV3(
|
||||||
|
pInvokeReceiver.Native,
|
||||||
|
out var nativeFrame,
|
||||||
|
IntPtr.Zero,
|
||||||
|
IntPtr.Zero,
|
||||||
|
(uint)Math.Max(0, timeoutMs));
|
||||||
|
|
||||||
|
if (frameType != NdiNative.FrameType.Video || nativeFrame.p_data == IntPtr.Zero)
|
||||||
|
{
|
||||||
|
// Even non-video frame types may need to be released; recv_free_video_v2 is safe on zeroed structs.
|
||||||
|
if (nativeFrame.p_data != IntPtr.Zero)
|
||||||
|
NdiNative.RecvFreeVideoV2(pInvokeReceiver.Native, ref nativeFrame);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var pixelFormat = nativeFrame.FourCC switch
|
||||||
|
{
|
||||||
|
NdiNative.FourCC.BGRA or NdiNative.FourCC.BGRX => PixelFormat.Bgra,
|
||||||
|
NdiNative.FourCC.UYVY => PixelFormat.Uyvy,
|
||||||
|
NdiNative.FourCC.RGBA => PixelFormat.Rgba,
|
||||||
|
_ => PixelFormat.Bgra
|
||||||
|
};
|
||||||
|
|
||||||
|
// Compute total byte length. line_stride_in_bytes is the union slot for video frames.
|
||||||
|
var byteLength = nativeFrame.line_stride_in_bytes * nativeFrame.yres;
|
||||||
|
if (byteLength <= 0)
|
||||||
|
{
|
||||||
|
_logger.LogWarning("NDI video frame had non-positive byte length ({Length}); skipping.", byteLength);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
var managed = new byte[byteLength];
|
||||||
|
Marshal.Copy(nativeFrame.p_data, managed, 0, byteLength);
|
||||||
|
|
||||||
|
return new RawFrame(
|
||||||
|
Width: nativeFrame.xres,
|
||||||
|
Height: nativeFrame.yres,
|
||||||
|
TimestampTicks: nativeFrame.timestamp,
|
||||||
|
Pixels: managed,
|
||||||
|
Format: pixelFormat);
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
NdiNative.RecvFreeVideoV2(pInvokeReceiver.Native, ref nativeFrame);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---- Send ----
|
||||||
|
|
||||||
|
public NdiSenderHandle CreateSender(string outputName)
|
||||||
|
{
|
||||||
|
var nameUtf8 = Marshal.StringToHGlobalAnsi(outputName);
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var settings = new NdiNative.SendCreateSettings
|
||||||
|
{
|
||||||
|
p_ndi_name = nameUtf8,
|
||||||
|
p_groups = IntPtr.Zero,
|
||||||
|
clock_video = true,
|
||||||
|
clock_audio = false
|
||||||
|
};
|
||||||
|
var native = NdiNative.SendCreate(ref settings);
|
||||||
|
if (native == IntPtr.Zero)
|
||||||
|
throw new InvalidOperationException($"NDIlib_send_create returned null for output '{outputName}'.");
|
||||||
|
return new NdiPInvokeSenderHandle(native, outputName);
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
Marshal.FreeHGlobal(nameUtf8);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void SendFrame(NdiSenderHandle sender, ProcessedFrame frame)
|
||||||
|
{
|
||||||
|
var pInvokeSender = (NdiPInvokeSenderHandle)sender;
|
||||||
|
|
||||||
|
var fourcc = frame.Format switch
|
||||||
|
{
|
||||||
|
PixelFormat.Bgra => NdiNative.FourCC.BGRA,
|
||||||
|
PixelFormat.Rgba => NdiNative.FourCC.RGBA,
|
||||||
|
PixelFormat.Uyvy => NdiNative.FourCC.UYVY,
|
||||||
|
_ => NdiNative.FourCC.BGRA
|
||||||
|
};
|
||||||
|
|
||||||
|
// Pin the managed buffer so the native call can read it directly.
|
||||||
|
var pixels = frame.Pixels.ToArray();
|
||||||
|
var handle = GCHandle.Alloc(pixels, GCHandleType.Pinned);
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var nativeFrame = new NdiNative.VideoFrameV2
|
||||||
|
{
|
||||||
|
xres = frame.Width,
|
||||||
|
yres = frame.Height,
|
||||||
|
FourCC = fourcc,
|
||||||
|
frame_rate_N = 60000,
|
||||||
|
frame_rate_D = 1001, // 59.94 default; controller-supplied later
|
||||||
|
picture_aspect_ratio = 0.0f, // 0 = derive from xres/yres
|
||||||
|
frame_format_type = NdiNative.FrameFormatType.Progressive,
|
||||||
|
timecode = NdiTimecodeSynthesize, // synthesize
|
||||||
|
p_data = handle.AddrOfPinnedObject(),
|
||||||
|
line_stride_in_bytes = frame.Width * BytesPerPixel(frame.Format),
|
||||||
|
p_metadata = IntPtr.Zero,
|
||||||
|
timestamp = frame.TimestampTicks
|
||||||
|
};
|
||||||
|
NdiNative.SendSendVideoV2(pInvokeSender.Native, ref nativeFrame);
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
handle.Free();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private const long NdiTimecodeSynthesize = unchecked((long)0x8000000000000000UL); // NDIlib_send_timecode_synthesize
|
||||||
|
|
||||||
|
private static int BytesPerPixel(PixelFormat fmt) => fmt switch
|
||||||
|
{
|
||||||
|
PixelFormat.Bgra or PixelFormat.Rgba => 4,
|
||||||
|
PixelFormat.Uyvy => 2,
|
||||||
|
_ => 4
|
||||||
|
};
|
||||||
|
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
if (_disposed) return;
|
||||||
|
if (_initialized)
|
||||||
|
{
|
||||||
|
NdiNative.Destroy();
|
||||||
|
_initialized = false;
|
||||||
|
}
|
||||||
|
_disposed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue