Algorithm-Repo/SourceCode/CVAlgorithmUtils/LesionDescription/SegmentDescribDemo/ReadDataBase.cs

using System;
using System.IO;
using System.Text;
using System.Collections.Generic;
using System.Drawing;
//using System.Data;
using System.Data.SQLite;
using Newtonsoft.Json;

using System.Security.Cryptography;
using System.Windows;
using System.Globalization;
//using AI.Common;
using AI.Common.Interface;
using RawImageShowUtilsLib;

namespace SegmentDescribDemo
{
    public enum EnumDesShapeValue
    {
        /// <summary>
        /// 椭圆形
        /// </summary>
        Oval,
        /// <summary>
        /// 类圆形
        /// </summary>
        Round,
        /// <summary>
        /// 不规则形
        /// </summary>
        Irregular,
    }

    /// <summary>
    /// 方向描述可能的结果值
    /// </summary>
    public enum EnumDesOrientationValue
    {
        /// <summary>
        /// 平行
        /// </summary>
        Parallel,
        /// <summary>
        /// 非平行
        /// </summary>
        NonParallel,
    }

    /// <summary>
    /// 内部回声描述可能的结果
    /// </summary>
    public enum EnumDesEchoPatternValue
    {
        /// <summary>
        /// 无回声
        /// </summary>
        Anechoic,
        /// <summary>
        /// 低回声
        /// </summary>
        Hypoechoic,
        /// <summary>
        /// 等回声
        /// </summary>
        Isoechoic,
        /// <summary>
        /// 高回声
        /// </summary>
        Hyperechoic,
        /// <summary>
        /// 混合回声
        /// </summary>
        Complex,
        /// <summary>
        /// 强回声
        /// </summary>
        Strongechoic,
    }

    /// <summary>
    /// 病灶边界描述可能的结果
    /// </summary>
    public enum EnumDesLesionBoundaryValue
    {
        /// <summary>
        /// 清晰
        /// </summary>
        AbruptInterface,
        /// <summary>
        /// 模糊
        /// </summary>
        EchogenicHalo,
    }

    /// <summary>
    /// 病灶边缘描述的可能结果
    /// </summary>
    public enum EnumDesMarginValue
    {
        /// <summary>
        /// 光整
        /// </summary>
        Circumscribed,
        /// <summary>
        /// 不光整
        /// </summary>
        NonCircumscribed,
    }

    /// <summary>
    /// 病灶描述GT图像信息
    /// </summary>
    public class LesionDesGTImgInfo
    {
        public string ImageId { get; set; }

        /// <summary>
        /// 病灶分级
        /// </summary>
        public List<int> Label { get; set; }
        /// <summary>
        /// 该图上所有病灶的标准描述
        /// </summary>
        public List<LesionInfo> Lesions { get; set; }

    }

    /// <summary>
    /// 病灶信息
    /// </summary>
    public class LesionInfo
    {
        /// <summary>
        /// 病灶轮廓
        /// </summary>
        public List<MyPoint> Contour { get; set; } = new List<MyPoint>();

        /// <summary>
        /// 病灶形状
        /// </summary>
        public EnumDesShapeValue Shape { get; set; } = EnumDesShapeValue.Oval;

        /// <summary>
        /// 方向
        /// </summary>
        public EnumDesOrientationValue Orientation { get; set; } = EnumDesOrientationValue.Parallel;

        /// <summary>
        /// 回声类型
        /// </summary>
        public EnumDesEchoPatternValue EchoPattern { get; set; } = EnumDesEchoPatternValue.Hypoechoic;

        /// <summary>
        /// 边界清晰不清晰
        /// </summary>
        public EnumDesLesionBoundaryValue Boundary { get; set; } = EnumDesLesionBoundaryValue.AbruptInterface;

        /// <summary>
        /// 边缘光整不光整
        /// </summary>
        public EnumDesMarginValue Margin { get; set; } = EnumDesMarginValue.Circumscribed;

        /// <summary>
        /// 横径起点
        /// </summary>
        public Point2D HorizontalPoint1 { get; set; } = new Point2D();

        /// <summary>
        /// 横径终点
        /// </summary>
        public Point2D HorizontalPoint2 { get; set; } = new Point2D();

        /// <summary>
        /// 纵径起点
        /// </summary>
        public Point2D VerticalPoint1 { get; set; } = new Point2D();

        /// <summary>
        /// 纵径终点
        /// </summary>
        public Point2D VerticalPoint2 { get; set; } = new Point2D();

        /// <summary>
        /// 内容完整
        /// </summary>
        public bool IsCompleted
        {
            get
            {
                return Contour.Count > 0 && HorizontalPoint1 != HorizontalPoint2 &&
                    VerticalPoint1 != VerticalPoint2;
            }
        }
    }


    class DataBase
    {
        #region 数据库相关变量
        private readonly string _dataFolder = System.IO.Path.Combine(System.IO.Path.GetDirectoryName(System.IO.Path.GetDirectoryName(
                                AppDomain.CurrentDomain.BaseDirectory)), "LesionDescriptionGTDatas");
        public readonly string _dbPath;
        public readonly string _imageDataFolder;
        public List<LesionDesGTImgInfo> _allImages = new List<LesionDesGTImgInfo>();
        private volatile int _imgIndex = 0;
        public List<LesionInfo> _lesions = new List<LesionInfo>();
        private volatile int _lesionIndex = 0;
        private System.Windows.Point _mouseStartPos;
        // 绘制中的所有轮廓点的集合
        private List<System.Windows.Point> _drawingPoints = new List<System.Windows.Point>();
        private volatile bool _canAutoFinish = false;
        #endregion


        public DataBase(string dbpath, string imgPath)
        {
            _dbPath = dbpath;
            _imageDataFolder = imgPath;
        }

        public DataBase(string dbpath)
        {
            _dbPath = dbpath;
        }

        /// <summary>
        /// 读入数据库
        /// </summary>
        public void ReadDataBase()
        {
            try
            {
                if (!File.Exists(_dbPath))
                {
                    MessageBox.Show("未找到存放待标注数据的文件：" + _dbPath);
                    return;
                }
                SQLiteConnection dbCon = new SQLiteConnection("Data Source=" + _dbPath);
                dbCon.Open();
                SQLiteCommand dbCmd;
                dbCmd = dbCon.CreateCommand();
                dbCmd.CommandText = "select * from LesionDesGTDatas";
                SQLiteDataReader dbReader;
                dbReader = dbCmd.ExecuteReader();
                if (!dbReader.HasRows)
                {
                    dbReader.Close();
                    dbCmd.Dispose();
                    dbCon.Close();
                    dbCon.Dispose();
                    MessageBox.Show("待标注的数据为空.");
                    return;
                }
                int index = 0;
                while (dbReader.Read())
                {
                    string imgId = ((string)dbReader[0]).TrimEnd();
                    string strLesions = ((string)dbReader[1]).TrimEnd();
                    List<LesionInfo> lesions = JsonConvert.DeserializeObject<List<LesionInfo>>(strLesions);

                    string strLesionsLabels = ((string)dbReader[2]).TrimEnd();
                    List<int> lesionsLabels = JsonConvert.DeserializeObject<List<int>>(strLesionsLabels);
                    index += 1;
                    _allImages.Add(new LesionDesGTImgInfo { ImageId = imgId, Lesions = lesions, Label = lesionsLabels });
                }
                dbReader.Close();
                dbCmd.Dispose();

                dbCmd = dbCon.CreateCommand();
                dbCmd.CommandText = "select * from LastModification";
                dbReader = dbCmd.ExecuteReader();
                if (!dbReader.HasRows)
                {
                    _imgIndex = 0;
                }
                else
                {
                    dbReader.Read();
                    string imgId = ((string)dbReader[0]).TrimEnd();
                    int imgIndex = _allImages.FindIndex(x => x.ImageId == imgId);
                    if (imgIndex == -1)
                    {
                        _imgIndex = 0;
                    }
                    else
                    {
                        _imgIndex = imgIndex;
                    }
                }
                dbReader.Close();
                dbCmd.Dispose();
                dbCon.Close();
                dbCon.Dispose();
                //return ;
            }
            catch (Exception excep)
            {
                MessageBox.Show("读入数据库时出错：" + excep);
            }
        }
        private string ComputeHashCode(byte[] input)
        {
            MD5 md5 = MD5.Create();
            byte[] hash = md5.ComputeHash(input);
            StringBuilder sb = new StringBuilder();
            foreach (var b in hash)
            {
                sb.Append(b.ToString("x2"));
            }
            string hashstr = sb.ToString();
            sb.Clear();
            return hashstr;
        }

        /// 创建数据库
        public void CreateDataBase()
        {
            if (File.Exists(_dbPath))
            {
                throw new Exception("数据库已存在，请检查.");
            }
            // 创建数据库文件
            SQLiteConnection dbCon = new SQLiteConnection("Data Source=" + _dbPath);
            dbCon.Open();
            SQLiteCommand dbCmd;
            dbCmd = dbCon.CreateCommand();
            dbCmd.CommandText = "create table if not exists GTDatas(ImageId varchar(16),Lesions varchar(16),ThyroidContour varchar(16))";
            dbCmd.ExecuteNonQuery();
            dbCmd.Dispose();

            dbCmd = dbCon.CreateCommand();
            dbCmd.CommandText = "create table if not exists LastModification(ImageId varchar(16))";
            dbCmd.ExecuteNonQuery();
            dbCmd.Dispose();

            dbCon.Close();
            dbCon.Dispose();
        }

    }
}