337 lines
14 KiB
C#
337 lines
14 KiB
C#
|
|
|||
|
|
using System.ComponentModel;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using System.Windows;
|
|||
|
|
using System.Windows.Markup;
|
|||
|
|
|
|||
|
|
using Autodesk.Revit.Attributes;
|
|||
|
|
using Autodesk.Revit.DB;
|
|||
|
|
using Autodesk.Revit.DB.Structure;
|
|||
|
|
|
|||
|
|
using FuzzySharp;
|
|||
|
|
|
|||
|
|
using JiebaNet.Segmenter;
|
|||
|
|
|
|||
|
|
using Microsoft.Win32;
|
|||
|
|
|
|||
|
|
using Nice3point.Revit.Toolkit.External;
|
|||
|
|
|
|||
|
|
using Szmedi.RevitToolkit.Approval.Controls.ProgressWrapper;
|
|||
|
|
using Szmedi.RevitToolkit.Approval.Models;
|
|||
|
|
|
|||
|
|
namespace Szmedi.RevitToolkit.Approval.Commands
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 建筑标识预处理
|
|||
|
|
/// </summary>
|
|||
|
|
[Transaction(TransactionMode.Manual)]
|
|||
|
|
[Regeneration(RegenerationOption.Manual)]
|
|||
|
|
public class ProcessArchiSignCmd : ExternalCommand
|
|||
|
|
{
|
|||
|
|
private Major major;
|
|||
|
|
private string majorName;
|
|||
|
|
public override void Execute()
|
|||
|
|
{
|
|||
|
|
StringBuilder sb = new StringBuilder();
|
|||
|
|
const string signParam = "深圳构件标识";
|
|||
|
|
var dir = Path.Combine(GlobalAssists.DirAssembly, "mvdlite");
|
|||
|
|
major = Document.GetMajor(out majorName);
|
|||
|
|
var result = MessageBox.Show(
|
|||
|
|
$"""
|
|||
|
|
通过各专业专业独有的构件
|
|||
|
|
(建筑-门窗，结构-梁，暖通-风管，电气-桥架，给排水-无风管有水管)
|
|||
|
|
|
|||
|
|
当前专业为：{majorName}，
|
|||
|
|
是 - 继续，否 - 手动选择mvdlite文件
|
|||
|
|
""",
|
|||
|
|
"提示",
|
|||
|
|
MessageBoxButton.YesNoCancel,
|
|||
|
|
MessageBoxImage.Question);
|
|||
|
|
string fileContent;
|
|||
|
|
if (result == MessageBoxResult.Yes)
|
|||
|
|
{
|
|||
|
|
fileContent = Directory.GetFiles(dir, "*.mvdlite")
|
|||
|
|
.FirstOrDefault(f => f.Contains(majorName)) switch
|
|||
|
|
{
|
|||
|
|
string file when !string.IsNullOrEmpty(file) => File.ReadAllText(file),
|
|||
|
|
_ => throw new FileNotFoundException("未找到对应的MVDLite文件，请检查目录或文件名。")
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
else if (result == MessageBoxResult.No)
|
|||
|
|
{
|
|||
|
|
OpenFileDialog dialog = new OpenFileDialog()
|
|||
|
|
{
|
|||
|
|
Filter = "MVDLite 文件 (*.mvdlite)|*.mvdlite",
|
|||
|
|
Title = "选择 MVDLite 文件",
|
|||
|
|
InitialDirectory = dir,
|
|||
|
|
RestoreDirectory = true
|
|||
|
|
};
|
|||
|
|
if (dialog.ShowDialog() != true)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
fileContent = File.ReadAllText(dialog.FileName);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
Document.InvokeGroup(
|
|||
|
|
_ =>
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
if (!NeedRemoveParameter(signParam))
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var componentSignSet = UiApplication.Application.Create.NewCategorySet();
|
|||
|
|
var builtInCategories = Enum.GetValues(typeof(BuiltInCategory));
|
|||
|
|
foreach (BuiltInCategory cate in builtInCategories)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
var category = Category.GetCategory(Document, cate);
|
|||
|
|
if (category is { CategoryType: CategoryType.Model, AllowsBoundParameters: true })
|
|||
|
|
{
|
|||
|
|
if (cate == BuiltInCategory.OST_Rooms || cate == BuiltInCategory.OST_Areas)
|
|||
|
|
{
|
|||
|
|
continue; //空间和区域不需要标识
|
|||
|
|
}
|
|||
|
|
componentSignSet.Insert(category);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//有的类别不支持这个方法获取
|
|||
|
|
catch (Exception)
|
|||
|
|
{
|
|||
|
|
//sb.AppendLine($"{ex.Message}:{cate}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Document.AddSharedParameter(signParam, componentSignSet, BuiltInParameterGroup.PG_IFC);
|
|||
|
|
|
|||
|
|
var allElements = Document.OfParentModelCollector().ToElements();
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
//sb.AppendLine($"标识,优化标识,匹配,比率,");
|
|||
|
|
MvdData data = MvdLiteAssist.Parse(fileContent);
|
|||
|
|
var groups = allElements.GroupBy(e => e.GetTypeId());
|
|||
|
|
Document.Invoke(
|
|||
|
|
_ =>
|
|||
|
|
{
|
|||
|
|
foreach (var g in groups)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
string mapperName = GetMapperSign(g, data);
|
|||
|
|
|
|||
|
|
//if (elementGroup.Key == ElementId.InvalidElementId)//可能是地形
|
|||
|
|
//{
|
|||
|
|
// string name = string.Empty;
|
|||
|
|
// if (elementGroup.FirstOrDefault().Category != null)
|
|||
|
|
// {
|
|||
|
|
// name = elementGroup.FirstOrDefault().Category.name;
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// {
|
|||
|
|
// name = elementGroup.FirstOrDefault().name;
|
|||
|
|
// }
|
|||
|
|
// mapperName = GetMapperSign(name, data);
|
|||
|
|
//}
|
|||
|
|
//else
|
|||
|
|
//{
|
|||
|
|
// mapperName = GetMapperSign(type.name, data);
|
|||
|
|
|
|||
|
|
// if (string.IsNullOrEmpty(mapperName))//如果类型名找不到标识，使用族名称
|
|||
|
|
// {
|
|||
|
|
// mapperName = GetMapperSign(type.FamilyName, data);
|
|||
|
|
// }
|
|||
|
|
//}
|
|||
|
|
|
|||
|
|
foreach (var elem in g)
|
|||
|
|
{
|
|||
|
|
var param = elem.LookupParameter(signParam);
|
|||
|
|
param.SetValue(mapperName);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}, $"填写标识");
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
MessageBox.Show(ex.Message);
|
|||
|
|
}
|
|||
|
|
}, $"填写 {majorName} 构件标识");
|
|||
|
|
MessageBox.Show("处理完成", "提示", MessageBoxButton.OK, MessageBoxImage.Information);
|
|||
|
|
//File.WriteAllText(Environment.GetFolderPath(Environment.SpecialFolder.Desktop) + "\\匹配结果" + ".csv",
|
|||
|
|
// sb.ToString(), Encoding.UTF8);
|
|||
|
|
}
|
|||
|
|
class Mapper(string name, string sign, int ratio)
|
|||
|
|
{
|
|||
|
|
public string Name { get; } = name;
|
|||
|
|
public string Sign { get; } = sign;
|
|||
|
|
public int Ratio { get; } = ratio;
|
|||
|
|
public override string ToString()
|
|||
|
|
{
|
|||
|
|
return $"{Name} - {Sign} ({Ratio})";
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//private StringBuilder sb = new StringBuilder();
|
|||
|
|
//private List<Mapper> Mappers { get; componentSignSet; } = new List<Mapper>();
|
|||
|
|
private string GetMapperSign(IGrouping<ElementId, Element> elementGroup, MvdData data)
|
|||
|
|
{
|
|||
|
|
if (elementGroup.Key == ElementId.InvalidElementId)
|
|||
|
|
{
|
|||
|
|
return string.Empty;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var definitions = MvdLiteAssist.GetElementSigns(data);
|
|||
|
|
var categoryName = elementGroup.FirstOrDefault().Category?.Name;
|
|||
|
|
|
|||
|
|
//如果没有匹配到，使用类型名进行匹配
|
|||
|
|
var type = Document.GetElement(elementGroup.Key) as ElementType;
|
|||
|
|
var famName = type.FamilyName;
|
|||
|
|
ConfigManager.ConfigFileBaseDir = Path.Combine(RvApp.AddInDirectory, "configs");
|
|||
|
|
var segmenter = new JiebaSegmenter();
|
|||
|
|
var subText = $"{type.Name} {famName} {categoryName}";
|
|||
|
|
var replaceText = subText.ExtractChinese();
|
|||
|
|
var segments = segmenter.Cut(replaceText).Distinct();
|
|||
|
|
var combinedText = string.Join(" ", segments);
|
|||
|
|
string resultName = string.Empty;
|
|||
|
|
double minRatio = 30;
|
|||
|
|
|
|||
|
|
foreach (var defin in definitions)
|
|||
|
|
{
|
|||
|
|
var defText = defin.Replace(majorName, "");
|
|||
|
|
var segmentsDef = segmenter.Cut(defText);
|
|||
|
|
defText = string.Join(" ", segmentsDef);
|
|||
|
|
var tempRatio = Fuzz.TokenSetRatio(defText, combinedText);
|
|||
|
|
//var tempRatio = Fuzz.Ratio(defin, segment);
|
|||
|
|
if (tempRatio >= minRatio)
|
|||
|
|
{
|
|||
|
|
//sb.AppendLine($"{defin},{defText},{combinedText},{tempRatio},");
|
|||
|
|
minRatio = tempRatio;
|
|||
|
|
resultName = defin;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return resultName;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//private static string ExtractChinese(string text)
|
|||
|
|
//{
|
|||
|
|
// if (string.IsNullOrEmpty(text)) return string.Empty;
|
|||
|
|
|
|||
|
|
// // 方法1：正则（简单常用）
|
|||
|
|
// return Regex.Replace(text, @"[^\u4e00-\u9fa5]", "");
|
|||
|
|
|
|||
|
|
// // 方法2：遍历（可扩展）
|
|||
|
|
// // var sb = new StringBuilder();
|
|||
|
|
// // foreach (char c in text)
|
|||
|
|
// // if (c >= 0x4e00 && c <= 0x9fa5)
|
|||
|
|
// // sb.Append(c);
|
|||
|
|
// // return sb.ToString();
|
|||
|
|
//}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 清理字符串中的括号、下划线、减号、加号等符号
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="input">原始字符串</param>
|
|||
|
|
/// <returns>清理后的字符串</returns>
|
|||
|
|
public static string CleanSpecialChars(string input, string replacestr = " ")
|
|||
|
|
{
|
|||
|
|
if (string.IsNullOrEmpty(input))
|
|||
|
|
return input;
|
|||
|
|
|
|||
|
|
// 正则模式说明：
|
|||
|
|
// [()\[\]{}<>_-+] 匹配以下符号：
|
|||
|
|
// 圆括号 ()、方括号 []、花括号 {}、尖括号 <>、下划线 _、减号 -、加号 +
|
|||
|
|
// 注意：方括号内的特殊字符（如 [ ] - ）需要转义或放在安全位置（如末尾）
|
|||
|
|
string pattern = @"[()\[\]{}<>_+-]";
|
|||
|
|
return Regex.Replace(input, pattern, replacestr);
|
|||
|
|
}
|
|||
|
|
private string GetBestMatchingSign(IGrouping<ElementId, Element> g, HashSet<string> definitions, string categoryName, string famName, ref string resultName)
|
|||
|
|
{
|
|||
|
|
//初步匹配
|
|||
|
|
List<string> strings = [];
|
|||
|
|
if (major == Major.Architecture || major == Major.Structure || major == Major.MasterPlan)
|
|||
|
|
{
|
|||
|
|
int mapperRatio = 30;
|
|||
|
|
foreach (var mapper in definitions)
|
|||
|
|
{
|
|||
|
|
//var sourceIndex = Levenshtein.GetMatchingBlocks(mapper, categoryName).FirstOrDefault().SourcePos;
|
|||
|
|
//var pos = mapper.Length - sourceIndex;
|
|||
|
|
|
|||
|
|
var tempRatio = Fuzz.WeightedRatio(mapper, $"{categoryName}");
|
|||
|
|
if (tempRatio > 30)
|
|||
|
|
{
|
|||
|
|
if (tempRatio > mapperRatio)
|
|||
|
|
{
|
|||
|
|
mapperRatio = tempRatio;
|
|||
|
|
resultName = mapper;
|
|||
|
|
}
|
|||
|
|
strings.Add(mapper);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
strings = [.. definitions];
|
|||
|
|
}
|
|||
|
|
var subText = g.FirstOrDefault().Name;
|
|||
|
|
|
|||
|
|
var replaceText = subText.Replace('-', ' ').Replace('_', ' ').Replace('—', ' ');
|
|||
|
|
var combinedText = $"{famName} {replaceText}";
|
|||
|
|
int minRatio = 20;
|
|||
|
|
foreach (var defin in strings)
|
|||
|
|
{
|
|||
|
|
//var tempRatio = Fuzz.TokenSortRatio(defin, replaceText);
|
|||
|
|
var tempRatio = Fuzz.WeightedRatio(defin, combinedText);
|
|||
|
|
if (tempRatio > minRatio)
|
|||
|
|
{
|
|||
|
|
//Mappers.Add(new(replaceText, defin, tempRatio));
|
|||
|
|
minRatio = tempRatio;
|
|||
|
|
resultName = defin;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//if (string.IsNullOrEmpty(resultName))
|
|||
|
|
//{
|
|||
|
|
// resultName = "无标识";
|
|||
|
|
//}
|
|||
|
|
return resultName;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 需要移除现有的标识参数
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="paramName"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
private bool NeedRemoveParameter(string paramName)
|
|||
|
|
{
|
|||
|
|
var dict = SharedParameterAssists.GetParameterElementBindings(Document)
|
|||
|
|
.Where(d => d.Key.Name == paramName);
|
|||
|
|
if (dict.Any())
|
|||
|
|
{
|
|||
|
|
var result = MessageBox.Show(
|
|||
|
|
$"已存在一个或多个{paramName}的参数定义，将全部进行删除，并重新添加，是否继续？",
|
|||
|
|
"警告",
|
|||
|
|
MessageBoxButton.YesNo,
|
|||
|
|
MessageBoxImage.Warning);
|
|||
|
|
if (result == MessageBoxResult.Yes)
|
|||
|
|
{
|
|||
|
|
Document.Invoke(
|
|||
|
|
_ =>
|
|||
|
|
{
|
|||
|
|
Document.Delete(dict.Select(d => d.Key).Select(p => p.Id).ToList());
|
|||
|
|
},
|
|||
|
|
$"删除现有{paramName}参数");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|