コード例 #1
0
        public static void FindCommonAncestorsWithinListWorks()
        {
            // Arrange
            var list = new List <TestElement>();

            list.Add(new TestElement("root", -1));
            list.Add(new TestElement("A", 0));
            var b0 = new TestElement("B", 0);
            var b1 = new TestElement("Bchild", 1);
            var b2 = new TestElement("Bchildchild", 2);

            list.Add(b0);
            list.Add(b1);
            list.Add(b2);

            var c0 = new TestElement("C", 0);

            list.Add(c0);

            var f0 = new TestElement("F", 0);
            var f1 = new TestElement("Fchild", 1);
            var f2 = new TestElement("Fchildchild", 2);

            list.Add(f0);
            list.Add(f1);
            list.Add(f2);

            // Init tree structure: set children and parent properties
            TreeElement.ListToTree(list);

            // Single element
            TestElement[] input          = { b1 };
            TestElement[] expectedResult = { b1 };
            var           result         = TreeElement.FindCommonAncestorsWithinList(input).ToArray();

            Assert.IsTrue(UnityEditor.ArrayUtility.ArrayEquals(expectedResult, result), "Single input should return single output");

            // Single sub tree
            input          = new[] { b1, b2 };
            expectedResult = new[] { b1 };
            result         = TreeElement.FindCommonAncestorsWithinList(input).ToArray();
            Assert.IsTrue(UnityEditor.ArrayUtility.ArrayEquals(expectedResult, result), "Common ancestor should only be b1 ");

            // Multiple sub trees
            input          = new[] { b0, b2, f0, f2, c0 };
            expectedResult = new[] { b0, f0, c0 };
            result         = TreeElement.FindCommonAncestorsWithinList(input).ToArray();
            Assert.IsTrue(UnityEditor.ArrayUtility.ArrayEquals(expectedResult, result), "Common ancestor should only be b0, f0, c0");
        }
コード例 #2
0
ファイル: TreeModel.cs プロジェクト: MUDV587/StratusFramework
        /// <summary>
        /// Removes the given elements
        /// </summary>
        /// <param name="elements"></param>
        public void RemoveElements(IList <T> elements)
        {
            foreach (var element in elements)
            {
                if (element == this.root)
                {
                    throw new ArgumentException("It is not allowed to remove the root element");
                }
            }

            var commonAncestors = TreeElement.FindCommonAncestorsWithinList(elements);

            foreach (var element in commonAncestors)
            {
                element.parent.children.Remove(element);
                element.parent = null;
            }

            TreeElement.TreeToList(this.root, this.data);

            OnChanged();
        }