海康摄像头取流示例初始签入
This commit is contained in:
140
SHH.CameraSdk/Core/Memory/FramePool.cs
Normal file
140
SHH.CameraSdk/Core/Memory/FramePool.cs
Normal file
@@ -0,0 +1,140 @@
|
||||
using OpenCvSharp;
|
||||
|
||||
namespace SHH.CameraSdk;
|
||||
|
||||
/// <summary>
|
||||
/// [零延迟核心] 智能帧对象池
|
||||
/// 功能:预分配并复用 SmartFrame 实例,杜绝频繁 new Mat() 与 GC 回收,消除内存分配停顿
|
||||
/// 核心策略:
|
||||
/// <para>1. 预热分配:启动时创建初始数量帧,避免运行时内存申请</para>
|
||||
/// <para>2. 上限控制:最大池大小限制内存占用,防止内存溢出</para>
|
||||
/// <para>3. 背压丢帧:池空时返回 null,强制丢帧保证实时性,不阻塞生产端</para>
|
||||
/// </summary>
|
||||
public class FramePool : IDisposable
|
||||
{
|
||||
#region --- 私有资源与配置 (Private Resources & Configurations) ---
|
||||
|
||||
/// <summary> 可用帧队列(线程安全):存储待借出的空闲智能帧 </summary>
|
||||
private readonly ConcurrentQueue<SmartFrame> _availableFrames = new();
|
||||
|
||||
/// <summary> 所有已分配帧列表:用于统一销毁释放内存 </summary>
|
||||
private readonly List<SmartFrame> _allAllocatedFrames = new();
|
||||
|
||||
/// <summary> 创建新帧锁:确保多线程下创建新帧的线程安全 </summary>
|
||||
private readonly object _lock = new();
|
||||
|
||||
/// <summary> 帧宽度(与相机输出分辨率一致) </summary>
|
||||
private readonly int _width;
|
||||
|
||||
/// <summary> 帧高度(与相机输出分辨率一致) </summary>
|
||||
private readonly int _height;
|
||||
|
||||
/// <summary> 帧数据类型(如 CV_8UC3 对应 RGB 彩色图像) </summary>
|
||||
private readonly MatType _type;
|
||||
|
||||
/// <summary> 池最大容量:限制最大分配帧数,防止内存占用过高 </summary>
|
||||
private readonly int _maxPoolSize;
|
||||
|
||||
#endregion
|
||||
|
||||
#region --- 构造与预热 (Constructor & Warm-Up) ---
|
||||
|
||||
/// <summary>
|
||||
/// 初始化智能帧对象池
|
||||
/// </summary>
|
||||
/// <param name="width">帧宽度</param>
|
||||
/// <param name="height">帧高度</param>
|
||||
/// <param name="type">帧数据类型</param>
|
||||
/// <param name="initialSize">初始预热帧数(默认5)</param>
|
||||
/// <param name="maxSize">池最大容量(默认10)</param>
|
||||
public FramePool(int width, int height, MatType type, int initialSize = 5, int maxSize = 10)
|
||||
{
|
||||
_width = width;
|
||||
_height = height;
|
||||
_type = type;
|
||||
_maxPoolSize = maxSize;
|
||||
|
||||
// 预热:启动时预分配初始数量帧,避免运行时动态申请内存
|
||||
for (int i = 0; i < initialSize; i++)
|
||||
{
|
||||
CreateNewFrame();
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 创建新智能帧并加入池(内部调用,加锁保护)
|
||||
/// </summary>
|
||||
private void CreateNewFrame()
|
||||
{
|
||||
var frame = new SmartFrame(this, _width, _height, _type);
|
||||
_allAllocatedFrames.Add(frame);
|
||||
_availableFrames.Enqueue(frame);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region --- 帧借出与归还 (Frame Borrow & Return) ---
|
||||
|
||||
/// <summary>
|
||||
/// 从池借出一个智能帧(O(1) 时间复杂度)
|
||||
/// </summary>
|
||||
/// <returns>可用智能帧 / 池空且达上限时返回 null(触发背压丢帧)</returns>
|
||||
public SmartFrame Get()
|
||||
{
|
||||
// 1. 优先从可用队列取帧,无锁快速路径
|
||||
if (_availableFrames.TryDequeue(out var frame))
|
||||
{
|
||||
frame.Activate();
|
||||
return frame;
|
||||
}
|
||||
|
||||
// 2. 可用队列为空,检查是否达最大容量
|
||||
if (_allAllocatedFrames.Count < _maxPoolSize)
|
||||
{
|
||||
// 加锁创建新帧,避免多线程重复创建
|
||||
lock (_lock)
|
||||
{
|
||||
// 双重检查:防止等待锁期间其他线程已创建新帧
|
||||
if (_allAllocatedFrames.Count < _maxPoolSize)
|
||||
{
|
||||
CreateNewFrame();
|
||||
}
|
||||
}
|
||||
// 递归重试取帧
|
||||
return Get();
|
||||
}
|
||||
|
||||
// 3. 背压策略:池空且达上限,返回 null 强制丢帧,保证生产端不阻塞
|
||||
// 适用场景:消费端处理过慢导致帧堆积,丢帧保实时性
|
||||
return null;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// [系统内部调用] 将帧归还至池(由 SmartFrame.Dispose 自动触发)
|
||||
/// </summary>
|
||||
/// <param name="frame">待归还的智能帧</param>
|
||||
public void Return(SmartFrame frame)
|
||||
{
|
||||
_availableFrames.Enqueue(frame);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region --- 资源释放 (Resource Disposal) ---
|
||||
|
||||
/// <summary>
|
||||
/// 释放帧池所有资源,销毁所有 Mat 内存
|
||||
/// </summary>
|
||||
public void Dispose()
|
||||
{
|
||||
// 遍历所有已分配帧,释放 OpenCV Mat 底层内存
|
||||
foreach (var frame in _allAllocatedFrames)
|
||||
{
|
||||
frame.InternalMat.Dispose();
|
||||
}
|
||||
_allAllocatedFrames.Clear();
|
||||
_availableFrames.Clear();
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
Reference in New Issue
Block a user