XdCxRhDW54/XdCxRhDW.Api/AddIns/信号仿真/DataEmulationHelper.cs

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

namespace XdCxRhDW.Api
{
    public static class DataEmulationHelper
    {
        private const string dll = @"AddIns\信号仿真\SignalEmulation.dll";

        #region cpp dll Interop 
        [DllImport(dll, EntryPoint = "genSignalsWithDtDf", CallingConvention = CallingConvention.Cdecl)]
        private extern static void GenSignalsWithDtDfCore(string file1, string file2, long fsHz, int sigLenMs, double dt, double df, int snr2, int snr1);

        //生成噪声,文件，采样率，时长ms,最大幅度
        [DllImport(dll, EntryPoint = "genNoizes", CallingConvention = CallingConvention.Cdecl)]
        private extern static void GenNoizes(string file1, long fsHz, int tlenms, int Am);
        #endregion

        /// <summary>
        /// 生成一组常规信号文件
        /// </summary>
        /// <param name="file1">输出文件1</param>
        /// <param name="file2">输出文件2</param>
        /// <param name="fsHz">采样率Hz</param>
        /// <param name="sigTimeLenMs">信号时长(毫秒)</param>
        /// <param name="dt">时差秒</param>
        /// <param name="df">频差Hz</param>
        /// <param name="snr1">文件1信噪比</param>
        /// <param name="snr2">文件2信噪比</param>
        public static void GenNormalFiles(string file1, string file2, long fsHz, int sigTimeLenMs, double dt, double df, int snr1, int snr2)
        {
            GenSignalsWithDtDfCore(file1, file2, fsHz, sigTimeLenMs, dt, df, snr2, snr1);
        }

        /// <summary>
        /// 生成单个噪声文件
        /// </summary>
        /// <param name="file1"></param>
        /// <param name="fsHz"></param>
        /// <param name="timeLenMs"></param>
        /// <param name="am"></param>
        public static void GenNoizesFile(string file1, long fsHz, int timeLenMs, int am)
        {
            GenNoizes(file1, fsHz, timeLenMs, am); 
        }
        /// <summary>
        /// 生成一组突发信号文件
        /// </summary>
        /// <param name="file1">输出文件1</param>
        /// <param name="file2">输出文件2</param>
        /// <param name="fsHz">采样率Hz</param>
        /// <param name="brustCount">突发个数</param>
        /// <param name="dt">时差秒</param>
        /// <param name="df">频差Hz</param>
        public static void GenBrustFiles(string file1, string file2, long fsHz, int brustCount, double dt, double df)
        {
            GenNoizes("n11.dat", fsHz, 900, 40);
            GenNoizes("n12.dat", fsHz, 900, 40);
            GenNormalFiles("s1.dat", "s2.dat", fsHz, 100, dt, df, 5, -5);
            GenNoizes("n21.dat", fsHz, 900, 40);
            GenNoizes("n22.dat", fsHz, 900, 40);
            var n11 = File.ReadAllBytes("n11.dat");
            var n12 = File.ReadAllBytes("n12.dat");
            var s1 = File.ReadAllBytes("s1.dat");
            var s2 = File.ReadAllBytes("s2.dat");
            var n21 = File.ReadAllBytes("n21.dat");
            var n22 = File.ReadAllBytes("n22.dat");
            using (FileStream fs = new FileStream(file1, FileMode.Create))
            {
                for (int i = 0; i < brustCount; i++)
                {
                    fs.Write(n11, 0, n11.Length);
                    fs.Write(s1, 0, s1.Length);
                    fs.Write(n21, 0, n21.Length);
                }
            
            }
            using (FileStream fs = new FileStream(file2, FileMode.Create))
            {
                for (int i = 0; i < brustCount; i++)
                {
                    fs.Write(n12, 0, n12.Length);
                    fs.Write(s2, 0, s2.Length);
                    fs.Write(n22, 0, n22.Length);
                }
            }
            File.Delete("n11.dat");
            File.Delete("n12.dat");
            File.Delete("s1.dat");
            File.Delete("s2.dat");
            File.Delete("n21.dat");
            File.Delete("n22.dat");
        }
    }
}