org.d2ab.collection.LongMemoizer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sequence Show documentation

A lightweight alternative to Java 8 sequential Stream

There is a newer version: 2.3.0

/*
 * Copyright 2016 Daniel Skogquist Åborg
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.d2ab.collection;

import java.util.BitSet;
import java.util.HashSet;
import java.util.Set;

/**
 * Tracks presence of longs, for example used to calculate distinct entries. Uses a mix between a {@link Set} and a
 * {@link BitSet} for remembering values. Uses a {@link BitSet} up to a threshold value, then uses a regular
 * {@link Set}.
 */
public class LongMemoizer {
	private final int threshold;
	private final Set high = new HashSet<>();
	private final BitSet low;

	public LongMemoizer(int threshold) {
		this.threshold = threshold;
		low = new BitSet(threshold);
	}

	public boolean add(long l) {
		if (l >= 0 && l < threshold)
			return add(low, (int) l);
		else
			return high.add(l);
	}

	public static boolean add(BitSet bitSet, int index) {
		boolean add = !bitSet.get(index);
		if (add)
			bitSet.set(index);
		return add;
	}
}