WingAIDiagnosisService/Manage/ThyroidDiagnosis.cs

using System.Collections.Generic;
using System.Linq;
using WingInterfaceLibrary.Enum;
namespace WingAIDiagnosisService.Manage
{
    /// <summary>
    /// 甲状腺诊断
    /// </summary>
    public class ThyroidDiagnosis : BaseDiagnosis
    {
        private AIThyroidLabelEnum? _mimanLabel;

        private DiagnosisPerImageModel _mimanDiagnosisImage = null;

        public ThyroidDiagnosis(List<DiagnosisPerImageModel> recordResult, DiagnosisOrganEnum organ)
        : base(recordResult, organ)
        {

        }

        #region Private

        /// <summary>
        /// 判断是否弥漫性图像
        /// </summary>
        /// <param name="label"></param>
        /// <returns></returns>
        private bool IsMiMan(int label)
        {
            switch (label)
            {
                case (int)AIThyroidLabelEnum.DiffuseDisease://弥漫性图像，无描述
                    return true;
            }
            return false;
        }

        private AIDetectedObject IsValidDetectedObjectInfo(AIDetectedObject detectedObjectInfo)
        {
            if (detectedObjectInfo.Label == 0)
            {
                return detectedObjectInfo;
            }
            if (IsMiMan(detectedObjectInfo.Label))
            {
                return detectedObjectInfo;
            }
            if (detectedObjectInfo.Descriptions == null || !detectedObjectInfo.Descriptions.Any())
            {
                return null;
            }

            var lesionSizeDescription = detectedObjectInfo.Descriptions.FirstOrDefault(d => d.Type == DiagnosisDescription.LesionSize && !string.IsNullOrWhiteSpace(d.Value));
            if (lesionSizeDescription == null)
            {
                return null;
            }

            var lessionSizeInfo = Newtonsoft.Json.JsonConvert.DeserializeObject<AIDiagnosisLesionSize>(lesionSizeDescription.Value);
            if (lessionSizeInfo.VerticalLengthInPixel == 0 || lessionSizeInfo.HorizontalLengthInPixel == 0)
            {
                return null;
            }

            return detectedObjectInfo;
        }

        private bool CheckIsThyroid(List<DiagnosisPerImageModel> result)
        {
            var thyroidCount = result.Where(a => a.PriorityScore > 0).Count();
            var totalCount = result.Count;
            var thyroidRate = thyroidCount * 1.0 / totalCount * 1.0;
            return thyroidRate > 1 / 3.0;
        }

        private bool ExistAIData(IEnumerable<AIDiagnosisResultPerOrgan> organDiagResults, AIThyroidLabelEnum label)
        {
            if (organDiagResults == null)
                return false;
            foreach (var record in organDiagResults)
            {
                if (record.DetectedObjects == null)
                    continue;
                if (record.Organ != DiagnosisOrganEnum.Thyroid)
                    continue;

                if (label == AIThyroidLabelEnum.TIRADS0)//TODO 是否未见异常
                {
                    if (record.DetectedObjects.Any(a => a.Label == (int)label))
                        return true;
                }
                else
                {
                    if (record.DetectedObjects.Any(a => a.Label == (int)label) && record.DetectedObjects.All(d => IsValidDetectedObjectInfo(d) != null))
                        return true;
                }
            }
            return false;
        }

        #endregion

        public override AIDiagnosisResultPerOrgan CheckResultValid(AIDiagnosisResultPerOrgan message)
        {
            if (message.Organ != DiagnosisOrganEnum.Thyroid)
            {
                return null;
            }
            var result = message;
            var validDetectedObjects = new List<AIDetectedObject>();
            foreach (var obj in message.DetectedObjects)
            {
                var obj_new = IsValidDetectedObjectInfo(obj);
                if (obj_new != null)
                {
                    validDetectedObjects.Add(obj_new);
                }
            }
            if (validDetectedObjects.Any())
            {
                result.DetectedObjects = validDetectedObjects.ToList();
                return result;
            }
            return null;
        }

        public override DiagnosisConclusion GetAIStatus()
        {
            var status = DiagnosisConclusion.Unrecognized;
            bool existMalignant = false;
            bool existBenign = false;
            if (RecordDiagnosisResult != null && RecordDiagnosisResult.Any())
            {
                if (!CheckIsThyroid(RecordDiagnosisResult))
                    return status;
                foreach (var diagResult in RecordDiagnosisResult)
                {
                    if (existMalignant && existBenign)
                        break;
                    if (diagResult.DiagResultsForEachOrgan != null && diagResult.PriorityScore > 0)
                    {
                        foreach (var organ in diagResult.DiagResultsForEachOrgan)
                        {
                            if (organ.DetectedObjects != null)
                            {
                                if (!existMalignant)
                                {
                                    if (organ.DetectedObjects.Any(a => a.Label >= 4))
                                    {
                                        existMalignant = true;
                                    }
                                }

                                if (!existBenign)
                                {
                                    if (organ.DetectedObjects.Any(a => a.Label >= 1 && a.Label < 4))
                                    {
                                        existBenign = true;
                                    }
                                }
                            }
                        }

                        if (existMalignant)
                        {
                            status = existBenign ? DiagnosisConclusion.BenignAndMalignant : DiagnosisConclusion.Malignant;
                        }
                        else
                        {
                            status = existBenign ? DiagnosisConclusion.Benign : DiagnosisConclusion.NoObviousLesion;
                        }
                    }
                }
            }
            return status;
        }

        /// <summary>
        /// 是否恶性病灶
        /// </summary>
        /// <param name="label"></param>
        /// <returns></returns>
        protected override bool IsMalignant(AIDetectedObject detectedObject)
        {
            return (detectedObject.Label == (int)AIThyroidLabelEnum.TIRADS4a ||
            detectedObject.Label == (int)AIThyroidLabelEnum.TIRADS4b ||
            detectedObject.Label == (int)AIThyroidLabelEnum.TIRADS4c ||
            detectedObject.Label == (int)AIThyroidLabelEnum.TIRADS5);
        }

        public override List<DiagnosisPerImageModel> GetReportResults()
        {
            if (!RecordDiagnosisResult.Any())
            {
                return RecordDiagnosisResult;
            }
            var reportResults = new List<DiagnosisPerImageModel>();
            for (var i = 1; i <= (int)AIThyroidLabelEnum.DiffuseDisease; i++)
            {
                var label = (AIThyroidLabelEnum)i;
                var data = RecordDiagnosisResult.FirstOrDefault(a => ExistAIData(a.DiagResultsForEachOrgan, label));
                if (data != null)
                {
                    if (!reportResults.Any(a => a.RemedicalCode == data.RemedicalCode && a.Index == data.Index))
                    {
                        reportResults.Add(data);
                    }
                }
            }
            if (!reportResults.Any())
            {

                var label = AIThyroidLabelEnum.TIRADS0; //未见异常
                var data = RecordDiagnosisResult.FirstOrDefault(a => ExistAIData(a.DiagResultsForEachOrgan, label));
                if (data != null)
                {
                    if (!reportResults.Any(a => a.RemedicalCode == data.RemedicalCode && a.Index == data.Index))
                    {
                        reportResults.Add(data);
                    }
                }
            }
            RecordDiagnosisResult = reportResults;
            InitialAIImage();
            return RecordDiagnosisResult;
        }


    }
}