qin
2024-12-15 9d75f0401985156299472e66d2fff07a854d9c67
HStation.RevitDev/RevitDataExport/Utility/CacheUtil.cs
@@ -1,9 +1,16 @@
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;
using Autodesk.Revit.UI;
using static System.Windows.Forms.VisualStyles.VisualStyleElement.Window;
using System.Net.Http;
namespace HStation.RevitDev.RevitDataExport.Utility
{
@@ -21,7 +28,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));
            }
        }
@@ -47,8 +54,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).Close();
            }
            var cache = File.ReadAllText(GlobalResource.RecordFilePath);
            Records records = new Records();
            if (!string.IsNullOrEmpty(cache))
@@ -62,11 +74,183 @@
            }
            else
            {
                configRecord = record;
                configRecord.Record = record.Record;
            }
            var config = Newtonsoft.Json.JsonConvert.SerializeObject(records);
            File.WriteAllText(GlobalResource.RecordFilePath, config);
        }
        public static bool UploadFile(Document _doc, string desc, out string err)
        {
            err = "";
            var tempDir = Path.Combine(Path.GetDirectoryName(GlobalResource.BinDirectory), "temp"); //"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";
            try
            {
                var title = _doc.Title + "_" + DateTime.Now.ToString("yyyy-MM-dd-HH-mm-ss") + ".ywrvt";
                var zipFilePath = tempDir + "\\" + title;
                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 strConfig = File.ReadAllText(GlobalResource.SettingFilePath);
                var ret = Newtonsoft.Json.JsonConvert.DeserializeObject<ConfigSettingModel>(strConfig);
                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);
                File.Delete(otherFile);
                using (var client = new HttpClient())
                {
                    var content = new MultipartFormDataContent();
                    content.Add(new StringContent("循环水Revit"), "UploadUserName");
                    content.Add(new StringContent(""), "FileCode");
                    content.Add(new StringContent(desc), "Description");
                    content.Add(new ByteArrayContent(File.ReadAllBytes(zipFilePath)), "File", title);
                    var res = client.PostAsync(ret.UploadUrl, content).Result.Content.ReadAsStringAsync().Result;
                    var result = JsonHelper.ToObject<HttpResultViewModel>(res);
                    File.Delete(zipFilePath);
                    if (result.Code == 0)
                        return true;
                    else
                    {
                        err = result.Message;
                        return false;
                    }
                }
            }
            catch (Exception ex)
            {
                File.Delete(structFile);
                File.Delete(otherFile);
                File.Delete(modelFile);
                File.Delete(settingFile);
                File.Delete(otherFile);
                err = JsonHelper.ToJson(ex);
                return false;
            }
        }
        public static bool ExportZipFile(Document _doc, out string err)
        {
            FolderBrowserDialog dialog = new FolderBrowserDialog();
            err = "";
            if (dialog.ShowDialog() == DialogResult.OK)
            {
                var tempDir = Path.Combine(Path.GetDirectoryName(GlobalResource.BinDirectory), "temp"); //"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";
                var userFile = GlobalResource.LoginUserFilePath;
                try
                {
                    var zipFilePath = dialog.SelectedPath + "\\model.ywrvt";
                    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, userFile };
                    ZipFiles(files, zipFilePath);
                    File.Delete(structFile);
                    File.Delete(otherFile);
                    File.Delete(modelFile);
                    File.Delete(settingFile);
                    File.Delete(otherFile);
                    return true;
                }
                catch (Exception ex)
                {
                    File.Delete(structFile);
                    File.Delete(otherFile);
                    File.Delete(modelFile);
                    File.Delete(settingFile);
                    File.Delete(otherFile);
                    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)
@@ -78,19 +262,66 @@
            {
                return;
            }
            GlobalResource.HideMode = !GlobalResource.HideMode;
            using (Transaction trans = new Transaction(document, "hide elements"))
            try
            {
                trans.Start();
                if (GlobalResource.HideMode)
                GlobalResource.HideMode = !GlobalResource.HideMode;
                using (Transaction trans = new Transaction(document, "hide elements"))
                {
                    document.ActiveView.HideElements(ids);
                    trans.Start();
                    if (GlobalResource.HideMode)
                    {
                        document.ActiveView.HideElements(ids);
                    }
                    else
                    {
                        document.ActiveView.UnhideElements(ids);
                    }
                    trans.Commit();
                }
                else
            }
            catch (Exception ex)
            {
                TaskDialog.Show("错误", "显隐失败,可能存在不能设置隐藏的构件!");
            }
        }
        public static void HideOrShowOtherModels(Document document)
        {
            if (document == null) { return; }
            //var ids = GlobalResource.RevitModels.GetOtherIds()?.Select(x => new ElementId(int.Parse(x))).ToList();
            var notContains = new List<string>() { "图例构件", "楼板", "墙", "栏杆扶手", "楼梯" };
            var nameContains = new List<string>() { "混凝土底座" };
            var c3 = new Autodesk.Revit.DB.FilteredElementCollector(document).WhereElementIsNotElementType();//.WhereElementIsElementType();
            var c4 = new Autodesk.Revit.DB.FilteredElementCollector(document).WhereElementIsElementType();
            c3.Concat(c4);
            //var ids = c3.Where(c => c.Category != null && !notContains.Any(c.Category.Name.Contains)).Select(c => new ElementId(int.Parse(c.Id.ToString()))).Distinct().ToList();
            var ids = c3.Where(c => (c.Category != null && notContains.Any(c.Category.Name.Contains)) || nameContains.Any(c.Name.Contains)).Select(c => new ElementId(int.Parse(c.Id.ToString()))).Distinct().ToList();
            if (ids.Count == 0)
            {
                return;
            }
            try
            {
                GlobalResource.IsOtherHidden = !GlobalResource.IsOtherHidden;
                using (Transaction trans = new Transaction(document, "hide other elements"))
                {
                    document.ActiveView.UnhideElements(ids);
                    trans.Start();
                    if (GlobalResource.IsOtherHidden)
                    {
                        document.ActiveView.HideElements(ids);
                    }
                    else
                    {
                        document.ActiveView.UnhideElements(ids);
                    }
                    trans.Commit();
                }
                trans.Commit();
            }
            catch (Exception ex)
            {
                //TaskDialog.Show("错误", "显隐失败,可能存在不能设置隐藏的构件!");
            }
        }
    }