Shrlalgo.RvKits/ShrlAlgoToolkit.RevitAddins/RvFamily/FamilyLibrary/RevitFileUtil.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.RvFamily.FamilyLibrary
{
    /// <summary>
    ///     直接读取文件的信息（不依赖RevitAPI）
    /// </summary>
    internal static class RevitFileUtil
    {
        private static string _fileName;

        public static Image PreviewImage { get; set; }
        public static List<FamilySymbolType> SymbolTypes { get; set; }

        public static StringBuilder FileInfo { get; set; }
        public static StringBuilder Content { get; set; }
        public static StringBuilder TransMissionData { get; set; }
        public static string SafeName { get; set; }
        public static string Product { get; set; }
        public static string RevitVersion { get; set; }
        public static string UpdateTime { get; set; }

        /// <summary>
        ///     图片资源转字节
        /// </summary>
        /// <param name="bitmapSource"></param>
        /// <returns></returns>
        public static byte[] BitSourceToArray(BitmapSource bitmapSource)
        {
            BitmapEncoder encoder = new JpegBitmapEncoder();
            using (MemoryStream ms = new MemoryStream())
            {
                encoder.Frames.Add(BitmapFrame.Create(bitmapSource));
                encoder.Save(ms);
                return ms.ToArray();
            }
        }

        public static StringBuilder GetFamilyInfo(List<FamilySymbolType> symbols)
        {
            StringBuilder sb = new StringBuilder();
            foreach (FamilySymbolType symbol in symbols)
            {
                sb.Append($"族类型：{symbol.Name}{symbol.Value}\r\n");
                foreach (Parameter param in symbol.Parameters)
                {
                    sb.Append($"   {param.Name}:{param.Value}\r\n");
                }
            }

            return sb;
        }

        public static int GetPngStartingOffset(byte[] previewData)
        {
            bool markerFound = false;
            int startingOffset = 0;
            int previousValue = 0;
            using (MemoryStream ms = new MemoryStream(previewData))
            {
                for (int i = 0; i < previewData.Length; i++)
                {
                    int 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>
        public static Image GetPreviewAsImage(byte[] previewData)
        {
            if (previewData == null || previewData.Length <= 0)
            {
                using (Bitmap newBitmap = new Bitmap(128, 128))
                {
                    return newBitmap.Clone() as Bitmap;
                }
            }

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

            try
            {
                byte[] pngDataBuffer = new byte[previewData.GetUpperBound(0) - startingOffset + 1];
                // read the PNG image data into a byte array
                using (MemoryStream 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 (MemoryStream ms = new MemoryStream(pngDataBuffer))
                    {
                        PngBitmapDecoder 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 (MemoryStream ms = new MemoryStream(decoderData))
                    {
                        using (Bitmap newBitmap = new Bitmap(ms))
                        {
                            using (Image newImage = newBitmap)
                            {
                                return newImage.Clone() as Image;
                            }
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                Debug.WriteLine(ex.Message);
            }

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


        public static string ParseDetailInfo(string detailInfo)
        {
            try
            {
                detailInfo = detailInfo.Trim();
                int index = detailInfo.IndexOf(":");
                string detailValue = detailInfo.Substring(detailInfo.IndexOf(":") + 1);
                string 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;
            //}
        }

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

            string[] basicFileInfoParts = unicodeData.Split('\0');

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

            return s;
        }

        /// <summary>
        ///     解析参数
        /// </summary>
        /// <param name="streamInfo"></param>
        public static List<FamilySymbolType> ParseParameter(StreamInfo streamInfo)
        {
            XmlDocument document = new XmlDocument();

            string xmlStr;
            byte[] streamData;
            using (Stream 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);

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

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

                XmlNodeList 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;

                List<FamilySymbolType> symbols = new List<FamilySymbolType>();
                foreach (XmlNode xn in xnlist)
                {
                    //XmlAttributeCollection attriCol = xn.Attributes;
                    //foreach (XmlAttribute xmlAttri in attriCol)
                    //{
                    //    string name = xmlAttri.Name;
                    //    string value = xmlAttri.Value;

                    //}
                    FamilySymbolType symbol = new FamilySymbolType();
                    if (xn.HasChildNodes)
                    {
                        foreach (XmlNode child in xn.ChildNodes)
                        {
                            Parameter p = new Parameter();
                            if (child.Name == "title")
                            {
                                symbol.Name = child.InnerText;
                                continue;
                            }

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

                    symbols.Add(symbol);
                }

                return symbols;
            }
        }

        public static void ParserRevitFile(string filename)
        {
            if (filename.EndsWith(".rvt") || filename.EndsWith(".rte") || filename.EndsWith(".rfa") ||
                filename.EndsWith(".rft"))
            {
                _fileName = filename;

                //获取Storageroot Object
                BindingFlags bindingFlags = BindingFlags.Static | BindingFlags.Instance | BindingFlags.Public |
                                            BindingFlags.NonPublic | BindingFlags.InvokeMethod;

                Type 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
                };
                StorageInfo Storage = (StorageInfo)storageRootType.InvokeMember("Open", bindingFlags, null, null, file);

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

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

                //var x = Storage.ThumbnailImage.GetPreviewAsImage();
                //读取结构化存储文件
                StreamInfo[] streams = Storage.GetStreams();
                foreach (StreamInfo streamInfo in streams)
                {
                    string unicodeData;
                    byte[] streamData = null;
                    using (Stream 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 = ParseParameter(streamInfo);
                            //Console.WriteLine("族参数:\r\n" + GetFamilyInfo(symbols));
                        }
                    }

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

                    if (streamInfo.Name.ToUpper().Equals("REVITPREVIEW4.0"))
                    {
                        PreviewImage = ParsePreview(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);
            //    }
            //}
        }

        public static StringBuilder ParseTransMissionData(string unicodeData)
        {
            StringBuilder s = new StringBuilder();

            string[] basicFileInfoParts = unicodeData.Split('\0');

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

            return s;
        }

        public static void ParseXmlDocument(XmlDocument document)
        {
            XmlNode rootNode = document.SelectSingleNode("A");
            XmlNodeList xmlNodeList = rootNode?.ChildNodes;
            if (xmlNodeList == null) return;
            foreach (XmlNode xn in xmlNodeList)
            {
                if (true)
                {
                }

                XmlAttributeCollection xnAttributes = xn.Attributes;
                if (xnAttributes != null)
                    foreach (XmlAttribute attribute in xnAttributes)
                    {
                        string name = attribute.Name;
                        string value = attribute.Value;
                    }

                if (xn.HasChildNodes)
                {
                }
            }
        }

        /// <summary>
        ///     清理无效字符
        /// </summary>
        /// <param name="oldValue"></param>
        /// <returns></returns>
        public static string PurgeUnprintableCharacters(string oldValue)
        {
            StringBuilder sb = new StringBuilder();
            char[] oldValueArray = oldValue.ToCharArray();
            foreach (char 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;
        }

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

            string[] basicFileInfoParts = unicodeData.Split('\0');

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

            return s;
        }

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

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

            return s;
        }

        private static Image ParsePreview(StreamInfo stream)
        {
            byte[] streamData = null;
            using (Stream 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);
        }
    }
}