1

13693261870

2024-07-22 7fb323da219c84d0275e2e1138747899d8cdc40a

1

 SimuTools/Tools/Handle.cs

@@ -9,7 +9,6 @@
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace SimuTools.Tools
@@ -350,8 +349,8 @@
            List<string> vyFiles = GetFiles(flowPath, "vy");
            if (null == vxFiles || null == vyFiles || vxFiles.Count != vyFiles.Count || vxFiles.Count != layer.waters.data.Count) return;

            //Parallel.For(0, vxFiles.Count, i =>
            for (int i = 0; i < vxFiles.Count; i++)
            Parallel.For(0, vxFiles.Count, i =>
            //for (int i = 0; i < vxFiles.Count; i++)
            {
                Dataset vxDs = null, vyDs = null;
                try
@@ -368,7 +367,7 @@
                    if (null != vxDs) vxDs.Dispose();
                    if (null != vyDs) vyDs.Dispose();
                }
            }
            });
        }

        /// <summary>
@@ -414,33 +413,14 @@
                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);

                Bitmap image = new Bitmap(width, height);
                Graphics graphic = Graphics.FromImage(image);
                graphic.Clear(Color.Transparent); // 填充透明色
                //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];

                // 用 R通道表示，流向为归一化的二维向量（x,y），G通道表示为 x *255 , B通道表示为 y * 255
                for (int x = 0; x < width; x++)
                {
                    for (int y = 0; y < height; y++)
                    {
                        int offset = x + y * width;
                        float fx = GetFloatValue(vxBuffer[offset]);
                        float fy = GetFloatValue(vyBuffer[offset]);

                        double dv = GetXyValue(fx, fy);
                        double dx = float.IsNaN(fx) ? 0 : fx * 0.5 + 0.5;
                        double dy = float.IsNaN(fy) ? 0 : fy * 0.5 + 0.5;

                        int r = Convert.ToInt32(dv * 255);
                        int g = Convert.ToInt32(dx * 255);
                        int b = Convert.ToInt32(dy * 255);

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

                image.Save(png, System.Drawing.Imaging.ImageFormat.Png);
                CreateFlowPng(vxBuffer, vyBuffer, png, width, height);
            }
            finally
            {
@@ -452,6 +432,40 @@
        }

        /// <summary>
        /// 创建流速流向PNG
        /// </summary>
        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);
            graphic.Clear(Color.Transparent); // 填充透明色

            // 用 R通道表示，流向为归一化的二维向量（x,y），G通道表示为 x *255 , B通道表示为 y * 255
            for (int x = 0; x < width; x++)
            {
                for (int y = 0; y < height; y++)
                {
                    int offset = x + y * width;
                    float fx = GetFloatValue(vxBuffer[offset]);
                    float fy = GetFloatValue(vyBuffer[offset]);

                    double dv = GetXyValue(fx, fy);
                    double dx = float.IsNaN(fx) ? 0 : fx * 0.5 + 0.5;
                    double dy = float.IsNaN(fy) ? 0 : fy * 0.5 + 0.5;

                    int r = Convert.ToInt32(dv * 255);
                    if (r > 255) r = 255;
                    int g = Convert.ToInt32(dx * 255);
                    int b = Convert.ToInt32(dy * 255);

                    Color color = Color.FromArgb(127, r, g, b);
                    image.SetPixel(x, y, color);
                }
            }
            image.Save(png, System.Drawing.Imaging.ImageFormat.Png);
        }

        /// <summary>
        /// 获取Float值
        /// </summary>
        private static float GetFloatValue(float val)