Civil 3d 纵断面标签添加（ai写的代码）

qq群友讨论如何添在纵断面图中添加纵断面标签。

经过搜索，

在au英文社区中找到了这个帖子，

Jeff_M给出了样例代码，
但不是很完整，

于是用ai写了一段相对完成的代码，

分享到这里！

using System;
using System.Collections.Generic;
using Autodesk.AutoCAD.ApplicationServices;
using Autodesk.AutoCAD.DatabaseServices;
using Autodesk.AutoCAD.EditorInput;
using Autodesk.AutoCAD.Runtime;
using Autodesk.Civil;
using Autodesk.Civil.ApplicationServices;
using Autodesk.Civil.DatabaseServices;
using Autodesk.Civil.DatabaseServices.Styles;
using AcadApp = Autodesk.AutoCAD.ApplicationServices.Application;
namespace Widgets
{
    /// <summary>
    /// 纵断面图标签添加（简化版，无反射）。
    /// 参照 Autodesk 论坛 ProfileLabelSetStyle 方案：
    /// https://forums.autodesk.com/t5/civil-3d-customization-forum/add-profile-label-set-to-profile-in-3d-civil-using-vb-net-api/m-p/9408920
    /// </summary>
    public static class ProfileViewLabelsSimple
    {
        /// <summary>命令入口。</summary>
        public static void AddLabels()
        {
            AddLabelsInteractive();
        }
        /// <summary>
        /// 拾取纵断面图，按 ProfileLabelSetStyle 为各纵断面创建标签组。
        /// </summary>
        public static void AddLabelsInteractive()
        {
            Document doc = AcadApp.DocumentManager.MdiActiveDocument;
            Editor ed = doc.Editor;
            if (!TrySelectProfileView(ed, out ObjectId profileViewId))
                return;
            PromptKeywordOptions pko = new PromptKeywordOptions(
                "\n操作 [全部(A)/替换(R)/列表(E)]: ", "All Replace List");
            pko.Keywords.Default = "All";
            PromptResult pkr = ed.GetKeywords(pko);
            if (pkr.Status != PromptStatus.OK)
                return;
            using (Transaction tr = doc.TransactionManager.StartTransaction())
            {
                ProfileView profileView = tr.GetObject(profileViewId, OpenMode.ForRead) as ProfileView;
                if (profileView == null)
                {
                    ed.WriteMessage("\n无法读取纵断面图。");
                    return;
                }
                if (pkr.StringResult == "List")
                {
                    ListExistingLabels(ed, tr, profileView);
                    tr.Commit();
                    return;
                }
                bool replaceExisting = pkr.StringResult == "Replace";
                ObjectId labelSetStyleId = ResolveLabelSetStyleId(ed);
                if (labelSetStyleId.IsNull)
                {
                    ed.WriteMessage("\n未找到可用的 ProfileLabelSetStyle。");
                    return;
                }
                ProfileLabelSetStyle labelSetStyle =
                    tr.GetObject(labelSetStyleId, OpenMode.ForRead) as ProfileLabelSetStyle;
                if (labelSetStyle == null)
                {
                    ed.WriteMessage("\n无法读取 ProfileLabelSetStyle。");
                    return;
                }
                ed.WriteMessage($"\n使用标签集样式: {labelSetStyle.Name}，共 {labelSetStyle.Count} 项。");
                int created = ApplyLabelSetToProfileView(
                    tr, profileViewId, profileView, labelSetStyle, replaceExisting, ed);
                tr.Commit();
                ed.WriteMessage($"\n完成，新建标签组 {created} 个。");
            }
        }
        /// <summary>
        /// 遍历 GraphOverrides，为每个 ProfileOverride 应用标签集。
        /// </summary>
        public static int ApplyLabelSetToProfileView(
            Transaction tr,
            ObjectId profileViewId,
            ProfileView profileView,
            ProfileLabelSetStyle labelSetStyle,
            bool replaceExisting,
            Editor ed = null)
        {
            if (profileView == null || labelSetStyle == null)
                return 0;
            ProfileOverrideCollection overrides = profileView.GraphOverrides;
            if (overrides == null || overrides.Count == 0)
            {
                ed?.WriteMessage("\n纵断面图中没有纵断面（GraphOverrides 为空）。");
                return 0;
            }
            ProfileViewStyle profileViewStyle = null;
            if (!profileView.StyleId.IsNull)
            {
                profileViewStyle = tr.GetObject(profileView.StyleId, OpenMode.ForRead) as ProfileViewStyle;
            }
            int created = 0;
            foreach (ProfileOverride profileOverride in overrides)
            {
                if (profileOverride == null || profileOverride.ProfileId.IsNull)
                    continue;
                ObjectId profileId = profileOverride.ProfileId;
                Profile profile = tr.GetObject(profileId, OpenMode.ForRead) as Profile;
                string profileName = profile?.Name ?? profileOverride.ProfileName ?? profileId.ToString();
                ed?.WriteMessage($"\n纵断面 \"{profileName}\"：");
                if (replaceExisting)
                {
                    EraseExistingLabelGroups(tr, profileViewId, profileId, ed);
                }
                int count = ApplyLabelSetToProfile(
                    tr, profileViewId, profileId, profileViewStyle, labelSetStyle, ed);
                created += count;
                ed?.WriteMessage($"  新建 {count} 个标签组。");
            }
            return created;
        }
        /// <summary>
        /// 按标签集定义，为指定纵断面 + 纵断面图创建标签组。
        /// </summary>
        public static int ApplyLabelSetToProfile(
            Transaction tr,
            ObjectId profileViewId,
            ObjectId profileId,
            ProfileViewStyle profileViewStyle,
            ProfileLabelSetStyle labelSetStyle,
            Editor ed = null)
        {
            if (labelSetStyle == null || labelSetStyle.Count == 0)
                return 0;
            int created = 0;
            // 逆序遍历，与论坛示例一致，尽量保持 UI 中标签组顺序
            for (int i = labelSetStyle.Count - 1; i >= 0; i--)
            {
                ProfileLabelSetItem item = labelSetStyle[i];
                if (item == null)
                    continue;
                ObjectId labelGroupId = CreateLabelGroupFromSetItem(
                    tr, profileViewId, profileId, profileViewStyle, item);
                if (labelGroupId.IsNull)
                {
                    ed?.WriteMessage($"\n  跳过 {item.LabelStyleType}（当前版本不支持或创建失败）。");
                    continue;
                }
                ApplyLabelGroupOptions(tr, labelGroupId, item);
                created++;
                ed?.WriteMessage($"\n  已添加 {item.LabelStyleType}。");
            }
            return created;
        }
        public static void ListExistingLabels(Editor ed, Transaction tr, ProfileView profileView)
        {
            ProfileOverrideCollection overrides = profileView.GraphOverrides;
            ed.WriteMessage($"\n纵断面图 \"{profileView.Name}\"，GraphOverrides 数量: {overrides?.Count ?? 0}");
            if (overrides == null)
                return;
            foreach (ProfileOverride profileOverride in overrides)
            {
                if (profileOverride == null)
                    continue;
                ObjectIdCollection labelGroupIds = profileOverride.GetProfileLabelGroupIds();
                ed.WriteMessage($"\n  纵断面: {profileOverride.ProfileName}  标签组: {labelGroupIds?.Count ?? 0}");
                if (labelGroupIds == null)
                    continue;
                foreach (ObjectId lgId in labelGroupIds)
                {
                    if (lgId.IsNull || lgId.IsErased)
                        continue;
                    LabelGroup lg = tr.GetObject(lgId, OpenMode.ForRead) as LabelGroup;
                    if (lg == null)
                        continue;
                    ed.WriteMessage(
                        $"\n    - {lg.GetType().Name}  样式: {lg.StyleName}  子标签数: {lg.SubEntityCount}");
                }
            }
        }
        private static void EraseExistingLabelGroups(
            Transaction tr,
            ObjectId profileViewId,
            ObjectId profileId,
            Editor ed)
        {
            RXClass labelGroupClass = ProfileLabelGroup.GetClass(typeof(ProfileLabelGroup));
            ObjectIdCollection existingIds = ProfileLabelGroup.GetAvailableLabelGroupIds(
                labelGroupClass, profileViewId, profileId, true);
            if (existingIds == null || existingIds.Count == 0)
                return;
            foreach (ObjectId id in existingIds)
            {
                if (id.IsNull || id.IsErased)
                    continue;
                ProfileLabelGroup grp = tr.GetObject(id, OpenMode.ForWrite) as ProfileLabelGroup;
                grp?.Erase();
            }
            ed?.WriteMessage($"\n  已清除已有标签组 {existingIds.Count} 个。");
        }
        private static ObjectId CreateLabelGroupFromSetItem(
            Transaction tr,
            ObjectId profileViewId,
            ObjectId profileId,
            ProfileViewStyle profileViewStyle,
            ProfileLabelSetItem item)
        {
            ObjectId styleId = item.LabelStyleId;
            if (styleId.IsNull)
                return ObjectId.Null;
            try
            {
                switch (item.LabelStyleType)
                {
                    case LabelStyleType.ProfileCrestCurve:
                        return ProfileCrestCurveLabelGroup.Create(profileViewId, profileId, styleId);
                    case LabelStyleType.ProfileSagCurve:
                        return ProfileSagCurveLabelGroup.Create(profileViewId, profileId, styleId);
                    // 论坛示例：ProfileGradeBreaks 使用 ProfilePVILabelGroup.Create
                    case LabelStyleType.ProfileGradeBreaks:
                        return ProfilePVILabelGroup.Create(profileViewId, profileId, styleId);
                    case LabelStyleType.ProfileHorizontalGeometryPoint:
                    {
                        ObjectId labelGroupId = ProfileHorizontalGeometryPointLabelGroup.Create(
                            profileViewId, profileId, styleId);
                        if (labelGroupId.IsNull)
                            return ObjectId.Null;
                        ApplyHorizontalGeometryPointOptions(tr, labelGroupId, item);
                        return labelGroupId;
                    }
                    case LabelStyleType.ProfileLine:
                        return ProfileLineLabelGroup.Create(profileViewId, profileId, styleId);
                    case LabelStyleType.ProfileMajorStation:
                        if (profileViewStyle == null)
                            return ObjectId.Null;
                        return ProfileStationLabelGroup.CreateMajor(
                            profileViewId,
                            profileId,
                            styleId,
                            profileViewStyle.BottomAxis.MajorTickStyle.Interval);
                    case LabelStyleType.ProfileMinorStation:
                        if (profileViewStyle == null)
                            return ObjectId.Null;
                        return ProfileMinorStationLabelGroup.Create(
                            profileViewId,
                            styleId,
                            profileViewStyle.BottomAxis.MinorTickStyle.Interval);
                    default:
                        return ObjectId.Null;
                }
            }
            catch
            {
                return ObjectId.Null;
            }
        }
        private static void ApplyHorizontalGeometryPointOptions(
            Transaction tr,
            ObjectId labelGroupId,
            ProfileLabelSetItem item)
        {
            ProfileHorizontalGeometryPointLabelGroup group =
                tr.GetObject(labelGroupId, OpenMode.ForWrite) as ProfileHorizontalGeometryPointLabelGroup;
            if (group == null)
                return;
            var geometryPoints = group.GetGeometryPointsOptions();
            Dictionary<AlignmentPointType, bool> labeledPoints = item.GetLabeledAlignmentGeometryPoints();
            if (geometryPoints == null || labeledPoints == null)
                return;
            geometryPoints.UnSelectAll();
            foreach (KeyValuePair<AlignmentPointType, bool> pair in labeledPoints)
            {
                if (!pair.Value)
                    continue;
                geometryPoints[pair.Key].Selected = true;
            }
            group.SetGeometryPointsOptions(geometryPoints);
        }
        private static void ApplyLabelGroupOptions(
            Transaction tr,
            ObjectId labelGroupId,
            ProfileLabelSetItem item)
        {
            try
            {
                ProfileLabelGroup group = tr.GetObject(labelGroupId, OpenMode.ForWrite) as ProfileLabelGroup;
                if (group == null)
                    return;
                group.DefaultDimensionAnchorOption =
                    (DimensionAnchorOptionType)item.DimensionAnchorOption;
                group.DefaultDimensionAnchorValue = item.DimensionAnchorValue;
            }
            catch
            {
                // 部分标签类型不支持锚点选项
            }
        }
        private static ObjectId ResolveLabelSetStyleId(Editor ed)
        {
            PromptStringOptions pso = new PromptStringOptions("\nProfileLabelSetStyle 名称 [Standard]: ");
            pso.AllowSpaces = true;
            pso.UseDefaultValue = true;
            pso.DefaultValue = "Standard";
            PromptResult psr = ed.GetString(pso);
            if (psr.Status != PromptStatus.OK)
                return ObjectId.Null;
            string inputName = string.IsNullOrWhiteSpace(psr.StringResult) ? "Standard" : psr.StringResult.Trim();
            CivilDocument civilDoc = CivilApplication.ActiveDocument;
            if (civilDoc == null)
                return ObjectId.Null;
            ProfileLabelSetStyleCollection styles = civilDoc.Styles.LabelSetStyles.ProfileLabelSetStyles;
            if (styles == null)
                return ObjectId.Null;
            string[] styleNames = { inputName, "Standard", "标准" };
            foreach (string name in styleNames)
            {
                try
                {
                    ObjectId id = styles[name];
                    if (!id.IsNull)
                        return id;
                }
                catch
                {
                    // 尝试下一个名称
                }
            }
            return ObjectId.Null;
        }
        private static bool TrySelectProfileView(Editor ed, out ObjectId profileViewId)
        {
            profileViewId = ObjectId.Null;
            PromptEntityOptions peo = new PromptEntityOptions("\n选择纵断面图: ");
            peo.SetRejectMessage("\n请选择 ProfileView");
            peo.AddAllowedClass(typeof(ProfileView), true);
            PromptEntityResult per = ed.GetEntity(peo);
            if (per.Status != PromptStatus.OK)
                return false;
            profileViewId = per.ObjectId;
            return true;
        }
    }
}