/* * (c) the authors Licensed under the Apache License, Version 2.0. */ package com.fr.third.bitmap.roaringbitmap; import com.fr.third.bitmap.roaringbitmap.buffer.ImmutableRoaringBitmap; import com.fr.third.bitmap.roaringbitmap.buffer.MappeableContainerPointer; import com.fr.third.bitmap.roaringbitmap.buffer.MutableRoaringBitmap; import java.io.DataInput; import java.io.DataOutput; import java.io.Externalizable; import java.io.IOException; import java.io.ObjectInput; import java.io.ObjectOutput; import java.io.Serializable; import java.util.Iterator; /** * RoaringBitmap, a compressed alternative to the BitSet. *

*

 * {@code
 *      import com.fr.swift.bitmap.roaringbitmap.*;
 *
 *      //...
 *
 *      RoaringBitmap rr = RoaringBitmap.bitmapOf(1,2,3,1000);
 *      RoaringBitmap rr2 = new RoaringBitmap();
 *      for(int k = 4000; k<4255;++k) rr2.add(k);
 *      RoaringBitmap rror = RoaringBitmap.or(rr, rr2);
 *
 *      //...
 *      DataOutputStream wheretoserialize = ...
 *      rr.runOptimize(); // can help compression
 *      rr.serialize(wheretoserialize);
 * }
 * 
*

* Integers are added in unsigned sorted order. That is, they are treated as unsigned integers (see * Java 8's Integer.toUnsignedLong function). *

* Bitmaps are limited to a maximum of Integer.MAX_VALUE entries. Trying to create larger bitmaps * could result in undefined behaviors. */ public class RoaringBitmap implements Cloneable, Serializable, Iterable, Externalizable, ImmutableBitmapDataProvider { private static final long serialVersionUID = 6L; RoaringArray highLowContainer = null; /** * Create an empty bitmap */ public RoaringBitmap() { highLowContainer = new RoaringArray(); } /** * Create a RoaringBitmap from a MutableRoaringBitmap or ImmutableRoaringBitmap. The source is not * modified. * * @param rb the original bitmap */ public RoaringBitmap(ImmutableRoaringBitmap rb) { highLowContainer = new RoaringArray(); MappeableContainerPointer cp = rb.getContainerPointer(); while (cp.getContainer() != null) { highLowContainer.append(cp.key(), cp.getContainer().toContainer()); cp.advance(); } } private static void rangeSanityCheck(final long rangeStart, final long rangeEnd) { if (rangeStart < 0 || rangeStart > (1L << 32) - 1) { throw new IllegalArgumentException("rangeStart=" + rangeStart + " should be in [0, 0xffffffff]"); } if (rangeEnd > (1L << 32) || rangeEnd < 0) { throw new IllegalArgumentException("rangeEnd=" + rangeEnd + " should be in [0, 0xffffffff + 1]"); } } /** * Generate a new bitmap with all integers in [rangeStart,rangeEnd) added. * * @param rb initial bitmap (will not be modified) * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new bitmap */ public static RoaringBitmap add(RoaringBitmap rb, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return rb.clone(); // empty range } final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); RoaringBitmap answer = new RoaringBitmap(); answer.highLowContainer.appendCopiesUntil(rb.highLowContainer, (short) hbStart); if (hbStart == hbLast) { final int i = rb.highLowContainer.getIndex((short) hbStart); final Container c = i >= 0 ? rb.highLowContainer.getContainerAtIndex(i).add(lbStart, lbLast + 1) : Container.rangeOfOnes(lbStart, lbLast + 1); answer.highLowContainer.append((short) hbStart, c); answer.highLowContainer.appendCopiesAfter(rb.highLowContainer, (short) hbLast); return answer; } int ifirst = rb.highLowContainer.getIndex((short) hbStart); int ilast = rb.highLowContainer.getIndex((short) hbLast); { final Container c = ifirst >= 0 ? rb.highLowContainer.getContainerAtIndex(ifirst).add(lbStart, Util.maxLowBitAsInteger() + 1) : Container.rangeOfOnes(lbStart, Util.maxLowBitAsInteger() + 1); answer.highLowContainer.append((short) hbStart, c); } for (int hb = hbStart + 1; hb < hbLast; ++hb) { Container c = Container.rangeOfOnes(0, Util.maxLowBitAsInteger() + 1); answer.highLowContainer.append((short) hb, c); } { final Container c = ilast >= 0 ? rb.highLowContainer.getContainerAtIndex(ilast).add(0, lbLast + 1) : Container.rangeOfOnes(0, lbLast + 1); answer.highLowContainer.append((short) hbLast, c); } answer.highLowContainer.appendCopiesAfter(rb.highLowContainer, (short) hbLast); return answer; } /** * Generate a new bitmap with all integers in [rangeStart,rangeEnd) added. * * @param rb initial bitmap (will not be modified) * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new bitmap * @deprecated use the version where longs specify the range */ @Deprecated public static RoaringBitmap add(RoaringBitmap rb, final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { return add(rb, (long) rangeStart, (long) rangeEnd); } // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve return add(rb, rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } /** * Bitwise AND (intersection) operation. The provided bitmaps are *not* modified. This operation * is thread-safe as long as the provided bitmaps remain unchanged. *

* If you have more than 2 bitmaps, consider using the FastAggregation class. * * @param x1 first bitmap * @param x2 other bitmap * @return result of the operation * @see FastAggregation#and(RoaringBitmap...) */ public static RoaringBitmap and(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); int pos1 = 0, pos2 = 0; while (pos1 < length1 && pos2 < length2) { final short s1 = x1.highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); final Container c = c1.and(c2); if (c.getCardinality() > 0) { answer.highLowContainer.append(s1, c); } ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1 = x1.highLowContainer.advanceUntil(s2, pos1); } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } return answer; } /** * Cardinality of Bitwise AND (intersection) operation. The provided bitmaps are *not* modified. * This operation is thread-safe as long as the provided bitmaps remain unchanged. * * @param x1 first bitmap * @param x2 other bitmap * @return as if you did and(x2,x2).getCardinality() * @see FastAggregation#and(RoaringBitmap...) */ public static int andCardinality(final RoaringBitmap x1, final RoaringBitmap x2) { int answer = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); int pos1 = 0, pos2 = 0; while (pos1 < length1 && pos2 < length2) { final short s1 = x1.highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); // TODO: could be made faster if we did not have to materialize container answer += c1.andCardinality(c2); ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1 = x1.highLowContainer.advanceUntil(s2, pos1); } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } return answer; } /** * Bitwise ANDNOT (difference) operation. The provided bitmaps are *not* modified. This operation * is thread-safe as long as the provided bitmaps remain unchanged. * * @param x1 first bitmap * @param x2 other bitmap * @return result of the operation */ public static RoaringBitmap andNot(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); int pos1 = 0, pos2 = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); while (pos1 < length1 && pos2 < length2) { final short s1 = x1.highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); final Container c = c1.andNot(c2); if (c.getCardinality() > 0) { answer.highLowContainer.append(s1, c); } ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 final int nextPos1 = x1.highLowContainer.advanceUntil(s2, pos1); answer.highLowContainer.appendCopy(x1.highLowContainer, pos1, nextPos1); pos1 = nextPos1; } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } if (pos2 == length2) { answer.highLowContainer.appendCopy(x1.highLowContainer, pos1, length1); } return answer; } /** * Generate a bitmap with the specified values set to true. The provided integers values don't * have to be in sorted order, but it may be preferable to sort them from a performance point of * view. * * @param dat set values * @return a new bitmap */ public static RoaringBitmap bitmapOf(final int... dat) { final RoaringBitmap ans = new RoaringBitmap(); ans.add(dat); return ans; } /** * Complements the bits in the given range, from rangeStart (inclusive) rangeEnd (exclusive). The * given bitmap is unchanged. * * @param bm bitmap being negated * @param rangeStart inclusive beginning of range, in [0, 0xffffffff] * @param rangeEnd exclusive ending of range, in [0, 0xffffffff + 1] * @return a new Bitmap */ public static RoaringBitmap flip(RoaringBitmap bm, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return bm.clone(); } RoaringBitmap answer = new RoaringBitmap(); final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); // copy the containers before the active area answer.highLowContainer.appendCopiesUntil(bm.highLowContainer, (short) hbStart); for (int hb = hbStart; hb <= hbLast; ++hb) { final int containerStart = (hb == hbStart) ? lbStart : 0; final int containerLast = (hb == hbLast) ? lbLast : Util.maxLowBitAsInteger(); final int i = bm.highLowContainer.getIndex((short) hb); final int j = answer.highLowContainer.getIndex((short) hb); assert j < 0; if (i >= 0) { Container c = bm.highLowContainer.getContainerAtIndex(i).not(containerStart, containerLast + 1); if (c.getCardinality() > 0) { answer.highLowContainer.insertNewKeyValueAt(-j - 1, (short) hb, c); } } else { // *think* the range of ones must never be // empty. answer.highLowContainer.insertNewKeyValueAt(-j - 1, (short) hb, Container.rangeOfOnes(containerStart, containerLast + 1)); } } // copy the containers after the active area. answer.highLowContainer.appendCopiesAfter(bm.highLowContainer, (short) hbLast); return answer; } /** * Complements the bits in the given range, from rangeStart (inclusive) rangeEnd (exclusive). The * given bitmap is unchanged. * * @param rb bitmap being negated * @param rangeStart inclusive beginning of range, in [0, 0xffffffff] * @param rangeEnd exclusive ending of range, in [0, 0xffffffff + 1] * @return a new Bitmap * @deprecated use the version where longs specify the range */ @Deprecated public static RoaringBitmap flip(RoaringBitmap rb, final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { return flip(rb, (long) rangeStart, (long) rangeEnd); } // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve return flip(rb, rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } /** * Checks whether the two bitmaps intersect. This can be much faster than calling "and" and * checking the cardinality of the result. * * @param x1 first bitmap * @param x2 other bitmap * @return true if they intersect */ public static boolean intersects(final RoaringBitmap x1, final RoaringBitmap x2) { final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); int pos1 = 0, pos2 = 0; while (pos1 < length1 && pos2 < length2) { final short s1 = x1.highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); if (c1.intersects(c2)) { return true; } ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1 = x1.highLowContainer.advanceUntil(s2, pos1); } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } return false; } // important: inputs should not have been computed lazily protected static RoaringBitmap lazyor(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); int pos1 = 0, pos2 = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = x1.highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { answer.highLowContainer.append(s1, x1.highLowContainer.getContainerAtIndex(pos1) .lazyOR(x2.highLowContainer.getContainerAtIndex(pos2))); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 answer.highLowContainer.appendCopy(x1.highLowContainer, pos1); pos1++; if (pos1 == length1) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 answer.highLowContainer.appendCopy(x2.highLowContainer, pos2); pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { answer.highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } else if (pos2 == length2) { answer.highLowContainer.appendCopy(x1.highLowContainer, pos1, length1); } return answer; } // important: inputs should not be reused protected static RoaringBitmap lazyorfromlazyinputs(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); int pos1 = 0, pos2 = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = x1.highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); if ((c2 instanceof BitmapContainer) && (!(c1 instanceof BitmapContainer))) { Container tmp = c1; c1 = c2; c2 = tmp; } answer.highLowContainer.append(s1, c1.lazyIOR(c2)); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 Container c1 = x1.highLowContainer.getContainerAtIndex(pos1); answer.highLowContainer.append(s1, c1); pos1++; if (pos1 == length1) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); answer.highLowContainer.append(s2, c2); pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { answer.highLowContainer.append(x2.highLowContainer, pos2, length2); } else if (pos2 == length2) { answer.highLowContainer.append(x1.highLowContainer, pos1, length1); } return answer; } /** * Compute overall OR between bitmaps. *

* (Effectively calls {@link FastAggregation#or}) * * @param bitmaps input bitmaps * @return aggregated bitmap */ public static RoaringBitmap or(Iterator bitmaps) { return FastAggregation.or(bitmaps); } /** * Compute overall OR between bitmaps. *

* (Effectively calls {@link FastAggregation#or}) * * @param bitmaps input bitmaps * @return aggregated bitmap */ public static RoaringBitmap or(RoaringBitmap... bitmaps) { return FastAggregation.or(bitmaps); } /** * Bitwise OR (union) operation. The provided bitmaps are *not* modified. This operation is * thread-safe as long as the provided bitmaps remain unchanged. *

* If you have more than 2 bitmaps, consider using the FastAggregation class. * * @param x1 first bitmap * @param x2 other bitmap * @return result of the operation * @see FastAggregation#or(RoaringBitmap...) * @see FastAggregation#horizontal_or(RoaringBitmap...) */ public static RoaringBitmap or(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); int pos1 = 0, pos2 = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = x1.highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { answer.highLowContainer.append(s1, x1.highLowContainer.getContainerAtIndex(pos1) .or(x2.highLowContainer.getContainerAtIndex(pos2))); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 answer.highLowContainer.appendCopy(x1.highLowContainer, pos1); pos1++; if (pos1 == length1) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 answer.highLowContainer.appendCopy(x2.highLowContainer, pos2); pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { answer.highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } else if (pos2 == length2) { answer.highLowContainer.appendCopy(x1.highLowContainer, pos1, length1); } return answer; } /** * Cardinality of the bitwise OR (union) operation. The provided bitmaps are *not* modified. This * operation is thread-safe as long as the provided bitmaps remain unchanged. *

* If you have more than 2 bitmaps, consider using the FastAggregation class. * * @param x1 first bitmap * @param x2 other bitmap * @return cardinality of the union * @see FastAggregation#or(RoaringBitmap...) * @see FastAggregation#horizontal_or(RoaringBitmap...) */ public static int orCardinality(final RoaringBitmap x1, final RoaringBitmap x2) { // we use the fact that the cardinality of the bitmaps is known so that // the union is just the total cardinality minus the intersection return x1.getCardinality() + x2.getCardinality() - andCardinality(x1, x2); } /** * Generate a new bitmap with all integers in [rangeStart,rangeEnd) removed. * * @param rb initial bitmap (will not be modified) * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new bitmap */ public static RoaringBitmap remove(RoaringBitmap rb, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return rb.clone(); // empty range } final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); RoaringBitmap answer = new RoaringBitmap(); answer.highLowContainer.appendCopiesUntil(rb.highLowContainer, (short) hbStart); if (hbStart == hbLast) { final int i = rb.highLowContainer.getIndex((short) hbStart); if (i >= 0) { final Container c = rb.highLowContainer.getContainerAtIndex(i).remove(lbStart, lbLast + 1); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbStart, c); } } answer.highLowContainer.appendCopiesAfter(rb.highLowContainer, (short) hbLast); return answer; } int ifirst = rb.highLowContainer.getIndex((short) hbStart); int ilast = rb.highLowContainer.getIndex((short) hbLast); if ((ifirst >= 0) && (lbStart != 0)) { final Container c = rb.highLowContainer.getContainerAtIndex(ifirst).remove(lbStart, Util.maxLowBitAsInteger() + 1); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbStart, c); } } if ((ilast >= 0) && (lbLast != Util.maxLowBitAsInteger())) { final Container c = rb.highLowContainer.getContainerAtIndex(ilast).remove(0, lbLast + 1); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbLast, c); } } answer.highLowContainer.appendCopiesAfter(rb.highLowContainer, (short) hbLast); return answer; } /** * Generate a new bitmap with all integers in [rangeStart,rangeEnd) removed. * * @param rb initial bitmap (will not be modified) * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new bitmap * @deprecated use the version where longs specify the range */ @Deprecated public static RoaringBitmap remove(RoaringBitmap rb, final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { return remove(rb, (long) rangeStart, (long) rangeEnd); } // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve return remove(rb, rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } /** * Bitwise XOR (symmetric difference) operation. The provided bitmaps are *not* modified. This * operation is thread-safe as long as the provided bitmaps remain unchanged. *

* If you have more than 2 bitmaps, consider using the FastAggregation class. * * @param x1 first bitmap * @param x2 other bitmap * @return result of the operation * @see FastAggregation#xor(RoaringBitmap...) * @see FastAggregation#horizontal_xor(RoaringBitmap...) */ public static RoaringBitmap xor(final RoaringBitmap x1, final RoaringBitmap x2) { final RoaringBitmap answer = new RoaringBitmap(); int pos1 = 0, pos2 = 0; final int length1 = x1.highLowContainer.size(), length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = x1.highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { final Container c = x1.highLowContainer.getContainerAtIndex(pos1) .xor(x2.highLowContainer.getContainerAtIndex(pos2)); if (c.getCardinality() > 0) { answer.highLowContainer.append(s1, c); } pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 answer.highLowContainer.appendCopy(x1.highLowContainer, pos1); pos1++; if (pos1 == length1) { break main; } s1 = x1.highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 answer.highLowContainer.appendCopy(x2.highLowContainer, pos2); pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { answer.highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } else if (pos2 == length2) { answer.highLowContainer.appendCopy(x1.highLowContainer, pos1, length1); } return answer; } /** * Computes AND between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bitmap */ public static RoaringBitmap and(@SuppressWarnings("rawtypes") final Iterator bitmaps, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); Iterator bitmapsIterator; bitmapsIterator = selectRangeWithoutCopy(bitmaps, rangeStart, rangeEnd); return FastAggregation.and(bitmapsIterator); } /** * Computes AND between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bitmap * @deprecated use the version where longs specify the range. Negative range end are illegal. */ @Deprecated public static RoaringBitmap and(@SuppressWarnings("rawtypes") final Iterator bitmaps, final int rangeStart, final int rangeEnd) { return and(bitmaps, (long) rangeStart, (long) rangeEnd); } /** * Bitwise ANDNOT (difference) operation for the given range, rangeStart (inclusive) and rangeEnd * (exclusive). The provided bitmaps are *not* modified. This operation is thread-safe as long as * the provided bitmaps remain unchanged. * * @param x1 first bitmap * @param x2 other bitmap * @param rangeStart starting point of the range (inclusive) * @param rangeEnd end point of the range (exclusive) * @return result of the operation */ public static RoaringBitmap andNot(final RoaringBitmap x1, final RoaringBitmap x2, long rangeStart, long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); RoaringBitmap rb1 = selectRangeWithoutCopy(x1, rangeStart, rangeEnd); RoaringBitmap rb2 = selectRangeWithoutCopy(x2, rangeStart, rangeEnd); return andNot(rb1, rb2); } /** * Bitwise ANDNOT (difference) operation for the given range, rangeStart (inclusive) and rangeEnd * (exclusive). The provided bitmaps are *not* modified. This operation is thread-safe as long as * the provided bitmaps remain unchanged. * * @param x1 first bitmap * @param x2 other bitmap * @param rangeStart starting point of the range (inclusive) * @param rangeEnd end point of the range (exclusive) * @return result of the operation * @deprecated use the version where longs specify the range. Negative values for range * endpoints are not allowed. */ @Deprecated public static RoaringBitmap andNot(final RoaringBitmap x1, final RoaringBitmap x2, final int rangeStart, final int rangeEnd) { return andNot(x1, x2, (long) rangeStart, (long) rangeEnd); } /** * Computes OR between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bitmap */ public static RoaringBitmap or(@SuppressWarnings("rawtypes") final Iterator bitmaps, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); Iterator bitmapsIterator; bitmapsIterator = selectRangeWithoutCopy(bitmaps, rangeStart, rangeEnd); return or(bitmapsIterator); } /** * Computes OR between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bitmap * @deprecated use the version where longs specify the range. * Negative range points are forbidden. */ @Deprecated public static RoaringBitmap or(@SuppressWarnings("rawtypes") final Iterator bitmaps, final int rangeStart, final int rangeEnd) { return or(bitmaps, (long) rangeStart, (long) rangeEnd); } /** * Assume that one wants to store "cardinality" integers in [0, universe_size), this function * returns an upper bound on the serialized size in bytes. * * @param cardinality maximal cardinality * @param universe_size maximal value * @return upper bound on the serialized size in bytes of the bitmap */ public static long maximumSerializedSize(long cardinality, long universe_size) { long contnbr = (universe_size + 65535) / 65536; if (contnbr > cardinality) { contnbr = cardinality; // we can't have more containers than we have values } final long headermax = Math.max(8, 4 + (contnbr + 7) / 8) + 8 * contnbr; final long valsarray = 2 * cardinality; final long valsbitmap = contnbr * 8192; final long valsbest = Math.min(valsarray, valsbitmap); return valsbest + headermax; } /** * Return new iterator with only values from rangeStart (inclusive) to rangeEnd (exclusive) * * @param bitmaps bitmaps iterator * @param rangeStart inclusive * @param rangeEnd exclusive * @return new iterator of bitmaps */ private static Iterator selectRangeWithoutCopy(final Iterator bitmaps, final long rangeStart, final long rangeEnd) { Iterator bitmapsIterator; bitmapsIterator = new Iterator() { @Override public boolean hasNext() { return bitmaps.hasNext(); } @Override public RoaringBitmap next() { RoaringBitmap next = (RoaringBitmap) bitmaps.next(); return selectRangeWithoutCopy(next, rangeStart, rangeEnd); } @Override public void remove() { throw new UnsupportedOperationException("Remove not supported"); } }; return bitmapsIterator; } /* * In testing, original int-range code failed an assertion with some negative ranges * so presumably nobody relies on negative ranges. rangeEnd=0 also failed. */ /** * Extracts the values in the specified range, rangeStart (inclusive) and rangeEnd (exclusive) * while avoiding copies as much as possible. * * @param rb input bitmap * @param rangeStart inclusive * @param rangeEnd exclusive * @return new bitmap */ // had formerly failed if rangeEnd==0 private static RoaringBitmap selectRangeWithoutCopy(RoaringBitmap rb, final long rangeStart, final long rangeEnd) { final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); RoaringBitmap answer = new RoaringBitmap(); assert (rangeStart >= 0 && rangeEnd >= 0); if (rangeEnd <= rangeStart) { return answer; } if (hbStart == hbLast) { final int i = rb.highLowContainer.getIndex((short) hbStart); if (i >= 0) { final Container c = rb.highLowContainer.getContainerAtIndex(i).remove(0, lbStart) .iremove(lbLast + 1, Util.maxLowBitAsInteger() + 1); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbStart, c); } } return answer; } int ifirst = rb.highLowContainer.getIndex((short) hbStart); int ilast = rb.highLowContainer.getIndex((short) hbLast); if (ifirst >= 0) { final Container c = rb.highLowContainer.getContainerAtIndex(ifirst).remove(0, lbStart); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbStart, c); } } // revised to loop on ints for (int hb = hbStart + 1; hb <= hbLast - 1; ++hb) { final int i = rb.highLowContainer.getIndex((short) hb); final int j = answer.highLowContainer.getIndex((short) hb); assert j < 0; if (i >= 0) { final Container c = rb.highLowContainer.getContainerAtIndex(i); answer.highLowContainer.insertNewKeyValueAt(-j - 1, (short) hb, c); } } if (ilast >= 0) { final Container c = rb.highLowContainer.getContainerAtIndex(ilast).remove(lbLast + 1, Util.maxLowBitAsInteger() + 1); if (c.getCardinality() > 0) { answer.highLowContainer.append((short) hbLast, c); } } return answer; } /** * Computes XOR between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bitmap */ public static RoaringBitmap xor(@SuppressWarnings("rawtypes") final Iterator bitmaps, final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); Iterator bitmapsIterator; bitmapsIterator = selectRangeWithoutCopy(bitmaps, rangeStart, rangeEnd); return FastAggregation.xor(bitmapsIterator); } /** * Computes XOR between input bitmaps in the given range, from rangeStart (inclusive) to rangeEnd * (exclusive) * * @param bitmaps input bitmaps, these are not modified * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @return new result bi * @deprecated use the version where longs specify the range. * Negative values not allowed for rangeStart and rangeEnd */ @Deprecated public static RoaringBitmap xor(@SuppressWarnings("rawtypes") final Iterator bitmaps, final int rangeStart, final int rangeEnd) { return xor(bitmaps, (long) rangeStart, (long) rangeEnd); } /** * Set all the specified values to true. This can be expected to be slightly * faster than calling "add" repeatedly. The provided integers values don't * have to be in sorted order, but it may be preferable to sort them from a performance point of * view. * * @param dat set values */ public void add(final int... dat) { Container currentcont = null; short currenthb = 0; int currentcontainerindex = 0; int j = 0; if (j < dat.length) { int val = dat[j]; currenthb = Util.highbits(val); currentcontainerindex = highLowContainer.getIndex(currenthb); if (currentcontainerindex >= 0) { currentcont = highLowContainer.getContainerAtIndex(currentcontainerindex); Container newcont = currentcont.add(Util.lowbits(val)); if (newcont != currentcont) { highLowContainer.setContainerAtIndex(currentcontainerindex, newcont); currentcont = newcont; } } else { currentcontainerindex = -currentcontainerindex - 1; final ArrayContainer newac = new ArrayContainer(); currentcont = newac.add(Util.lowbits(val)); highLowContainer.insertNewKeyValueAt(currentcontainerindex, currenthb, currentcont); } j++; } for (; j < dat.length; ++j) { int val = dat[j]; short newhb = Util.highbits(val); if (currenthb == newhb) {// easy case // this could be quite frequent Container newcont = currentcont.add(Util.lowbits(val)); if (newcont != currentcont) { highLowContainer.setContainerAtIndex(currentcontainerindex, newcont); currentcont = newcont; } } else { currenthb = newhb; currentcontainerindex = highLowContainer.getIndex(currenthb); if (currentcontainerindex >= 0) { currentcont = highLowContainer.getContainerAtIndex(currentcontainerindex); Container newcont = currentcont.add(Util.lowbits(val)); if (newcont != currentcont) { highLowContainer.setContainerAtIndex(currentcontainerindex, newcont); currentcont = newcont; } } else { currentcontainerindex = -currentcontainerindex - 1; final ArrayContainer newac = new ArrayContainer(); currentcont = newac.add(Util.lowbits(val)); highLowContainer.insertNewKeyValueAt(currentcontainerindex, currenthb, currentcont); } } } } /** * Add the value to the container (set the value to "true"), whether it already appears or not. * * @param x integer value */ public void add(final int x) { final short hb = Util.highbits(x); final int i = highLowContainer.getIndex(hb); if (i >= 0) { highLowContainer.setContainerAtIndex(i, highLowContainer.getContainerAtIndex(i).add(Util.lowbits(x))); } else { final ArrayContainer newac = new ArrayContainer(); highLowContainer.insertNewKeyValueAt(-i - 1, hb, newac.add(Util.lowbits(x))); } } /** * Add to the current bitmap all integers in [rangeStart,rangeEnd). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range */ public void add(final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return; // empty range } final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); for (int hb = hbStart; hb <= hbLast; ++hb) { // first container may contain partial range final int containerStart = (hb == hbStart) ? lbStart : 0; // last container may contain partial range final int containerLast = (hb == hbLast) ? lbLast : Util.maxLowBitAsInteger(); final int i = highLowContainer.getIndex((short) hb); if (i >= 0) { final Container c = highLowContainer.getContainerAtIndex(i).iadd(containerStart, containerLast + 1); highLowContainer.setContainerAtIndex(i, c); } else { highLowContainer.insertNewKeyValueAt(-i - 1, (short) hb, Container.rangeOfOnes(containerStart, containerLast + 1)); } } } /** * Add to the current bitmap all integers in [rangeStart,rangeEnd). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @deprecated use the version where longs specify the range */ @Deprecated public void add(final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { add((long) rangeStart, (long) rangeEnd); } // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve add(rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } /** * In-place bitwise AND (intersection) operation. The current bitmap is modified. * * @param x2 other bitmap */ public void and(final RoaringBitmap x2) { int pos1 = 0, pos2 = 0, intersectionSize = 0; final int length1 = highLowContainer.size(), length2 = x2.highLowContainer.size(); while (pos1 < length1 && pos2 < length2) { final short s1 = highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); final Container c = c1.iand(c2); if (c.getCardinality() > 0) { highLowContainer.replaceKeyAndContainerAtIndex(intersectionSize++, s1, c); } ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1 = highLowContainer.advanceUntil(s2, pos1); } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } highLowContainer.resize(intersectionSize); } /** * In-place bitwise ANDNOT (difference) operation. The current bitmap is modified. * * @param x2 other bitmap */ public void andNot(final RoaringBitmap x2) { int pos1 = 0, pos2 = 0, intersectionSize = 0; final int length1 = highLowContainer.size(), length2 = x2.highLowContainer.size(); while (pos1 < length1 && pos2 < length2) { final short s1 = highLowContainer.getKeyAtIndex(pos1); final short s2 = x2.highLowContainer.getKeyAtIndex(pos2); if (s1 == s2) { final Container c1 = highLowContainer.getContainerAtIndex(pos1); final Container c2 = x2.highLowContainer.getContainerAtIndex(pos2); final Container c = c1.iandNot(c2); if (c.getCardinality() > 0) { highLowContainer.replaceKeyAndContainerAtIndex(intersectionSize++, s1, c); } ++pos1; ++pos2; } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 if (pos1 != intersectionSize) { final Container c1 = highLowContainer.getContainerAtIndex(pos1); highLowContainer.replaceKeyAndContainerAtIndex(intersectionSize, s1, c1); } ++intersectionSize; ++pos1; } else { // s1 > s2 pos2 = x2.highLowContainer.advanceUntil(s1, pos2); } } if (pos1 < length1) { highLowContainer.copyRange(pos1, length1, intersectionSize); intersectionSize += length1 - pos1; } highLowContainer.resize(intersectionSize); } /** * Add the value to the container (set the value to "true"), whether it already appears or not. * * @param x integer value * @return true if the added int wasn't already contained in the bitmap. False otherwise. */ public boolean checkedAdd(final int x) { final short hb = Util.highbits(x); final int i = highLowContainer.getIndex(hb); if (i >= 0) { Container c = highLowContainer.getContainerAtIndex(i); int oldCard = c.getCardinality(); // we need to keep the newContainer if a switch between containers type // occur, in order to get the new cardinality Container newCont = c.add(Util.lowbits(x)); highLowContainer.setContainerAtIndex(i, newCont); if (newCont.getCardinality() > oldCard) { return true; } } else { final ArrayContainer newac = new ArrayContainer(); highLowContainer.insertNewKeyValueAt(-i - 1, hb, newac.add(Util.lowbits(x))); return true; } return false; } /** * If present remove the specified integer (effectively, sets its bit value to false) * * @param x integer value representing the index in a bitmap * @return true if the unset bit was already in the bitmap */ public boolean checkedRemove(final int x) { final short hb = Util.highbits(x); final int i = highLowContainer.getIndex(hb); if (i < 0) { return false; } Container C = highLowContainer.getContainerAtIndex(i); int oldcard = C.getCardinality(); C.remove(Util.lowbits(x)); int newcard = C.getCardinality(); if (newcard == oldcard) { return false; } if (newcard > 0) { highLowContainer.setContainerAtIndex(i, C); } else { highLowContainer.removeAtIndex(i); } return true; } /** * reset to an empty bitmap; result occupies as much space a newly created bitmap. */ public void clear() { highLowContainer = new RoaringArray(); // lose references } @Override public RoaringBitmap clone() { try { final RoaringBitmap x = (RoaringBitmap) super.clone(); x.highLowContainer = highLowContainer.clone(); return x; } catch (final CloneNotSupportedException e) { throw new RuntimeException("shouldn't happen with clone", e); } } /** * Checks whether the value in included, which is equivalent to checking if the corresponding bit * is set (get in BitSet class). * * @param x integer value * @return whether the integer value is included. */ @Override public boolean contains(final int x) { final short hb = Util.highbits(x); final Container c = highLowContainer.getContainer(hb); return c != null && c.contains(Util.lowbits(x)); } /** * Deserialize (retrieve) this bitmap. *

* The current bitmap is overwritten. * * @param in the DataInput stream * @throws IOException Signals that an I/O exception has occurred. */ public void deserialize(DataInput in) throws IOException { this.highLowContainer.deserialize(in); } @Override public boolean equals(Object o) { if (o instanceof RoaringBitmap) { final RoaringBitmap srb = (RoaringBitmap) o; return srb.highLowContainer.equals(this.highLowContainer); } return false; } /** * Add the value if it is not already present, otherwise remove it. * * @param x integer value */ public void flip(final int x) { final short hb = Util.highbits(x); final int i = highLowContainer.getIndex(hb); if (i >= 0) { Container c = highLowContainer.getContainerAtIndex(i).flip(Util.lowbits(x)); if (c.getCardinality() > 0) { highLowContainer.setContainerAtIndex(i, c); } else { highLowContainer.removeAtIndex(i); } } else { final ArrayContainer newac = new ArrayContainer(); highLowContainer.insertNewKeyValueAt(-i - 1, hb, newac.add(Util.lowbits(x))); } } /** * Modifies the current bitmap by complementing the bits in the given range, from rangeStart * (inclusive) rangeEnd (exclusive). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range */ public void flip(final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return; // empty range } final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); // TODO:this can be accelerated considerably for (int hb = hbStart; hb <= hbLast; ++hb) { // first container may contain partial range final int containerStart = (hb == hbStart) ? lbStart : 0; // last container may contain partial range final int containerLast = (hb == hbLast) ? lbLast : Util.maxLowBitAsInteger(); final int i = highLowContainer.getIndex((short) hb); if (i >= 0) { final Container c = highLowContainer.getContainerAtIndex(i).inot(containerStart, containerLast + 1); if (c.getCardinality() > 0) { highLowContainer.setContainerAtIndex(i, c); } else { highLowContainer.removeAtIndex(i); } } else { highLowContainer.insertNewKeyValueAt(-i - 1, (short) hb, Container.rangeOfOnes(containerStart, containerLast + 1)); } } } /** * Modifies the current bitmap by complementing the bits in the given range, from rangeStart * (inclusive) rangeEnd (exclusive). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @deprecated use the version where longs specify the range */ @Deprecated public void flip(final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { flip((long) rangeStart, (long) rangeEnd); } else { // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve flip(rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } } /** * Returns the number of distinct integers added to the bitmap (e.g., number of bits set). * * @return the cardinality */ @Override public long getLongCardinality() { long size = 0; for (int i = 0; i < this.highLowContainer.size(); i++) { size += this.highLowContainer.getContainerAtIndex(i).getCardinality(); } return size; } @Override public int getCardinality() { return (int) getLongCardinality(); } @Override public void forEach(IntConsumer ic) { for (int i = 0; i < this.highLowContainer.size(); i++) { this.highLowContainer.getContainerAtIndex(i).forEach(this.highLowContainer.keys[i], ic); } } /** * Return a low-level container pointer that can be used to access the underlying data structure. * * @return container pointer */ public ContainerPointer getContainerPointer() { return this.highLowContainer.getContainerPointer(); } /** * For better performance, consider the Use the {@link #forEach forEach} method. * * @return a custom iterator over set bits, the bits are traversed in ascending sorted order */ @Override public PeekableIntIterator getIntIterator() { return new RoaringIntIterator(); } /** * @return a custom iterator over set bits, the bits are traversed in descending sorted order */ @Override public IntIterator getReverseIntIterator() { return new RoaringReverseIntIterator(); } /** * Estimate of the memory usage of this data structure. This can be expected to be within 1% of * the true memory usage. * * @return estimated memory usage. */ @Override public long getLongSizeInBytes() { long size = 8; for (int i = 0; i < this.highLowContainer.size(); i++) { final Container c = this.highLowContainer.getContainerAtIndex(i); size += 2 + c.getSizeInBytes(); } return size; } @Override public int getSizeInBytes() { return (int) getLongSizeInBytes(); } @Override public int hashCode() { return highLowContainer.hashCode(); } /** * Check whether this bitmap has had its runs compressed. * * @return whether this bitmap has run compression */ public boolean hasRunCompression() { for (int i = 0; i < this.highLowContainer.size(); i++) { Container c = this.highLowContainer.getContainerAtIndex(i); if (c instanceof RunContainer) { return true; } } return false; } /** * Checks whether the bitmap is empty. * * @return true if this bitmap contains no set bit */ @Override public boolean isEmpty() { return highLowContainer.size() == 0; } /** * iterate over the positions of the true values. * * @return the iterator */ @Override public Iterator iterator() { return new Iterator() { private int hs = 0; private ShortIterator iter; private int pos = 0; private int x; @Override public boolean hasNext() { return pos < RoaringBitmap.this.highLowContainer.size(); } private Iterator init() { if (pos < RoaringBitmap.this.highLowContainer.size()) { iter = RoaringBitmap.this.highLowContainer.getContainerAtIndex(pos).getShortIterator(); hs = RoaringBitmap.this.highLowContainer.getKeyAtIndex(pos) << 16; } return this; } @Override public Integer next() { x = iter.nextAsInt() | hs; if (!iter.hasNext()) { ++pos; init(); } return x; } @Override public void remove() { if ((x & hs) == hs) {// still in same container iter.remove(); } else { RoaringBitmap.this.remove(x); } } }.init(); } // don't forget to call repairAfterLazy() afterward // important: x2 should not have been computed lazily protected void lazyor(final RoaringBitmap x2) { int pos1 = 0, pos2 = 0; int length1 = highLowContainer.size(); final int length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { this.highLowContainer.setContainerAtIndex(pos1, highLowContainer.getContainerAtIndex(pos1) .lazyIOR(x2.highLowContainer.getContainerAtIndex(pos2))); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1++; if (pos1 == length1) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 highLowContainer.insertNewKeyValueAt(pos1, s2, x2.highLowContainer.getContainerAtIndex(pos2).clone()); pos1++; length1++; pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } } // don't forget to call repairAfterLazy() afterward // important: x2 should not have been computed lazily // this method is like lazyor except that it will convert // the current container to a bitset public void naivelazyor(RoaringBitmap x2) { int pos1 = 0, pos2 = 0; int length1 = highLowContainer.size(); final int length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { Container c1 = highLowContainer.getContainerAtIndex(pos1); c1 = c1.toBitmapContainer(); this.highLowContainer.setContainerAtIndex(pos1, c1.lazyIOR(x2.highLowContainer.getContainerAtIndex(pos2))); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1++; if (pos1 == length1) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 highLowContainer.insertNewKeyValueAt(pos1, s2, x2.highLowContainer.getContainerAtIndex(pos2).clone()); pos1++; length1++; pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } } /** * Create a new Roaring bitmap containing at most maxcardinality integers. * * @param maxcardinality maximal cardinality * @return a new bitmap with cardinality no more than maxcardinality */ @Override public RoaringBitmap limit(int maxcardinality) { RoaringBitmap answer = new RoaringBitmap(); int currentcardinality = 0; for (int i = 0; (currentcardinality < maxcardinality) && (i < this.highLowContainer.size()); i++) { Container c = this.highLowContainer.getContainerAtIndex(i); if (c.getCardinality() + currentcardinality <= maxcardinality) { answer.highLowContainer.appendCopy(this.highLowContainer, i); currentcardinality += c.getCardinality(); } else { int leftover = maxcardinality - currentcardinality; Container limited = c.limit(leftover); answer.highLowContainer.append(this.highLowContainer.getKeyAtIndex(i), limited); break; } } return answer; } /** * In-place bitwise OR (union) operation. The current bitmap is modified. * * @param x2 other bitmap */ public void or(final RoaringBitmap x2) { int pos1 = 0, pos2 = 0; int length1 = highLowContainer.size(); final int length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { this.highLowContainer.setContainerAtIndex(pos1, highLowContainer.getContainerAtIndex(pos1) .ior(x2.highLowContainer.getContainerAtIndex(pos2))); pos1++; pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1++; if (pos1 == length1) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 highLowContainer.insertNewKeyValueAt(pos1, s2, x2.highLowContainer.getContainerAtIndex(pos2).clone()); pos1++; length1++; pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } } /** * Rank returns the number of integers that are smaller or equal to x (Rank(infinity) would be * GetCardinality()). * * @param x upper limit * @return the rank */ @Override public long rankLong(int x) { long size = 0; short xhigh = Util.highbits(x); for (int i = 0; i < this.highLowContainer.size(); i++) { short key = this.highLowContainer.getKeyAtIndex(i); if (Util.compareUnsigned(key, xhigh) < 0) { size += this.highLowContainer.getContainerAtIndex(i).getCardinality(); } else { return size + this.highLowContainer.getContainerAtIndex(i).rank(Util.lowbits(x)); } } return size; } @Override public int rank(int x) { return (int) rankLong(x); } @Override public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException { this.highLowContainer.readExternal(in); } /** * If present remove the specified integer (effectively, sets its bit value to false) * * @param x integer value representing the index in a bitmap */ public void remove(final int x) { final short hb = Util.highbits(x); final int i = highLowContainer.getIndex(hb); if (i < 0) { return; } highLowContainer.setContainerAtIndex(i, highLowContainer.getContainerAtIndex(i).remove(Util.lowbits(x))); if (highLowContainer.getContainerAtIndex(i).getCardinality() == 0) { highLowContainer.removeAtIndex(i); } } /** * Remove from the current bitmap all integers in [rangeStart,rangeEnd). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range */ public void remove(final long rangeStart, final long rangeEnd) { rangeSanityCheck(rangeStart, rangeEnd); if (rangeStart >= rangeEnd) { return; // empty range } final int hbStart = Util.toIntUnsigned(Util.highbits(rangeStart)); final int lbStart = Util.toIntUnsigned(Util.lowbits(rangeStart)); final int hbLast = Util.toIntUnsigned(Util.highbits(rangeEnd - 1)); final int lbLast = Util.toIntUnsigned(Util.lowbits(rangeEnd - 1)); if (hbStart == hbLast) { final int i = highLowContainer.getIndex((short) hbStart); if (i < 0) { return; } final Container c = highLowContainer.getContainerAtIndex(i).iremove(lbStart, lbLast + 1); if (c.getCardinality() > 0) { highLowContainer.setContainerAtIndex(i, c); } else { highLowContainer.removeAtIndex(i); } return; } int ifirst = highLowContainer.getIndex((short) hbStart); int ilast = highLowContainer.getIndex((short) hbLast); if (ifirst >= 0) { if (lbStart != 0) { final Container c = highLowContainer.getContainerAtIndex(ifirst).iremove(lbStart, Util.maxLowBitAsInteger() + 1); if (c.getCardinality() > 0) { highLowContainer.setContainerAtIndex(ifirst, c); ifirst++; } } } else { ifirst = -ifirst - 1; } if (ilast >= 0) { if (lbLast != Util.maxLowBitAsInteger()) { final Container c = highLowContainer.getContainerAtIndex(ilast).iremove(0, lbLast + 1); if (c.getCardinality() > 0) { highLowContainer.setContainerAtIndex(ilast, c); } else { ilast++; } } else { ilast++; } } else { ilast = -ilast - 1; } highLowContainer.removeIndexRange(ifirst, ilast); } /** * Remove from the current bitmap all integers in [rangeStart,rangeEnd). * * @param rangeStart inclusive beginning of range * @param rangeEnd exclusive ending of range * @deprecated use the version where longs specify the range */ @Deprecated public void remove(final int rangeStart, final int rangeEnd) { if (rangeStart >= 0) { remove((long) rangeStart, (long) rangeEnd); } // rangeStart being -ve and rangeEnd being positive is not expected) // so assume both -ve remove(rangeStart & 0xFFFFFFFFL, rangeEnd & 0xFFFFFFFFL); } /** * Remove run-length encoding even when it is more space efficient * * @return whether a change was applied */ public boolean removeRunCompression() { boolean answer = false; for (int i = 0; i < this.highLowContainer.size(); i++) { Container c = this.highLowContainer.getContainerAtIndex(i); if (c instanceof RunContainer) { Container newc = ((RunContainer) c).toBitmapOrArrayContainer(c.getCardinality()); this.highLowContainer.setContainerAtIndex(i, newc); answer = true; } } return answer; } // to be used with lazyor public void repairAfterLazy() { for (int k = 0; k < highLowContainer.size(); ++k) { Container c = highLowContainer.getContainerAtIndex(k); highLowContainer.setContainerAtIndex(k, c.repairAfterLazy()); } } /** * Use a run-length encoding where it is more space efficient * * @return whether a change was applied */ public boolean runOptimize() { boolean answer = false; for (int i = 0; i < this.highLowContainer.size(); i++) { Container c = this.highLowContainer.getContainerAtIndex(i).runOptimize(); if (c instanceof RunContainer) { answer = true; } this.highLowContainer.setContainerAtIndex(i, c); } return answer; } /** * Return the jth value stored in this bitmap. * * @param j index of the value * @return the value */ @Override public int select(int j) { int leftover = j; for (int i = 0; i < this.highLowContainer.size(); i++) { Container c = this.highLowContainer.getContainerAtIndex(i); int thiscard = c.getCardinality(); if (thiscard > leftover) { int keycontrib = this.highLowContainer.getKeyAtIndex(i) << 16; int lowcontrib = Util.toIntUnsigned(c.select(leftover)); return lowcontrib + keycontrib; } leftover -= thiscard; } throw new IllegalArgumentException( "select " + j + " when the cardinality is " + this.getCardinality()); } /** * Serialize this bitmap. *

* Consider calling {@link #runOptimize} before serialization to improve compression. *

* The current bitmap is not modified. *

* Advanced example: To serialize your bitmap to a ByteBuffer, you can do the following. *

*

     * {@code
     *   //r is your bitmap
     *
     *   r.runOptimize(); // might improve compression
     *   // next we create the ByteBuffer where the data will be stored
     *   ByteBuffer outbb = ByteBuffer.allocate(r.serializedSizeInBytes());
     *   // then we can serialize on a custom OutputStream
     *   mrb.serialize(new DataOutputStream(new OutputStream(){
     *       ByteBuffer mBB;
     *       OutputStream init(ByteBuffer mbb) {mBB=mbb; return this;}
     *       public void close() {}
     *       public void flush() {}
     *       public void write(int b) {
     *         mBB.put((byte) b);}
     *       public void write(byte[] b) {mBB.put(b);}
     *       public void write(byte[] b, int off, int l) {mBB.put(b,off,l);}
     *   }.init(outbb)));
     *   // outbuff will now contain a serialized version of your bitmap
     * }
     * 
*

* Note: Java's data structures are in big endian format. Roaring serializes to a little endian * format, so the bytes are flipped by the library during serialization to ensure that what is * stored is in little endian---despite Java's big endianness. You can defeat this process by * reflipping the bytes again in a custom DataOutput which could lead to serialized Roaring * objects with an incorrect byte order. * * @param out the DataOutput stream * @throws IOException Signals that an I/O exception has occurred. */ @Override public void serialize(DataOutput out) throws IOException { this.highLowContainer.serialize(out); } /** * Report the number of bytes required to serialize this bitmap. This is the number of bytes * written out when using the serialize method. When using the writeExternal method, the count * will be higher due to the overhead of Java serialization. * * @return the size in bytes */ @Override public int serializedSizeInBytes() { return this.highLowContainer.serializedSizeInBytes(); } /** * Return the set values as an array, if the cardinality is smaller than 2147483648. * The integer values are in sorted order. * * @return array representing the set values. */ @Override public int[] toArray() { final int[] array = new int[(int) this.getCardinality()]; int pos = 0, pos2 = 0; while (pos < this.highLowContainer.size()) { final int hs = this.highLowContainer.getKeyAtIndex(pos) << 16; Container c = this.highLowContainer.getContainerAtIndex(pos++); c.fillLeastSignificant16bits(array, pos2, hs); pos2 += c.getCardinality(); } return array; } /** * Convert (copies) to a mutable roaring bitmap. * * @return a copy of this bitmap as a MutableRoaringBitmap */ public MutableRoaringBitmap toMutableRoaringBitmap() { return new MutableRoaringBitmap(this); } /** * A string describing the bitmap. * * @return the string */ @Override public String toString() { final StringBuilder answer = new StringBuilder(); final IntIterator i = this.getIntIterator(); answer.append("{"); if (i.hasNext()) { answer.append(i.next() & 0xFFFFFFFFL); } while (i.hasNext()) { answer.append(","); // to avoid using too much memory, we limit the size if (answer.length() > 0x80000) { answer.append("..."); break; } answer.append(i.next() & 0xFFFFFFFFL); } answer.append("}"); return answer.toString(); } /** * Recover allocated but unused memory. */ public void trim() { for (int i = 0; i < this.highLowContainer.size(); i++) { this.highLowContainer.getContainerAtIndex(i).trim(); } } @Override public void writeExternal(ObjectOutput out) throws IOException { this.highLowContainer.writeExternal(out); } /** * In-place bitwise XOR (symmetric difference) operation. The current bitmap is modified. * * @param x2 other bitmap */ public void xor(final RoaringBitmap x2) { int pos1 = 0, pos2 = 0; int length1 = highLowContainer.size(); final int length2 = x2.highLowContainer.size(); main: if (pos1 < length1 && pos2 < length2) { short s1 = highLowContainer.getKeyAtIndex(pos1); short s2 = x2.highLowContainer.getKeyAtIndex(pos2); while (true) { if (s1 == s2) { final Container c = highLowContainer.getContainerAtIndex(pos1) .ixor(x2.highLowContainer.getContainerAtIndex(pos2)); if (c.getCardinality() > 0) { this.highLowContainer.setContainerAtIndex(pos1, c); pos1++; } else { highLowContainer.removeAtIndex(pos1); --length1; } pos2++; if ((pos1 == length1) || (pos2 == length2)) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); s2 = x2.highLowContainer.getKeyAtIndex(pos2); } else if (Util.compareUnsigned(s1, s2) < 0) { // s1 < s2 pos1++; if (pos1 == length1) { break main; } s1 = highLowContainer.getKeyAtIndex(pos1); } else { // s1 > s2 highLowContainer.insertNewKeyValueAt(pos1, s2, x2.highLowContainer.getContainerAtIndex(pos2).clone()); pos1++; length1++; pos2++; if (pos2 == length2) { break main; } s2 = x2.highLowContainer.getKeyAtIndex(pos2); } } } if (pos1 == length1) { highLowContainer.appendCopy(x2.highLowContainer, pos2, length2); } } private final class RoaringIntIterator implements PeekableIntIterator { private int hs = 0; private PeekableShortIterator iter; private int pos = 0; private RoaringIntIterator() { nextContainer(); } @Override public PeekableIntIterator clone() { try { RoaringIntIterator x = (RoaringIntIterator) super.clone(); x.iter = this.iter.clone(); return x; } catch (CloneNotSupportedException e) { return null;// will not happen } } @Override public boolean hasNext() { return pos < RoaringBitmap.this.highLowContainer.size(); } @Override public int next() { final int x = iter.nextAsInt() | hs; if (!iter.hasNext()) { ++pos; nextContainer(); } return x; } private void nextContainer() { if (pos < RoaringBitmap.this.highLowContainer.size()) { iter = RoaringBitmap.this.highLowContainer.getContainerAtIndex(pos).getShortIterator(); hs = RoaringBitmap.this.highLowContainer.getKeyAtIndex(pos) << 16; } } @Override public void advanceIfNeeded(int minval) { while (hasNext() && ((hs >>> 16) < (minval >>> 16))) { ++pos; nextContainer(); } if (hasNext() && ((hs >>> 16) == (minval >>> 16))) { iter.advanceIfNeeded(Util.lowbits(minval)); if (!iter.hasNext()) { ++pos; nextContainer(); } } } @Override public int peekNext() { return Util.toIntUnsigned(iter.peekNext()) | hs; } } private final class RoaringReverseIntIterator implements IntIterator { int hs = 0; ShortIterator iter; // don't need an int because we go to 0, not Short.MAX_VALUE, and signed shorts underflow well // below zero short pos = (short) (RoaringBitmap.this.highLowContainer.size() - 1); private RoaringReverseIntIterator() { nextContainer(); } @Override public IntIterator clone() { try { RoaringReverseIntIterator clone = (RoaringReverseIntIterator) super.clone(); clone.iter = this.iter.clone(); return clone; } catch (CloneNotSupportedException e) { return null;// will not happen } } @Override public boolean hasNext() { return pos >= 0; } @Override public int next() { final int x = iter.nextAsInt() | hs; if (!iter.hasNext()) { --pos; nextContainer(); } return x; } private void nextContainer() { if (pos >= 0) { iter = RoaringBitmap.this.highLowContainer.getContainerAtIndex(pos).getReverseShortIterator(); hs = RoaringBitmap.this.highLowContainer.getKeyAtIndex(pos) << 16; } } } }