sclass FreeList { MultiSetMap byLength = multiSetMap_innerCustomCompactTreeSet_outerTreeMap(intRangeComparator()); CompactTreeSet inOrder = new(intRangeComparator()); int totalFree; bool changed; void add(IntRange r) { if (empty(r)) ret; // only special case here is merging with previous and/or following entry IntRange before = inOrder.floor(r), after = inOrder.ceiling(r); printVars("FreeList add", +r, +before, +after); if (before != null && before.end == r.start) { r = IntRange(before.start, r.end); printVars("FreeList merge", +r, +before); remove(before); } if (after != null && after.start == r.end) { r = IntRange(r.start, after.end); printVars("FreeList merge", +r, +after); remove(after); } internalAdd(r); } void internalAdd(IntRange r) { inOrder.add(r); byLength.add(r.length(), r); totalFree += r.length(); set changed; } void remove(IntRange r) { if (empty(r)) ret; IntRange r2 = inOrder.floor(r); if (r2 == null) fail("User error"); if (eq(r, r2)) { printVars("FreeList easy remove", r2); inOrder.remove(r2); byLength.remove(r2.length(), r2); totalFree -= r2.length(); set changed; ret; } // remove full range, add back left and right remainders printVars("FreeList remove", +r, +r2); remove(r2); add(IntRange(r2.start, r.start)); add(IntRange(r.end, r2.end)); printVars("FreeList remove done"); } IntRange findFreeSpace(int minSize) { Map.Entry> e = ((NavigableMap) byLength.data).ceilingEntry(minSize); ret e == null ? null : first(e.getValue()); } int totalFree() { ret totalFree; } void clear { byLength.clear(); inOrder.clear(); totalFree = 0; set changed; } IntRange highestFreeRange() { ret last(inOrder); } L asList() { ret cloneList(inOrder); } static FreeList fromSortedList(L l) { new FreeList fl; fOr (IntRange r : l) if (!r.isEmpty()) fl.internalAdd(r); ret fl; } int[] toIntArray() { ret intRangesToIntArray_startAndLength(inOrder); } static FreeList fromIntArray(int[] a) { ret fromSortedList(intArrayToIntRanges_startAndLength(a)); } int size() { ret l(inOrder); } L lengthStats() { ret map toIntPair(multiSetMapToMultiSetPairs(byLength)); } void sanityTest { assertEquals((long) totalFree, totalIntRangesLength(inOrder)); } }