Shrlalgo.RvKits/ShrlAlgoToolkit.RevitAddins/Assists/RevitFileAssist.cs

using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.IO.Packaging;
using System.Reflection;
using System.Text;
using System.Windows.Media.Imaging;
using System.Xml;

namespace ShrlAlgoToolkit.RevitAddins.Assists
{
    /// <summary>
    ///     直接读取文件的信息（不依赖RevitAPI）
    /// </summary>
    public class RevitFileAssist
    {
        public RevitFileAssist(string filePath)
        {
            if (filePath.EndsWith(".rvt") || filePath.EndsWith(".rte") || filePath.EndsWith(".rfa") ||
                filePath.EndsWith(".rft"))
            {
                ParserRevitFile(filePath);
            }
            throw new Exception("文件格式不正确！");
        }
        /// <summary>
        ///     图片资源转字节
        /// </summary>
        /// <param name="bitmapSource"></param>
        /// <returns></returns>
        private static byte[] BitSourceToArray(BitmapSource bitmapSource)
        {
            BitmapEncoder encoder = new JpegBitmapEncoder();
            using (var ms = new MemoryStream())
            {
                encoder.Frames.Add(BitmapFrame.Create(bitmapSource));
                encoder.Save(ms);
                return ms.ToArray();
            }
        }

        private static int GetPngStartingOffset(byte[] previewData)
        {
            var markerFound = false;
            var startingOffset = 0;
            var previousValue = 0;
            using (var ms = new MemoryStream(previewData))
            {
                for (var i = 0; i < previewData.Length; i++)
                {
                    var currentValue = ms.ReadByte();
                    // possible start of PNG file data
                    if (currentValue == 137) // 0x89
                    {
                        markerFound = true;
                        startingOffset = i;
                        previousValue = currentValue;
                        continue;
                    }

                    switch (currentValue)
                    {
                        case 80: // 0x50
                            if (markerFound && previousValue == 137)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;

                        case 78: // 0x4E
                            if (markerFound && previousValue == 80)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;

                        case 71: // 0x47
                            if (markerFound && previousValue == 78)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;

                        case 13: // 0x0D
                            if (markerFound && previousValue == 71)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;

                        case 10: // 0x0A
                            if (markerFound && previousValue == 26)
                            {
                                return startingOffset;
                            }

                            if (markerFound && previousValue == 13)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;

                        case 26: // 0x1A
                            if (markerFound && previousValue == 10)
                            {
                                previousValue = currentValue;
                                continue;
                            }

                            markerFound = false;
                            break;
                    }
                }
            }

            return 0;
        }


        /// <summary>
        ///     获取缩略图
        /// </summary>
        /// <param name="previewData"></param>
        /// <returns></returns>
        private static Image GetPreviewAsImage(byte[] previewData)
        {
            if (previewData == null || previewData.Length <= 0)
            {
                using (var newBitmap = new Bitmap(128, 128))
                {
                    return newBitmap.Clone() as Bitmap;
                }
            }

            // read past the Revit metadata to the start of the PNG image
            var startingOffset = GetPngStartingOffset(previewData);
            if (startingOffset == 0)
            {
                using (var newBitmap = new Bitmap(128, 128))
                {
                    return newBitmap.Clone() as Bitmap;
                }
            }

            try
            {
                var pngDataBuffer = new byte[previewData.GetUpperBound(0) - startingOffset + 1];
                // read the PNG image data into a byte array
                using (var ms = new MemoryStream(previewData))
                {
                    ms.Position = startingOffset;
                    ms.Read(pngDataBuffer, 0, pngDataBuffer.Length);
                }

                byte[] decoderData = null;

                // if the image data is valid
                if (pngDataBuffer != null)
                {
                    // use a memory stream to decode the PNG image data
                    // and copy the decoded data into a byte array
                    using (var ms = new MemoryStream(pngDataBuffer))
                    {
                        var decoder = new PngBitmapDecoder(ms, BitmapCreateOptions.PreservePixelFormat,
                            BitmapCacheOption.Default);
                        decoderData = BitSourceToArray(decoder.Frames[0]);
                    }
                }

                // if the decoded data is valie
                if (decoderData != null && decoderData.Length > 0)
                {
                    // use another memory stream to create a Bitmap
                    // and then an Image from that Bitmap
                    using (var ms = new MemoryStream(decoderData))
                    {
                        using (var newBitmap = new Bitmap(ms))
                        {
                            using (Image newImage = newBitmap)
                            {
                                return newImage.Clone() as Image;
                            }
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                Debug.WriteLine(ex.Message);
            }

            using (var newBitmap = new Bitmap(128, 128))
            {
                return newBitmap.Clone() as Bitmap;
            }
        }

        private static StringBuilder ParseContents(string unicodeData)
        {
            var s = new StringBuilder();

            var basicFileInfoParts = unicodeData.Split('\0');

            foreach (var basicFileInfoPart in basicFileInfoParts)
            {
                if (basicFileInfoPart.IndexOf("\r\n", StringComparison.Ordinal) >= 0)
                {
                    var detailInfoParts = basicFileInfoPart.Split(new[] { "\r\n" }, new StringSplitOptions());
                    foreach (var detailPart in detailInfoParts)
                    {
                        s.Append(ParseDetailInfo(detailPart) + "\r\n");
                    }
                }
            }

            return s;
        }


        private static string ParseDetailInfo(string detailInfo)
        {
            try
            {
                detailInfo = detailInfo.Trim();
                var index = detailInfo.IndexOf(":");
                var detailValue = detailInfo.Substring(detailInfo.IndexOf(":") + 1);
                var detailKey = detailInfo.Substring(0, detailInfo.IndexOf(":"));
                detailKey = detailKey.Trim().ToUpper().Replace(" ", string.Empty);
                detailKey = PurgeUnprintableCharacters(detailKey);
                detailValue = PurgeUnprintableCharacters(detailValue);
                return $"{detailKey}:{detailValue}";
                //Console.WriteLine($"{detailKey}:{detailValue}");
            }
            catch (Exception)
            {
                return detailInfo;
            }

            //switch (detailKey)
            //{
            //    case "WORKSHARING":
            //        if (string.IsNullOrEmpty(detailValue))
            //        {
            //            WorkSharing = WorkSharingMode.HCLight;
            //            return;
            //        }

            //        string workSharing = detailValue.Replace(" ", string.Empty).Trim().ToUpper();
            //        switch (workSharing)
            //        {
            //            case "NOTENABLED":
            //                WorkSharing = WorkSharingMode.NotEnabled;
            //                break;

            //            case "LOCAL":
            //                WorkSharing = WorkSharingMode.Local;
            //                break;

            //            case "CENTRAL":
            //                WorkSharing = WorkSharingMode.Central;
            //                break;

            //            default:
            //                WorkSharing = WorkSharingMode.HCLight;
            //                break;
            //        }
            //        break;

            //    case "USERNAME":
            //        UserName = detailValue.Trim();
            //        break;

            //    case "CENTRALFILEPATH":
            //        CentralFilePath = detailValue.Trim();
            //        break;

            //    case "REVITBUILD":
            //        RevitBuild = detailValue.Trim();
            //        break;

            //    case "LASTSAVEPATH":
            //        LastSavedpath = detailValue.Trim();
            //        break;

            //    case "OPENWORKSETDEFAULT":
            //        OpenWorksetDefault = Convert.ToInt32(detailValue.Trim());
            //        break;

            //    default:
            //        Console.WriteLine($"{detailKey}:{detailValue}");
            //        //Debug.Assert(false, string.Format("{0} was not found in the case tests.", detailKey));
            //        break;
            //}
        }

        private static StringBuilder ParseFileInfo(string unicodeData)
        {
            var s = new StringBuilder();

            var basicFileInfoParts = unicodeData.Split('\0');

            foreach (var basicFileInfoPart in basicFileInfoParts)
            {
                if (basicFileInfoPart.IndexOf("\r\n", StringComparison.Ordinal) >= 0)
                {
                    var detailInfoParts = basicFileInfoPart.Split(new[] { "\r\n" }, new StringSplitOptions());
                    foreach (var detailPart in detailInfoParts)
                    {
                        s.Append(ParseDetailInfo(detailPart) + "\r\n");
                    }
                }
            }

            return s;
        }

        /// <summary>
        ///     解析参数
        /// </summary>
        /// <param name="streamInfo"></param>
        private List<FamilyTypeDefinition> ParseParameterDefinition(StreamInfo streamInfo)
        {
            var document = new XmlDocument();

            string xmlStr;
            byte[] streamData;
            using (var streamReader = streamInfo.GetStream(FileMode.Open, FileAccess.Read))
            {
                streamData = new byte[streamReader.Length];
                streamReader.Read(streamData, 0, streamData.Length);
                xmlStr = Encoding.UTF8.GetString(streamData);
                document.LoadXml(xmlStr);
                //string xmldocpath = Path.GetFileNameWithoutExtension(FileName) + ".xml";
                //document.Save(xmldocpath);

                //读取参数信息
                var root = document.DocumentElement;
                //节点前缀的命名空间
                var nameSpace = root.GetNamespaceOfui:InputAssist.Prefix("A");
                //string nameSpace = root.NamespaceURI;
                var nsmgr = new XmlNamespaceManager(document.NameTable);
                ;
                nsmgr.AddNamespace("entry", nameSpace);

                //族类型
                var xnlist = document.GetElementsByTagName("A:part");

                var fileinfo = document.GetElementsByTagName("A:design-file");
                foreach (XmlNode xn in fileinfo)
                {
                    if (xn.HasChildNodes)
                    {
                        foreach (XmlNode child in xn.ChildNodes)
                        {
                            if (child.Name == "A:title")
                            {
                                SafeName = child.InnerText;
                            }

                            if (child.Name == "A:product")
                            {
                                Product = child.InnerText;
                            }

                            if (child.Name == "A:product-version")
                            {
                                RevitVersion = child.InnerText;
                            }

                            if (child.Name == "A:product-updated")
                            {
                                UpdateTime = child.InnerText;
                            }
                        }
                    }
                }
                //XmlNode rootnode = document.SelectSingleNode("/entry/A: family/A:part", nsmgr);
                //XmlNodeList xnlist = rootnode.ChildNodes;

                var symbols = new List<FamilyTypeDefinition>();
                foreach (XmlNode xn in xnlist)
                {
                    //XmlAttributeCollection attriCol = xn.Attributes;
                    //foreach (XmlAttribute xmlAttri in attriCol)
                    //{
                    //    string name = xmlAttri.Name;
                    //    string value = xmlAttri.Value;
                    //}
                    var symbol = new FamilyTypeDefinition();
                    if (xn.HasChildNodes)
                    {
                        foreach (XmlNode child in xn.ChildNodes)
                        {
                            var p = new ParameterDefinition();
                            if (child.Name == "title")
                            {
                                symbol.Name = child.InnerText;
                                continue;
                            }

                            //族类型节点
                            p.Name = child.Name;
                            //族名称
                            p.Value = child.InnerText;
                            symbol.ParameterDefinitions.Add(p);
                        }
                    }

                    symbols.Add(symbol);
                }

                return symbols;
            }
        }

        private static StringBuilder ParsePartAtom(string unicodeData)
        {
            var s = new StringBuilder();
            var basicFileInfoParts = unicodeData.Split('\0');

            foreach (var basicFileInfoPart in basicFileInfoParts)
            {
                if (basicFileInfoPart.IndexOf("\r\n", StringComparison.Ordinal) >= 0)
                {
                    var detailInfoParts = basicFileInfoPart.Split(new[] { "\r\n" }, new StringSplitOptions());
                    foreach (var detailPart in detailInfoParts)
                    {
                        s.Append(ParseDetailInfo(detailPart) + "\r\n");
                    }
                }
            }

            return s;
        }
        /// <summary>
        /// 解析预览图像
        /// </summary>
        /// <param name="stream"></param>
        /// <returns></returns>
        private static Image ParsePreviewImage(StreamInfo stream)
        {
            byte[] streamData = null;
            using (var streamReader = stream.GetStream(FileMode.Open, FileAccess.Read))
            {
                streamData = new byte[streamReader.Length];
                streamReader.Read(streamData, 0, streamData.Length);
                //unicodeData = Encoding.Unicode.GetString(streamData);
                //unicodeData = Encoding.UTF8.GetString(streamData);
            }

            return GetPreviewAsImage(streamData);
        }
        /// <summary>
        /// 解析传输数据
        /// </summary>
        /// <param name="unicodeData"></param>
        /// <returns></returns>
        private static StringBuilder ParseTransMissionData(string unicodeData)
        {
            var s = new StringBuilder();

            var basicFileInfoParts = unicodeData.Split('\0');

            foreach (var basicFileInfoPart in basicFileInfoParts)
            {
                if (basicFileInfoPart.IndexOf("\r\n") >= 0)
                {
                    var detailInfoParts = basicFileInfoPart.Split(new[] { "\r\n" }, new StringSplitOptions());
                    foreach (var detailPart in detailInfoParts)
                    {
                        s.Append(ParseDetailInfo(detailPart) + "\r\n");
                    }
                }
            }

            return s;
        }
        //public static StringBuilder GetFamilyInfo(List<FamilyTypeDefinition> symbols)
        //{
        //    var sb = new StringBuilder();
        //    foreach (var symbol in symbols)
        //    {
        //        sb.Append($"族类型：{symbol.Name}{symbol.Value}\r\n");
        //        foreach (var param in symbol.ParameterDefinitions)
        //        {
        //            sb.Append($"   {param.Name}:{param.Value}\r\n");
        //        }
        //    }
        //    return sb;
        //}
        /// <summary>
        /// 解析Revit文件信息
        /// </summary>
        /// <param name="filename"></param>
        private void ParserRevitFile(string filename)
        {
            //获取Storageroot Object
            var bindingFlags = BindingFlags.Static | BindingFlags.Instance | BindingFlags.Public |
                                        BindingFlags.NonPublic | BindingFlags.InvokeMethod;

            var storageRootType = typeof(StorageInfo).Assembly.GetType("System.IO.Packaging.StorageRoot", true, false);
            //Type storageRootType = typeof(StorageInfo).Assembly.GetType("System.IO.Packaging", true, false);
            object[] file =
            {
                    filename,
                    FileMode.Open,
                    FileAccess.Read,
                    FileShare.Read
                };
            var Storage = (StorageInfo)storageRootType.InvokeMember("Open", bindingFlags, null, null, file);

            //if (Storage != null)
            //{
            //    storageRootType.InvokeMember("CloseTrigger", bindingFlags, null, Storage, file);

            //    //Console.Write($"{filePath}文件无法作为结构存储，并打开打开");
            //}

            //var x = Storage.ThumbnailImage.GetPreviewAsImage();
            //读取结构化存储文件
            var streams = Storage.GetStreams();
            foreach (var streamInfo in streams)
            {
                string unicodeData;
                byte[] streamData = null;
                using (var streamReader = streamInfo.GetStream(FileMode.Open, FileAccess.Read))
                {
                    streamData = new byte[streamReader.Length];
                    streamReader.Read(streamData, 0, streamData.Length);
                    unicodeData = Encoding.Unicode.GetString(streamData);
                    //unicodeData = Encoding.UTF8.GetString(streamData);
                }

                if (streamInfo.Name.ToUpper().Equals("BASICFILEINFO"))
                {
                    FileInfo = ParseFileInfo(unicodeData);
                }

                if (streamInfo.Name.ToUpper().Equals("PARTATOM"))
                {
                    //Console.WriteLine("部分原子:\r\n" + ParsePartAtom(unicodeData));
                    if (filename.EndsWith(".rfa"))
                    {
                        SymbolTypes = ParseParameterDefinition(streamInfo);
                        //Console.WriteLine("族参数:\r\n" + GetFamilyInfo(symbols));
                    }
                }

                if (streamInfo.Name.ToUpper().Equals("TRANSMISSIONDATA"))
                {
                    TransMissionData = ParseTransMissionData(unicodeData);
                }

                if (streamInfo.Name.ToUpper().Equals("REVITPREVIEW4.0"))
                {
                    PreviewImage = ParsePreviewImage(streamInfo);
                }

                if (streamInfo.Name.ToUpper().Equals("CONTENTS"))
                {
                    Content = ParseContents(unicodeData);
                }
            }
            //if (Storage != null)
            //{
            //    try
            //    {
            //        storageRootType.InvokeMember("CloseTrigger", bindingFlags, null, Storage, file);
            //    }
            //    catch (Exception ex)
            //    {
            //        MessageBox.ShowAhead(ex.ViewMessage);
            //    }
            //}
        }

        /// <summary>
        ///     清理无效字符
        /// </summary>
        /// <param name="oldValue"></param>
        /// <returns></returns>
        public static string PurgeUnprintableCharacters(string oldValue)
        {
            var sb = new StringBuilder();
            var oldValueArray = oldValue.ToCharArray();
            foreach (var letter in oldValueArray)
            {
                int decimalValue = letter;
                if (decimalValue >= 32 && decimalValue <= 126)
                {
                    sb.Append(letter);
                }
            }

            oldValue = sb.ToString();
            sb.Length = 0;
            sb.Capacity = 0;
            sb = null;
            return oldValue;
        }

        public StringBuilder Content { get; private set; }

        public StringBuilder FileInfo { get; private set; }
        public Image PreviewImage { get; private set; }
        public string Product { get; private set; }
        public string RevitVersion { get; private set; }
        public string SafeName { get; private set; }
        public List<FamilyTypeDefinition> SymbolTypes { get; private set; }
        public StringBuilder TransMissionData { get; private set; }
        public string UpdateTime { get; private set; }

        public class FamilyTypeDefinition
        {
            public string Name { get; set; }
            public List<ParameterDefinition> ParameterDefinitions { get; set; } = new();
            public string Value { get; set; }
        }

        public class ParameterDefinition
        {
            public string Name { get; set; }
            public string TypeOfParameterDefinition { get; set; }
            public string Value { get; set; }
        }
    }
}