73 lines
2.3 KiB
C#
73 lines
2.3 KiB
C#
using NetMQ;
|
||
using SHH.Contracts;
|
||
|
||
namespace SHH.CameraService
|
||
{
|
||
/// <summary>
|
||
/// 负责将业务契约转换为 ZeroMQ 传输协议
|
||
/// </summary>
|
||
public static class NetMQProtocolExtensions
|
||
{
|
||
private const string PROTOCOL_HEADER = "SHH_V1";
|
||
|
||
/// <summary>
|
||
/// 扩展方法：将 Payload 转为 NetMQMessage
|
||
/// 使用方法：var msg = payload.ToNetMqMessage();
|
||
/// </summary>
|
||
public static NetMQMessage ToNetMqMessage(this VideoPayload payload)
|
||
{
|
||
var msg = new NetMQMessage();
|
||
|
||
// Frame 0: 协议魔数
|
||
msg.Append(PROTOCOL_HEADER);
|
||
|
||
// Frame 1: 元数据 JSON
|
||
msg.Append(payload.GetMetadataJson());
|
||
|
||
// Frame 2: 原始图 (保持帧位对齐，无数据则发空帧)
|
||
if (payload.HasOriginalImage && payload.OriginalImageBytes != null)
|
||
msg.Append(payload.OriginalImageBytes);
|
||
else
|
||
msg.Append(Array.Empty<byte>());
|
||
|
||
// Frame 3: 处理图
|
||
if (payload.HasTargetImage && payload.TargetImageBytes != null)
|
||
msg.Append(payload.TargetImageBytes);
|
||
else
|
||
msg.Append(Array.Empty<byte>());
|
||
|
||
return msg;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 扩展方法：从 NetMQMessage 还原 Payload
|
||
/// </summary>
|
||
public static VideoPayload ToVideoPayload(this NetMQMessage msg)
|
||
{
|
||
if (msg == null || msg.FrameCount < 4) return null;
|
||
|
||
// Frame 0 Check
|
||
if (msg[0].ConvertToString() != PROTOCOL_HEADER) return null;
|
||
|
||
// Frame 1: Metadata
|
||
string json = msg[1].ConvertToString();
|
||
var payload = VideoPayload.FromMetadataJson(json);
|
||
if (payload == null) return null;
|
||
|
||
// Frame 2: Raw Image
|
||
// 利用 BufferSize 避免不必要的内存拷贝，如果长度为0则跳过
|
||
if (payload.HasOriginalImage && msg[2].BufferSize > 0)
|
||
{
|
||
payload.OriginalImageBytes = msg[2].ToByteArray();
|
||
}
|
||
|
||
// Frame 3: Processed Image
|
||
if (payload.HasTargetImage && msg[3].BufferSize > 0)
|
||
{
|
||
payload.TargetImageBytes = msg[3].ToByteArray();
|
||
}
|
||
|
||
return payload;
|
||
}
|
||
}
|
||
} |