From a8028c5fe78d8e9a05f171407ac129c4fbd65e87 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 24 七月 2024 16:47:01 +0800
Subject: [PATCH] 1

---
 SimuTools/Tools/Handle.cs |  178 +++++++++++++++++++++++++++++++++--------------------------
 1 files changed, 100 insertions(+), 78 deletions(-)

diff --git a/SimuTools/Tools/Handle.cs b/SimuTools/Tools/Handle.cs
index febc768..a7a4aa6 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;
@@ -72,9 +76,16 @@
         /// </summary>
         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";
+
             Geometry minPoint = GdalHelper.GetMinPoint(ds);
             Geometry maxPoint = GdalHelper.GetMaxPoint(ds);
-            layer.extension = new Extension(minPoint.GetX(0), minPoint.GetY(0), maxPoint.GetX(0), maxPoint.GetY(0), double.MaxValue, double.MinValue);
+            double minx = GetMinVal(minPoint.GetX(0), 10000000);
+            double miny = GetMinVal(minPoint.GetY(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);
 
             Band band = ds.GetRasterBand(1);
             double[] mm = new double[2];
@@ -85,17 +96,17 @@
         /// <summary>
         /// 鑾峰彇鏈�灏忓��
         /// </summary>
-        private static double GetMinVal(double val)
+        private static double GetMinVal(double val, double radix = 1000)
         {
-            return Convert.ToInt32(Math.Floor(val * 1000)) / 1000.0;
+            return Convert.ToInt32(Math.Floor(val * radix)) / radix;
         }
 
         /// <summary>
         /// 鑾峰彇鏈�澶у��
         /// </summary>
-        private static double GetMaxVal(double val)
+        private static double GetMaxVal(double val, double radix = 1000)
         {
-            return Convert.ToInt32(Math.Ceiling(val * 1000)) / 1000.0;
+            return Convert.ToInt32(Math.Ceiling(val * radix)) / radix;
         }
 
         /// <summary>
@@ -130,20 +141,12 @@
             //string[] options = Gdal.ParseCommandLine(args);
             string[] options = new string[]
             {
-                "-t_srs",
-                "EPSG:4326",
-                "-ts",
-                width.ToString(),
-                height.ToString(),
-                "-te",
-                layer.extension.minx.ToString(),
-                layer.extension.miny.ToString(),
-                layer.extension.maxx.ToString(),
-                layer.extension.maxy.ToString(),
-                "-r",
-                "bilinear",
-                "-of",
-                "GTiff"
+                "-t_srs", "EPSG:4326", // layer.terrain.epsg,
+                "-ts", width.ToString(), height.ToString(),
+                "-te", layer.extension.minx.ToString(), layer.extension.miny.ToString(), layer.extension.maxx.ToString(), layer.extension.maxy.ToString(),
+                "-te_srs", "EPSG:4326", // layer.terrain.epsg,
+                "-r", "bilinear",
+                "-of", "GTiff"
             };
             GDALWarpAppOptions warpAppOptions = new GDALWarpAppOptions(options);
 
@@ -170,7 +173,8 @@
                 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;
                 for (int x = 0; x < width; x++)
                 {
                     for (int y = 0; y < height; y++)
@@ -178,10 +182,10 @@
                         int offset = x + y * width;
                         if (float.IsNaN(buffer[offset]) || buffer[offset] < -999) 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 val = Convert.ToInt32((buffer[offset] - layer.terrain.minHeight) * 100 / perHeight);
+                        int val = Convert.ToInt32((buffer[offset] - layer.terrain.minHeight) / differ * 65536);
+                        int r = 0;
+                        int g = val / 256;
                         int b = val % 256;
 
                         Color color = Color.FromArgb(127, r, g, b);
@@ -193,7 +197,7 @@
             finally
             {
                 if (null != ds) ds.Dispose();
-                if (File.Exists(tif)) File.Delete(tif);
+                //if (File.Exists(tif)) File.Delete(tif);
             }
         }
         #endregion
@@ -330,7 +334,7 @@
                 ds = Gdal.Open(tif, Access.GA_ReadOnly);
                 if (null == ds || 0 == ds.RasterCount) return;
 
-                OSGeo.GDAL.Band band = ds.GetRasterBand(1);
+                Band band = ds.GetRasterBand(1);
                 float[] buffer = new float[width * height];
                 band.ReadRaster(0, 0, width, height, buffer, width, height, 0, 0);
 
@@ -361,7 +365,7 @@
             finally
             {
                 if (null != ds) ds.Dispose();
-                if (File.Exists(tif)) File.Delete(tif);
+                //if (File.Exists(tif)) File.Delete(tif);
             }
         }
         #endregion
@@ -440,29 +444,21 @@
                 vxDs.GetRasterBand(1).ReadRaster(0, 0, width, height, vxBuffer, width, height, 0, 0);
                 vyDs.GetRasterBand(1).ReadRaster(0, 0, width, height, vyBuffer, width, height, 0, 0);
 
-                double[] vxMm = new double[2];
-                vxDs.GetRasterBand(1).ComputeRasterMinMax(vxMm, 0);
-                double[] vyMm = new double[2];
-                vyDs.GetRasterBand(1).ComputeRasterMinMax(vyMm, 0);
-                double min = vxMm[0] < vyMm[0] ? vxMm[0] : vyMm[0];
-                double max = vxMm[1] > vyMm[1] ? vxMm[1] : vyMm[1];
-                double perHeight = (max - min) / 255;
-
-                CreateFlowPng(vxBuffer, vyBuffer, min, perHeight, png, width, height);
+                CreateFlowPng(vxBuffer, vyBuffer, png, width, height);
             }
             finally
             {
                 if (null != vxDs) vxDs.Dispose();
                 if (null != vyDs) vyDs.Dispose();
-                if (File.Exists(vxTif)) File.Delete(vxTif);
-                if (File.Exists(vyTif)) File.Delete(vyTif);
+                //if (File.Exists(vxTif)) File.Delete(vxTif);
+                //if (File.Exists(vyTif)) File.Delete(vyTif);
             }
         }
 
         /// <summary>
         /// 鍒涘缓娴侀�熸祦鍚慞NG
         /// </summary>
-        private static void CreateFlowPng(float[] vxBuffer, float[] vyBuffer, double minHeight, double perHeight, string png, int width, int height)
+        private static void CreateFlowPng(float[] vxBuffer, float[] vyBuffer, string png, int width, int height)
         {
             Bitmap image = new Bitmap(width, height);
             Graphics graphic = Graphics.FromImage(image);
@@ -476,17 +472,17 @@
                     int offset = x + y * width;
                     float fx = GetFloatValue(vxBuffer[offset]);
                     float fy = GetFloatValue(vyBuffer[offset]);
-                    double rv = GetRVal(fx, fy);
-                    double dv = GetXyValue(fx, fy, rv);
+                    if (float.IsNaN(fx) && float.IsNaN(fy) || (fx == 0 && fy == 0)) continue;
 
-                    int r= Convert.ToInt32(double.IsNaN(dv) || dv == 0 ? 0 : (dv * 0.5 + 0.5) * 255);
-                    int g = float.IsNaN(fx) ? 0 : Convert.ToInt32((fx - minHeight) / perHeight);
-                    int b = float.IsNaN(fy) ? 0 : Convert.ToInt32((fy - minHeight) / perHeight);
-                    if (r < 0 || r > 255) r = 255;
-                    if (g < 0 || g > 255) g = 255;
-                    if (b < 0 || b > 255) b = 255;
+                    fx = float.IsNaN(fx) ? 0 : fx;
+                    fy = float.IsNaN(fy) ? 0 : fy;
+                    double dr = Math.Sqrt(Math.Pow(fx, 2) + Math.Pow(fy, 2));
 
-                    Color color = Color.FromArgb(127, r, g, b);
+                    int r = Convert.ToInt32(dr / 4 * 255);
+                    int g = Convert.ToInt32((fx / dr * 0.5 + 0.5) * 255);
+                    int b = Convert.ToInt32((fy / dr * 0.5 + 0.5) * 255);
+
+                    Color color = Color.FromArgb(127, GetSafeValue(r), GetSafeValue(g), GetSafeValue(b));
                     image.SetPixel(x, y, color);
                 }
             }
@@ -502,38 +498,14 @@
         }
 
         /// <summary>
-        /// 鑾峰彇R鍊�
+        /// 鑾峰彇瀹夊叏鍊�
         /// </summary>
-        private static double GetRVal(float fx, float fy)
+        private static int GetSafeValue(int val)
         {
-            if (float.IsNaN(fx) && float.IsNaN(fy)) return double.NaN;
+            if (val < 0) return 0;
+            if (val > 255) return 255;
 
-            if (float.IsNaN(fx)) return Math.Abs(fy);
-
-            if (float.IsNaN(fy)) return Math.Abs(fx);
-
-            return Math.Sqrt(Math.Pow(fx, 2) + Math.Pow(fy, 2));
-        }
-
-        /// <summary>
-        /// 鑾峰彇XY鍊�
-        /// </summary>
-        private static double GetXyValue(float fx, float fy, double rv)
-        {
-            if ((float.IsNaN(fx) && float.IsNaN(fy)) || double.IsNaN(rv)) return 0;
-
-            if (float.IsNaN(fx))
-            {
-                return fy / rv;
-            }
-            if (float.IsNaN(fy))
-            {
-                return fx / rv;
-            }
-
-            double dv = fx / rv + fy / rv;
-
-            return dv < 0 ? -Math.Sqrt(-dv) : Math.Sqrt(dv);
+            return val;
         }
         #endregion
 
@@ -557,7 +529,57 @@
 
         #region 鏆傛椂涓嶇敤
         /// <summary>
-        /// 閲嶉噰鏍� *
+        /// 璁剧疆鍦板舰淇℃伅
+        /// </summary>
+        private static void SetTerrainInfo(Dataset ds, string outPath, Domain.Layer layer)
+        {
+            string tempPath = Path.Combine(outPath, "temp");
+            if (!Directory.Exists(tempPath)) Directory.CreateDirectory(tempPath);
+
+            string tif = Path.Combine(tempPath, Path.GetFileNameWithoutExtension(ds.GetDescription()) + ".tif");
+            Resample(ds, tif);
+            if (!File.Exists(tif)) throw new Exception("楂樼▼鏂囦欢閲嶆姇褰卞け璐ワ紒");
+
+            Dataset tifDS = null;
+            try
+            {
+                tifDS = Gdal.Open(tif, Access.GA_ReadOnly);
+                if (null == tifDS || 0 == tifDS.RasterCount || null == tifDS.GetSpatialRef()) return;
+
+                double[] transform = new double[6];
+                tifDS.GetGeoTransform(transform);
+                double minx = transform[0];
+                double miny = transform[3] - tifDS.RasterYSize * transform[1];
+                double maxx = transform[0] + (tifDS.RasterYSize * transform[1]);
+                double maxy = transform[3];
+
+                layer.extension = new Extension(minx, miny, maxx, maxy, double.MaxValue, double.MinValue);
+
+                Band band = tifDS.GetRasterBand(1);
+                double[] mm = new double[2];
+                band.ComputeRasterMinMax(mm, 0);
+                layer.terrain.SetHeight(GetMinVal(mm[0]), GetMaxVal(mm[1]));
+            }
+            finally
+            {
+                if (null != tifDS) tifDS.Dispose();
+            }
+        }
+
+        /// <summary>
+        /// 閲嶉噰鏍�
+        /// </summary>
+        private static void Resample(Dataset ds, string dest)
+        {
+            string[] options = new string[] { "-t_srs", "EPSG:4326", "-r", "bilinear", "-of", "GTiff" };
+            GDALWarpAppOptions warpAppOptions = new GDALWarpAppOptions(options);
+
+            Dataset destDs = Gdal.Warp(dest, new Dataset[] { ds }, warpAppOptions, null, null);
+            destDs.Dispose();
+        }
+
+        /// <summary>
+        /// 閲嶉噰鏍�
         /// </summary>
         private static void Resample(string source, string dest, int width, int height)
         {

--
Gitblit v1.9.3