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 | 179 +++++++++++++++++++++++++++++++++--------------------------
1 files changed, 99 insertions(+), 80 deletions(-)
diff --git a/SimuTools/Tools/Handle.cs b/SimuTools/Tools/Handle.cs
index 579d7a8..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,20 +472,17 @@
int offset = x + y * width;
float fx = GetFloatValue(vxBuffer[offset]);
float fy = GetFloatValue(vyBuffer[offset]);
+ if (float.IsNaN(fx) && float.IsNaN(fy) || (fx == 0 && fy == 0)) continue;
- double rv = GetRVal(fx, fy);
- double dv = GetXyValue(fx, fy, rv);
- int r = Convert.ToInt32(double.IsNaN(dv) || dv == 0 ? 0 : (dv * 0.5 + 0.5) * 255);
- if (r < 0 || r > 255) r = 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));
- //int g = Convert.ToInt32((float.IsNaN(fx) ? 0 : fx * 0.5 + 0.5) * 255);
- //int b = Convert.ToInt32((float.IsNaN(fy) ? 0 : fy * 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 (g < 0 || g > 255) g = 255;
- if (b < 0 || b > 255) b = 255;
+ 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, r, g, b);
+ Color color = Color.FromArgb(127, GetSafeValue(r), GetSafeValue(g), GetSafeValue(b));
image.SetPixel(x, y, color);
}
}
@@ -505,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 1.0;
+ 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 v = Math.Sqrt(Math.Pow(fx / rv, 2) + Math.Pow(fy / rv, 2));
-
- return (fx > 0 && fy > 0) || (fx < 0 && fy < 0) ? v : -v;
+ return val;
}
#endregion
@@ -560,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