并行处理水面数据

13693261870

2024-07-19 e2102fe4a286a495a7932b0559125e2318b802cc

 SimuTools/Tools/Handle.cs

@@ -11,6 +11,7 @@
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace SimuTools.Tools
{
@@ -35,7 +36,7 @@
            string temp = Path.Combine(outPath, "temp");
            if (!Directory.Exists(temp)) Directory.CreateDirectory(temp);

            //CopeTerrain(terrainFile, outPath, layer);
            CopeTerrain(terrainFile, outPath, layer);
            CopeWater(waterPath, outPath, layer);
            CopeFlow(flowPath, outPath, layer);
            CopeLayerJson(outPath, layer);
@@ -186,7 +187,7 @@
                {
                    for (int y = 0; y < height; y++)
                    {
                        if (buffer[x * y] == double.NaN || buffer[x * y] == -9999)
                        if (float.IsNaN(buffer[x * y]) || buffer[x * y] == -9999)
                        {
                            image.SetPixel(x, y, Color.Transparent);
                            continue;
@@ -208,6 +209,7 @@
            finally
            {
                if (null != ds) ds.Dispose();
                if (File.Exists(tif)) File.Delete(tif);
            }
        }

@@ -216,13 +218,67 @@
        /// </summary>
        private static void CopeWater(string waterPath, string outPath, Domain.Layer layer)
        {
            string watersPath = Path.Combine(outPath, "waters");
            if (!Directory.Exists(watersPath)) Directory.CreateDirectory(watersPath);

            List<string> files = GetFiles(waterPath);
            if (null == files || files.Count == 0) return;

            SetWaterData(layer, files);

            /*for (int i = 0, c = files.Count; i < c; i++)
            {
                Dataset ds = null;
                try
                {
                    ds = Gdal.Open(files[i], Access.GA_ReadOnly);
                    if (null == ds || 0 == ds.RasterCount || null == ds.GetSpatialRef()) continue;

                    CreateWaterPng(ds, layer, outPath, layer.waters.data[i]);
                }
                finally
                {
                    if (null != ds) ds.Dispose();
                }
            }*/

            Parallel.For(0, files.Count, i =>
            {
                Dataset ds = null;
                try
                {
                    ds = Gdal.Open(files[i], Access.GA_ReadOnly);
                    if (null == ds || 0 == ds.RasterCount || null == ds.GetSpatialRef()) return;

                    CreateWaterPng(ds, layer, outPath, layer.waters.data[i]);
                }
                finally
                {
                    if (null != ds) ds.Dispose();
                }
            });
        }

        /// <summary>
        /// 创建地形图层
        /// </summary>
        private static void CreateWaterPng(Dataset ds, Domain.Layer layer, string outPath, long ticks)
        {
            string tempPath = Path.Combine(outPath, "temp");
            if (!Directory.Exists(tempPath)) Directory.CreateDirectory(tempPath);
            string waterPath = Path.Combine(outPath, "waters", ticks.ToString());
            if (!Directory.Exists(waterPath)) Directory.CreateDirectory(waterPath);

            foreach (int[] sizes in layer.terrain.size)
            {
                string fileName = Path.GetFileNameWithoutExtension(ds.GetDescription()) + "_" + sizes[0] + "_" + sizes[1];
                string tif = Path.Combine(tempPath, fileName + ".tif");
                Resample(ds.GetDescription(), tif, sizes[0], sizes[1]);
                if (!File.Exists(tif))
                {
                    continue;
                }

                string png = Path.Combine(waterPath, sizes[0] + "_" + sizes[1] + ".png");
                Terrain2Png(layer, tif, png, sizes[0], sizes[1]);
            }
        }

        /// <summary>
@@ -245,7 +301,7 @@
        private static void SetWaterData(Domain.Layer layer, List<string> files)
        {
            DateTime now = DateTime.Now;
            long startTicks = new DateTime(1970, 1, 1, 0, 0, 0).Ticks;
            long startTicks = new DateTime(1970, 1, 1, 8, 0, 0).Ticks;

            foreach (string file in files)
            {