添加Tif值转存至Png

13693261870

2024-07-19 5d77b35448990a62846b29ce96ceb0cf2faa20f7

 SimuTools/Tools/Handle.cs

@@ -4,15 +4,22 @@
using SimuTools.Domain;
using System;
using System.Collections.Generic;
using System.Configuration;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace SimuTools.Tools
{
    public class Handle
    {
        public static readonly int MAX = 256 * 256;

        public static readonly string GdalPath = ConfigurationManager.AppSettings["gdalPath"];

        public static readonly string BaseDir = AppDomain.CurrentDomain.BaseDirectory;

        /// <summary>
@@ -25,10 +32,15 @@
            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);
        }

        /// <summary>
@@ -60,7 +72,7 @@
            Geometry maxPoint = GdalHelper.GetMaxPoint(ds);
            layer.extension = new Extension(minPoint.GetX(0), minPoint.GetY(0), maxPoint.GetX(0), maxPoint.GetY(0));

            Band band = ds.GetRasterBand(1);
            OSGeo.GDAL.Band band = ds.GetRasterBand(1);
            double[] mm = new double[2];
            band.ComputeRasterMinMax(mm, 0);
            layer.extension.SetHeight(mm[0], mm[1]);
@@ -71,9 +83,131 @@
        /// </summary>
        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;
                }

                string png = Path.Combine(terrainPath, sizes[0] + "_" + sizes[1] + ".png");
                Terrain2Png(layer, tif, png, sizes[0], sizes[1]);
            }
        }

        /// <summary>
        /// 重采样
        /// </summary>
        private static void Resample(string source, string dest, int width, int height)
        {
            // https://blog.51cto.com/u_16099346/6691820
            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);
        }

        /// <summary>
        /// 执行命令
        /// </summary>
        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;
        }

        /// <summary>
        /// Tif值转存至Png
        /// </summary>
        private static void Terrain2Png(Domain.Layer layer, string tif, string png, int width, int height)
        {
            Dataset ds = null;
            try
            {
                ds = Gdal.Open(tif, Access.GA_ReadOnly);
                if (null == ds || 0 == ds.RasterCount) return;

                OSGeo.GDAL.Band band = ds.GetRasterBand(1);
                float[] buffer = new float[width * height];
                band.ReadRaster(0, 0, width, height, buffer, width, height, 0, 0);

                Bitmap image = new Bitmap(width, height);
                //Graphics graphic = Graphics.FromImage(image);
                //graphic.Clear(Color.Transparent); // 填充透明色

                double perHeight = (layer.extension.maxHeight - layer.extension.minHeight) * 100 / 65536;

                for (int x = 0; x < width; x++)
                {
                    for (int y = 0; y < height; y++)
                    {
                        if (buffer[x * y] == double.NaN || buffer[x * y] == -9999)
                        {
                            image.SetPixel(x, y, Color.Transparent);
                            continue;
                        }

                        //int val = Convert.ToInt32(buffer[x * y]* 100); //Convert.ToInt32((buffer[x * y] - layer.extension.minHeight) * 100);
                        int val = Convert.ToInt32((buffer[x * y] - layer.extension.minHeight) * 100 / perHeight);
                        int r = val / 65536;
                        int g = (val - r * 65536) / 256;
                        int b = val % 256;

                        Color color = Color.FromArgb(0, r, g, b);
                        image.SetPixel(x, y, color);
                    }
                }

                image.Save(png, System.Drawing.Imaging.ImageFormat.Png);
            }
            finally
            {
                if (null != ds) ds.Dispose();
            }
        }

@@ -108,5 +242,30 @@
                sw.Write(json);
            }
        }

        #region 暂时不用
        /// <summary>
        /// 重采样 *
        /// </summary>
        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
    }
}