using 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
|
{
|
public class CacheUtil
|
{
|
public static void InitCache(Document doc)
|
{
|
var filePath = doc.PathName;
|
var cache = File.ReadAllText(GlobalResource.RecordFilePath);
|
if (string.IsNullOrEmpty(cache)) { return; }
|
|
Records records = Newtonsoft.Json.JsonConvert.DeserializeObject<Records>(cache);
|
if (records == null || records.ConfigRecords == null || records.ConfigRecords.Count == 0) { return; }
|
|
ConfigRecord record = records.ConfigRecords.Where(x => x.FilePath == filePath)?.FirstOrDefault();
|
if (record != null)
|
{
|
GlobalResource.RevitModels.Add(new Tuple<string, Dictionary<RevitType, List<string>>>(doc.Title, record.Record));
|
}
|
}
|
|
public static void SaveCache(Document doc)
|
{
|
GlobalResource.LastFilePath = doc.PathName;
|
|
ExportModelHelper exportModelHelper = new ExportModelHelper(doc);
|
var revitJson = exportModelHelper.Export();
|
|
var dir = Path.GetDirectoryName(GlobalResource.ExportFilePath);
|
if (!Directory.Exists(dir))
|
{
|
Directory.CreateDirectory(dir);
|
}
|
|
if (File.Exists(GlobalResource.ExportFilePath))
|
{
|
File.Delete(GlobalResource.ExportFilePath);
|
}
|
|
File.WriteAllText(GlobalResource.ExportFilePath, revitJson);
|
var record = new ConfigRecord
|
{
|
FilePath = GlobalResource.LastFilePath,
|
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))
|
{
|
records = Newtonsoft.Json.JsonConvert.DeserializeObject<Records>(cache);
|
}
|
var configRecord = records.ConfigRecords?.Where(x => x.FilePath == doc.PathName)?.FirstOrDefault();
|
if (configRecord == null)
|
{
|
records.ConfigRecords.Add(record);
|
}
|
else
|
{
|
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";
|
if (File.Exists(structFile)) File.Delete(structFile);
|
if (File.Exists(modelFile)) File.Delete(modelFile);
|
if (!File.Exists(settingFile)) File.Create(settingFile).Close();
|
File.Copy(_doc.PathName, modelFile);
|
File.Copy(GlobalResource.ExportFilePath, structFile);
|
|
var setting = new SettingsViewModel() { Version = "1" };
|
File.WriteAllText(settingFile, JsonHelper.ToJson(setting));
|
|
var files = new string[] { structFile, modelFile, settingFile };
|
ZipFiles(files, zipFilePath);
|
|
File.Delete(structFile);
|
File.Delete(modelFile);
|
File.Delete(settingFile);
|
return true;
|
}
|
catch (Exception ex)
|
{
|
err = JsonHelper.ToJson(ex);
|
return false;
|
}
|
|
}
|
return true;
|
|
}
|
|
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; }
|
|
var ids = GlobalResource.RevitModels.GetIds()?.Select(x => new ElementId(int.Parse(x))).ToList();
|
if (ids.Count == 0)
|
{
|
return;
|
}
|
GlobalResource.HideMode = !GlobalResource.HideMode;
|
using (Transaction trans = new Transaction(document, "hide elements"))
|
{
|
trans.Start();
|
if (GlobalResource.HideMode)
|
{
|
document.ActiveView.HideElements(ids);
|
}
|
else
|
{
|
document.ActiveView.UnhideElements(ids);
|
}
|
trans.Commit();
|
}
|
}
|
}
|
}
|
