导出功能

qin

2024-08-15 763de811220b4b724f355965c156ce95d62d64e5

 HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs

@@ -1,70 +1,174 @@
using Autodesk.Revit.DB;
using HStation.RevitDev.RevitDataExport.Entity;
using System.IO;
using System.Linq;

namespace HStation.RevitDev.RevitDataExport.Utility
{
    public class CacheUtil
    {
        public static void ReadCache(Document doc)
        {
            var filePath = doc.PathName;
            var cache = File.ReadAllText(Common.GlobalResource.ConfigFilePath);
            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)
            {
                Common.GlobalResource.RevitModels = record.Record;
            }
        }

        public static void SaveCache(Document doc)
        {
            Common.GlobalResource.LastFilePath = doc.PathName;

            ExportModelHelper exportModelHelper = new ExportModelHelper(doc);
            var revitJson = exportModelHelper.Export();

            var dir = Path.GetDirectoryName(Common.GlobalResource.ExportFilePath);
            if (!Directory.Exists(dir))
            {
                Directory.CreateDirectory(dir);
            }

            if (File.Exists(Common.GlobalResource.ExportFilePath))
            {
                File.Delete(Common.GlobalResource.ExportFilePath);
            }

            File.WriteAllText(Common.GlobalResource.ExportFilePath, revitJson);
            var record = new ConfigRecord
            {
                FilePath = Common.GlobalResource.LastFilePath,
                Record = Common.GlobalResource.RevitModels
            };
            var cache = File.ReadAllText(Common.GlobalResource.ConfigFilePath);
            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;
            }

            var config = Newtonsoft.Json.JsonConvert.SerializeObject(records);
            File.WriteAllText(Common.GlobalResource.ConfigFilePath, config);
        }
    }
}
using Autodesk.Revit.DB;

using HStation.RevitDev.RevitDataExport.Common;

using HStation.RevitDev.RevitDataExport.Entity;

using ICSharpCode.SharpZipLib.Zip;

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 = 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

            };

            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;

            }



            var config = Newtonsoft.Json.JsonConvert.SerializeObject(records);

            File.WriteAllText(GlobalResource.RecordFilePath, config);

        }



        public static bool ExportZipFile(Document _doc)

        {

            FolderBrowserDialog dialog = new FolderBrowserDialog();

            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.RecordFilePath, 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)
                {
                    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();

            }

        }

    }

}