MeterVision_v2/MeterVision/FreeAi/FaImport.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;

namespace MeterVision.FreeAi
{
    public class FaImport
    {
        //beforeAI结构体
        //[StructLayout(LayoutKind.Sequential, Pack = 1)]
        //public struct BeforeAI
        //{
        //    // 01 （配置信息）数字个数(黑色+红色)
        //    public byte NUM_NUMBERS;          // uint8_t 对应 byte

        //    // 01 （配置信息）数字个数(黑色),人工决定的时候不需要配置，=NUM_NUMBERS
        //    public byte NUM_BLACK_NUMBERS;    // uint8_t 对应 byte

        //    // 02（配置信息）指针个数
        //    public byte NUM_INDICATORS;       // uint8_t 对应 byte

        //    // 1立方=1000L = 10000分升,1分升=1/10升=1/10000立方
        //    // 尾数单位的含义,最小单位1分升,1dL, 0.1L（万分之一立方)
        //    public uint UNIT_OF_THE_LAST_NUMBER; // uint32_t 对应 uint

        //    // 04（配置信息）强制拉回的概率阈值（初始=50%）
        //    public byte BEST_CONF;            // uint8_t 对应 byte

        //    // 05（配置信息）识别可信度的概率阈值（初始=5%）
        //    public byte GOOD_CONF;            // uint8_t 对应 byte

        //    // 06（配置信息）单独每一位概率阈值（初始=90%）
        //    public byte BIT_BEST_CONF;        // uint8_t 对应 byte

        //    // 07（配置信息）是否颠倒安装（0=正常安装；1=上下颠倒安装）
        //    public byte IS_UPDOWN;            // uint8_t 对应 byte

        //    // 每小时的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
        //    public ulong MAX_FLOW_PER_HOUR;   // uint64_t 对应 ulong

        //    // 当前时间距离上次准确识别时间内的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
        //    public ulong MAX_FLOW_threshold;  // uint64_t 对应 ulong

        //    // 09（本次结果）图像倾斜角度
        //    public short lean_angle;          // int16_t 对应 short

        //    // 8字节整数,表示识别结果，最小单位分升,即1dL=0.1L=1m³/10000),万分之一立方;
        //    public ulong AI_result;           // uint64_t 对应 ulong

        //    // 传入的ai_result的值是否是可信的前值(来源于AfterAI);
        //    public byte is_OK_preValue;       // uint8_t 对应 byte

        //    // 11（上次结果）数字区域的xmin、ymin、xmax、ymax坐标值
        //    [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
        //    public ushort[] num_region;       // uint16_t 对应 ushort, 4 个元素的数组

        //    // 12 是否需要上传整图？0=不需要；1=需要；
        //    public byte is_get_whole_image;   // uint8_t 对应 byte

        //    // 上传日志的printf回调函数
        //    public IntPtr printfAiLog;        // 回调函数指针, 用 IntPtr 表示

        //    // 14（配置信息）公司平台上传的图片的存储大小？（1:BMP彩色图; >1:JPG彩色图）
        //    public ushort PingTai_image_size; // uint16_t 对应 ushort

        //    // 15（配置信息）上传图像尺寸大小 = 数字区域区域外扩 PingTai_image_margin 像素
        //    public byte PingTai_image_margin; // uint8_t 对应 byte

        //    // 16（配置信息）数字区域的xmin、ymin、xmax、ymax坐标值
        //    [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
        //    public ushort[] config_num_region; // uint16_t 对应 ushort, 4 个元素的数组

        //    // 17（配置信息）图像倾斜角度
        //    public short config_lean_angle;   // int16_t 对应 short

        //    // 18（配置信息）在识别过程中，是否使用 config_num_region、config_lean_angle变量值
        //    public byte is_use_config_info;   // uint8_t 对应 byte

        //    // 19（配置信息）是否上传CNN模型的检测结果到服务器？（0=不上传；1=上传）
        //    public byte is_upload_cnn_output; // uint8_t 对应 byte

        //    // 20（配置信息）对CNN输出进行解码时，使用的NMS阈值（默认值=10）
        //    public byte nms_threshold;        // uint8_t 对应 byte

        //    // 21（配置信息）是否使用检测到的大框来排除数字（0=不用；1=用）（默认值=1）
        //    public byte is_use_detected_num88;// uint8_t 对应 byte

        //    // 22（上次结果），存储每个指针区域的xmin、ymin、xmax、ymax坐标值。最多存储10个指针
        //    //[MarshalAs(UnmanagedType.ByValArray, SizeConst = 10)]
        //    //public ushort[,] indicators_regions; // uint16_t[10][4] 对应 ushort[10, 4]
        //    [MarshalAs(UnmanagedType.ByValArray, SizeConst = 40)]
        //    public ushort[] indicators_regions; // 对应 uint16_t[10][4] 展开成一维数组

        //    // 表具中心点x
        //    public float center_x;            // float 对应 float

        //    // 表具中心点y
        //    public float center_y;            // float 对应 float

        //    // 图片亮度
        //    public float bright_factor;       // float 对应 float

        //    // 仪表类型,模型类型
        //    public byte meter_type;           // uint8_t 对应 byte

        //    // 人工配置的水表参数使能开关
        //    public byte manual_parameter_enable; // uint8_t 对应 byte

        //    // 大图JPG图像的最大字节数
        //    public ushort big_jpg_max_size;   // uint16_t 对应 ushort

        //    // AI输出JPG图是,使用测试图标记,1:使用,0:不使用
        //    public byte use_test_jpg_flag;    // uint8_t 对应 byte

        //    // 数字水表图像放大倍率,中心点为图像中心
        //    public float number_image_zoomrate;   // float 对应 float

        //    // 指针水表图像放大倍率,中心点为拟合圆圆心
        //    public float indicator_image_zommrate;// float 对应 float

        //    // 水表中数字区域位于上半部？1=位于上半部分；0=位于下半部分。 默认值为1
        //    public byte is_nums_in_upper;     // uint8_t 对应 byte

        //    // 数字+指针水表中“数字”部分读数；全指针水表中“黑色”指针部分的读数；全数字水表中“数字”部分读数
        //    public ulong result_of_1st_part;  // uint64_t 对应 ulong

        //    // 数字+指针水表中“指针”部分读数；全指针水表中“红色”指针部分的读数；用来判断此水表是否在动
        //    public ulong result_of_2st_part;  // uint64_t 对应 ulong

        //    // 压缩JPG的索引,来自after_struct,首次索引为0
        //    public byte compress_jpg_index;   // uint8_t 对应 byte

        //    // 是否上传红色指针部分的数据(0:不上传;1:上传;默认1)
        //    public byte is_upload_red_ind_part;// uint8_t 对应 byte
        //}


        [StructLayout(LayoutKind.Sequential, Pack = 1)]
        public struct BeforeAI
        {
            // 01 （配置信息）数字个数(黑色+红色)
            public byte NUM_NUMBERS;          // uint8_t 对应 byte

            // 01 （配置信息）数字个数(黑色),人工决定的时候不需要配置，=NUM_NUMBERS
            public byte NUM_BLACK_NUMBERS;    // uint8_t 对应 byte

            // 02（配置信息）指针个数
            public byte NUM_INDICATORS;       // uint8_t 对应 byte

            // 1立方=1000L = 10000分升,1分升=1/10升=1/10000立方
            // 尾数单位的含义,最小单位1分升,1dL, 0.1L（万分之一立方)
            public uint UNIT_OF_THE_LAST_NUMBER; // uint32_t 对应 uint

            // 04（配置信息）强制拉回的概率阈值（初始=50%）
            public byte BEST_CONF;            // uint8_t 对应 byte

            // 05（配置信息）识别可信度的概率阈值（初始=5%）
            public byte GOOD_CONF;            // uint8_t 对应 byte

            // 06（配置信息）单独每一位概率阈值（初始=90%）
            public byte BIT_BEST_CONF;        // uint8_t 对应 byte

            // 07（配置信息）是否颠倒安装（0=正常安装；1=上下颠倒安装）
            public byte IS_UPDOWN;            // uint8_t 对应 byte

            // 每小时的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
            public ulong MAX_FLOW_PER_HOUR;   // uint64_t 对应 ulong

            // 当前时间距离上次准确识别时间内的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
            public ulong MAX_FLOW_threshold;  // uint64_t 对应 ulong

            // 09（本次结果）图像倾斜角度
            public short lean_angle;          // int16_t 对应 short

            // 8字节整数,表示识别结果，最小单位分升,即1dL=0.1L=1m³/10000),万分之一立方;
            public ulong AI_result;           // uint64_t 对应 ulong

            // 传入的ai_result的值是否是可信的前值(来源于AfterAI);
            public byte is_OK_preValue;       // uint8_t 对应 byte

            // 11（上次结果）数字区域的xmin、ymin、xmax、ymax坐标值
            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
            public ushort[] num_region;       // uint16_t 对应 ushort, 4 个元素的数组

            // 12 是否需要上传整图？0=不需要；1=需要；
            public byte is_get_whole_image;   // uint8_t 对应 byte

            // 上传日志的printf回调函数
            public IntPtr printfAiLog;        // 回调函数指针, 用 IntPtr 表示

            // 14（配置信息）公司平台上传的图片的存储大小？（1:BMP彩色图; >1:JPG彩色图）
            public ushort PingTai_image_size; // uint16_t 对应 ushort

            // 15（配置信息）上传图像尺寸大小 = 数字区域区域外扩 PingTai_image_margin 像素
            public byte PingTai_image_margin; // uint8_t 对应 byte

            // 16（配置信息）数字区域的xmin、ymin、xmax、ymax坐标值
            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
            public ushort[] config_num_region; // uint16_t 对应 ushort, 4 个元素的数组

            // 17（配置信息）图像倾斜角度
            public short config_lean_angle;   // int16_t 对应 short

            // 18（配置信息）在识别过程中，是否使用 config_num_region、config_lean_angle变量值
            public byte is_use_config_info;   // uint8_t 对应 byte

            // 19（配置信息）是否上传CNN模型的检测结果到服务器？（0=不上传；1=上传）
            public byte is_upload_cnn_output; // uint8_t 对应 byte

            // 20（配置信息）对CNN输出进行解码时，使用的NMS阈值（默认值=10）
            public byte nms_threshold;        // uint8_t 对应 byte

            // 21（配置信息）是否使用检测到的大框来排除数字（0=不用；1=用）（默认值=1）
            public byte is_use_detected_num88;// uint8_t 对应 byte

            // 22（上次结果），存储每个指针区域的xmin、ymin、xmax、ymax坐标值。最多存储10个指针
            //[MarshalAs(UnmanagedType.ByValArray, SizeConst = 10)]
            //public ushort[,] indicators_regions; // uint16_t[10][4] 对应 ushort[10, 4]
            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 40)]
            public ushort[] indicators_regions; // 对应 uint16_t[10][4] 展开成一维数组

            // 图片亮度
            public float bright_factor;       // float 对应 float

            // 仪表类型,模型类型
            public byte meter_type;           // uint8_t 对应 byte

            // 人工配置的水表参数使能开关
            public byte manual_parameter_enable; // uint8_t 对应 byte

            // 大图JPG图像的最大字节数
            public ushort big_jpg_max_size;   // uint16_t 对应 ushort

            // AI输出JPG图是,使用测试图标记,1:使用,0:不使用
            public byte use_test_jpg_flag;    // uint8_t 对应 byte

            // 表具中心点x
            public float center_x;            // float 对应 float

            // 表具中心点y
            public float center_y;            // float 对应 float

            // 数字水表图像放大倍率,中心点为图像中心
            public float number_image_zoomrate;   // float 对应 float

            // 指针水表图像放大倍率,中心点为拟合圆圆心
            public float indicator_image_zommrate;// float 对应 float

            public float image_zoomrate;//图像放大倍率,中心点为拟合圆圆心

            // 水表中数字区域位于上半部？1=位于上半部分；0=位于下半部分。 默认值为1
            public byte is_nums_in_upper;     // uint8_t 对应 byte

            // 数字+指针水表中“数字”部分读数；全指针水表中“黑色”指针部分的读数；全数字水表中“数字”部分读数
            public ulong result_of_1st_part;  // uint64_t 对应 ulong

            // 数字+指针水表中“指针”部分读数；全指针水表中“红色”指针部分的读数；用来判断此水表是否在动
            public ulong result_of_2st_part;  // uint64_t 对应 ulong

            // 压缩JPG的索引,来自after_struct,首次索引为0
            public byte compress_jpg_index;   // uint8_t 对应 byte

            // 是否上传红色指针部分的数据(0:不上传;1:上传;默认1)
            public byte is_upload_red_ind_part;// uint8_t 对应 byte


            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 200)]
            public ulong[] history_result; // 对应 uint16_t[10][4] 展开成一维数组


            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 16)]
            public ulong[] history_digit_statistics;
        }


            //After结构体
            [StructLayout(LayoutKind.Sequential, Pack = 1)]
        public struct AfterAI
        {
            public byte NUM_DETECTED;    //01（本次结果）检测到的数字个数(黑色+红色)
            public byte IND_DETECTED;    //01（本次结果）在数字+指针水表中，检测到的指针个数，全指针水表不用此值(2024.05.06增加)
            public byte NUM_BLACK_DETECTED; //01 （本次信息）检测到的 数字个数(黑色)

            public ulong AI_result;      //识别最终结果，经过校正步骤后的最终结果，单位分升dL
            public byte is_OK_preValue;  //ai_result的值是否是可信的值, add by djs 20240417
            public ulong init_AI_result; //网络直接输出的、没有经过校正的最初识别结果，单位分升dL

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 24)] // AI_result_array[2][12] 转成一维数组
            public byte[] AI_result_array;    //03（本次结果）int数组的识别结果（2维数组，第一行元素为类别、第二行为识别概率）

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
            public ushort[] num_region;    //04（本次结果）数字区域的xmin、ymin、xmax、ymax坐标值

            public short lean_angle;    //05（本次结果）图像倾斜角度，和Before_AI_struct_RX结构体中的brightness一起维护图像倾斜角度

            public byte is_get_ShuiLiSuo_image;  //06（本次结果）20220620 new add by lyj 是否生成了水利所的小图， 0=没有生产 1=生成了
            public byte is_get_KaoJiPingTai_image; //07（本次结果）20220620 new add by lyj 是否生成了拷机平台的中图，0=没有生成 1=生成了

            public byte is_get_whole_image; //08（本次结果）是否生成了上传的整图？0=没有生成；1=生成了
                                            //生成大图的条件：邓金生要求生成、或廉永健识别失败0.91、0.92
            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
            public byte[] AI_ver;  //由日期改为->类型加版本号

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 1200)] // CNN_output_array[200][6] 转成一维数组
            public ushort[] CNN_output_array; //10（本次结果）CNN网络的输出数据

            public byte num_conf_valid_target; //11（本次结果）CNN网络的输出数据中有效目标的个数

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 40)] // indicators_regions[10][4] 转成一维数组
            public ushort[] indicators_regions;    //12（本次结果）存储每个指针区域的xmin、ymin、xmax、ymax坐标值

            public ulong MAX_FLOW_PER_HOUR;   //每小时的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
            public ulong MAX_FLOW_threshold;  //当前时间距离上次准确识别时间内的最大流量,最小单位0.1L(1m³/10000), (万分之一立方)
            public uint UNIT_OF_THE_LAST_NUMBER;   //尾数单位的含义,最小单位1分升,1dL, 0.1L（万分之一立方)

            public float center_x;          //表具中心点x
            public float center_y;          //表具中心点y
            public float image_zoomrate;       //图像放大倍率，中心点为拟合圆圆心 
            public float bright_factor; //图片亮度
            public byte meter_type;    //仪表类型(0:不是水表图片,1:数字+指针,2:全指针 3:全数字)

            public ulong result_of_1st_part;  //数字+指针水表中“数字”部分读数；全指针水表中“黑色”指针部分的读数；全数字水表中“数字”部分读数
            public ulong result_of_2st_part;  //数字+指针水表中“指针”部分读数；全指针水表中“红色”指针部分的读数；全数字水表中“数字”部分读数
            public ulong offset_1st_of_result; //暂时不用，数字+指针水表中“数字”部分的变化量；全指针水表中“黑色”指针部分的变化量
            public ulong offset_2nd_of_result; //数字+指针水表中“指针”部分的变化量绝对值；全指针水表中“红色”指针部分的变化量
            public byte compress_jpg_index;    //本次压缩的使用的索引

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 16)]
            public ulong[] history_digit_statistics;

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 3)]
            public ulong[] predicate_value;

            public byte is_clean_history;
        }

        [StructLayout(LayoutKind.Sequential, Pack = 1)]
        public struct SmallImage
        {
            public ushort size_of_JPEG_data;  //01（本次结果）数字区域的JPG数据长度

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 841)]
            public byte[] JPEG_data;  //02（本次结果）数字区域的JPG数字部分

            public byte number_region_index_of_header; //03（本次结果）JPG压缩头编号 （如果此值0-9，说明是彩色图，JPG文件头589字节；10-19，说明是灰度图，JPG文件头514字节）

            public int image_height;  //04（本次结果）图像高度
            public int image_width;   //05（本次结果）图像宽度
        }


        [StructLayout(LayoutKind.Sequential, Pack = 1)]
        public struct BigImage
        {
            public ushort size_of_JPEG_data;  //01（本次结果）JPG格式的整张图片数据长度 （包括JPG数据， 包括文件头）

            [MarshalAs(UnmanagedType.ByValArray, SizeConst = 50 * 1024)]
            public byte[] JPEG_data;  //02（本次结果）JPG格式的整张图片（包括JPG数据， 包括文件头）

            public byte whole_image_index_of_header;  //03（本次结果）JPG压缩头编号。（如果此值0-9，说明是彩色图，JPG文件头589字节；10-19，说明是灰度图，JPG文件头514字节）

            public ushort image_height;  //04（本次结果）图像高度

            public ushort image_width;   //05（本次结果）图像宽度
        }

        // 声明打印回调的委托
        public delegate void PrintfCallback(string message);

        //已改为使用动态加载方式，暂时把静态的注释掉
        // 引入 C 函数
        [DllImport("freeAI.dll", CharSet = CharSet.Unicode, CallingConvention = CallingConvention.Cdecl)]
        public static extern int recognition(
            [In] short[] rgb565,
            ref BeforeAI beforeAI,
            ref AfterAI afterAI,
            ref SmallImage smallJpg,
            ref BigImage bigJpg,
            [MarshalAs(UnmanagedType.LPWStr)] string modelPath,
            PrintfCallback callback
        );

    }
}