zhao
2021-06-24 02ca96debc6056275d58e55d97f7885a195542d0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
using System;
using System.Collections.Generic;
using System.Text;
using QiHe.CodeLib;
 
namespace HH.WMS.Utils.ExcelLibrary.CompoundDocumentFormat
{
    public class DirectoryTree
    {
        public static void Build(DirectoryEntry rootEntry)
        {
            if (rootEntry.Members.Count > 0)
            {
                rootEntry.MembersTreeNodeDID = BuildStorageEntry(rootEntry);
            }
        }
 
        private static int BuildStorageEntry(DirectoryEntry storageEntry)
        {
            // direct members of each storage are organised in a separate red-black tree
            RedBlackTree<DirectoryEntry> rbTree = new RedBlackTree<DirectoryEntry>();
            foreach (DirectoryEntry entry in storageEntry.Members.Values)
            {
                rbTree.Add(entry);
            }
 
            foreach (RedBlackTreeNode<DirectoryEntry> node in rbTree.InorderTreeWalk(rbTree.Root))
            {
                DirectoryEntry entry = node.Data;
                entry.NodeColor = GetNodeColor(node.Color);
                entry.LeftChildDID = GetNodeID(node.Left);
                entry.RightChildDID = GetNodeID(node.Right);
 
                if (entry.Members.Count > 0)
                {
                    entry.EntryType = EntryType.Storage;
                    entry.MembersTreeNodeDID = BuildStorageEntry(entry);
                }
                else
                {
                    entry.EntryType = EntryType.Stream;
                    entry.MembersTreeNodeDID = -1;
                }
            }
 
            return rbTree.Root.Data.ID;
        }
 
        private static int GetNodeID(RedBlackTreeNode<DirectoryEntry> node)
        {
            if (node == RedBlackTreeNode<DirectoryEntry>.Nil)
            {
                return -1;
            }
            else
            {
                return node.Data.ID;
            }
        }
 
        private static byte GetNodeColor(QiHe.CodeLib.NodeColor nodeColor)
        {
            switch (nodeColor)
            {
                case QiHe.CodeLib.NodeColor.Black:
                    return NodeColor.Black;
                case QiHe.CodeLib.NodeColor.Red:
                    return NodeColor.Red;
                default:
                    throw new ArgumentException();
            }
        }
    }
}