// #define USING_HASH_SET// ==++==// // Copyright (c) Microsoft Corporation. All rights reserved.// // ==--==/*============================================================**** Class: SortedSet**** Purpose: A generic sorted set.**** Date: August 15, 2008** ===========================================================*/namespace System.Collections.Generic { using System; using System.Diagnostics; using System.Diagnostics.CodeAnalysis; using System.Runtime.Serialization; // // A binary search tree is a red-black tree if it satisfies the following red-black properties: // 1. Every node is either red or black // 2. Every leaf (nil node) is black // 3. If a node is red, then both its children are black // 4. Every simple path from a node to a descendant leaf contains the same number of black nodes // // The basic idea of red-black tree is to represent 2-3-4 trees as standard BSTs but to add one extra bit of information // per node to encode 3-nodes and 4-nodes. // 4-nodes will be represented as: B // R R // 3 -node will be represented as: B or B // R B B R // // For a detailed description of the algorithm, take a look at "Algorithms" by Robert Sedgewick. // internal delegate bool TreeWalkPredicate<T>(SortedSet<T>.Node node); internal enum TreeRotation { LeftRotation = 1, RightRotation = 2, RightLeftRotation = 3, LeftRightRotation = 4, } [SuppressMessage("Microsoft.Naming", "CA1710:IdentifiersShouldHaveCorrectSuffix", Justification = "by design name choice")] [DebuggerTypeProxy(typeof(System.Collections.Generic.SortedSetDebugView<>))] [DebuggerDisplay("Count = {Count}")]#if !FEATURE_NETCORE [Serializable] public class SortedSet<T> : ISet<T>, ICollection<T>, ICollection, ISerializable, IDeserializationCallback, IReadOnlyCollection<T> {#else public class SortedSet<T> : ISet<T>, ICollection<T>, ICollection, IReadOnlyCollection<T> {#endif //!FEATURE_NETCORE #region local variables/constants Node root; IComparer<T> comparer; int count; int version; private Object _syncRoot; private const String ComparerName = "Comparer"; private const String CountName = "Count"; private const String ItemsName = "Items"; private const String VersionName = "Version"; //needed for enumerator private const String TreeName = "Tree"; private const String NodeValueName = "Item"; private const String EnumStartName = "EnumStarted"; private const String ReverseName = "Reverse"; private const String EnumVersionName = "EnumVersion";#if !FEATURE_NETCORE //needed for TreeSubset private const String minName = "Min"; private const String maxName = "Max"; private const String lBoundActiveName = "lBoundActive"; private const String uBoundActiveName = "uBoundActive"; private SerializationInfo siInfo; //A temporary variable which we need during deserialization. #endif internal const int StackAllocThreshold = 100; #endregion #region Constructors public SortedSet() { this.comparer = Comparer<T>.Default; } public SortedSet(IComparer<T> comparer) { if (comparer == null) { this.comparer = Comparer<T>.Default; } else { this.comparer = comparer; } } public SortedSet(IEnumerable<T> collection) : this(collection, Comparer<T>.Default) { } public SortedSet(IEnumerable<T> collection, IComparer<T> comparer) : this(comparer) { if (collection == null) { throw new ArgumentNullException("collection"); } // these are explicit type checks in the mould of HashSet. It would have worked better // with something like an ISorted<T> (we could make this work for SortedList.Keys etc) SortedSet<T> baseSortedSet = collection as SortedSet<T>; SortedSet<T> baseTreeSubSet = collection as TreeSubSet; if (baseSortedSet != null && baseTreeSubSet == null && AreComparersEqual(this, baseSortedSet)) { //breadth first traversal to recreate nodes if (baseSortedSet.Count == 0) { count = 0; version = 0; root = null; return; } //pre order way to replicate nodes Stack<Node> theirStack = new Stack<SortedSet<T>.Node>(2 * log2(baseSortedSet.Count) + 2); Stack<Node> myStack = new Stack<SortedSet<T>.Node>(2 * log2(baseSortedSet.Count) + 2); Node theirCurrent = baseSortedSet.root; Node myCurrent = (theirCurrent != null ? new SortedSet<T>.Node(theirCurrent.Item, theirCurrent.IsRed) : null); root = myCurrent; while (theirCurrent != null) { theirStack.Push(theirCurrent); myStack.Push(myCurrent); myCurrent.Left = (theirCurrent.Left != null ? new SortedSet<T>.Node(theirCurrent.Left.Item, theirCurrent.Left.IsRed) : null); theirCurrent = theirCurrent.Left; myCurrent = myCurrent.Left; } while (theirStack.Count != 0) { theirCurrent = theirStack.Pop(); myCurrent = myStack.Pop(); Node theirRight = theirCurrent.Right; Node myRight = null; if (theirRight != null) { myRight = new SortedSet<T>.Node(theirRight.Item, theirRight.IsRed); } myCurrent.Right = myRight; while (theirRight != null) { theirStack.Push(theirRight); myStack.Push(myRight); myRight.Left = (theirRight.Left != null ? new SortedSet<T>.Node(theirRight.Left.Item, theirRight.Left.IsRed) : null); theirRight = theirRight.Left; myRight = myRight.Left; } } count = baseSortedSet.count; version = 0; } else { //As it stands, you're doing an NlogN sort of the collection List<T> els = new List<T>(collection); els.Sort(this.comparer); for (int i = 1; i < els.Count; i++) { if (comparer.Compare(els[i], els[i - 1]) == 0) { els.RemoveAt(i); i--; } } root = ConstructRootFromSortedArray(els.ToArray(), 0, els.Count - 1, null); count = els.Count; version = 0; } }#if !FEATURE_NETCORE protected SortedSet(SerializationInfo info, StreamingContext context) { siInfo = info; }#endif #endregion #region Bulk Operation Helpers private void AddAllElements(IEnumerable<T> collection) { foreach (T item in collection) { if (!this.Contains(item)) Add(item); } } private void RemoveAllElements(IEnumerable<T> collection) { T min = this.Min; T max = this.Max; foreach (T item in collection) { if (!(comparer.Compare(item, min) < 0 || comparer.Compare(item, max) > 0) && this.Contains(item)) this.Remove(item); } } private bool ContainsAllElements(IEnumerable<T> collection) { foreach (T item in collection) { if (!this.Contains(item)) { return false; } } return true; } // // Do a in order walk on tree and calls the delegate for each node. // If the action delegate returns false, stop the walk. // // Return true if the entire tree has been walked. // Otherwise returns false. // internal bool InOrderTreeWalk(TreeWalkPredicate<T> action) { return InOrderTreeWalk(action, false); } // Allows for the change in traversal direction. Reverse visits nodes in descending order internal virtual bool InOrderTreeWalk(TreeWalkPredicate<T> action, bool reverse) { if (root == null) { return true; } // The maximum height of a red-black tree is 2*lg(n+1). // See page 264 of "Introduction to algorithms" by Thomas H. Cormen // note: this should be logbase2, but since the stack grows itself, we // don't want the extra cost Stack<Node> stack = new Stack<Node>(2 * (int)(SortedSet<T>.log2(Count + 1))); Node current = root; while (current != null) { stack.Push(current); current = (reverse ? current.Right : current.Left); } while (stack.Count != 0) { current = stack.Pop(); if (!action(current)) { return false; } Node node = (reverse ? current.Left : current.Right); while (node != null) { stack.Push(node); node = (reverse ? node.Right : node.Left); } } return true; } // // Do a left to right breadth first walk on tree and // calls the delegate for each node. // If the action delegate returns false, stop the walk. // // Return true if the entire tree has been walked. // Otherwise returns false. // internal virtual bool BreadthFirstTreeWalk(TreeWalkPredicate<T> action) { if (root == null) { return true; } List<Node> processQueue = new List<Node>(); processQueue.Add(root); Node current; while (processQueue.Count != 0) { current = processQueue[0]; processQueue.RemoveAt(0); if (!action(current)) { return false; } if (current.Left != null) { processQueue.Add(current.Left); } if (current.Right != null) { processQueue.Add(current.Right); } } return true; } #endregion #region Properties public int Count { get { VersionCheck(); return count; } } public IComparer<T> Comparer { get { return comparer; } } bool ICollection<T>.IsReadOnly { get { return false; } } bool ICollection.IsSynchronized { get { return false; } } object ICollection.SyncRoot { get { if (_syncRoot == null) { System.Threading.Interlocked.CompareExchange(ref _syncRoot, new Object(), null); } return _syncRoot; } } #endregion #region Subclass helpers //virtual function for subclass that needs to update count internal virtual void VersionCheck() { } //virtual function for subclass that needs to do range checks internal virtual bool IsWithinRange(T item) { return true; } #endregion #region ICollection<T> Members /// <summary> /// Add the value ITEM to the tree, returns true if added, false if duplicate /// </summary> /// <param name="item">item to be added</param> public bool Add(T item) { return AddIfNotPresent(item); } void ICollection<T>.Add(T item) { AddIfNotPresent(item); } /// <summary> /// Adds ITEM to the tree if not already present. Returns TRUE if value was successfully added /// or FALSE if it is a duplicate /// </summary> internal virtual bool AddIfNotPresent(T item) { if (root == null) { // empty tree root = new Node(item, false); count = 1; version++; return true; } // // Search for a node at bottom to insert the new node. // If we can guanratee the node we found is not a 4-node, it would be easy to do insertion. // We split 4-nodes along the search path. // Node current = root; Node parent = null; Node grandParent = null; Node greatGrandParent = null; //even if we don't actually add to the set, we may be altering its structure (by doing rotations //and such). so update version to disable any enumerators/subsets working on it version++; int order = 0; while (current != null) { order = comparer.Compare(item, current.Item); if (order == 0) { // We could have changed root node to red during the search process. // We need to set it to black before we return. root.IsRed = false; return false; } // split a 4-node into two 2-nodes if (Is4Node(current)) { Split4Node(current); // We could have introduced two consecutive red nodes after split. Fix that by rotation. if (IsRed(parent)) { InsertionBalance(current, ref parent, grandParent, greatGrandParent); } } greatGrandParent = grandParent; grandParent = parent; parent = current; current = (order < 0) ? current.Left : current.Right; } Debug.Assert(parent != null, "Parent node cannot be null here!"); // ready to insert the new node Node node = new Node(item); if (order > 0) { parent.Right = node; } else { parent.Left = node; } // the new node will be red, so we will need to adjust the colors if parent node is also red if (parent.IsRed) { InsertionBalance(node, ref parent, grandParent, greatGrandParent); } // Root node is always black root.IsRed = false; ++count; return true; } /// <summary> /// Remove the T ITEM from this SortedSet. Returns true if successfully removed. /// </summary> /// <param name="item"></param> /// <returns></returns> public bool Remove(T item) { return this.DoRemove(item); // hack so it can be made non-virtual } internal virtual bool DoRemove(T item) { if (root == null) { return false; } // Search for a node and then find its succesor. // Then copy the item from the succesor to the matching node and delete the successor. // If a node doesn't have a successor, we can replace it with its left child (if not empty.) // or delete the matching node. // // In top-down implementation, it is important to make sure the node to be deleted is not a 2-node. // Following code will make sure the node on the path is not a 2 Node. //even if we don't actually remove from the set, we may be altering its structure (by doing rotations //and such). so update version to disable any enumerators/subsets working on it version++; Node current = root; Node parent = null; Node grandParent = null; Node match = null; Node parentOfMatch = null; bool foundMatch = false; while (current != null) { if (Is2Node(current)) { // fix up 2-Node if (parent == null) { // current is root. Mark it as red current.IsRed = true; } else { Node sibling = GetSibling(current, parent); if (sibling.IsRed) { // If parent is a 3-node, flip the orientation of the red link. // We can acheive this by a single rotation // This case is converted to one of other cased below. Debug.Assert(!parent.IsRed, "parent must be a black node!"); if (parent.Right == sibling) { RotateLeft(parent); } else { RotateRight(parent); } parent.IsRed = true; sibling.IsRed = false; // parent's color // sibling becomes child of grandParent or root after rotation. Update link from grandParent or root ReplaceChildOfNodeOrRoot(grandParent, parent, sibling); // sibling will become grandParent of current node grandParent = sibling; if (parent == match) { parentOfMatch = sibling; } // update sibling, this is necessary for following processing sibling = (parent.Left == current) ? parent.Right : parent.Left; } Debug.Assert(sibling != null || sibling.IsRed == false, "sibling must not be null and it must be black!"); if (Is2Node(sibling)) { Merge2Nodes(parent, current, sibling); } else { // current is a 2-node and sibling is either a 3-node or a 4-node. // We can change the color of current to red by some rotation. TreeRotation rotation = RotationNeeded(parent, current, sibling); Node newGrandParent = null; switch (rotation) { case TreeRotation.RightRotation: Debug.Assert(parent.Left == sibling, "sibling must be left child of parent!"); Debug.Assert(sibling.Left.IsRed, "Left child of sibling must be red!"); sibling.Left.IsRed = false; newGrandParent = RotateRight(parent); break; case TreeRotation.LeftRotation: Debug.Assert(parent.Right == sibling, "sibling must be left child of parent!"); Debug.Assert(sibling.Right.IsRed, "Right child of sibling must be red!"); sibling.Right.IsRed = false; newGrandParent = RotateLeft(parent); break; case TreeRotation.RightLeftRotation: Debug.Assert(parent.Right == sibling, "sibling must be left child of parent!"); Debug.Assert(sibling.Left.IsRed, "Left child of sibling must be red!"); newGrandParent = RotateRightLeft(parent); break; case TreeRotation.LeftRightRotation: Debug.Assert(parent.Left == sibling, "sibling must be left child of parent!"); Debug.Assert(sibling.Right.IsRed, "Right child of sibling must be red!"); newGrandParent = RotateLeftRight(parent); break; } newGrandParent.IsRed = parent.IsRed; parent.IsRed = false; current.IsRed = true; ReplaceChildOfNodeOrRoot(grandParent, parent, newGrandParent); if (parent == match) { parentOfMatch = newGrandParent; } grandParent = newGrandParent; } } } // we don't need to compare any more once we found the match int order = foundMatch ? -1 : comparer.Compare(item, current.Item); if (order == 0) { // save the matching node foundMatch = true; match = current; parentOfMatch = parent; } grandParent = parent; parent = current; if (order < 0) { current = current.Left; } else { current = current.Right; // continue the search in right sub tree after we find a match } } // move successor to the matching node position and replace links if (match != null) { ReplaceNode(match, parentOfMatch, parent, grandParent); --count; } if (root != null) { root.IsRed = false; } return foundMatch; } public virtual void Clear() { root = null; count = 0; ++version; } public virtual bool Contains(T item) { return FindNode(item) != null; } public void CopyTo(T[] array) { CopyTo(array, 0, Count); } public void CopyTo(T[] array, int index) { CopyTo(array, index, Count); } public void CopyTo(T[] array, int index, int count) { if (array == null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.array); } if (index < 0) { ThrowHelper.ThrowArgumentOutOfRangeException(ExceptionArgument.index); } if (count < 0) { throw new ArgumentOutOfRangeException("count", SR.GetString(SR.ArgumentOutOfRange_NeedNonNegNum)); } // will array, starting at arrayIndex, be able to hold elements? Note: not // checking arrayIndex >= array.Length (consistency with list of allowing // count of 0; subsequent check takes care of the rest) if (index > array.Length || count > array.Length - index) { throw new ArgumentException(SR.GetString(SR.Arg_ArrayPlusOffTooSmall)); } //upper bound count += index; InOrderTreeWalk(delegate(Node node) { if (index >= count) { return false; } else { array[index++] = node.Item; return true; } }); } void ICollection.CopyTo(Array array, int index) { if (array == null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.array); } if (array.Rank != 1) { ThrowHelper.ThrowArgumentException(ExceptionResource.Arg_RankMultiDimNotSupported); } if (array.GetLowerBound(0) != 0) { ThrowHelper.ThrowArgumentException(ExceptionResource.Arg_NonZeroLowerBound); } if (index < 0) { ThrowHelper.ThrowArgumentOutOfRangeException(ExceptionArgument.arrayIndex, ExceptionResource.ArgumentOutOfRange_NeedNonNegNum); } if (array.Length - index < Count) { ThrowHelper.ThrowArgumentException(ExceptionResource.Arg_ArrayPlusOffTooSmall); } T[] tarray = array as T[]; if (tarray != null) { CopyTo(tarray, index); } else { object[] objects = array as object[]; if (objects == null) { ThrowHelper.ThrowArgumentException(ExceptionResource.Argument_InvalidArrayType); } try { InOrderTreeWalk(delegate(Node node) { objects[index++] = node.Item; return true; }); } catch (ArrayTypeMismatchException) { ThrowHelper.ThrowArgumentException(ExceptionResource.Argument_InvalidArrayType); } } } #endregion #region IEnumerable<T> members public Enumerator GetEnumerator() { return new Enumerator(this); } IEnumerator<T> IEnumerable<T>.GetEnumerator() { return new Enumerator(this); } IEnumerator IEnumerable.GetEnumerator() { return new Enumerator(this); } #endregion #region Tree Specific Operations private static Node GetSibling(Node node, Node parent) { if (parent.Left == node) { return parent.Right; } return parent.Left; } // After calling InsertionBalance, we need to make sure current and parent up-to-date. // It doesn't matter if we keep grandParent and greatGrantParent up-to-date // because we won't need to split again in the next node. // By the time we need to split again, everything will be correctly set. // private void InsertionBalance(Node current, ref Node parent, Node grandParent, Node greatGrandParent) { Debug.Assert(grandParent != null, "Grand parent cannot be null here!"); bool parentIsOnRight = (grandParent.Right == parent); bool currentIsOnRight = (parent.Right == current); Node newChildOfGreatGrandParent; if (parentIsOnRight == currentIsOnRight) { // same orientation, single rotation newChildOfGreatGrandParent = currentIsOnRight ? RotateLeft(grandParent) : RotateRight(grandParent); } else { // different orientaton, double rotation newChildOfGreatGrandParent = currentIsOnRight ? RotateLeftRight(grandParent) : RotateRightLeft(grandParent); // current node now becomes the child of greatgrandparent parent = greatGrandParent; } // grand parent will become a child of either parent of current. grandParent.IsRed = true; newChildOfGreatGrandParent.IsRed = false; ReplaceChildOfNodeOrRoot(greatGrandParent, grandParent, newChildOfGreatGrandParent); } private static bool Is2Node(Node node) { Debug.Assert(node != null, "node cannot be null!"); return IsBlack(node) && IsNullOrBlack(node.Left) && IsNullOrBlack(node.Right); } private static bool Is4Node(Node node) { return IsRed(node.Left) && IsRed(node.Right); } private static bool IsBlack(Node node) { return (node != null && !node.IsRed); } private static bool IsNullOrBlack(Node node) { return (node == null || !node.IsRed); } private static bool IsRed(Node node) { return (node != null && node.IsRed); } private static void Merge2Nodes(Node parent, Node child1, Node child2) { Debug.Assert(IsRed(parent), "parent must be be red"); // combing two 2-nodes into a 4-node parent.IsRed = false; child1.IsRed = true; child2.IsRed = true; } // Replace the child of a parent node. // If the parent node is null, replace the root. private void ReplaceChildOfNodeOrRoot(Node parent, Node child, Node newChild) { if (parent != null) { if (parent.Left == child) { parent.Left = newChild; } else { parent.Right = newChild; } } else { root = newChild; } } // Replace the matching node with its succesor. private void ReplaceNode(Node match, Node parentOfMatch, Node succesor, Node parentOfSuccesor) { if (succesor == match) { // this node has no successor, should only happen if right child of matching node is null. Debug.Assert(match.Right == null, "Right child must be null!"); succesor = match.Left; } else { Debug.Assert(parentOfSuccesor != null, "parent of successor cannot be null!"); Debug.Assert(succesor.Left == null, "Left child of succesor must be null!"); Debug.Assert((succesor.Right == null && succesor.IsRed) || (succesor.Right.IsRed && !succesor.IsRed), "Succesor must be in valid state"); if (succesor.Right != null) { succesor.Right.IsRed = false; } if (parentOfSuccesor != match) { // detach succesor from its parent and set its right child parentOfSuccesor.Left = succesor.Right; succesor.Right = match.Right; } succesor.Left = match.Left; } if (succesor != null) { succesor.IsRed = match.IsRed; } ReplaceChildOfNodeOrRoot(parentOfMatch, match, succesor); } internal virtual Node FindNode(T item) { Node current = root; while (current != null) { int order = comparer.Compare(item, current.Item); if (order == 0) { return current; } else { current = (order < 0) ? current.Left : current.Right; } } return null; } //used for bithelpers. Note that this implementation is completely different //from the Subset's. The two should not be mixed. This indexes as if the tree were an array. //http://en.wikipedia.org/wiki/Binary_Tree#Methods_for_storing_binary_trees internal virtual int InternalIndexOf(T item) { Node current = root; int count = 0; while (current != null) { int order = comparer.Compare(item, current.Item); if (order == 0) { return count; } else { current = (order < 0) ? current.Left : current.Right; count = (order < 0) ? (2 * count + 1) : (2 * count + 2); } } return -1; } internal Node FindRange(T from, T to) { return FindRange(from, to, true, true); } internal Node FindRange(T from, T to, bool lowerBoundActive, bool upperBoundActive) { Node current = root; while (current != null) { if (lowerBoundActive && comparer.Compare(from, current.Item) > 0) { current = current.Right; } else { if (upperBoundActive && comparer.Compare(to, current.Item) < 0) { current = current.Left; } else { return current; } } } return null; } internal void UpdateVersion() { ++version; } private static Node RotateLeft(Node node) { Node x = node.Right; node.Right = x.Left; x.Left = node; return x; } private static Node RotateLeftRight(Node node) { Node child = node.Left; Node grandChild = child.Right; node.Left = grandChild.Right; grandChild.Right = node; child.Right = grandChild.Left; grandChild.Left = child; return grandChild; } private static Node RotateRight(Node node) { Node x = node.Left; node.Left = x.Right; x.Right = node; return x; } private static Node RotateRightLeft(Node node) { Node child = node.Right; Node grandChild = child.Left; node.Right = grandChild.Left; grandChild.Left = node; child.Left = grandChild.Right; grandChild.Right = child; return grandChild; } /// <summary> /// Testing counter that can track rotations /// </summary> private static TreeRotation RotationNeeded(Node parent, Node current, Node sibling) { Debug.Assert(IsRed(sibling.Left) || IsRed(sibling.Right), "sibling must have at least one red child"); if (IsRed(sibling.Left)) { if (parent.Left == current) { return TreeRotation.RightLeftRotation; } return TreeRotation.RightRotation; } else { if (parent.Left == current) { return TreeRotation.LeftRotation; } return TreeRotation.LeftRightRotation; } } /// <summary> /// Used for deep equality of SortedSet testing /// </summary> /// <returns></returns> public static IEqualityComparer<SortedSet<T>> CreateSetComparer() { return new SortedSetEqualityComparer<T>(); } /// <summary> /// Create a new set comparer for this set, where this set's members' equality is defined by the /// memberEqualityComparer. Note that this equality comparer's definition of equality must be the /// same as this set's Comparer's definition of equality /// </summary> public static IEqualityComparer<SortedSet<T>> CreateSetComparer(IEqualityComparer<T> memberEqualityComparer) { return new SortedSetEqualityComparer<T>(memberEqualityComparer); } /// <summary> /// Decides whether these sets are the same, given the comparer. If the EC's are the same, we can /// just use SetEquals, but if they aren't then we have to manually check with the given comparer /// </summary> internal static bool SortedSetEquals(SortedSet<T> set1, SortedSet<T> set2, IComparer<T> comparer) { // handle null cases first if (set1 == null) { return (set2 == null); } else if (set2 == null) { // set1 != null return false; } if (AreComparersEqual(set1, set2)) { if (set1.Count != set2.Count) return false; return set1.SetEquals(set2); } else { bool found = false; foreach (T item1 in set1) { found = false; foreach (T item2 in set2) { if (comparer.Compare(item1, item2) == 0) { found = true; break; } } if (!found) return false; } return true; } } //This is a little frustrating because we can't support more sorted structures private static bool AreComparersEqual(SortedSet<T> set1, SortedSet<T> set2) { return set1.Comparer.Equals(set2.Comparer); } private static void Split4Node(Node node) { node.IsRed = true; node.Left.IsRed = false; node.Right.IsRed = false; } /// <summary> /// Copies this to an array. Used for DebugView /// </summary> /// <returns></returns> internal T[] ToArray() { T[] newArray = new T[Count]; CopyTo(newArray); return newArray; } #endregion #region ISet Members /// <summary> /// Transform this set into its union with the IEnumerable OTHER ///Attempts to insert each element and rejects it if it exists. /// NOTE: The caller object is important as UnionWith uses the Comparator ///associated with THIS to check equality /// Throws ArgumentNullException if OTHER is null /// </summary> /// <param name="other"></param> public void UnionWith(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } SortedSet<T> s = other as SortedSet<T>; TreeSubSet t = this as TreeSubSet; if (t != null) VersionCheck(); if (s != null && t == null && this.count == 0) { SortedSet<T> dummy = new SortedSet<T>(s, this.comparer); this.root = dummy.root; this.count = dummy.count; this.version++; return; } if (s != null && t == null && AreComparersEqual(this, s) && (s.Count > this.Count / 2)) { //this actually hurts if N is much greater than M the /2 is arbitrary //first do a merge sort to an array. T[] merged = new T[s.Count + this.Count]; int c = 0; Enumerator mine = this.GetEnumerator(); Enumerator theirs = s.GetEnumerator(); bool mineEnded = !mine.MoveNext(), theirsEnded = !theirs.MoveNext(); while (!mineEnded && !theirsEnded) { int comp = Comparer.Compare(mine.Current, theirs.Current); if (comp < 0) { merged[c++] = mine.Current; mineEnded = !mine.MoveNext(); } else if (comp == 0) { merged[c++] = theirs.Current; mineEnded = !mine.MoveNext(); theirsEnded = !theirs.MoveNext(); } else { merged[c++] = theirs.Current; theirsEnded = !theirs.MoveNext(); } } if (!mineEnded || !theirsEnded) { Enumerator remaining = (mineEnded ? theirs : mine); do { merged[c++] = remaining.Current; } while (remaining.MoveNext()); } //now merged has all c elements //safe to gc the root, we have all the elements root = null; root = SortedSet<T>.ConstructRootFromSortedArray(merged, 0, c - 1, null); count = c; version++; } else { AddAllElements(other); } } private static Node ConstructRootFromSortedArray(T[] arr, int startIndex, int endIndex, Node redNode) { //what does this do? //you're given a sorted array... say 1 2 3 4 5 6 //2 cases: // If there are odd # of elements, pick the middle element (in this case 4), and compute // its left and right branches // If there are even # of elements, pick the left middle element, save the right middle element // and call the function on the rest // 1 2 3 4 5 6 -> pick 3, save 4 and call the fn on 1,2 and 5,6 // now add 4 as a red node to the lowest element on the right branch // 3 3 // 1 5 -> 1 5 // 2 6 2 4 6 // As we're adding to the leftmost of the right branch, nesting will not hurt the red-black properties // Leaf nodes are red if they have no sibling (if there are 2 nodes or if a node trickles // down to the bottom //the iterative way to do this ends up wasting more space than it saves in stack frames (at //least in what i tried) //so we're doing this recursively //base cases are described below int size = endIndex - startIndex + 1; if (size == 0) { return null; } Node root = null; if (size == 1) { root = new Node(arr[startIndex], false); if (redNode != null) { root.Left = redNode; } } else if (size == 2) { root = new Node(arr[startIndex], false); root.Right = new Node(arr[endIndex], false); root.Right.IsRed = true; if (redNode != null) { root.Left = redNode; } } else if (size == 3) { root = new Node(arr[startIndex + 1], false); root.Left = new Node(arr[startIndex], false); root.Right = new Node(arr[endIndex], false); if (redNode != null) { root.Left.Left = redNode; } } else { int midpt = ((startIndex + endIndex) / 2); root = new Node(arr[midpt], false); root.Left = ConstructRootFromSortedArray(arr, startIndex, midpt - 1, redNode); if (size % 2 == 0) { root.Right = ConstructRootFromSortedArray(arr, midpt + 2, endIndex, new Node(arr[midpt + 1], true)); } else { root.Right = ConstructRootFromSortedArray(arr, midpt + 1, endIndex, null); } } return root; } /// <summary> /// Transform this set into its intersection with the IEnumerable OTHER /// NOTE: The caller object is important as IntersectionWith uses the /// comparator associated with THIS to check equality /// Throws ArgumentNullException if OTHER is null /// </summary> /// <param name="other"></param> public virtual void IntersectWith(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (Count == 0) return; //HashSet<T> optimizations can't be done until equality comparers and comparers are related //Technically, this would work as well with an ISorted<T> SortedSet<T> s = other as SortedSet<T>; TreeSubSet t = this as TreeSubSet; if (t != null) VersionCheck(); //only let this happen if i am also a SortedSet, not a SubSet if (s != null && t == null && AreComparersEqual(this, s)) { //first do a merge sort to an array. T[] merged = new T[this.Count]; int c = 0; Enumerator mine = this.GetEnumerator(); Enumerator theirs = s.GetEnumerator(); bool mineEnded = !mine.MoveNext(), theirsEnded = !theirs.MoveNext(); T max = Max; T min = Min; while (!mineEnded && !theirsEnded && Comparer.Compare(theirs.Current, max) <= 0) { int comp = Comparer.Compare(mine.Current, theirs.Current); if (comp < 0) { mineEnded = !mine.MoveNext(); } else if (comp == 0) { merged[c++] = theirs.Current; mineEnded = !mine.MoveNext(); theirsEnded = !theirs.MoveNext(); } else { theirsEnded = !theirs.MoveNext(); } } //now merged has all c elements //safe to gc the root, we have all the elements root = null; root = SortedSet<T>.ConstructRootFromSortedArray(merged, 0, c - 1, null); count = c; version++; } else { IntersectWithEnumerable(other); } } internal virtual void IntersectWithEnumerable(IEnumerable<T> other) { // List<T> toSave = new List<T>(this.Count); foreach (T item in other) { if (this.Contains(item)) { toSave.Add(item); this.Remove(item); } } this.Clear(); AddAllElements(toSave); } /// <summary> /// Transform this set into its complement with the IEnumerable OTHER /// NOTE: The caller object is important as ExceptWith uses the /// comparator associated with THIS to check equality /// Throws ArgumentNullException if OTHER is null /// </summary> /// <param name="other"></param> public void ExceptWith(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (count == 0) return; if (other == this) { this.Clear(); return; } SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted)) { //outside range, no point doing anything if (!(comparer.Compare(asSorted.Max, this.Min) < 0 || comparer.Compare(asSorted.Min, this.Max) > 0)) { T min = this.Min; T max = this.Max; foreach (T item in other) { if (comparer.Compare(item, min) < 0) continue; if (comparer.Compare(item, max) > 0) break; Remove(item); } } } else { RemoveAllElements(other); } } /// <summary> /// Transform this set so it contains elements in THIS or OTHER but not both /// NOTE: The caller object is important as SymmetricExceptWith uses the /// comparator associated with THIS to check equality /// Throws ArgumentNullException if OTHER is null /// </summary> /// <param name="other"></param> public void SymmetricExceptWith(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (this.Count == 0) { this.UnionWith(other); return; } if (other == this) { this.Clear(); return; } SortedSet<T> asSorted = other as SortedSet<T>;#if USING_HASH_SET HashSet<T> asHash = other as HashSet<T>;#endif if (asSorted != null && AreComparersEqual(this, asSorted)) { SymmetricExceptWithSameEC(asSorted); }#if USING_HASH_SET else if (asHash != null && this.comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { SymmetricExceptWithSameEC(asHash); }#endif else { //need perf improvement on this T[] elements = (new List<T>(other)).ToArray(); Array.Sort(elements, this.Comparer); SymmetricExceptWithSameEC(elements); } } //OTHER must be a set internal void SymmetricExceptWithSameEC(ISet<T> other) { foreach (T item in other) { //yes, it is classier to say //if (!this.Remove(item))this.Add(item); //but this ends up saving on rotations if (this.Contains(item)) { this.Remove(item); } else { this.Add(item); } } } //OTHER must be a sorted array internal void SymmetricExceptWithSameEC(T[] other) { if (other.Length == 0) { return; } T last = other[0]; for (int i = 0; i < other.Length; i++) { while (i < other.Length && i != 0 && comparer.Compare(other[i], last) == 0) i++; if (i >= other.Length) break; if (this.Contains(other[i])) { this.Remove(other[i]); } else { this.Add(other[i]); } last = other[i]; } } /// <summary> /// Checks whether this Tree is a subset of the IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> [System.Security.SecuritySafeCritical] public bool IsSubsetOf(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (Count == 0) return true; SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted)) { if (this.Count > asSorted.Count) return false; return IsSubsetOfSortedSetWithSameEC(asSorted); } else { //worst case: mark every element in my set and see if i've counted all //O(MlogN) ElementCount result = CheckUniqueAndUnfoundElements(other, false); return (result.uniqueCount == Count && result.unfoundCount >= 0); } } private bool IsSubsetOfSortedSetWithSameEC(SortedSet<T> asSorted) { SortedSet<T> prunedOther = asSorted.GetViewBetween(this.Min, this.Max); foreach (T item in this) { if (!prunedOther.Contains(item)) return false; } return true; } /// <summary> /// Checks whether this Tree is a proper subset of the IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> [System.Security.SecuritySafeCritical] public bool IsProperSubsetOf(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if ((other as ICollection) != null) { if (Count == 0) return (other as ICollection).Count > 0; }#if USING_HASH_SET //do it one way for HashSets HashSet<T> asHash = other as HashSet<T>; if (asHash != null && comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { return asHash.IsProperSupersetOf(this); }#endif //another for sorted sets with the same comparer SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted)) { if (this.Count >= asSorted.Count) return false; return IsSubsetOfSortedSetWithSameEC(asSorted); } //worst case: mark every element in my set and see if i've counted all //O(MlogN). ElementCount result = CheckUniqueAndUnfoundElements(other, false); return (result.uniqueCount == Count && result.unfoundCount > 0); } /// <summary> /// Checks whether this Tree is a super set of the IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> public bool IsSupersetOf(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if ((other as ICollection) != null && (other as ICollection).Count == 0) return true; //do it one way for HashSets#if USING_HASH_SET HashSet<T> asHash = other as HashSet<T>; if (asHash != null && comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { return asHash.IsSubsetOf(this); }#endif //another for sorted sets with the same comparer SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted)) { if (this.Count < asSorted.Count) return false; SortedSet<T> pruned = GetViewBetween(asSorted.Min, asSorted.Max); foreach (T item in asSorted) { if (!pruned.Contains(item)) return false; } return true; } //and a third for everything else return ContainsAllElements(other); } /// <summary> /// Checks whether this Tree is a proper super set of the IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> [System.Security.SecuritySafeCritical] public bool IsProperSupersetOf(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (Count == 0) return false; if ((other as ICollection) != null && (other as ICollection).Count == 0) return true;#if USING_HASH_SET //do it one way for HashSets HashSet<T> asHash = other as HashSet<T>; if (asHash != null && comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { return asHash.IsProperSubsetOf(this); }#endif //another way for sorted sets SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(asSorted, this)) { if (asSorted.Count >= this.Count) return false; SortedSet<T> pruned = GetViewBetween(asSorted.Min, asSorted.Max); foreach (T item in asSorted) { if (!pruned.Contains(item)) return false; } return true; } //worst case: mark every element in my set and see if i've counted all //O(MlogN) //slight optimization, put it into a HashSet and then check can do it in O(N+M) //but slower in better cases + wastes space ElementCount result = CheckUniqueAndUnfoundElements(other, true); return (result.uniqueCount < Count && result.unfoundCount == 0); } /// <summary> /// Checks whether this Tree has all elements in common with IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> [System.Security.SecuritySafeCritical] public bool SetEquals(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); }#if USING_HASH_SET HashSet<T> asHash = other as HashSet<T>; if (asHash != null && comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { return asHash.SetEquals(this); }#endif SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted)) { IEnumerator<T> mine = this.GetEnumerator(); IEnumerator<T> theirs = asSorted.GetEnumerator(); bool mineEnded = !mine.MoveNext(); bool theirsEnded = !theirs.MoveNext(); while (!mineEnded && !theirsEnded) { if (Comparer.Compare(mine.Current, theirs.Current) != 0) { return false; } mineEnded = !mine.MoveNext(); theirsEnded = !theirs.MoveNext(); } return mineEnded && theirsEnded; } //worst case: mark every element in my set and see if i've counted all //O(N) by size of other ElementCount result = CheckUniqueAndUnfoundElements(other, true); return (result.uniqueCount == Count && result.unfoundCount == 0); } /// <summary> /// Checks whether this Tree has any elements in common with IEnumerable other /// </summary> /// <param name="other"></param> /// <returns></returns> public bool Overlaps(IEnumerable<T> other) { if (other == null) { throw new ArgumentNullException("other"); } if (this.Count == 0) return false; if ((other as ICollection<T> != null) && (other as ICollection<T>).Count == 0) return false; SortedSet<T> asSorted = other as SortedSet<T>; if (asSorted != null && AreComparersEqual(this, asSorted) && (comparer.Compare(Min, asSorted.Max) > 0 || comparer.Compare(Max, asSorted.Min) < 0)) { return false; }#if USING_HASH_SET HashSet<T> asHash = other as HashSet<T>; if (asHash != null && comparer.Equals(Comparer<T>.Default) && asHash.Comparer.Equals(EqualityComparer<T>.Default)) { return asHash.Overlaps(this); }#endif foreach (T item in other) { if (this.Contains(item)) { return true; } } return false; } /// <summary> /// This works similar to HashSet's CheckUniqueAndUnfound (description below), except that the bit /// array maps differently than in the HashSet. We can only use this for the bulk boolean checks. /// /// Determines counts that can be used to determine equality, subset, and superset. This /// is only used when other is an IEnumerable and not a HashSet. If other is a HashSet /// these properties can be checked faster without use of marking because we can assume /// other has no duplicates. /// /// The following count checks are performed by callers: /// 1. Equals: checks if unfoundCount = 0 and uniqueFoundCount = Count; i.e. everything /// in other is in this and everything in this is in other /// 2. Subset: checks if unfoundCount >= 0 and uniqueFoundCount = Count; i.e. other may /// have elements not in this and everything in this is in other /// 3. Proper subset: checks if unfoundCount > 0 and uniqueFoundCount = Count; i.e /// other must have at least one element not in this and everything in this is in other /// 4. Proper superset: checks if unfound count = 0 and uniqueFoundCount strictly less /// than Count; i.e. everything in other was in this and this had at least one element /// not contained in other. /// /// An earlier implementation used delegates to perform these checks rather than returning /// an ElementCount struct; however this was changed due to the perf overhead of delegates. /// </summary> /// <param name="other"></param> /// <param name="returnIfUnfound">Allows us to finish faster for equals and proper superset /// because unfoundCount must be 0.</param> /// <returns></returns> // <SecurityKernel Critical="True" Ring="0"> // <UsesUnsafeCode Name="Local bitArrayPtr of type: Int32*" /> // <ReferencesCritical Name="Method: BitHelper..ctor(System.Int32*,System.Int32)" Ring="1" /> // <ReferencesCritical Name="Method: BitHelper.IsMarked(System.Int32):System.Boolean" Ring="1" /> // <ReferencesCritical Name="Method: BitHelper.MarkBit(System.Int32):System.Void" Ring="1" /> // </SecurityKernel> [System.Security.SecurityCritical] private unsafe ElementCount CheckUniqueAndUnfoundElements(IEnumerable<T> other, bool returnIfUnfound) { ElementCount result; // need special case in case this has no elements. if (Count == 0) { int numElementsInOther = 0; foreach (T item in other) { numElementsInOther++; // break right away, all we want to know is whether other has 0 or 1 elements break; } result.uniqueCount = 0; result.unfoundCount = numElementsInOther; return result; } int originalLastIndex = Count; int intArrayLength = BitHelper.ToIntArrayLength(originalLastIndex); BitHelper bitHelper; if (intArrayLength <= StackAllocThreshold) { int* bitArrayPtr = stackalloc int[intArrayLength]; bitHelper = new BitHelper(bitArrayPtr, intArrayLength); } else { int[] bitArray = new int[intArrayLength]; bitHelper = new BitHelper(bitArray, intArrayLength); } // count of items in other not found in this int unfoundCount = 0; // count of unique items in other found in this int uniqueFoundCount = 0; foreach (T item in other) { int index = InternalIndexOf(item); if (index >= 0) { if (!bitHelper.IsMarked(index)) { // item hasn't been seen yet bitHelper.MarkBit(index); uniqueFoundCount++; } } else { unfoundCount++; if (returnIfUnfound) { break; } } } result.uniqueCount = uniqueFoundCount; result.unfoundCount = unfoundCount; return result; } public int RemoveWhere(Predicate<T> match) { if (match == null) { throw new ArgumentNullException("match"); } List<T> matches = new List<T>(this.Count); BreadthFirstTreeWalk(delegate(Node n) { if (match(n.Item)) { matches.Add(n.Item); } return true; }); // reverse breadth first to (try to) incur low cost int actuallyRemoved = 0; for (int i = matches.Count - 1; i >= 0; i--) { if (this.Remove(matches[i])) { actuallyRemoved++; } } return actuallyRemoved; } #endregion #region ISorted Members public T Min { get { T ret = default(T); InOrderTreeWalk(delegate(SortedSet<T>.Node n) { ret = n.Item; return false; }); return ret; } } public T Max { get { T ret = default(T); InOrderTreeWalk(delegate(SortedSet<T>.Node n) { ret = n.Item; return false; }, true); return ret; } } public IEnumerable<T> Reverse() { Enumerator e = new Enumerator(this, true); while (e.MoveNext()) { yield return e.Current; } } /// <summary> /// Returns a subset of this tree ranging from values lBound to uBound /// Any changes made to the subset reflect in the actual tree /// </summary> /// <param name="lowVestalue">Lowest Value allowed in the subset</param> /// <param name="highestValue">Highest Value allowed in the subset</param> public virtual SortedSet<T> GetViewBetween(T lowerValue, T upperValue) { if (Comparer.Compare(lowerValue, upperValue) > 0) { throw new ArgumentException("lowerBound is greater than upperBound"); } return new TreeSubSet(this, lowerValue, upperValue, true, true); }#if DEBUG /// <summary> /// debug status to be checked whenever any operation is called /// </summary> /// <returns></returns> internal virtual bool versionUpToDate() { return true; }#endif /// <summary> /// This class represents a subset view into the tree. Any changes to this view /// are reflected in the actual tree. Uses the Comparator of the underlying tree. /// </summary> /// <typeparam name="T"></typeparam> #if !FEATURE_NETCORE [Serializable] internal sealed class TreeSubSet : SortedSet<T>, ISerializable, IDeserializationCallback {#else internal sealed class TreeSubSet : SortedSet<T> {#endif SortedSet<T> underlying; T min, max; //these exist for unbounded collections //for instance, you could allow this subset to be defined for i>10. The set will throw if //anything <=10 is added, but there is no upperbound. These features Head(), Tail(), were punted //in the spec, and are not available, but the framework is there to make them available at some point. bool lBoundActive, uBoundActive; //used to see if the count is out of date #if DEBUG internal override bool versionUpToDate() { return (this.version == underlying.version); }#endif public TreeSubSet(SortedSet<T> Underlying, T Min, T Max, bool lowerBoundActive, bool upperBoundActive) : base(Underlying.Comparer) { underlying = Underlying; min = Min; max = Max; lBoundActive = lowerBoundActive; uBoundActive = upperBoundActive; root = underlying.FindRange(min, max, lBoundActive, uBoundActive); // root is first element within range count = 0; version = -1; VersionCheckImpl(); }#if !FEATURE_NETCORE /// <summary> /// For serialization and deserialization /// </summary> private TreeSubSet() { comparer = null; } [SuppressMessage("Microsoft.Usage", "CA2236:CallBaseClassMethodsOnISerializableTypes", Justification = "special case TreeSubSet serialization")] private TreeSubSet(SerializationInfo info, StreamingContext context) { siInfo = info; OnDeserializationImpl(info); }#endif // !FEATURE_NETCORE /// <summary> /// Additions to this tree need to be added to the underlying tree as well /// </summary> internal override bool AddIfNotPresent(T item) { if (!IsWithinRange(item)) { ThrowHelper.ThrowArgumentOutOfRangeException(ExceptionArgument.collection); } bool ret = underlying.AddIfNotPresent(item); VersionCheck();#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif return ret; } public override bool Contains(T item) { VersionCheck();#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif return base.Contains(item); } internal override bool DoRemove(T item) { // todo: uppercase this and others if (!IsWithinRange(item)) { return false; } bool ret = underlying.Remove(item); VersionCheck();#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif return ret; } public override void Clear() { if (count == 0) { return; } List<T> toRemove = new List<T>(); BreadthFirstTreeWalk(delegate(Node n) { toRemove.Add(n.Item); return true; }); while (toRemove.Count != 0) { underlying.Remove(toRemove[toRemove.Count - 1]); toRemove.RemoveAt(toRemove.Count - 1); } root = null; count = 0; version = underlying.version; } internal override bool IsWithinRange(T item) { int comp = (lBoundActive ? Comparer.Compare(min, item) : -1); if (comp > 0) { return false; } comp = (uBoundActive ? Comparer.Compare(max, item) : 1); if (comp < 0) { return false; } return true; } internal override bool InOrderTreeWalk(TreeWalkPredicate<T> action, Boolean reverse) { VersionCheck(); if (root == null) { return true; } // The maximum height of a red-black tree is 2*lg(n+1). // See page 264 of "Introduction to algorithms" by Thomas H. Cormen Stack<Node> stack = new Stack<Node>(2 * (int)SortedSet<T>.log2(count + 1)); //this is not exactly right if count is out of date, but the stack can grow Node current = root; while (current != null) { if (IsWithinRange(current.Item)) { stack.Push(current); current = (reverse ? current.Right : current.Left); } else if (lBoundActive && Comparer.Compare(min, current.Item) > 0) { current = current.Right; } else { current = current.Left; } } while (stack.Count != 0) { current = stack.Pop(); if (!action(current)) { return false; } Node node = (reverse ? current.Left : current.Right); while (node != null) { if (IsWithinRange(node.Item)) { stack.Push(node); node = (reverse ? node.Right : node.Left); } else if (lBoundActive && Comparer.Compare(min, node.Item) > 0) { node = node.Right; } else { node = node.Left; } } } return true; } internal override bool BreadthFirstTreeWalk(TreeWalkPredicate<T> action) { VersionCheck(); if (root == null) { return true; } List<Node> processQueue = new List<Node>(); processQueue.Add(root); Node current; while (processQueue.Count != 0) { current = processQueue[0]; processQueue.RemoveAt(0); if (IsWithinRange(current.Item) && !action(current)) { return false; } if (current.Left != null && (!lBoundActive || Comparer.Compare(min, current.Item) < 0)) { processQueue.Add(current.Left); } if (current.Right != null && (!uBoundActive || Comparer.Compare(max, current.Item) > 0)) { processQueue.Add(current.Right); } } return true; } internal override SortedSet<T>.Node FindNode(T item) { if (!IsWithinRange(item)) { return null; } VersionCheck();#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif return base.FindNode(item); } //this does indexing in an inefficient way compared to the actual sortedset, but it saves a //lot of space internal override int InternalIndexOf(T item) { int count = -1; foreach (T i in this) { count++; if (Comparer.Compare(item, i) == 0) return count; }#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif return -1; } /// <summary> /// checks whether this subset is out of date. updates if necessary. /// </summary> internal override void VersionCheck() { VersionCheckImpl(); } private void VersionCheckImpl() { Debug.Assert(underlying != null, "Underlying set no longer exists"); if (this.version != underlying.version) { this.root = underlying.FindRange(min, max, lBoundActive, uBoundActive); this.version = underlying.version; count = 0; InOrderTreeWalk(delegate(Node n) { count++; return true; }); } } //This passes functionality down to the underlying tree, clipping edges if necessary //There's nothing gained by having a nested subset. May as well draw it from the base //Cannot increase the bounds of the subset, can only decrease it public override SortedSet<T> GetViewBetween(T lowerValue, T upperValue) { if (lBoundActive && Comparer.Compare(min, lowerValue) > 0) { //lBound = min; throw new ArgumentOutOfRangeException("lowerValue"); } if (uBoundActive && Comparer.Compare(max, upperValue) < 0) { //uBound = max; throw new ArgumentOutOfRangeException("upperValue"); } TreeSubSet ret = (TreeSubSet)underlying.GetViewBetween(lowerValue, upperValue); return ret; } internal override void IntersectWithEnumerable(IEnumerable<T> other) { List<T> toSave = new List<T>(this.Count); foreach (T item in other) { if (this.Contains(item)) { toSave.Add(item); this.Remove(item); } } this.Clear(); this.AddAllElements(toSave);#if DEBUG Debug.Assert(this.versionUpToDate() && this.root == this.underlying.FindRange(min, max));#endif }#if !FEATURE_NETCORE void ISerializable.GetObjectData(SerializationInfo info, StreamingContext context) { GetObjectData(info, context); } protected override void GetObjectData(SerializationInfo info, StreamingContext context) { if (info == null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.info); } info.AddValue(maxName, max, typeof(T)); info.AddValue(minName, min, typeof(T)); info.AddValue(lBoundActiveName, lBoundActive); info.AddValue(uBoundActiveName, uBoundActive); base.GetObjectData(info, context); } void IDeserializationCallback.OnDeserialization(Object sender) { //don't do anything here as its already been done by the constructor //OnDeserialization(sender); } protected override void OnDeserialization(Object sender) { OnDeserializationImpl(sender); } private void OnDeserializationImpl(Object sender) { if (siInfo == null) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_InvalidOnDeser); } comparer = (IComparer<T>)siInfo.GetValue(ComparerName, typeof(IComparer<T>)); int savedCount = siInfo.GetInt32(CountName); max = (T)siInfo.GetValue(maxName, typeof(T)); min = (T)siInfo.GetValue(minName, typeof(T)); lBoundActive = siInfo.GetBoolean(lBoundActiveName); uBoundActive = siInfo.GetBoolean(uBoundActiveName); underlying = new SortedSet<T>(); if (savedCount != 0) { T[] items = (T[])siInfo.GetValue(ItemsName, typeof(T[])); if (items == null) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_MissingValues); } for (int i = 0; i < items.Length; i++) { underlying.Add(items[i]); } } underlying.version = siInfo.GetInt32(VersionName); count = underlying.count; version = underlying.version - 1; VersionCheck(); //this should update the count to be right and update root to be right if (count != savedCount) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_MismatchedCount); } siInfo = null; }#endif // !FEATURE_NETCORE } #endregion #region Serialization methods#if !FEATURE_NETCORE // LinkDemand here is unnecessary as this is a methodimpl and linkdemand from the interface should suffice void ISerializable.GetObjectData(SerializationInfo info, StreamingContext context) { GetObjectData(info, context); } protected virtual void GetObjectData(SerializationInfo info, StreamingContext context) { if (info == null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.info); } info.AddValue(CountName, count); //This is the length of the bucket array. info.AddValue(ComparerName, comparer, typeof(IComparer<T>)); info.AddValue(VersionName, version); if (root != null) { T[] items = new T[Count]; CopyTo(items, 0); info.AddValue(ItemsName, items, typeof(T[])); } } void IDeserializationCallback.OnDeserialization(Object sender) { OnDeserialization(sender); } protected virtual void OnDeserialization(Object sender) { if (comparer != null) { return; //Somebody had a dependency on this class and fixed us up before the ObjectManager got to it. } if (siInfo == null) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_InvalidOnDeser); } comparer = (IComparer<T>)siInfo.GetValue(ComparerName, typeof(IComparer<T>)); int savedCount = siInfo.GetInt32(CountName); if (savedCount != 0) { T[] items = (T[])siInfo.GetValue(ItemsName, typeof(T[])); if (items == null) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_MissingValues); } for (int i = 0; i < items.Length; i++) { Add(items[i]); } } version = siInfo.GetInt32(VersionName); if (count != savedCount) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_MismatchedCount); } siInfo = null; }#endif //!FEATURE_NETCORE #endregion #region Helper Classes internal class Node { public bool IsRed; public T Item; public Node Left; public Node Right; public Node(T item) { // The default color will be red, we never need to create a black node directly. this.Item = item; IsRed = true; } public Node(T item, bool isRed) { // The default color will be red, we never need to create a black node directly. this.Item = item; this.IsRed = isRed; } } [SuppressMessage("Microsoft.Performance", "CA1815:OverrideEqualsAndOperatorEqualsOnValueTypes", Justification = "not an expected scenario")]#if !FEATURE_NETCORE [Serializable] public struct Enumerator : IEnumerator<T>, IEnumerator, ISerializable, IDeserializationCallback {#else public struct Enumerator : IEnumerator<T>, IEnumerator {#endif private SortedSet<T> tree; private int version; private Stack<SortedSet<T>.Node> stack; private SortedSet<T>.Node current; static SortedSet<T>.Node dummyNode = new SortedSet<T>.Node(default(T)); private bool reverse;#if !FEATURE_NETCORE private SerializationInfo siInfo;#endif internal Enumerator(SortedSet<T> set) { tree = set; //this is a hack to make sure that the underlying subset has not been changed since // tree.VersionCheck(); version = tree.version; // 2lg(n + 1) is the maximum height stack = new Stack<SortedSet<T>.Node>(2 * (int)SortedSet<T>.log2(set.Count + 1)); current = null; reverse = false;#if !FEATURE_NETCORE siInfo = null;#endif Intialize(); } internal Enumerator(SortedSet<T> set, bool reverse) { tree = set; //this is a hack to make sure that the underlying subset has not been changed since // tree.VersionCheck(); version = tree.version; // 2lg(n + 1) is the maximum height stack = new Stack<SortedSet<T>.Node>(2 * (int)SortedSet<T>.log2(set.Count + 1)); current = null; this.reverse = reverse;#if !FEATURE_NETCORE siInfo = null;#endif Intialize(); }#if !FEATURE_NETCORE private Enumerator(SerializationInfo info, StreamingContext context) { tree = null; version = -1; current = null; reverse = false; stack = null; this.siInfo = info; } void ISerializable.GetObjectData(SerializationInfo info, StreamingContext context) { GetObjectData(info, context); } private void GetObjectData(SerializationInfo info, StreamingContext context) { if (info == null) { ThrowHelper.ThrowArgumentNullException(ExceptionArgument.info); } info.AddValue(TreeName, tree, typeof(SortedSet<T>)); info.AddValue(EnumVersionName, version); info.AddValue(ReverseName, reverse); info.AddValue(EnumStartName, !NotStartedOrEnded); info.AddValue(NodeValueName, (current == null ? dummyNode.Item : current.Item), typeof(T)); } void IDeserializationCallback.OnDeserialization(Object sender) { OnDeserialization(sender); } private void OnDeserialization(Object sender) { if (siInfo == null) { ThrowHelper.ThrowSerializationException(ExceptionResource.Serialization_InvalidOnDeser); } tree = (SortedSet<T>)siInfo.GetValue(TreeName, typeof(SortedSet<T>)); version = siInfo.GetInt32(EnumVersionName); reverse = siInfo.GetBoolean(ReverseName); bool EnumStarted = siInfo.GetBoolean(EnumStartName); stack = new Stack<SortedSet<T>.Node>(2 * (int)SortedSet<T>.log2(tree.Count + 1)); current = null; if (EnumStarted) { T item = (T)siInfo.GetValue(NodeValueName, typeof(T)); Intialize(); //go until it reaches the value we want while (this.MoveNext()) { if (tree.Comparer.Compare(this.Current, item) == 0) break; } } }#endif //!FEATURE_NETCORE private void Intialize() { current = null; SortedSet<T>.Node node = tree.root; Node next = null, other = null; while (node != null) { next = (reverse ? node.Right : node.Left); other = (reverse ? node.Left : node.Right); if (tree.IsWithinRange(node.Item)) { stack.Push(node); node = next; } else if (next == null || !tree.IsWithinRange(next.Item)) { node = other; } else { node = next; } } } public bool MoveNext() { //this is a hack to make sure that the underlying subset has not been changed since // tree.VersionCheck(); if (version != tree.version) { ThrowHelper.ThrowInvalidOperationException(ExceptionResource.InvalidOperation_EnumFailedVersion); } if (stack.Count == 0) { current = null; return false; } current = stack.Pop(); SortedSet<T>.Node node = (reverse ? current.Left : current.Right); Node next = null, other = null; while (node != null) { next = (reverse ? node.Right : node.Left); other = (reverse ? node.Left : node.Right); if (tree.IsWithinRange(node.Item)) { stack.Push(node); node = next; } else if (other == null || !tree.IsWithinRange(other.Item)) { node = next; } else { node = other; } } return true; } public void Dispose() { } public T Current { get { if (current != null) { return current.Item; } return default(T); } } object IEnumerator.Current { get { if (current == null) { ThrowHelper.ThrowInvalidOperationException(ExceptionResource.InvalidOperation_EnumOpCantHappen); } return current.Item; } } internal bool NotStartedOrEnded { get { return current == null; } } internal void Reset() { if (version != tree.version) { ThrowHelper.ThrowInvalidOperationException(ExceptionResource.InvalidOperation_EnumFailedVersion); } stack.Clear(); Intialize(); } void IEnumerator.Reset() { Reset(); } } internal struct ElementCount { internal int uniqueCount; internal int unfoundCount; } #endregion #region misc // used for set checking operations (using enumerables) that rely on counting private static int log2(int value) { //Contract.Requires(value>0) int c = 0; while (value > 0) { c++; value >>= 1; } return c; } #endregion } /// <summary> /// A class that generates an IEqualityComparer for this SortedSet. Requires that the definition of /// equality defined by the IComparer for this SortedSet be consistent with the default IEqualityComparer /// for the type T. If not, such an IEqualityComparer should be provided through the constructor. /// </summary> internal class SortedSetEqualityComparer<T> : IEqualityComparer<SortedSet<T>> { private IComparer<T> comparer; private IEqualityComparer<T> e_comparer; public SortedSetEqualityComparer() : this(null, null) { } public SortedSetEqualityComparer(IComparer<T> comparer) : this(comparer, null) { } public SortedSetEqualityComparer(IEqualityComparer<T> memberEqualityComparer) : this(null, memberEqualityComparer) { } /// <summary> /// Create a new SetEqualityComparer, given a comparer for member order and another for member equality (these /// must be consistent in their definition of equality) /// </summary> public SortedSetEqualityComparer(IComparer<T> comparer, IEqualityComparer<T> memberEqualityComparer) { if (comparer == null) this.comparer = Comparer<T>.Default; else this.comparer = comparer; if (memberEqualityComparer == null) e_comparer = EqualityComparer<T>.Default; else e_comparer = memberEqualityComparer; } // using comparer to keep equals properties in tact; don't want to choose one of the comparers public bool Equals(SortedSet<T> x, SortedSet<T> y) { return SortedSet<T>.SortedSetEquals(x, y, comparer); } //IMPORTANT: this part uses the fact that GetHashCode() is consistent with the notion of equality in //the set public int GetHashCode(SortedSet<T> obj) { int hashCode = 0; if (obj != null) { foreach (T t in obj) { hashCode = hashCode ^ (e_comparer.GetHashCode(t) & 0x7FFFFFFF); } } // else returns hashcode of 0 for null HashSets return hashCode; } // Equals method for the comparer itself. public override bool Equals(Object obj) { SortedSetEqualityComparer<T> comparer = obj as SortedSetEqualityComparer<T>; if (comparer == null) { return false; } return (this.comparer == comparer.comparer); } public override int GetHashCode() { return comparer.GetHashCode() ^ e_comparer.GetHashCode(); } }}
