SzmediTools/Szmedi.CADkits/TerrainTools.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.RegularExpressions;
using System.Windows.Forms;
using System.Windows.Forms.Integration;

using Autodesk.AutoCAD.ApplicationServices;
using Autodesk.AutoCAD.Colors;
using Autodesk.AutoCAD.DatabaseServices;
using Autodesk.AutoCAD.EditorInput;
using Autodesk.AutoCAD.Geometry;
using Autodesk.AutoCAD.Runtime;
using Autodesk.AutoCAD.Windows;

using Szmedi.CADkits;

using Application = Autodesk.AutoCAD.ApplicationServices.Application;

public class TerrainTools
{
    private static PaletteSet ps;
    public const string CheckLineLayer = "校对线";

    [CommandMethod("MoveTerrainElevation")]
    public void MoveTerrainElevation()
    {
        Document doc = Application.DocumentManager.MdiActiveDocument;
        if (doc == null) return;
        Database db = doc.Database;
        Editor ed = doc.Editor;

        try
        {
            // --- Phase 1: User Input ---
            #region User Input Section

            string targetBlockName = null; // 仅当选中块时使用
            string terrainLayerName;

            // 定义处理模式: 0-块, 1-点, 2-多段线
            int processingMode = 0;
            string textLayerName;
            double maxDistance = 5.0;

            // 1. 选择样例地形点（块 或 点）
            using (Transaction tr = db.TransactionManager.StartTransaction())
            {
                PromptEntityOptions peoEnt = new PromptEntityOptions("\n请选择一个样例地形点 (块/点/多段线（起终点重合）): ");
                peoEnt.SetRejectMessage("\n选择的对象无效。");
                peoEnt.AddAllowedClass(typeof(BlockReference), true);
                peoEnt.AddAllowedClass(typeof(DBPoint), true);
                peoEnt.AddAllowedClass(typeof(Polyline), true);

                PromptEntityResult perEnt = ed.GetEntity(peoEnt);
                if (perEnt.Status != PromptStatus.OK) { ed.WriteMessage("\n操作已取消。"); return; }

                Entity sampleEnt = (Entity)tr.GetObject(perEnt.ObjectId, OpenMode.ForRead);
                terrainLayerName = sampleEnt.Layer;

                if (sampleEnt is BlockReference sampleBlock)
                {
                    processingMode = 0;
                    targetBlockName = sampleBlock.IsDynamicBlock ? ((BlockTableRecord)tr.GetObject(sampleBlock.DynamicBlockTableRecord, OpenMode.ForRead)).Name : sampleBlock.Name;
                    ed.WriteMessage($"\n已选择模式: [处理块]。块名: '{targetBlockName}', 图层: '{terrainLayerName}'");
                }
                else if (sampleEnt is DBPoint)
                {
                    processingMode = 1;
                    ed.WriteMessage($"\n已选择模式: [处理点实体]。所在图层: '{terrainLayerName}'");
                }
                else if (sampleEnt is Polyline)
                {
                    processingMode = 2;
                    ed.WriteMessage("\n模式: [多段线]");
                }
                tr.Commit();
            }

            // 2. 选择样例文字
            using (Transaction tr = db.TransactionManager.StartTransaction())
            {
                PromptEntityOptions peoText = new PromptEntityOptions("\n请选择一个样例标高文字 (用于确定图层): ");
                peoText.SetRejectMessage("\n选择的不是 MText 或 DBText。");
                peoText.AddAllowedClass(typeof(MText), true);
                peoText.AddAllowedClass(typeof(DBText), true);
                PromptEntityResult perText = ed.GetEntity(peoText);
                if (perText.Status != PromptStatus.OK) { ed.WriteMessage("\n操作已取消。"); return; }
                Entity sampleTextEnt = (Entity)tr.GetObject(perText.ObjectId, OpenMode.ForRead);
                textLayerName = sampleTextEnt.Layer;
                ed.WriteMessage($"\n已选择样例文字，所在图层: '{textLayerName}'");
                tr.Commit();
            }

            // 3. 输入距离
            PromptDoubleOptions pdo = new PromptDoubleOptions($"\n请输入查找标高文字的最大距离(图形单位) <{maxDistance}>: ");
            pdo.AllowNegative = false; pdo.AllowZero = false; pdo.AllowNone = true; pdo.DefaultValue = maxDistance;
            PromptDoubleResult pdr = ed.GetDouble(pdo);
            if (pdr.Status != PromptStatus.OK && pdr.Status != PromptStatus.None) { ed.WriteMessage("\n操作已取消。"); return; }
            if (pdr.Status == PromptStatus.OK) maxDistance = pdr.Value;

            #endregion

            // --- Phase 2: Automatic Matching ---
            List<TerrainEntityInfo> terrainEntities = new List<TerrainEntityInfo>();
            int updatedCount = 0;

            using (Transaction tr = db.TransactionManager.StartTransaction())
            {
                BlockTableRecord btr = (BlockTableRecord)tr.GetObject(db.CurrentSpaceId, OpenMode.ForWrite);

                // 根据模式采集
                if (processingMode == 0)
                    terrainEntities = CollectBlocks(tr, btr, targetBlockName, terrainLayerName);
                else if (processingMode == 1)
                    terrainEntities = CollectPoints(tr, btr, terrainLayerName);
                else if (processingMode == 2)
                    terrainEntities = CollectPolylines(tr, btr, terrainLayerName);
                var elevationTexts = CollectFilteredTexts(tr, btr, textLayerName);

                if (elevationTexts.Count == 0) { ed.WriteMessage($"\n警告: 在图层 '{textLayerName}' 上未找到任何有效的标高文字。"); tr.Commit(); return; }

                ed.WriteMessage($"\n正在使用贪婪匹配算法分析 {terrainEntities.Count} 个地形点...");

                // 核心匹配算法不需要修改，因为它只依赖 ObjectId 和 Position，这在 TerrainEntityInfo 中都有
                var validPairs = GetScoredValidPairs(terrainEntities, elevationTexts, maxDistance);

                ed.WriteMessage($"\n分析完成。找到 {validPairs.Count} 个最优匹配对。");

                if (validPairs.Count > 0)
                {
                    CreateLayerIfNotExists(db, tr, CheckLineLayer, 8);
                    ProgressMeter pm = new ProgressMeter();
                    pm.SetLimit(validPairs.Count);
                    pm.Start("正在更新已自动匹配的实体...");

                    var textDataMap = elevationTexts.ToDictionary(t => t.ObjectId, t => t);

                    foreach (var pair in validPairs)
                    {
                        if (double.TryParse(textDataMap[pair.Value].TextContent, out double elevationZ))
                        {
                            // 更新位置逻辑：需要判断是块还是点
                            Entity ent = (Entity)tr.GetObject(pair.Key, OpenMode.ForWrite);
                            Point3d currentPos;

                            if (ent is BlockReference blk)
                            {
                                blk.Position = new Point3d(blk.Position.X, blk.Position.Y, elevationZ);
                                currentPos = blk.Position;
                            }
                            else if (ent is DBPoint pt)
                            {
                                pt.Position = new Point3d(pt.Position.X, pt.Position.Y, elevationZ);
                                currentPos = pt.Position;
                            }
                            else if (ent is Polyline pl)
                            {
                                // 对于 2D 多段线，修改 Elevation 属性
                                pl.Elevation = elevationZ;
                                Extents3d ext = pl.GeometricExtents;
                                currentPos = ext.MinPoint + (ext.MaxPoint - ext.MinPoint) * 0.5;
                                //currentPos = pl.GetPoint3dAt(0); // 取第一个点作为连线起点
                            }
                            else
                            {
                                continue;
                            }

                            Line line = new Line(currentPos, textDataMap[pair.Value].Position)
                            {
                                Layer = CheckLineLayer,
                                ColorIndex = 256
                            };
                            btr.AppendEntity(line);
                            tr.AddNewlyCreatedDBObject(line, true);
                            updatedCount++;
                        }
                    }
                    pm.Stop();
                }
                tr.Commit();
                ed.WriteMessage($"\n\n--- 自动处理报告 ---\n成功自动更新 {updatedCount} 个实体。\n{terrainEntities.Count - updatedCount} 个实体需要审核。");

                if (terrainEntities.Count > 0)
                {
                    if (ps == null) { ps = new PaletteSet("审核并校对高程点", new Guid("C4E61D33-3162-4889-A263-6A1D2C52D26C")); }
                    if (ps.Count > 0) { ps.Remove(0); }

                    // UI 控件初始化
                    var reviewControl = new ReviewAndMatchWindow(terrainEntities, validPairs, elevationTexts, CheckLineLayer, ps);
                    ElementHost host = new ElementHost { Dock = DockStyle.Fill, Child = reviewControl };
                    ps.Add("ReviewAndMatch", host);
                    ps.Visible = true;
                }
            }

           
        }
        catch (System.Exception ex) { ed.WriteMessage($"\n发生严重错误: {ex.Message}\n{ex.StackTrace}"); }
    }

    // 匹配算法逻辑保持不变，只需签名中的 List 类型改为 TerrainEntityInfo
    private Dictionary<ObjectId, ObjectId> GetScoredValidPairs(List<TerrainEntityInfo> entities, List<TextWithPosition> texts, double maxDist)
    {
        var textCompetition = new Dictionary<ObjectId, double>();
        // ... (原逻辑不变，ent.Position 属性是一样的) ...
        foreach (var text in texts)
        {
            var distancesToEnts = entities
                .Select(b => new Point2d(b.Position.X, b.Position.Y).GetDistanceTo(new Point2d(text.Position.X, text.Position.Y)))
                .OrderBy(d => d)
                .ToList();

            if (distancesToEnts.Count >= 2)
            {
                double d1 = distancesToEnts[0];
                double d2 = distancesToEnts[1];
                textCompetition[text.ObjectId] = (d2 > 0.001) ? (d1 / d2) : 1.0;
            }
            else
            {
                textCompetition[text.ObjectId] = 0.0;
            }
        }

        var allPossiblePairs = new List<Tuple<ObjectId, ObjectId, double>>();
        foreach (var ent in entities)
        {
            foreach (var text in texts)
            {
                double dist = new Point2d(ent.Position.X, ent.Position.Y)
                                .GetDistanceTo(new Point2d(text.Position.X, text.Position.Y));

                if (dist <= maxDist)
                {
                    double distanceScore = 1.0 / (dist + 0.001);
                    double uniquenessScore = 1.0 - textCompetition[text.ObjectId];
                    double finalScore = 0.7 * distanceScore + 0.3 * uniquenessScore * distanceScore;
                    allPossiblePairs.Add(new Tuple<ObjectId, ObjectId, double>(ent.ObjectId, text.ObjectId, finalScore));
                }
            }
        }

        var sortedPairs = allPossiblePairs.OrderByDescending(p => p.Item3).ToList();
        var validPairs = new Dictionary<ObjectId, ObjectId>();
        var matchedEnts = new HashSet<ObjectId>();
        var matchedTexts = new HashSet<ObjectId>();

        foreach (var pair in sortedPairs)
        {
            if (!matchedEnts.Contains(pair.Item1) && !matchedTexts.Contains(pair.Item2))
            {
                validPairs.Add(pair.Item1, pair.Item2);
                matchedEnts.Add(pair.Item1);
                matchedTexts.Add(pair.Item2);
            }
        }
        return validPairs;
    }

    /// <summary>
    /// 采集多段线方法 ---
    /// </summary>
    /// <param name="tr"></param>
    /// <param name="btr"></param>
    /// <param name="layerName"></param>
    /// <returns></returns>
    private List<TerrainEntityInfo> CollectPolylines(Transaction tr, BlockTableRecord btr, string layerName)
    {
        var list = new List<TerrainEntityInfo>();
        foreach (ObjectId id in btr)
        {
            if (id.ObjectClass.IsDerivedFrom(RXObject.GetClass(typeof(Polyline))))
            {
                var pl = (Polyline)tr.GetObject(id, OpenMode.ForRead);
                if (pl.Layer.Equals(layerName, StringComparison.OrdinalIgnoreCase))
                {
                    try
                    {
                        // 【核心修改】：对于圆弧/圆形多段线，取其几何中心作为参考点
                        // 这样即使多段线是由两个半圆组成的，中心点也会准确落在圆心
                        Extents3d ext = pl.GeometricExtents;
                        Point3d center = ext.MinPoint + (ext.MaxPoint - ext.MinPoint) * 0.5;

                        // 将计算出的“圆心”存入 TerrainEntityInfo
                        list.Add(new TerrainEntityInfo(id, center));
                    }
                    catch
                    {
                        // 如果某些特殊情况无法获取包围盒，则退而求其次取第一个点
                        list.Add(new TerrainEntityInfo(id, pl.GetPoint3dAt(0)));
                    }
                }
            }
        }
        return list;
    }
    /// 采集点
    /// </summary>
    /// <param name="tr"></param>
    /// <param name="btr"></param>
    /// <param name="layerName"></param>
    /// <returns></returns>
    private List<TerrainEntityInfo> CollectPoints(Transaction tr, BlockTableRecord btr, string layerName)
    {
        var list = new List<TerrainEntityInfo>();
        foreach (ObjectId id in btr)
        {
            if (id.ObjectClass.IsDerivedFrom(RXObject.GetClass(typeof(DBPoint))))
            {
                var pt = (DBPoint)tr.GetObject(id, OpenMode.ForRead);
                if (pt.Layer.Equals(layerName, StringComparison.OrdinalIgnoreCase))
                {
                    list.Add(new TerrainEntityInfo(id, pt.Position));
                }
            }
        }
        return list;
    }

    // 修改：采集块实体 (返回类型变更为 TerrainEntityInfo)
    private List<TerrainEntityInfo> CollectBlocks(Transaction tr, BlockTableRecord btr, string targetBlockName, string blockLayerName)
    {
        var list = new List<TerrainEntityInfo>();
        foreach (ObjectId id in btr)
            if (id.ObjectClass.IsDerivedFrom(RXObject.GetClass(typeof(BlockReference))))
            {
                var br = (BlockReference)tr.GetObject(id, OpenMode.ForRead);
                string name = br.IsDynamicBlock ? ((BlockTableRecord)tr.GetObject(br.DynamicBlockTableRecord, OpenMode.ForRead)).Name : br.Name;
                if (br.Layer.Equals(blockLayerName, StringComparison.OrdinalIgnoreCase) && name.Equals(targetBlockName, StringComparison.OrdinalIgnoreCase))
                    list.Add(new TerrainEntityInfo(id, br.Position));
            }
        return list;
    }

    private List<TextWithPosition> CollectFilteredTexts(Transaction tr, BlockTableRecord btr, string textLayerName) { var list = new List<TextWithPosition>(); Regex rgx = new Regex(@"[\u4e00-\u9fa5]"); foreach (ObjectId id in btr) { if (id.ObjectClass.IsDerivedFrom(RXObject.GetClass(typeof(DBText))) || id.ObjectClass.IsDerivedFrom(RXObject.GetClass(typeof(MText)))) { var ent = (Entity)tr.GetObject(id, OpenMode.ForRead); if (ent.Layer.Equals(textLayerName, StringComparison.OrdinalIgnoreCase)) { string txt = (ent is MText mt) ? mt.Contents : ((DBText)ent).TextString; if (!string.IsNullOrWhiteSpace(txt) && !rgx.IsMatch(txt) && txt.Contains(".") && double.TryParse(txt, out _)) { try { Extents3d ext = ent.GeometricExtents; Point3d center = ext.MinPoint + (ext.MaxPoint - ext.MinPoint) * 0.5; list.Add(new TextWithPosition(id, center, txt)); } catch { } } } } } return list; }
    private void CreateLayerIfNotExists(Database db, Transaction tr, string layerName, short colorIndex) { LayerTable lt = (LayerTable)tr.GetObject(db.LayerTableId, OpenMode.ForRead); if (!lt.Has(layerName)) { lt.UpgradeOpen(); var ltr = new LayerTableRecord { Name = layerName, Color = Color.FromColorIndex(ColorMethod.ByAci, colorIndex) }; lt.Add(ltr); tr.AddNewlyCreatedDBObject(ltr, true); } }

    // 清理功能保持不变
    public static void CleanAllCheckLines()
    {
        // ... (与原代码一致，此处略) ...
        // 请确保保留原 CleanAllCheckLines 实现
        Document doc = Application.DocumentManager.MdiActiveDocument;
        if (doc == null) return;
        Database db = doc.Database;
        Editor ed = doc.Editor;
        int cleanedCount = 0;
        using (doc.LockDocument())
        using (Transaction tr = db.TransactionManager.StartTransaction())
        {
            try
            {
                LayerTable lt = (LayerTable)tr.GetObject(db.LayerTableId, OpenMode.ForRead);
                if (!lt.Has(CheckLineLayer)) return;
                LayerTableRecord layer = (LayerTableRecord)tr.GetObject(lt[CheckLineLayer], OpenMode.ForRead);
                if (layer.IsLocked) { layer.UpgradeOpen(); layer.IsLocked = false; }
                BlockTableRecord btr = (BlockTableRecord)tr.GetObject(db.CurrentSpaceId, OpenMode.ForRead);
                foreach (ObjectId id in btr)
                {
                    if (id.ObjectClass.DxfName.Equals("LINE", StringComparison.OrdinalIgnoreCase))
                    {
                        Line line = tr.GetObject(id, OpenMode.ForRead) as Line;
                        if (line != null && line.Layer.Equals(CheckLineLayer, StringComparison.OrdinalIgnoreCase))
                        {
                            line.UpgradeOpen(); line.Erase(); cleanedCount++;
                        }
                    }
                }
                tr.Commit(); ed.WriteMessage($"\n已清理 {cleanedCount} 条校对线。"); ed.Regen();
            }
            catch { tr.Abort(); }
        }
    }
}