实战|用 AI 写 CAD 插件:设置标注(附源码 + 提示词)

点击蓝字，关注我们

using System;using System.Globalization;using System.Linq;using Autodesk.AutoCAD.ApplicationServices;using Autodesk.AutoCAD.DatabaseServices;using Autodesk.AutoCAD.EditorInput;using Autodesk.AutoCAD.Runtime;[assembly: CommandClass(typeof(KJDimPlugin.KJDimCommand))]namespaceKJDimPlugin{    publicclassKJDimCommand    {        privatestring _pendingStyleName;        // ═══════════════════════════════════════════════════════════        //  主命令        // ═══════════════════════════════════════════════════════════        [CommandMethod("KJDim")]        publicvoidExecuteKJDim()        {            var doc = Application.DocumentManager.MdiActiveDocument;            if (doc == null) return;            _pendingStyleName = null;            try            {                if (!IniConfigManager.ConfigExists())                {                    if (!RunStyleConfigFlow(doc, isFirstTime: true)) return;                    if (_pendingStyleName != null) return;                }                else                {                    var parameters = DimStyleService.LoadParamsFromIni();                    if (parameters.Count == 0)                    {                        doc.Editor.WriteMessage("\n配置文件损坏，重新初始化...\n");                        IniConfigManager.DeleteConfig();                        if (!RunStyleConfigFlow(doc, isFirstTime: true)) return;                        if (_pendingStyleName != null) return;                    }                    else                    {                        string std = DimStyleService.EnsureDimStyle(doc.Database, parameters);                        doc.Editor.WriteMessage($"\n标准样式 \"{std}\" 已就绪并置为当前。\n");                    }                }                RunDailyLoop(doc);            }            catch (System.Exception ex)            {                doc.Editor.WriteMessage($"\n运行错误：{ex.Message}\n");            }        }        // ═══════════════════════════════════════════════════════════        //  首次配置流程        // ═══════════════════════════════════════════════════════════        privateboolRunStyleConfigFlow(Document doc, bool isFirstTime)        {            var db = doc.Database;            var ed = doc.Editor;            var firstWin = new FirstTimeWindow(isFirstTime);            var r1 = Application.ShowModalWindow(firstWin);            if (r1 != true) returnfalse;            if (firstWin.Action == FirstTimeAction.CreateNew)            {                string newStyle = DoCreateNewStyle(doc);                if (string.IsNullOrEmpty(newStyle)) returnfalse;                returntrue;            }            elseif (firstWin.Action == FirstTimeAction.UseExisting)            {                string chosen = DoPickExistingStyle(doc);                if (string.IsNullOrEmpty(chosen)) returnfalse;                var parameters = DimStyleService.ReadDimStyleParams(db, chosen);                if (parameters == null) returnfalse;                DimStyleService.SaveParamsToIni(parameters);                DimStyleService.EnsureDimStyle(db, parameters);                ed.WriteMessage($"\n标准样式 \"{chosen}\" 已保存并置为当前。\n" +                                $"配置文件：{IniConfigManager.GetConfigPath()}\n");                returntrue;            }            returnfalse;        }        // ═══════════════════════════════════════════════════════════        //  新建样式        // ═══════════════════════════════════════════════════════════        privatestringDoCreateNewStyle(Document doc)        {            var db = doc.Database;            var ed = doc.Editor;            var nameWin = new NameInputWindow();            var nr = Application.ShowModalWindow(nameWin);            if (nr != true) returnnull;            string newName = nameWin.StyleName;            double dimScale = nameWin.DimScale;            if (DimStyleService.GetAllDimStyleNames(db).Contains(newName))            {                ed.WriteMessage($"\n样式 \"{newName}\" 已存在，请换一个名称。\n");                returnnull;            }            if (!CreateStyleFromDefaultTemplate(db, newName, dimScale))            {                ed.WriteMessage($"\n创建样式 \"{newName}\" 失败。\n");                returnnull;            }            _pendingStyleName = newName;            doc.CommandEnded -= OnDimStyleCommandEnded;            doc.CommandEnded += OnDimStyleCommandEnded;            ed.WriteMessage(                $"\n已创建样式 \"{newName}\"（比例 1:{dimScale:0.##}）并置为当前。\n" +                $"正在打开【标注样式管理器】...\n");            doc.SendStringToExecute("_.DIMSTYLE\n", true, false, false);            return newName;        }        privatevoidOnDimStyleCommandEnded(object sender, CommandEventArgs e)        {            if (!string.Equals(e.GlobalCommandName, "DIMSTYLE",                StringComparison.OrdinalIgnoreCase)) return;            var doc = sender as Document;            if (doc == null) return;            doc.CommandEnded -= OnDimStyleCommandEnded;            string styleName = _pendingStyleName;            _pendingStyleName = null;            if (string.IsNullOrEmpty(styleName)) return;            var db = doc.Database;            var ed = doc.Editor;            try            {                var parameters = DimStyleService.ReadDimStyleParams(db, styleName);                if (parameters == null) return;                DimStyleService.SaveParamsToIni(parameters);                DimStyleService.EnsureDimStyle(db, parameters);                ed.WriteMessage($"\n标准样式 \"{styleName}\" 已保存并置为当前。\n" +                                $"配置文件：{IniConfigManager.GetConfigPath()}\n");                RunDailyLoop(doc);            }            catch (System.Exception ex)            {                ed.WriteMessage($"\n保存参数失败：{ex.Message}\n");            }        }        // ═══════════════════════════════════════════════════════════        //  选取已有样式        // ═══════════════════════════════════════════════════════════        privatestringDoPickExistingStyle(Document doc)        {            var db = doc.Database;            var ed = doc.Editor;            var items = DimStyleService.GetAllDimStyleItems(db);            if (items.Count == 0)            {                ed.WriteMessage("\n当前图纸中没有标注样式。\n");                returnnull;            }            var pickWin = new StylePickWindow(items);            var pr = Application.ShowModalWindow(pickWin);            if (pr != true) returnnull;            string chosen = pickWin.SelectedStyleName;            using (var tr = db.TransactionManager.StartTransaction())            {                var table = (DimStyleTable)tr.GetObject(db.DimStyleTableId, OpenMode.ForRead);                if (table.Has(chosen))                {                    ObjectId sid = table[chosen];                    db.Dimstyle = sid;                    var rec = (DimStyleTableRecord)tr.GetObject(sid, OpenMode.ForRead);                    db.SetDimstyleData(rec);                }                tr.Commit();            }            return chosen;        }        // ═══════════════════════════════════════════════════════════        //  从系统模板创建        // ═══════════════════════════════════════════════════════════        privateboolCreateStyleFromDefaultTemplate(Database db, string newName, double dimScale)        {            using (var tr = db.TransactionManager.StartTransaction())            {                var table = (DimStyleTable)tr.GetObject(db.DimStyleTableId, OpenMode.ForRead);                if (table.Has(newName)) { tr.Commit(); returnfalse; }                table.UpgradeOpen();                var newRecord = new DimStyleTableRecord();                string baseName = (db.Measurement == MeasurementValue.English)                    ? "STANDARD" : "ISO-25";                if (table.Has(baseName))                {                    var baseRec = (DimStyleTableRecord)tr.GetObject(                        table[baseName], OpenMode.ForRead);                    newRecord.CopyFrom(baseRec);                }                newRecord.Name = newName;                newRecord.Dimscale = dimScale;                if (table.Has(newRecord.Name)) { tr.Commit(); returnfalse; }                ObjectId id = table.Add(newRecord);                tr.AddNewlyCreatedDBObject(newRecord, true);                db.Dimstyle = id;                db.SetDimstyleData(newRecord);                tr.Commit();            }            returntrue;        }        // ═══════════════════════════════════════════════════════════        //  日常循环        // ═══════════════════════════════════════════════════════════        privatevoidRunDailyLoop(Document doc)        {            while (true)            {                var db = doc.Database;                var ed = doc.Editor;                string std = IniConfigManager.ReadValue("Name", "KJ_STD");                double stdScale = 1.0;                double.TryParse(IniConfigManager.ReadValue("Dimscale", "1"),                    NumberStyles.Any, CultureInfo.InvariantCulture, out stdScale);                if (stdScale <= 0) stdScale = 1.0;                var allStyles = DimStyleService.GetAllDimStyleItems(db);                var win = new DailyUseWindow(std, stdScale, allStyles);                var res = Application.ShowModalWindow(win);                if (res != true) return;                switch (win.Action)                {                    case DailyAction.ReplaceAll:                        ReplaceAllDimensions(doc, win.SourceStyle, win.TargetStyle);                        return;                    case DailyAction.ReplaceSelection:                        ReplaceSelectionInteractive(doc, win.TargetStyle);                        return;                    case DailyAction.CreateNewStyle:                        DoCreateNewStyle(doc);                        return;                    case DailyAction.ChangeStandardStyle:                        ChangeStandardStyle(doc);                        break;                }            }        }        // ═══════════════════════════════════════════════════════════        //  更换标准样式（直接弹样式列表）        // ═══════════════════════════════════════════════════════════        privatevoidChangeStandardStyle(Document doc)        {            var db = doc.Database;            var ed = doc.Editor;            var items = DimStyleService.GetAllDimStyleItems(db);            if (items.Count == 0) return;            var pickWin = new StylePickWindow(items);            var pr = Application.ShowModalWindow(pickWin);            if (pr != true) return;            string chosen = pickWin.SelectedStyleName;            var p = DimStyleService.ReadDimStyleParams(db, chosen);            if (p == null) return;            DimStyleService.SaveParamsToIni(p);            DimStyleService.EnsureDimStyle(db, p);            ed.WriteMessage($"\n标准样式已更换为 \"{chosen}\"。\n");        }        // ═══════════════════════════════════════════════════════════        //  全图替换        // ═══════════════════════════════════════════════════════════        privatevoidReplaceAllDimensions(Document doc, string sourceStyle, string targetStyle)        {            var db = doc.Database;            var ed = doc.Editor;            int count = 0;            using (var tr = db.TransactionManager.StartTransaction())            {                var dimTable = (DimStyleTable)tr.GetObject(db.DimStyleTableId, OpenMode.ForRead);                if (!dimTable.Has(sourceStyle) || !dimTable.Has(targetStyle))                {                    ed.WriteMessage("\n源样式或目标样式在当前图纸中不存在。\n");                    tr.Commit();                    return;                }                ObjectId srcId = dimTable[sourceStyle];                ObjectId tgtId = dimTable[targetStyle];                var bt = (BlockTable)tr.GetObject(db.BlockTableId, OpenMode.ForRead);                foreach (ObjectId btrId in bt)                {                    var btr = (BlockTableRecord)tr.GetObject(btrId, OpenMode.ForRead);                    foreach (ObjectId entId in btr)                    {                        var ent = tr.GetObject(entId, OpenMode.ForRead);                        if (ent is Dimension dim && dim.DimensionStyle == srcId)                        {                            dim.UpgradeOpen();                            dim.DimensionStyle = tgtId;                            count++;                        }                    }                }                tr.Commit();            }            ed.WriteMessage($"\n已将 {count} 个标注从 \"{sourceStyle}\" 替换为 \"{targetStyle}\"。\n");        }        // ═══════════════════════════════════════════════════════════        //  交互式选择 + 替换        // ═══════════════════════════════════════════════════════════        privatevoidReplaceSelectionInteractive(Document doc, string targetStyle)        {            var db = doc.Database;            var ed = doc.Editor;            var opt = new PromptSelectionOptions            {                MessageForAdding = "\n请选择要替换的标注对象：",                AllowDuplicates = false            };            var res = ed.GetSelection(opt);            if (res.Status != PromptStatus.OK)            {                ed.WriteMessage("\n未选择任何对象，操作取消。\n");                return;            }            int count = 0;            using (var tr = db.TransactionManager.StartTransaction())            {                var dimTable = (DimStyleTable)tr.GetObject(db.DimStyleTableId, OpenMode.ForRead);                if (!dimTable.Has(targetStyle))                {                    ed.WriteMessage($"\n目标样式 \"{targetStyle}\" 在当前图纸中不存在。\n");                    tr.Commit();                    return;                }                ObjectId tgtId = dimTable[targetStyle];                foreach (SelectedObject so in res.Value)                {                    if (so == null) continue;                    var ent = tr.GetObject(so.ObjectId, OpenMode.ForRead);                    if (ent is Dimension dim)                    {                        dim.UpgradeOpen();                        dim.DimensionStyle = tgtId;                        count++;                    }                }                tr.Commit();            }            ed.WriteMessage($"\n已将选中的 {count} 个标注替换为 \"{targetStyle}\"。\n");        }    }}

说明：配置文件保存在插件同目录的KJDimStyle.ini，想更换标准样式删除配置文件重新拾取即可；代码默认包含文字高度、箭头大小、偏移、精度等核心参数，可根据出图标准自行扩展更多样式参数。