From e2102fe4a286a495a7932b0559125e2318b802cc Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 19 七月 2024 17:27:23 +0800
Subject: [PATCH] 并行处理水面数据
---
SimuTools/Tools/Handle.cs | 68 +++++++++++++++++++++++++++++++---
1 files changed, 62 insertions(+), 6 deletions(-)
diff --git a/SimuTools/Tools/Handle.cs b/SimuTools/Tools/Handle.cs
index e8f81ab..6ff3f22 100644
--- a/SimuTools/Tools/Handle.cs
+++ b/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)
{
--
Gitblit v1.9.3