From 9d25253a85cc4fb3aa0f4adc764626ad0e64d66a Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 25 七月 2024 14:40:01 +0800
Subject: [PATCH] 修改算法-1

---
 SimuTools/Tools/Handle.cs |  137 +++++++++++++++++++++++++++++----------------
 1 files changed, 88 insertions(+), 49 deletions(-)

diff --git a/SimuTools/Tools/Handle.cs b/SimuTools/Tools/Handle.cs
index 0a9c3fd..0b3938d 100644
--- a/SimuTools/Tools/Handle.cs
+++ b/SimuTools/Tools/Handle.cs
@@ -18,6 +18,10 @@
         #region 鎴愬憳鍙橀噺
         public static readonly int MAX = 256 * 256;
 
+        public static readonly double MAX_X_OFFSET = 0.002;
+
+        public static readonly double MIN_X_OFFSET = -0.00005;
+
         public static readonly double WaterHeightOffset = 1.0;
 
         public static readonly string BaseDir = AppDomain.CurrentDomain.BaseDirectory;
@@ -38,19 +42,19 @@
             string temp = Path.Combine(outPath, "temp");
             if (!Directory.Exists(temp)) Directory.CreateDirectory(temp);
 
-            CopeTerrain(terrainFile, outPath, layer);
+            CopeTerrain(terrainFile, waterPath, outPath, layer);
             CopeWater(waterPath, outPath, layer);
-            //CopeFlow(flowPath, outPath, layer);
+            CopeFlow(flowPath, outPath, layer);
             CopeLayerJson(outPath, layer);
 
-            //if (Directory.Exists(temp)) Directory.Delete(temp, true);
+            if (Directory.Exists(temp)) Directory.Delete(temp, true);
         }
 
         #region 鍦板舰
         /// <summary>
         /// 澶勭悊鍦板舰
         /// </summary>
-        private static void CopeTerrain(string terrainFile, string outPath, Domain.Layer layer)
+        private static void CopeTerrain(string terrainFile, string waterPath, string outPath, Domain.Layer layer)
         {
             Dataset ds = null;
             try
@@ -58,7 +62,8 @@
                 ds = Gdal.Open(terrainFile, Access.GA_ReadOnly);
                 if (null == ds || 0 == ds.RasterCount || null == ds.GetSpatialRef()) return;
 
-                SetTerrainInfo(ds, outPath, layer);
+                SetTerrainInfo(ds, layer);
+                SetWaterInfo(waterPath, layer);
                 CreateTerrainPng(ds, layer, outPath);
             }
             finally
@@ -70,7 +75,7 @@
         /// <summary>
         /// 璁剧疆鍦板舰淇℃伅
         /// </summary>
-        private static void SetTerrainInfo(Dataset ds, string outPath, Domain.Layer layer)
+        private static void SetTerrainInfo(Dataset ds, Domain.Layer layer)
         {
             //String srsName = ds.GetSpatialRef().GetName();
             //layer.terrain.epsg = !string.IsNullOrEmpty(srsName) && srsName.Contains(GdalHelper.CGCS2000) ? "EPSG:4490" : "EPSG:4326";
@@ -79,7 +84,7 @@
             Geometry maxPoint = GdalHelper.GetMaxPoint(ds);
             double minx = GetMinVal(minPoint.GetX(0), 10000000);
             double miny = GetMinVal(minPoint.GetY(0), 10000000);
-            double maxx = GetMaxVal(maxPoint.GetX(0), 10000000);
+            double maxx = GetMaxVal(maxPoint.GetX(0) + MAX_X_OFFSET, 10000000);
             double maxy = GetMaxVal(maxPoint.GetY(0), 10000000);
             layer.extension = new Extension(minx, miny, maxx, maxy, double.MaxValue, double.MinValue);
 
@@ -169,19 +174,30 @@
                 Graphics graphic = Graphics.FromImage(image);
                 graphic.Clear(Color.Transparent); // 濉厖閫忔槑鑹�
 
-                double perHeight = (layer.terrain.maxHeight - layer.terrain.minHeight) * 100 / 65536;
+                //double perHeight = (layer.terrain.maxHeight - layer.terrain.minHeight) * 100 / 65536;
+                //double differ = layer.terrain.maxHeight - layer.terrain.minHeight;
+                double differ = layer.extension.maxHeight - layer.extension.minHeight, minHeight = layer.extension.minHeight;
                 for (int x = 0; x < width; x++)
                 {
                     for (int y = 0; y < height; y++)
                     {
                         int offset = x + y * width;
-                        if (float.IsNaN(buffer[offset]) || buffer[offset] < -999) continue;
+                        if (float.IsNaN(buffer[offset]) || buffer[offset] < -999 || buffer[offset] < minHeight) continue;
 
-                        //int val = Convert.ToInt32(buffer[offset] * 100);
-                        int val = Convert.ToInt32((buffer[offset] - layer.terrain.minHeight) * 100 / perHeight);
-                        int r = val / 65536;
-                        int g = (val - r * 65536) / 256;
-                        int b = val % 256;
+                        //int val = Convert.ToInt32((buffer[offset] - layer.terrain.minHeight) * 100 / perHeight);
+                        //int val = Convert.ToInt32((buffer[offset] - layer.terrain.minHeight) / differ * 255); // 65535
+
+                        int r = 0, g, b;
+                        if (buffer[offset] - layer.extension.maxHeight > 0)
+                        {
+                            g = b = 255;
+                        }
+                        else
+                        {
+                            int val = Convert.ToInt32((buffer[offset] - minHeight) / differ * 65535);
+                            g = val / 256;
+                            b = val % 256;
+                        }
 
                         Color color = Color.FromArgb(127, r, g, b);
                         image.SetPixel(x, y, color);
@@ -199,18 +215,26 @@
 
         #region 姘撮潰
         /// <summary>
-        /// 澶勭悊姘撮潰
+        /// 璁剧疆姘撮潰淇℃伅
         /// </summary>
-        private static void CopeWater(string waterPath, string outPath, Domain.Layer layer)
+        private static void SetWaterInfo(string waterPath, Domain.Layer layer)
         {
-            List<string> files = GetFiles(waterPath);
+            List<string> files = layer.waters.files = GetFiles(waterPath);
             if (null == files || files.Count == 0) return;
 
             SetWaterData(layer, files);
             SetWaterHeight(layer, files);
-            if (files.Count != layer.waters.data.Count) return;
+        }
 
-            //ProcessWaters(files, outPath, layer);
+        /// <summary>
+        /// 澶勭悊姘撮潰
+        /// </summary>
+        private static void CopeWater(string waterPath, string outPath, Domain.Layer layer)
+        {
+            List<String> files = layer.waters.files;
+            if (files.Count == 0 || files.Count != layer.waters.data.Count) return;
+
+            ProcessWaters(files, outPath, layer);
         }
 
         /// <summary>
@@ -219,7 +243,7 @@
         private static List<String> GetFiles(string path, string prefix = "")
         {
             string[] files = Directory.GetFiles(path, prefix + "*.tif?", SearchOption.TopDirectoryOnly);
-            if (null == files || files.Length == 0) return null;
+            if (null == files || files.Length == 0) return new List<string>();
 
             List<string> list = files.ToList();
             list.Sort();
@@ -245,6 +269,8 @@
                 DateTime d = new DateTime(now.Year, now.Month, now.Day, hour, minute, second);
                 layer.waters.data.Add((d.Ticks - startTicks) / 10000);
             }
+            layer.duration.start = layer.waters.data[0];
+            layer.duration.end = layer.waters.data[layer.waters.data.Count - 1];
         }
 
         /// <summary>
@@ -269,8 +295,8 @@
                     if (null != ds) ds.Dispose();
                 }
             });
-            layer.extension.minHeight = GetMinVal(layer.extension.minHeight);
-            layer.extension.maxHeight = GetMaxVal(layer.extension.maxHeight + WaterHeightOffset);
+            layer.extension.minHeight = GetMinVal(layer.extension.minHeight - 1);
+            layer.extension.maxHeight = GetMaxVal(layer.extension.maxHeight + 1);
         }
 
         /// <summary>
@@ -337,19 +363,32 @@
                 Graphics graphic = Graphics.FromImage(image);
                 graphic.Clear(Color.Transparent); // 濉厖閫忔槑鑹�
 
-                double perHeight = (layer.extension.maxHeight - layer.extension.minHeight) * 100 / 65536;
+                //double perHeight = (layer.extension.maxHeight - layer.extension.minHeight) * 100 / 65536;
+                double differ = layer.extension.maxHeight - layer.extension.minHeight, minHeight = layer.extension.minHeight;
                 for (int x = 0; x < width; x++)
                 {
                     for (int y = 0; y < height; y++)
                     {
                         int offset = x + y * width;
-                        if (float.IsNaN(buffer[offset]) || buffer[offset] < -999) continue;
+                        if (float.IsNaN(buffer[offset]) || buffer[offset] < -999 || buffer[offset] < minHeight) continue;
 
                         //int val = Convert.ToInt32(buffer[offset] * 100);
-                        int val = Convert.ToInt32((buffer[offset] - layer.extension.minHeight + WaterHeightOffset) * 100 / perHeight);
-                        int r = val / 65536;
-                        int g = (val - r * 65536) / 256;
-                        int b = val % 256;
+                        //int val = Convert.ToInt32((buffer[offset] - layer.extension.minHeight + WaterHeightOffset) * 100 / perHeight);
+                        //int r = val / 65536;
+                        //int g = (val - r * 65536) / 256;
+                        //int b = val % 256;
+
+                        int r = 0, g, b;
+                        if (buffer[offset] - layer.extension.maxHeight > 0)
+                        {
+                            g = b = 255;
+                        }
+                        else
+                        {
+                            int val = Convert.ToInt32((buffer[offset] - minHeight) / differ * 65535);
+                            g = val / 256;
+                            b = val % 256;
+                        }
 
                         Color color = Color.FromArgb(127, r, g, b);
                         image.SetPixel(x, y, color);
@@ -498,7 +537,7 @@
         private static int GetSafeValue(int val)
         {
             if (val < 0) return 0;
-            if (val > 255) val = 255;
+            if (val > 255) return 255;
 
             return val;
         }
@@ -506,9 +545,27 @@
 
         #region 鍏冩暟鎹�
         /// <summary>
+        /// 澶勭悊鍏冩暟鎹�
+        /// </summary>
+        private static void CopeLayerJson(string outPath, Domain.Layer layer)
+        {
+            if (null == layer) return;
+
+            String json = JsonConvert.SerializeObject(layer);
+
+            string filePath = Path.Combine(outPath, "layer.json");
+            using (StreamWriter sw = new StreamWriter(filePath))
+            {
+                sw.Write(json);
+            }
+        }
+        #endregion
+
+        #region 鏆傛椂涓嶇敤
+        /// <summary>
         /// 璁剧疆鍦板舰淇℃伅
         /// </summary>
-        private static void SetTerrainInfo2(Dataset ds, string outPath, Domain.Layer layer)
+        private static void SetTerrainInfo(Dataset ds, string outPath, Domain.Layer layer)
         {
             string tempPath = Path.Combine(outPath, "temp");
             if (!Directory.Exists(tempPath)) Directory.CreateDirectory(tempPath);
@@ -556,25 +613,7 @@
         }
 
         /// <summary>
-        /// 澶勭悊鍏冩暟鎹�
-        /// </summary>
-        private static void CopeLayerJson(string outPath, Domain.Layer layer)
-        {
-            if (null == layer) return;
-
-            String json = JsonConvert.SerializeObject(layer);
-
-            string filePath = Path.Combine(outPath, "layer.json");
-            using (StreamWriter sw = new StreamWriter(filePath))
-            {
-                sw.Write(json);
-            }
-        }
-        #endregion
-
-        #region 鏆傛椂涓嶇敤
-        /// <summary>
-        /// 閲嶉噰鏍� *
+        /// 閲嶉噰鏍�
         /// </summary>
         private static void Resample(string source, string dest, int width, int height)
         {

--
Gitblit v1.9.3