123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462 |
- using System;
- using System.Reflection;
- using System.Text.Json;
- using Vinno.AI.CommonSDK.Enums;
- using Vinno.AI.CommonSDK.Interfaces;
- using Vinno.AI.CommonSDK.Models;
- using Vinno.AI.CommonSDK.Tools;
- using Vinno.AI.MyocardialDiagnosisSDK.Enums;
- using Vinno.AI.MyocardialDiagnosisSDK.Interfaces;
- using Vinno.AI.MyocardialDiagnosisSDK.Models;
- namespace Vinno.AI.MyocardialDiagnosisSDK
- {
- public class MyocardialDiagnosis : IMyocardialDiagnosis
- {
- private readonly IMyocardialDiagnosisService _myocardialDiagnosisService;
- private IAIImageProvider _imageProvider;
- private bool _initialized;
- private bool _disposed;
- /// <summary>
- /// Used For Detect One Image
- /// </summary>
- private PipeClient _singleImagePipeClient;
- /// <summary>
- /// Used For Provide Raw Image
- /// </summary>
- private PipeClient _rawImageProviderPipeClient;
- /// <summary>
- /// Used For Provide Byte Image
- /// </summary>
- private PipeClient _byteImageProviderPipeClient;
- /// <summary>
- /// Raised when the image evaluation is started.
- /// </summary>
- public event EventHandler StartEvaluationNotification;
- /// <summary>
- /// Raised when the image evaluation is finished.
- /// </summary>
- public event EventHandler<AIDetectedObject> FinishEvaluationNotification;
- public MyocardialDiagnosis()
- {
- _myocardialDiagnosisService = AIManager.Instance.AIDiagnosisSystemJsonRpcClientManager?.GetService<IMyocardialDiagnosisService>();
- if (AIManager.Instance.AINotificationManager != null)
- {
- AIManager.Instance.AINotificationManager.NotificationReceived += OnNotificationReceived;
- }
- _singleImagePipeClient = new PipeClient(AIDiagnosisSystemConsts.PipeForMyocardialDiagnosisSingleImage);
- _singleImagePipeClient.LogMsgThrow += OnLogMsgThrow;
- _singleImagePipeClient.Start();
- _rawImageProviderPipeClient = new PipeClient(AIDiagnosisSystemConsts.PipeForMyocardialDiagnosisRawImageProvider);
- _rawImageProviderPipeClient.LogMsgThrow += OnLogMsgThrow;
- _rawImageProviderPipeClient.Start();
- _byteImageProviderPipeClient = new PipeClient(AIDiagnosisSystemConsts.PipeForMyocardialDiagnosisByteImageProvider);
- _byteImageProviderPipeClient.LogMsgThrow += OnLogMsgThrow;
- _byteImageProviderPipeClient.Start();
- }
- /// <summary>
- /// 初始化Myocardial Diagnosis
- /// </summary>
- /// <param name="myocardialParameter">Myocardial Parameter</param>
- /// <param name="imageProvider">Image Provider</param>
- public void Initialize(MyocardialDiagnosisParameter myocardialParameter, IAIImageProvider imageProvider = null)
- {
- try
- {
- if (!_initialized)
- {
- _imageProvider = imageProvider;
- _myocardialDiagnosisService.Initialize(myocardialParameter, _imageProvider != null);
- _initialized = true;
- }
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// Start Image Provider
- /// </summary>
- public void Start()
- {
- if (_imageProvider != null)
- {
- _myocardialDiagnosisService.Start();
- _imageProvider.ByteImageProvided += OnByteImageProvided;
- _imageProvider.RawImageProvided += OnRawImageProvided;
- _imageProvider.Start();
- }
- }
- /// <summary>
- /// Stop Image Provider
- /// </summary>
- public void Stop()
- {
- if (_imageProvider != null)
- {
- _imageProvider.ByteImageProvided -= OnByteImageProvided;
- _imageProvider.RawImageProvided -= OnRawImageProvided;
- _imageProvider.Stop();
- }
- _myocardialDiagnosisService.Stop();
- }
- private void OnLogMsgThrow(object sender, AILogEventArgs e)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(e);
- }
- private void OnNotificationReceived(object sender, AINotificationArgs e)
- {
- switch (e.NotificationType)
- {
- case AIEnumNotificationType.MyocardialDiagnosisStartEvaluationRaised:
- StartEvaluationNotification?.Invoke(this, EventArgs.Empty);
- break;
- case AIEnumNotificationType.MyocardialDiagnosisFinishEvaluationRaised:
- var transAIDetectedObject = JsonSerializer.Deserialize<TransAIDetectedObject>(e.Params?.ToString());
- var detectedObject = AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(transAIDetectedObject);
- FinishEvaluationNotification?.Invoke(this, detectedObject);
- break;
- case AIEnumNotificationType.MyocardialDiagnosisNotifyLogRaised:
- var logEventArgs = JsonSerializer.Deserialize<AILogEventArgs>(e.Params?.ToString());
- AIManager.Instance.AILogManager?.WriteLogInfo(logEventArgs);
- break;
- }
- }
- /// <summary>
- /// 设置每秒图片吞吐量
- /// </summary>
- /// <param name="detectTps">每秒图片吞吐量,必须大于0</param>
- public void SetDetectTps(int detectTps)
- {
- if (detectTps <= 0)
- {
- throw new ArgumentOutOfRangeException($"DetectTps Must > 0. DetectTps : {detectTps}");
- }
- try
- {
- _myocardialDiagnosisService.SetDetectTps(detectTps);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// 设置间隔时间
- /// </summary>
- /// <param name="intervalTime"></param>
- public void SetIntervalTime(int intervalTime)
- {
- try
- {
- _myocardialDiagnosisService.SetIntervalTime(intervalTime);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// 设置检测模式
- /// </summary>
- /// <param name="detectMode"></param>
- public void SetDetectMode(AIEnumDetectMode detectMode)
- {
- try
- {
- _myocardialDiagnosisService.SetDetectMode(detectMode);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// 设置是否对心肌圆形轮廓进行补全
- /// </summary>
- /// <param name="isCompleteMyocardiaImg"></param>
- public void SetIsCompleteMyocardialImg(bool isCompleteMyocardiaImg)
- {
- try
- {
- _myocardialDiagnosisService.SetIsCompleteMyocardialImg(isCompleteMyocardiaImg);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// 检测单张Raw Image
- /// </summary>
- /// <param name="rawImage">图像资源</param>
- /// <param name="isCircle">是否是基底类图片</param>
- /// <returns></returns>
- public AIDetectedObject DetectOneImage(AIRawImage rawImage, bool isCircle)
- {
- if (rawImage == null)
- {
- throw new ArgumentNullException(nameof(rawImage));
- }
- try
- {
- _singleImagePipeClient.SendBytes(rawImage.DataBuffer);
- var result = _myocardialDiagnosisService.DetectOneRawImage(rawImage.Height, rawImage.Width, rawImage.ColorType, isCircle);
- return AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(result);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return null;
- }
- }
- /// <summary>
- /// 检测单张Byte Image
- /// </summary>
- /// <param name="byteImage">图像资源</param>
- /// <param name="isCircle">是否是基底类图片</param>
- /// <returns></returns>
- public AIDetectedObject DetectOneImage(byte[] byteImage, bool isCircle)
- {
- if (byteImage == null)
- {
- throw new ArgumentNullException(nameof(byteImage));
- }
- try
- {
- _singleImagePipeClient.SendBytes(byteImage);
- var result = _myocardialDiagnosisService.DetectOneByteImage(isCircle);
- return AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(result);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return null;
- }
- }
- /// <summary>
- /// 检测出心肌轮廓后,判断单帧图的心肌轮廓是否能够补全成完整圆
- /// </summary>
- /// <param name="inferResult">心肌检测结果</param>
- /// <returns></returns>
- public bool AbleToBeCompleted(AIDetectedObject inferResult)
- {
- if (inferResult == null)
- {
- throw new ArgumentNullException(nameof(inferResult));
- }
- try
- {
- var transAIDetectedObject = AICommonConvertHelper.ConvertAIDetectedObjectToTransAIDetectedObject(inferResult);
- return _myocardialDiagnosisService.AbleToBeCompleted(transAIDetectedObject);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return false;
- }
- }
- /// <summary>
- /// 检测心肌图片,给出补全完整的心肌轮廓。
- /// </summary>
- /// <param name="segResult">心肌轮廓检测结果,是DetectOneImage方法得出的不补全的检测结果</param>
- /// <returns></returns>
- public AIDetectedObject CompleteOneImage(AIDetectedObject segResult)
- {
- if (segResult == null)
- {
- throw new ArgumentNullException(nameof(segResult));
- }
- try
- {
- var transAIDetectedObject = AICommonConvertHelper.ConvertAIDetectedObjectToTransAIDetectedObject(segResult);
- var result = _myocardialDiagnosisService.CompleteOneImage(transAIDetectedObject);
- return AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(result);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return null;
- }
- }
- /// <summary>
- /// 根据分割结果,提取相应内外轮廓的值
- /// </summary>
- /// <param name="segResult">心肌轮廓检测结果</param>
- /// <returns></returns>
- public AIDetectedObject ExtractContours(AIDetectedObject segResult)
- {
- if (segResult == null)
- {
- throw new ArgumentNullException(nameof(segResult));
- }
- try
- {
- var transAIDetectedObject = AICommonConvertHelper.ConvertAIDetectedObjectToTransAIDetectedObject(segResult);
- var result = _myocardialDiagnosisService.ExtractContours(transAIDetectedObject);
- return AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(result);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return null;
- }
- }
- /// <summary>
- /// 对心肌的EDES帧的内外轮廓做后处理,当前仅供人的心肌使用
- /// </summary>
- /// <param name="segResult">分割结果,使用其中的轮廓点</param>
- /// <param name="type">输入的心肌切面的类型</param>
- /// <returns>最终的图像内外轮廓经过后处理的结果,比原始分割的心肌轮廓更接近真实轮廓,0表示外轮廓、1表示内轮廓</returns>
- public AIDetectedObject ModifyEDESFrame(AIDetectedObject segResult, AIEnumMyocardialType type)
- {
- if (segResult == null)
- {
- throw new ArgumentNullException(nameof(segResult));
- }
- try
- {
- var transAIDetectedObject = AICommonConvertHelper.ConvertAIDetectedObjectToTransAIDetectedObject(segResult);
- var result = _myocardialDiagnosisService.ModifyEDESFrame(transAIDetectedObject, type);
- return AICommonConvertHelper.ConvertTransAIDetectedObjectToAIDetectedObject(result);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- return null;
- }
- }
- private void OnByteImageProvided(object sender, byte[] byteImage)
- {
- try
- {
- if (byteImage == null)
- {
- return;
- }
- _byteImageProviderPipeClient?.SendBytes(byteImage);
- _myocardialDiagnosisService.SendByteImageData();
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- private void OnRawImageProvided(object sender, AIRawImage rawImage)
- {
- try
- {
- if (rawImage == null)
- {
- return;
- }
- _rawImageProviderPipeClient?.SendBytes(rawImage.DataBuffer);
- _myocardialDiagnosisService.SendRawImageData(rawImage.Height, rawImage.Width, rawImage.ColorType);
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- /// <summary>
- /// Close Myocardial Diagnosis
- /// </summary>
- public void Close()
- {
- try
- {
- if (_disposed)
- {
- return;
- }
- _initialized = false;
- if (_imageProvider != null)
- {
- _imageProvider.ByteImageProvided -= OnByteImageProvided;
- _imageProvider.RawImageProvided -= OnRawImageProvided;
- _imageProvider.Stop();
- _imageProvider = null;
- }
- _myocardialDiagnosisService.Close();
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- public void Dispose()
- {
- try
- {
- if (!_disposed)
- {
- _initialized = false;
- if (AIManager.Instance.AINotificationManager != null)
- {
- AIManager.Instance.AINotificationManager.NotificationReceived -= OnNotificationReceived;
- }
- if (_imageProvider != null)
- {
- _imageProvider.ByteImageProvided -= OnByteImageProvided;
- _imageProvider.RawImageProvided -= OnRawImageProvided;
- _imageProvider.Stop();
- _imageProvider = null;
- }
- if (_singleImagePipeClient != null)
- {
- _singleImagePipeClient.Dispose();
- _singleImagePipeClient.LogMsgThrow -= OnLogMsgThrow;
- _singleImagePipeClient = null;
- }
- if (_rawImageProviderPipeClient != null)
- {
- _rawImageProviderPipeClient.Dispose();
- _rawImageProviderPipeClient.LogMsgThrow -= OnLogMsgThrow;
- _rawImageProviderPipeClient = null;
- }
- if (_byteImageProviderPipeClient != null)
- {
- _byteImageProviderPipeClient.Dispose();
- _byteImageProviderPipeClient.LogMsgThrow -= OnLogMsgThrow;
- _byteImageProviderPipeClient = null;
- }
- _disposed = true;
- }
- }
- catch (Exception ex)
- {
- AIManager.Instance.AILogManager?.WriteLogInfo(new AILogEventArgs(AIEnumLogType.ErrorLog, $"错误方法名:{MethodBase.GetCurrentMethod().DeclaringType.Name}.{MethodBase.GetCurrentMethod().Name}, 错误信息:{ex}"));
- }
- }
- }
- }
|