using Newtonsoft.Json; using OSGeo.GDAL; using OSGeo.OGR; using SimuTools.Domain; using System; using System.Collections.Generic; using System.Configuration; using System.Diagnostics; using System.IO; using System.Linq; using System.Text; using System.Threading.Tasks; namespace SimuTools.Tools { public class Handle { public static readonly string GdalPath = ConfigurationManager.AppSettings["gdalPath"]; public static readonly string BaseDir = AppDomain.CurrentDomain.BaseDirectory; /// /// 运行 /// public static void Run(string terrainFile, string waterPath, string flowPath, string outPath) { Domain.Layer layer = new Domain.Layer(); layer.duration = new Duration(); layer.terrain = new Terrain(); layer.waters = new Water(); string temp = Path.Combine(outPath, "temp"); if (!Directory.Exists(temp)) Directory.CreateDirectory(temp); CopeTerrain(terrainFile, outPath, layer); CopeWater(waterPath, outPath, layer); CopeFlow(flowPath, outPath, layer); CopeLayerJson(outPath, layer); //if (Directory.Exists(temp)) Directory.Delete(temp, true); } /// /// 处理地形 /// private static void CopeTerrain(string terrainFile, string outPath, Domain.Layer layer) { Dataset ds = null; try { ds = Gdal.Open(terrainFile, Access.GA_ReadOnly); if (null == ds || 0 == ds.RasterCount || null == ds.GetSpatialRef()) return; SetTerrainInfo(ds, layer); CreateTerrainPng(ds, layer, outPath); } finally { if (null != ds) ds.Dispose(); } } /// /// 设置地形信息 /// private static void SetTerrainInfo(Dataset ds, Domain.Layer layer) { 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)); OSGeo.GDAL.Band band = ds.GetRasterBand(1); double[] mm = new double[2]; band.ComputeRasterMinMax(mm, 0); layer.extension.SetHeight(mm[0], mm[1]); } /// /// 创建地形图层 /// private static void CreateTerrainPng(Dataset ds, Domain.Layer layer, string outPath) { string tempPath = Path.Combine(outPath, "temp"); if (!Directory.Exists(tempPath)) Directory.CreateDirectory(tempPath); string terrainPath = Path.Combine(outPath, "terrain"); if (!Directory.Exists(terrainPath)) Directory.CreateDirectory(terrainPath); foreach (int[] sizes in layer.terrain.size) { //string filePath = Path.Combine(outPath, sizes[0] + "_" + sizes[1] + ".png"); //// 填充内存图像 //byte[] buffer = new byte[sizes[0] * sizes[1] * 3]; //for (int i = 0; i < buffer.Length; i++) //{ // buffer[i] = (byte)(i % 256); //} string tif = Path.Combine(tempPath, DateTime.Now.Ticks.ToString() + ".tif"); Resample(ds.GetDescription(), tif, sizes[0], sizes[1]); if (!File.Exists(tif)) { continue; } // } } /// /// 重采样 /// private static void Resample(string source, string dest, int width, int height) { string cmd = string.Format("{0}gdalwarp.exe -t_srs {1} -ts {2} {3} -r {4} -of GTiff \"{5}\" \"{6}\"", GdalPath, "EPSG:4326", width, height, "bilinear", source, dest); string err = ExecExe(cmd); } /// /// 执行命令 /// public static string ExecExe(string cmd) { string str = null; Process p = null; try { p = new Process(); p.StartInfo.FileName = "cmd.exe"; p.StartInfo.UseShellExecute = false; p.StartInfo.CreateNoWindow = true; p.StartInfo.RedirectStandardInput = true; p.StartInfo.RedirectStandardOutput = true; p.StartInfo.RedirectStandardError = true; p.Start(); StreamWriter si = p.StandardInput; StreamReader se = p.StandardError; si.AutoFlush = true; si.WriteLine(cmd); si.WriteLine("exit"); str = se.ReadToEnd(); se.Close(); si.Close(); } catch (Exception ex) { LogOut.Error(ex.Message + "\r\n" + ex.StackTrace); str = ex.Message; } finally { if (p != null) p.Close(); } return str; } /// /// 处理水面 /// private static void CopeWater(string waterPath, string outPath, Domain.Layer layer) { } /// /// 处理流速流向 /// private static void CopeFlow(string flowPath, string outPath, Domain.Layer layer) { } /// /// 处理元数据 /// 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); } } #region 暂时不用 /// /// 重采样 * /// private static void Resample(Dataset ds, string dest, int width, int height) { // 设置Warp的选项:https://blog.csdn.net/qq_43210879/article/details/121350561 string[] options = new string[] { "format=GTiff", "width=" + width, "height=" + height, "dstSRS=EPSG:4326" }; OSGeo.GDAL.Driver driver = Gdal.GetDriverByName("GTiff"); Dataset destDs = driver.Create(dest, width, height, ds.RasterCount, ds.GetRasterBand(1).DataType, null); GDALWarpAppOptions warpAppOptions = new GDALWarpAppOptions(options); Gdal.Warp(destDs, new Dataset[] { ds }, warpAppOptions, null, null); destDs.Dispose(); } #endregion } }