com.fathzer.imt.implementation.RoaringBitmap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of in-memory-table Show documentation
There is a newer version: 0.0.6
package com.fathzer.imt.implementation;

import java.io.Serializable;

import com.fathzer.imt.Bitmap;
import com.fathzer.imt.util.IntIterator;
import com.fathzer.imt.util.UnexpectedCloneNotSupportedException;

/** A Bitmap backed by the excellent Roaring library from D. Lemire. 
 * @author Jean-Marc Astesana
 */
public class RoaringBitmap implements Bitmap, Cloneable, Serializable {
	private static final long serialVersionUID = 1L;

	private org.roaringbitmap.RoaringBitmap set;
	private boolean isLocked;
	
	/** Constructor.
	 * 
Builds a new empty bitmap. 
	 */
	public RoaringBitmap() {
		this.set = new org.roaringbitmap.RoaringBitmap();
		this.isLocked = false;
	}

	@Override
	public int getCardinality() {
		return set.getCardinality();
	}
	
	@Override
	public void or(Bitmap bitmap) {
		check();
		this.set.or(((RoaringBitmap)bitmap).set);
	}

	@Override
	public void xor(Bitmap bitmap) {
		check();
		set.xor(((RoaringBitmap)bitmap).set);
	}

	@Override
	public void and(Bitmap bitmap) {
		check();
		this.set.and(((RoaringBitmap)bitmap).set);
	}

	@Override
	public void andNot(Bitmap bitmap) {
		check();
		set.andNot(((RoaringBitmap)bitmap).set);
	}

	@Override
	public void not(int size) {
		check();
		this.set.flip(0, size);
	}

	@Override
	public void trim() {
		check();
		set.trim();
	}
	
	@Override
	public IntIterator getIterator() {
		final org.roaringbitmap.IntIterator iter = set.getIntIterator();
		return new IntIterator() {
			@Override
			public boolean hasNext() {
				return iter.hasNext();
			}

			@Override
			public int next() {
				return iter.next();
			}
		};
	}

	@Override
	public boolean isEmpty() {
		return set.isEmpty();
	}

	@Override
	public boolean contains(int index) {
		return set.contains(index);
	}

	@Override
	public void add(int index) {
		check();
		set.add(index);
	}

	@Override
	public void remove(int index) {
		check();
		set.remove(index);
	}

	@Override
	public long getSizeInBytes() {
		return set.getSizeInBytes();
	}

	@Override
	public Bitmap getLocked() {
		if (isLocked) {
			return this;
		} else {
			RoaringBitmap result = (RoaringBitmap) clone();
			result.set.trim();
			result.isLocked = true;
			return result;
		}
	}
	
	/** Clones this bitmap.
	 * @return A unlocked copy of this bitmap 
	 */
	@Override
	public RoaringBitmap clone() {
		RoaringBitmap result;
		try {
			result = (RoaringBitmap) super.clone();
			result.set = this.set.clone();
			result.isLocked = false;
			return result;
		} catch (CloneNotSupportedException e) {
			throw new UnexpectedCloneNotSupportedException(e);
		}
	}

	@Override
	public boolean isLocked() {
		return isLocked;
	}

	private void check() {
		if (isLocked) {
			throw new IllegalStateException();
		}
	}

	@Override
	public String toString() {
		return set.toString();
	}
}