From 9162a61818966f17e3aa2952c39d5484af1147d7 Mon Sep 17 00:00:00 2001
From: qin <a@163.com>
Date: 星期四, 22 八月 2024 16:02:27 +0800
Subject: [PATCH] 优化导出逻辑
---
HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs | 105 +++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 102 insertions(+), 3 deletions(-)
diff --git a/HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs b/HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs
index aba5500..4ab42ba 100644
--- a/HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs
+++ b/HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs
@@ -1,8 +1,13 @@
锘縰sing Autodesk.Revit.DB;
+using HStation.RevitDev.Model.ModelEnum;
using HStation.RevitDev.RevitDataExport.Common;
using HStation.RevitDev.RevitDataExport.Entity;
+using ICSharpCode.SharpZipLib.Zip;
+using System.Collections.Generic;
+using System;
using System.IO;
using System.Linq;
+using System.Windows.Forms;
namespace HStation.RevitDev.RevitDataExport.Utility
{
@@ -20,7 +25,7 @@
ConfigRecord record = records.ConfigRecords.Where(x => x.FilePath == filePath)?.FirstOrDefault();
if (record != null)
{
- GlobalResource.RevitModels = record.Record;
+ GlobalResource.RevitModels.Add(new Tuple<string, Dictionary<RevitType, List<string>>>(doc.Title, record.Record));
}
}
@@ -46,8 +51,13 @@
var record = new ConfigRecord
{
FilePath = GlobalResource.LastFilePath,
- Record = GlobalResource.RevitModels
+ Record = GlobalResource.RevitModels.Find(x => x.Item1 == doc.Title)?.Item2
};
+ if (!File.Exists(GlobalResource.RecordFilePath))
+ {
+ File.Create(GlobalResource.RecordFilePath);
+ }
+
var cache = File.ReadAllText(GlobalResource.RecordFilePath);
Records records = new Records();
if (!string.IsNullOrEmpty(cache))
@@ -61,13 +71,102 @@
}
else
{
- configRecord = record;
+ configRecord.Record = record.Record;
}
var config = Newtonsoft.Json.JsonConvert.SerializeObject(records);
File.WriteAllText(GlobalResource.RecordFilePath, config);
}
+ public static bool ExportZipFile(Document _doc, out string err)
+ {
+ FolderBrowserDialog dialog = new FolderBrowserDialog();
+ err = "";
+ if (dialog.ShowDialog() == DialogResult.OK)
+ {
+ try
+ {
+ var zipFilePath = dialog.SelectedPath + "\\model.ywrvt";
+
+ var tempDir = "d:\\temp";
+ if (!Directory.Exists(tempDir))
+ Directory.CreateDirectory(tempDir);
+ var structFile = tempDir + "\\struct.json";
+ var modelFile = tempDir + "\\model.rvt";
+ var settingFile = tempDir + "\\setting.json";
+ var otherFile = tempDir + "\\struct_others.json";
+ if (File.Exists(structFile)) File.Delete(structFile);
+
+ if (File.Exists(modelFile)) File.Delete(modelFile);
+ if (!File.Exists(settingFile)) File.Create(settingFile).Close();
+ if (!File.Exists(otherFile)) File.Create(otherFile).Close();
+ File.Copy(_doc.PathName, modelFile);
+ File.Copy(GlobalResource.ExportFilePath, structFile);
+
+ var notContains = new List<string>() { "绠￠亾闄勪欢", "绠￠亾", "绠′欢 ", "绠￠亾闄勪欢", "绠￠亾绯荤粺" };
+ var c3 = new Autodesk.Revit.DB.FilteredElementCollector(_doc).WhereElementIsNotElementType();//.WhereElementIsElementType();
+ var c4 = new Autodesk.Revit.DB.FilteredElementCollector(_doc).WhereElementIsElementType();
+
+ c3.Concat(c4);
+ var r = c3.Where(c => c.Category != null && !notContains.Any(c.Category.Name.Contains)).Select(c => new { ID = c.Id.ToString(), Name = c.Name, CategoryName = c.Category.Name, CategoryID = c.Category.Id.ToString() }).Distinct().ToList();
+ var t = JsonHelper.ToJson(r);
+ File.WriteAllText(otherFile, t);
+
+ var setting = new SettingsViewModel() { Version = "1" };
+ File.WriteAllText(settingFile, JsonHelper.ToJson(setting));
+
+ var files = new string[] { structFile, modelFile, settingFile, otherFile };
+ ZipFiles(files, zipFilePath);
+
+ File.Delete(structFile);
+ File.Delete(otherFile);
+ File.Delete(modelFile);
+ File.Delete(settingFile);
+ return true;
+ }
+ catch (Exception ex)
+ {
+ err = JsonHelper.ToJson(ex);
+ return false;
+ }
+ }
+ else return false;
+
+ }
+
+ private static void ZipFiles(string[] files, string zipFilePath)
+ {
+ using (var zipOutputStream = new ZipOutputStream(File.Create(zipFilePath)))
+ {
+ zipOutputStream.SetLevel(9); // 璁剧疆鍘嬬缉绾у埆
+
+ foreach (var file in files)
+ {
+ var entry = new ZipEntry(Path.GetFileName(file))
+ {
+ DateTime = DateTime.Now,
+ Size = File.ReadAllBytes(file).Length
+ };
+
+ zipOutputStream.PutNextEntry(entry);
+
+ using (var fileStream = File.OpenRead(file))
+ {
+ int sourceBytes;
+ byte[] buffer = new byte[4096];
+
+ while ((sourceBytes = fileStream.Read(buffer, 0, buffer.Length)) > 0)
+ {
+ zipOutputStream.Write(buffer, 0, sourceBytes);
+ }
+ }
+ }
+
+ zipOutputStream.Finish();
+ zipOutputStream.Close();
+ }
+ }
+
public static void HideOrShowModels(Document document)
{
if (document == null) { return; }
--
Gitblit v1.9.3